From 8dd245ef30417ce65b807dfbe8aa3bbcbec2484f Mon Sep 17 00:00:00 2001
From: Eric <illuminating.me@gmail.com>
Date: Sun, 19 Nov 2023 02:03:33 +0800
Subject: [PATCH] feat: update graphql query to read token (#829)

---
 ee/tabby-webserver/Cargo.toml             | 2 +-
 ee/tabby-webserver/graphql/schema.graphql | 1 +
 ee/tabby-webserver/src/schema.rs          | 5 +++++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/ee/tabby-webserver/Cargo.toml b/ee/tabby-webserver/Cargo.toml
index 4f6f37e..99be2be 100644
--- 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
index 31c58c8..6b67428 100644
--- 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
index 21019c5..44e0f04 100644
--- 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)]