feat: storing chat messages and fetching latest messages

crates/daemon/src/chatpersistence.rs

@@ -1,11 +1,13 @@
 use anyhow::Result;
 use directories::ProjectDirs;
+use shared::ai::ChatMessage as CMessage;
 use sqlx::sqlite::SqliteConnectOptions;
 use sqlx::Row;
 use sqlx::SqlitePool;
 use tokio::fs;
 use tonic::async_trait;
 
+#[derive(Debug, sqlx::FromRow)]
 pub struct ChatMessage {
     pub id: i64,
     pub text: String,
@@ -14,8 +16,16 @@ pub struct ChatMessage {
 
 #[async_trait]
 pub trait ChatRepository {
-    async fn save_message(&self, text: &str, is_user: &bool) -> Result<()>;
-    async fn get_all_messages(&self) -> Result<Vec<ChatMessage>>;
+    async fn save_message(&self, text: &str, is_user: &bool) -> Result<ChatMessage>;
+    async fn get_latest_messages(&self) -> Result<Vec<ChatMessage>>;
+}
+
+pub fn message_to_dto(msg: &ChatMessage) -> CMessage {
+    CMessage {
+        id: msg.id,
+        text: msg.text.clone(),
+        is_user: msg.is_user,
+    }
 }
 
 pub struct SqliteChatRepository {
@@ -40,14 +50,15 @@ impl SqliteChatRepository {
         .await?;
 
         sqlx::query(
-            "CREATE TABLE IF NOT EXISTS message (
+            "CREATE TABLE IF NOT EXISTS messages (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 text TEXT NOT NULL,
                 is_user BOOL NOT NULL
             )",
         )
         .execute(&pool)
-        .await?;
+        .await
+        .inspect_err(|e| println!("sql error: {}", e))?;
 
         Ok(Self { pool })
     }
@@ -55,19 +66,35 @@ impl SqliteChatRepository {
 
 #[async_trait]
 impl ChatRepository for SqliteChatRepository {
-    async fn save_message(&self, text: &str, is_user: &bool) -> Result<()> {
-        sqlx::query("INSERT INTO messages (text, is_user) values (?, ?)")
-            .bind(text)
-            .bind(is_user)
-            .execute(&self.pool)
-            .await?;
-        Ok(())
+    async fn save_message(&self, text: &str, is_user: &bool) -> Result<ChatMessage> {
+        let result = sqlx::query_as::<_, ChatMessage>(
+            r#"
+            INSERT INTO messages (text, is_user)
+            VALUES (?, ?)
+            RETURNING id, text, is_user
+            "#,
+        )
+        .bind(text)
+        .bind(is_user)
+        .fetch_one(&self.pool)
+        .await
+        .inspect_err(|e| println!("sql error: {}", e))?;
+        Ok(result)
     }
 
-    async fn get_all_messages(&self) -> Result<Vec<ChatMessage>> {
-        let rows = sqlx::query("SELECT id, text, is_user FROM messages ORDER BY id DESC LIMIT 10")
-            .fetch_all(&self.pool)
-            .await?;
+    async fn get_latest_messages(&self) -> Result<Vec<ChatMessage>> {
+        let rows = sqlx::query(
+            r#"
+            SELECT * FROM (
+            SELECT id, text, is_user
+                FROM messages
+                ORDER BY id DESC
+                LIMIT 10
+            ) AS subquery ORDER BY id ASC"#,
+        )
+        .fetch_all(&self.pool)
+        .await
+        .inspect_err(|e| println!("sql error: {}", e))?;
 
         let messages = rows
             .into_iter()