refactor(document): add web document service api

ee/tabby-schema/graphql/schema.graphql

@@ -242,6 +242,16 @@ type CompletionStats {
   selects: Int!
 }
 
+type CustomDocumentConnection {
+  edges: [CustomDocumentEdge!]!
+  pageInfo: PageInfo!
+}
+
+type CustomDocumentEdge {
+  node: CustomWebDocument!
+  cursor: String!
+}
+
 type CustomWebDocument {
   url: String!
   name: String!
@@ -516,10 +526,21 @@ type PageInfo {
   endCursor: String
 }
 
+type PresetDocumentConnection {
+  edges: [PresetDocumentEdge!]!
+  pageInfo: PageInfo!
+}
+
+type PresetDocumentEdge {
+  node: PresetWebDocument!
+  cursor: String!
+}
+
 type PresetWebDocument {
   name: String!
   id: ID!
-  jobInfo: JobInfo!
+  active: Boolean!
+  jobInfo: JobInfo
 }
 
 type ProvidedRepository {
@@ -593,8 +614,8 @@ type Query {
     Thread is public within an instance, so no need to check for ownership.
   """
   threadMessages(threadId: ID!, after: String, before: String, first: Int, last: Int): MessageConnection!
-  customWebDocuments(after: String, before: String, first: Int, last: Int): WebDocumentConnection!
-  presetWebDocuments(active: Boolean!): [PresetWebDocument!]!
+  customWebDocuments(after: String, before: String, first: Int, last: Int): CustomDocumentConnection!
+  presetWebDocuments(after: String, before: String, first: Int, last: Int, active: Boolean!): PresetDocumentConnection!
 }
 
 type RefreshTokenResponse {
@@ -739,13 +760,3 @@ type WebCrawlerUrlEdge {
   node: WebCrawlerUrl!
   cursor: String!
 }
-
-type WebDocumentConnection {
-  edges: [WebDocumentEdge!]!
-  pageInfo: PageInfo!
-}
-
-type WebDocumentEdge {
-  node: CustomWebDocument!
-  cursor: String!
-}

ee/tabby-schema/src/schema/mod.rs

@@ -590,11 +590,23 @@ impl Query {
         )
         .await
     }
-    async fn preset_web_documents(ctx: &Context, active: bool) -> Result<Vec<PresetWebDocument>> {
-        ctx.locator
-            .web_documents()
-            .list_preset_web_documents(active)
-            .await
+    async fn preset_web_documents(ctx: &Context,
+                                  after: Option<String>,
+                                  before: Option<String>,
+                                  first: Option<i32>,
+                                  last: Option<i32>,
+                                  active: bool) -> Result<Connection<PresetWebDocument>> {
+        query_async(
+            after,
+            before,
+            first,
+            last,
+            |after, before, first, last| async move {
+                ctx.locator
+                    .web_documents()
+                    .list_preset_web_documents(after, before, first, last, active)
+                    .await
+            }).await
     }
 }
 

ee/tabby-schema/src/schema/web_documents.rs

@@ -20,7 +20,8 @@ pub struct CustomWebDocument {
 pub struct PresetWebDocument {
     pub name: String,
     pub id: ID,
-    pub job_info: JobInfo,
+    pub active: bool,
+    pub job_info: Option<JobInfo>,
 }
 
 impl CustomWebDocument {
@@ -35,15 +36,23 @@ impl CustomWebDocument {
 
 #[derive(Validate, GraphQLInputObject)]
 pub struct CreateCustomDocumentInput {
-    #[validate(url(code = "name", message = "Invalid Name"))]
+    #[validate(regex(
+        code = "name",
+        path = "*crate::schema::constants::USERNAME_REGEX",
+        message = "Invalid repository name"
+    ))]
     pub name: String,
     #[validate(url(code = "url", message = "Invalid URL"))]
     pub url: String,
 }
 
 #[derive(Validate, GraphQLInputObject)]
 pub struct SetPresetDocumentActiveInput {
-    #[validate(url(code = "name", message = "Invalid Name"))]
+    #[validate(regex(
+        code = "name",
+        path = "*crate::schema::constants::USERNAME_REGEX",
+        message = "Invalid repository name"
+    ))]
     pub name: String,
     pub active: bool,
 }
@@ -56,11 +65,27 @@ impl NodeType for CustomWebDocument {
     }
 
     fn connection_type_name() -> &'static str {
-        "WebDocumentConnection"
+        "CustomDocumentConnection"
+    }
+
+    fn edge_type_name() -> &'static str {
+        "CustomDocumentEdge"
+    }
+}
+
+impl NodeType for PresetWebDocument {
+    type Cursor = String;
+
+    fn cursor(&self) -> Self::Cursor {
+        self.name.clone()
+    }
+
+    fn connection_type_name() -> &'static str {
+        "PresetDocumentConnection"
     }
 
     fn edge_type_name() -> &'static str {
-        "WebDocumentEdge"
+        "PresetDocumentEdge"
     }
 }
 
@@ -76,6 +101,11 @@ pub trait WebDocumentService: Send + Sync {
 
     async fn create_custom_web_document(&self, name: String, url: String) -> Result<ID>;
     async fn delete_custom_web_document(&self, id: ID) -> Result<()>;
-    async fn list_preset_web_documents(&self, active: bool) -> Result<Vec<PresetWebDocument>>;
+    async fn list_preset_web_documents(&self,
+                                       after: Option<String>,
+                                       before: Option<String>,
+                                       first: Option<usize>,
+                                       last: Option<usize>,
+                                       active: bool) -> Result<Vec<PresetWebDocument>>;
     async fn set_preset_web_documents_active(&self, name: String, active: bool) -> Result<ID>;
 }

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

@@ -40,7 +40,12 @@ impl WebDocumentService for WebDocumentServiceImpl {
         Ok(())
     }
 
-    async fn list_preset_web_documents(&self, active: bool) -> Result<Vec<PresetWebDocument>> {
+    async fn list_preset_web_documents(&self,
+                                       after: Option<String>,
+                                       before: Option<String>,
+                                       first: Option<usize>,
+                                       last: Option<usize>,
+                                       active: bool) -> Result<Vec<PresetWebDocument>> {
         Ok(vec![])
     }