zhenyi
821537186e
- Reorganized import statements in adapter tests for better readability - Replaced or_insert_with(Vec::new) with or_default() in test closures - Updated Cargo.lock with new dependency versions and checksums - Added TLS features to tonic dependency configuration - Included sqlx, chrono, and uuid dependencies with specific features - Added jsonwebtoken and arc-swap as project dependencies - Reformatted assertion statements to comply with line length limits - Adjusted base64 import order in engine codec module - Updated protobuf include statement formatting
106 lines
3.6 KiB
Rust
106 lines
3.6 KiB
Rust
//! Draft event handlers on `MessageService`.
|
|
//!
|
|
//! Drafts are per-user private data — no permission checks beyond auth needed.
|
|
|
|
use std::sync::Arc;
|
|
|
|
use uuid::Uuid;
|
|
|
|
use crate::ImksError;
|
|
use crate::socket::socket::Socket;
|
|
|
|
use super::message::MessageService;
|
|
|
|
impl MessageService {
|
|
/// Handle `draft:save` — upsert a draft.
|
|
pub async fn save_draft(
|
|
&self,
|
|
socket: Arc<Socket>,
|
|
data: &serde_json::Value,
|
|
) -> crate::ImksResult<()> {
|
|
let user_id = self.user_id(&socket)?;
|
|
let arr = data
|
|
.as_array()
|
|
.and_then(|a| a.first())
|
|
.ok_or_else(|| ImksError::InvalidInput("Expected [payload] array".into()))?;
|
|
|
|
let channel_id: Uuid = Self::parse_field(arr, "channel_id")?;
|
|
let body: String = Self::parse_field(arr, "body")?;
|
|
let thread_id: Option<Uuid> = Self::parse_optional(arr, "thread_id")?;
|
|
let reply_to_message_id: Option<Uuid> = Self::parse_optional(arr, "reply_to_message_id")?;
|
|
let metadata: Option<serde_json::Value> = Self::parse_optional(arr, "metadata")?;
|
|
let channel_id_str = channel_id.to_string();
|
|
let user_id_str = user_id.to_string();
|
|
|
|
self.validate_channel_write(&channel_id_str, &user_id_str)
|
|
.await?;
|
|
|
|
self.repo
|
|
.upsert_draft(
|
|
channel_id,
|
|
user_id,
|
|
thread_id,
|
|
&body,
|
|
reply_to_message_id,
|
|
metadata,
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
/// Handle `draft:get` — retrieve a draft and send it back to the requesting socket.
|
|
pub async fn get_draft(
|
|
&self,
|
|
socket: Arc<Socket>,
|
|
data: &serde_json::Value,
|
|
) -> crate::ImksResult<()> {
|
|
let user_id = self.user_id(&socket)?;
|
|
let arr = data
|
|
.as_array()
|
|
.and_then(|a| a.first())
|
|
.ok_or_else(|| ImksError::InvalidInput("Expected [payload] array".into()))?;
|
|
|
|
let channel_id: Uuid = Self::parse_field(arr, "channel_id")?;
|
|
let thread_id: Option<Uuid> = Self::parse_optional(arr, "thread_id")?;
|
|
let channel_id_str = channel_id.to_string();
|
|
let user_id_str = user_id.to_string();
|
|
|
|
self.ensure_readable(&channel_id_str, &user_id_str).await?;
|
|
self.ensure_member(&channel_id_str, &user_id_str).await?;
|
|
|
|
let draft = self.repo.get_draft(channel_id, user_id, thread_id).await?;
|
|
if let Some(d) = draft {
|
|
let _ = socket.emit("draft:loaded", serde_json::to_value(&d).unwrap_or_default());
|
|
} else {
|
|
let _ = socket.emit("draft:loaded", serde_json::json!(null));
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
/// Handle `draft:delete` — delete a draft after sending.
|
|
pub async fn delete_draft(
|
|
&self,
|
|
socket: Arc<Socket>,
|
|
data: &serde_json::Value,
|
|
) -> crate::ImksResult<()> {
|
|
let user_id = self.user_id(&socket)?;
|
|
let arr = data
|
|
.as_array()
|
|
.and_then(|a| a.first())
|
|
.ok_or_else(|| ImksError::InvalidInput("Expected [payload] array".into()))?;
|
|
|
|
let channel_id: Uuid = Self::parse_field(arr, "channel_id")?;
|
|
let thread_id: Option<Uuid> = Self::parse_optional(arr, "thread_id")?;
|
|
let channel_id_str = channel_id.to_string();
|
|
let user_id_str = user_id.to_string();
|
|
|
|
self.ensure_readable(&channel_id_str, &user_id_str).await?;
|
|
self.ensure_member(&channel_id_str, &user_id_str).await?;
|
|
|
|
self.repo
|
|
.delete_draft(channel_id, user_id, thread_id)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
}
|