summaryrefslogtreecommitdiff
path: root/src/event/repo/message.rs
diff options
context:
space:
mode:
authorOwen Jacobson <owen@grimoire.ca>2024-10-02 00:41:25 -0400
committerOwen Jacobson <owen@grimoire.ca>2024-10-02 00:41:38 -0400
commit357116366c1307bedaac6a3dfe9c5ed8e0e0c210 (patch)
treed701378187d8b0f99d524991925e8348c6cab0d6 /src/event/repo/message.rs
parentf878f0b5eaa44e8ee8d67cbfd706926ff2119113 (diff)
First pass on reorganizing the backend.
This is primarily renames and repackagings.
Diffstat (limited to 'src/event/repo/message.rs')
-rw-r--r--src/event/repo/message.rs188
1 files changed, 188 insertions, 0 deletions
diff --git a/src/event/repo/message.rs b/src/event/repo/message.rs
new file mode 100644
index 0000000..f051fec
--- /dev/null
+++ b/src/event/repo/message.rs
@@ -0,0 +1,188 @@
+use sqlx::{sqlite::Sqlite, SqliteConnection, Transaction};
+
+use crate::{
+ channel::{self, Channel},
+ clock::DateTime,
+ event::{types, Sequence},
+ login::{self, Login},
+ message::{self, Message},
+};
+
+pub trait Provider {
+ fn message_events(&mut self) -> Events;
+}
+
+impl<'c> Provider for Transaction<'c, Sqlite> {
+ fn message_events(&mut self) -> Events {
+ Events(self)
+ }
+}
+
+pub struct Events<'t>(&'t mut SqliteConnection);
+
+impl<'c> Events<'c> {
+ pub async fn create(
+ &mut self,
+ sender: &Login,
+ channel: &Channel,
+ sent_at: &DateTime,
+ sent_sequence: Sequence,
+ body: &str,
+ ) -> Result<types::ChannelEvent, sqlx::Error> {
+ let id = message::Id::generate();
+
+ let message = sqlx::query!(
+ r#"
+ insert into message
+ (id, channel, sender, sent_at, sent_sequence, body)
+ values ($1, $2, $3, $4, $5, $6)
+ returning
+ id as "id: message::Id",
+ sender as "sender: login::Id",
+ sent_at as "sent_at: DateTime",
+ sent_sequence as "sent_sequence: Sequence",
+ body
+ "#,
+ id,
+ channel.id,
+ sender.id,
+ sent_at,
+ sent_sequence,
+ body,
+ )
+ .map(|row| types::ChannelEvent {
+ sequence: row.sent_sequence,
+ at: row.sent_at,
+ data: types::MessageEvent {
+ channel: channel.clone(),
+ sender: sender.clone(),
+ message: Message {
+ id: row.id,
+ body: row.body,
+ },
+ }
+ .into(),
+ })
+ .fetch_one(&mut *self.0)
+ .await?;
+
+ Ok(message)
+ }
+
+ pub async fn delete(
+ &mut self,
+ channel: &Channel,
+ message: &message::Id,
+ deleted_at: &DateTime,
+ deleted_sequence: Sequence,
+ ) -> Result<types::ChannelEvent, sqlx::Error> {
+ sqlx::query_scalar!(
+ r#"
+ delete from message
+ where id = $1
+ returning 1 as "row: i64"
+ "#,
+ message,
+ )
+ .fetch_one(&mut *self.0)
+ .await?;
+
+ Ok(types::ChannelEvent {
+ sequence: deleted_sequence,
+ at: *deleted_at,
+ data: types::MessageDeletedEvent {
+ channel: channel.clone(),
+ message: message.clone(),
+ }
+ .into(),
+ })
+ }
+
+ pub async fn expired(
+ &mut self,
+ expire_at: &DateTime,
+ ) -> Result<Vec<(Channel, message::Id)>, sqlx::Error> {
+ let messages = sqlx::query!(
+ r#"
+ select
+ channel.id as "channel_id: channel::Id",
+ channel.name as "channel_name",
+ channel.created_at as "channel_created_at: DateTime",
+ channel.created_sequence as "channel_created_sequence: Sequence",
+ message.id as "message: message::Id"
+ from message
+ join channel on message.channel = channel.id
+ join login as sender on message.sender = sender.id
+ where sent_at < $1
+ "#,
+ expire_at,
+ )
+ .map(|row| {
+ (
+ Channel {
+ id: row.channel_id,
+ name: row.channel_name,
+ created_at: row.channel_created_at,
+ created_sequence: row.channel_created_sequence,
+ },
+ row.message,
+ )
+ })
+ .fetch_all(&mut *self.0)
+ .await?;
+
+ Ok(messages)
+ }
+
+ pub async fn replay(
+ &mut self,
+ resume_at: Option<Sequence>,
+ ) -> Result<Vec<types::ChannelEvent>, sqlx::Error> {
+ let events = sqlx::query!(
+ r#"
+ select
+ message.id as "id: message::Id",
+ channel.id as "channel_id: channel::Id",
+ channel.name as "channel_name",
+ channel.created_at as "channel_created_at: DateTime",
+ channel.created_sequence as "channel_created_sequence: Sequence",
+ sender.id as "sender_id: login::Id",
+ sender.name as sender_name,
+ message.sent_at as "sent_at: DateTime",
+ message.sent_sequence as "sent_sequence: Sequence",
+ message.body
+ from message
+ join channel on message.channel = channel.id
+ join login as sender on message.sender = sender.id
+ where coalesce(message.sent_sequence > $1, true)
+ order by sent_sequence asc
+ "#,
+ resume_at,
+ )
+ .map(|row| types::ChannelEvent {
+ sequence: row.sent_sequence,
+ at: row.sent_at,
+ data: types::MessageEvent {
+ channel: Channel {
+ id: row.channel_id,
+ name: row.channel_name,
+ created_at: row.channel_created_at,
+ created_sequence: row.channel_created_sequence,
+ },
+ sender: Login {
+ id: row.sender_id,
+ name: row.sender_name,
+ },
+ message: Message {
+ id: row.id,
+ body: row.body,
+ },
+ }
+ .into(),
+ })
+ .fetch_all(&mut *self.0)
+ .await?;
+
+ Ok(events)
+ }
+}