feat(storage): add load_messages_after_timestamp, load_session_timelines, get_max_message_seq

src/storage/memory.rs

@@ -250,6 +250,30 @@ impl super::Storage {
 
         Ok(result.rows_affected())
     }
+
+    /// Load timeline entries for a specific session.
+    pub async fn load_session_timelines(
+        &self,
+        session_id: &str,
+        limit: usize,
+    ) -> Result<Vec<MemoryEntry>, StorageError> {
+        let rows = sqlx::query(
+            r#"
+            SELECT id, key, content, category, importance,
+                   session_id, created_at, updated_at
+            FROM memories
+            WHERE session_id = ? AND category = 'timeline'
+            ORDER BY created_at DESC
+            LIMIT ?
+            "#,
+        )
+        .bind(session_id)
+        .bind(limit as i64)
+        .fetch_all(self.pool())
+        .await?;
+
+        parse_memory_rows(&rows)
+    }
 }
 
 fn parse_memory_rows(

src/storage/mod.rs

@@ -571,6 +571,52 @@ impl Storage {
             .collect())
     }
 
+    pub async fn get_max_message_seq(&self, session_id: &str) -> Result<i64, StorageError> {
+        let row = sqlx::query(
+            "SELECT COALESCE(MAX(seq), 0) as max_seq FROM messages WHERE session_id = ?",
+        )
+        .bind(session_id)
+        .fetch_one(self.pool())
+        .await?;
+        Ok(row.get::<i64, _>("max_seq"))
+    }
+
+    pub async fn load_messages_after_timestamp(
+        &self,
+        session_id: &str,
+        after_ts: i64,
+    ) -> Result<Vec<crate::storage::message::MessageMeta>, StorageError> {
+        let rows = sqlx::query(
+            r#"
+            SELECT id, session_id, seq, role, content, media_refs, tool_call_id, tool_name, tool_calls, source, created_at
+            FROM messages
+            WHERE session_id = ? AND created_at > ?
+            ORDER BY seq ASC
+            "#,
+        )
+        .bind(session_id)
+        .bind(after_ts)
+        .fetch_all(self.pool())
+        .await?;
+
+        Ok(rows
+            .into_iter()
+            .map(|row| crate::storage::message::MessageMeta {
+                id: row.get("id"),
+                session_id: row.get("session_id"),
+                seq: row.get("seq"),
+                role: row.get("role"),
+                content: row.get("content"),
+                media_refs: row.get("media_refs"),
+                tool_call_id: row.get("tool_call_id"),
+                tool_name: row.get("tool_name"),
+                tool_calls: row.get("tool_calls"),
+                source: row.get("source"),
+                created_at: row.get("created_at"),
+            })
+            .collect())
+    }
+
     pub async fn list_all_active_sessions(
         &self,
         limit: i64,