feat: update graphql query to read token (#829)

TabbyML · Nov 18, 2023 · 8dd245e · 8dd245e
1 parent 530e142
commit 8dd245e
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/ee/tabby-webserver/Cargo.toml b/ee/tabby-webserver/Cargo.toml
@@ -25,7 +25,7 @@ serde.workspace = true
 tabby-common = { path = "../../crates/tabby-common" }
 tarpc = { version = "0.33.0", features = ["serde-transport"] }
 thiserror.workspace = true
-tokio.workspace = true
+tokio = { workspace = true, features = ["fs"] }
 tokio-rusqlite = "0.4.0"
 tokio-tungstenite = "0.20.1"
 tracing.workspace = true

diff --git a/ee/tabby-webserver/graphql/schema.graphql b/ee/tabby-webserver/graphql/schema.graphql
@@ -9,6 +9,7 @@ type Mutation {
 
 type Query {
   workers: [Worker!]!
+  registrationToken: String!
 }
 
 type Worker {

diff --git a/ee/tabby-webserver/src/schema.rs b/ee/tabby-webserver/src/schema.rs
@@ -13,6 +13,11 @@ impl Query {
     async fn workers(ctx: &ServerContext) -> Vec<Worker> {
         ctx.list_workers().await
     }
+
+    async fn registration_token(ctx: &ServerContext) -> FieldResult<String> {
+        let token = ctx.read_registration_token().await?;
+        Ok(token)
+    }
 }
 
 #[derive(Default)]