chore(graphql): support query user by email

Signed-off-by: Wei Zhang <[email protected]>
TabbyML · Dec 3, 2024 · 793ada4 · 793ada4
1 parent e5096a9
commit 793ada4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/ee/tabby-schema/graphql/schema.graphql b/ee/tabby-schema/graphql/schema.graphql
@@ -681,6 +681,7 @@ type Query {
   me: UserSecured!
   "List users, accessible for all login users."
   users(after: String, before: String, first: Int, last: Int): UserConnection!
+  user(email: String): User!
   invitations(after: String, before: String, first: Int, last: Int): InvitationConnection!
   jobRuns(ids: [ID!], jobs: [String!], after: String, before: String, first: Int, last: Int): JobRunConnection!
   jobRunStats(jobs: [String!]): JobStats!

diff --git a/ee/tabby-schema/src/schema/mod.rs b/ee/tabby-schema/src/schema/mod.rs
@@ -287,6 +287,20 @@ impl Query {
         .await
     }
 
+    async fn user(ctx: &Context, email: Option<String>) -> Result<UserValue> {
+        check_user(ctx).await?;
+
+        if email.is_none() {
+            return Err(CoreError::InvalidInput(ValidationErrors::new()));
+        }
+
+        ctx.locator
+            .auth()
+            .get_user_by_email(email.unwrap().as_str())
+            .await
+            .map(UserValue::UserSecured)
+    }
+
     async fn invitations(
         ctx: &Context,
         after: Option<String>,