feat(db): add code_source_id column to thread_messages table (TabbyML…

…#3571)

* feat(db): add code_source_id column to thread_messages table

* feat(graphql/thread): add code_source_id to thread message attachments

* refactor(answer.rs): make test pass

* [autofix.ci] apply automated fixes

* [autofix.ci] apply automated fixes (attempt 2/3)

* feat(testutils): enhance mock AccessPolicy and add sample CodeSearch hits

* chore: make the test with reason Fake Repository and Useful ContextHelperInfo

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Jackson Chen <[email protected]>

ee/tabby-db/migrations/0040_add-code-source-id-column.down.sql

@@ -0,0 +1,2 @@
+ALTER TABLE thread_messages
+  DROP code_source_id;

ee/tabby-db/migrations/0040_add-code-source-id-column.up.sql

@@ -0,0 +1,2 @@
+ALTER TABLE thread_messages
+  ADD code_source_id VARCHAR(255);

ee/tabby-db/schema/schema.sql

@@ -182,6 +182,7 @@ CREATE TABLE thread_messages(
   doc_attachments BLOB,
   created_at TIMESTAMP NOT NULL DEFAULT(DATETIME('now')),
   updated_at TIMESTAMP NOT NULL DEFAULT(DATETIME('now')),
+  code_source_id VARCHAR(255),
   FOREIGN KEY(thread_id) REFERENCES threads(id) ON DELETE CASCADE
 );
 CREATE TABLE web_documents(

ee/tabby-db/src/threads.rs

@@ -23,6 +23,8 @@ pub struct ThreadMessageDAO {
     pub role: String,
     pub content: String,
 
+    pub code_source_id: Option<String>,
+
     pub code_attachments: Option<Json<Vec<ThreadMessageAttachmentCode>>>,
     pub client_code_attachments: Option<Json<Vec<ThreadMessageAttachmentClientCode>>>,
     pub doc_attachments: Option<Json<Vec<ThreadMessageAttachmentDoc>>>,
@@ -223,12 +225,14 @@ impl DbConn {
     pub async fn update_thread_message_code_attachments(
         &self,
         message_id: i64,
+        code_source_id: &str,
         code_attachments: &[ThreadMessageAttachmentCode],
     ) -> Result<()> {
         let code_attachments = Json(code_attachments);
         query!(
-            "UPDATE thread_messages SET code_attachments = ?, updated_at = DATETIME('now') WHERE id = ?",
+            "UPDATE thread_messages SET code_attachments = ?, code_source_id = ?, updated_at = DATETIME('now') WHERE id = ?",
             code_attachments,
+            code_source_id,
             message_id
         )
         .execute(&self.pool)
@@ -296,6 +300,7 @@ impl DbConn {
                 thread_id,
                 role,
                 content,
+                code_source_id,
                 code_attachments as "code_attachments: Json<Vec<ThreadMessageAttachmentCode>>",
                 client_code_attachments as "client_code_attachments: Json<Vec<ThreadMessageAttachmentClientCode>>",
                 doc_attachments as "doc_attachments: Json<Vec<ThreadMessageAttachmentDoc>>",
@@ -329,6 +334,7 @@ impl DbConn {
                 "thread_id",
                 "role",
                 "content",
+                "code_source_id",
                 "code_attachments" as "code_attachments: Json<Vec<ThreadMessageAttachmentCode>>",
                 "client_code_attachments" as "client_code_attachments: Json<Vec<ThreadMessageAttachmentClientCode>>",
                 "doc_attachments" as "doc_attachments: Json<Vec<ThreadMessageAttachmentDoc>>",

ee/tabby-schema/graphql/schema.graphql

@@ -812,6 +812,7 @@ type Thread {
 }
 
 type ThreadAssistantMessageAttachmentsCode {
+  codeSourceId: String!
   hits: [MessageCodeSearchHit!]!
 }
 

ee/tabby-schema/src/schema/thread/types.rs

@@ -241,6 +241,7 @@ pub struct ThreadAssistantMessageCreated {
 
 #[derive(GraphQLObject)]
 pub struct ThreadAssistantMessageAttachmentsCode {
+    pub code_source_id: String,
     pub hits: Vec<MessageCodeSearchHit>,
 }
 

ee/tabby-webserver/src/service/access_policy.rs

@@ -64,11 +64,8 @@ pub mod testutils {
 
     use super::*;
 
-    pub async fn make_policy() -> AccessPolicy {
-        AccessPolicy::new(
-            DbConn::new_in_memory().await.unwrap(),
-            &ID::from("nihao".to_string()),
-            false,
-        )
+    /// create a mock `AccessPolicy` for testing with admin permissions
+    pub async fn make_policy(db: DbConn) -> AccessPolicy {
+        AccessPolicy::new(db.clone(), &ID::from("nihao".to_string()), true)
     }
 }