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
84 lines
2.4 KiB
Rust
84 lines
2.4 KiB
Rust
//! Attachment CRUD operations on `MessageRepo`.
|
|
|
|
use uuid::Uuid;
|
|
|
|
use crate::ImksResult;
|
|
use crate::models::message_attachment::{AttachmentSummary, MessageAttachment};
|
|
|
|
use super::message_repo::MessageRepo;
|
|
|
|
impl MessageRepo {
|
|
/// Create a single attachment record.
|
|
#[allow(clippy::too_many_arguments)]
|
|
pub async fn create_attachment(
|
|
&self,
|
|
message_id: Uuid,
|
|
filename: &str,
|
|
content_type: Option<&str>,
|
|
size: i64,
|
|
url: &str,
|
|
storage_key: Option<&str>,
|
|
width: Option<i32>,
|
|
height: Option<i32>,
|
|
spoiler: bool,
|
|
) -> ImksResult<MessageAttachment> {
|
|
let id = Uuid::now_v7();
|
|
|
|
sqlx::query_as::<_, MessageAttachment>(
|
|
r#"
|
|
INSERT INTO message_attachment (
|
|
id, message_id, filename, content_type, size, url,
|
|
storage_key, width, height, spoiler
|
|
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
|
|
RETURNING *
|
|
"#,
|
|
)
|
|
.bind(id)
|
|
.bind(message_id)
|
|
.bind(filename)
|
|
.bind(content_type)
|
|
.bind(size)
|
|
.bind(url)
|
|
.bind(storage_key)
|
|
.bind(width)
|
|
.bind(height)
|
|
.bind(spoiler)
|
|
.fetch_one(self.pool())
|
|
.await
|
|
.map_err(Into::into)
|
|
}
|
|
|
|
/// Get all attachments for a message.
|
|
pub async fn get_attachments(&self, message_id: Uuid) -> ImksResult<Vec<MessageAttachment>> {
|
|
sqlx::query_as::<_, MessageAttachment>(
|
|
"SELECT * FROM message_attachment WHERE message_id = $1 ORDER BY created_at",
|
|
)
|
|
.bind(message_id)
|
|
.fetch_all(self.pool())
|
|
.await
|
|
.map_err(Into::into)
|
|
}
|
|
|
|
/// Get lightweight attachment summaries (no URLs/storage keys).
|
|
pub async fn get_attachment_summaries(
|
|
&self,
|
|
message_id: Uuid,
|
|
) -> ImksResult<Vec<AttachmentSummary>> {
|
|
let attachments = self.get_attachments(message_id).await?;
|
|
Ok(attachments
|
|
.into_iter()
|
|
.map(AttachmentSummary::from)
|
|
.collect())
|
|
}
|
|
|
|
/// Delete a single attachment.
|
|
pub async fn delete_attachment(&self, attachment_id: Uuid) -> ImksResult<bool> {
|
|
let result = sqlx::query("DELETE FROM message_attachment WHERE id = $1")
|
|
.bind(attachment_id)
|
|
.execute(self.pool())
|
|
.await?;
|
|
|
|
Ok(result.rows_affected() > 0)
|
|
}
|
|
}
|