Adding repo::scheduling module

2026-06-18 13:42:09 -04:00
parent d1bc5eb830
commit 7227a260a4
1 changed files with 73 additions and 0 deletions
@@ -25,3 +25,76 @@ pub async fn insert(
        Err(_) => Err(sqlx::Error::RowNotFound),
    }
 }
+
+pub async fn get(
+    pool: &sqlx::PgPool,
+    id: &uuid::Uuid,
+) -> Result<textsender_models::message::scheduling::ScheduledMessage, sqlx::Error> {
+    match sqlx::query(
+        r#"
+        SELECT id, scheduled, created, status, user_id FROM "scheduled_messages"
+        WHERE
+            id = $1;
+        "#,
+    )
+    .bind(id)
+    .fetch_one(pool)
+    .await
+    {
+        Ok(row) => {
+            let id: uuid::Uuid = row.try_get("id")?;
+            let scheduled: time::OffsetDateTime = row.try_get("scheduled")?;
+            let status: String = row.try_get("status")?;
+            let created: time::OffsetDateTime = row.try_get("created")?;
+            let user_id: uuid::Uuid = row.try_get("user_id")?;
+            Ok(textsender_models::message::scheduling::ScheduledMessage {
+                id,
+                scheduled: Some(scheduled),
+                created: Some(created),
+                status,
+                user_id,
+            })
+        }
+        Err(_) => Err(sqlx::Error::RowNotFound),
+    }
+}
+
+pub async fn get_with_user_id(
+    pool: &sqlx::PgPool,
+    user_id: &uuid::Uuid,
+) -> Result<Vec<textsender_models::message::scheduling::ScheduledMessage>, sqlx::Error> {
+    match sqlx::query(
+        r#"
+        SELECT id, scheduled, created, status, user_id FROM "scheduled_messages"
+        WHERE
+            user_id = $1;
+        "#,
+    )
+    .bind(user_id)
+    .fetch_all(pool)
+    .await
+    {
+        Ok(rows) => {
+            let mut messages: Vec<textsender_models::message::scheduling::ScheduledMessage> = Vec::new();
+            for row in rows {
+            let id: uuid::Uuid = row.try_get("id")?;
+            let scheduled: time::OffsetDateTime = row.try_get("scheduled")?;
+            let status: String = row.try_get("status")?;
+            let created: time::OffsetDateTime = row.try_get("created")?;
+            let user_id: uuid::Uuid = row.try_get("user_id")?;
+
+                let message = textsender_models::message::scheduling::ScheduledMessage {
+                    id,
+                    scheduled: Some(scheduled),
+                    created: Some(created),
+                    status,
+                    user_id,
+                };
+                messages.push(message);
+            }
+
+            Ok(messages)
+        }
+        Err(_) => Err(sqlx::Error::RowNotFound),
+    }
+}