WIP: check auth status

sourcegraph · May 15, 2023 · 1282e91 · 1282e91
1 parent f6dfaf3
commit 1282e91
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/client/cody-shared/src/sourcegraph-api/graphql/client.ts b/client/cody-shared/src/sourcegraph-api/graphql/client.ts
@@ -12,7 +12,7 @@ import {
     SEARCH_EMBEDDINGS_QUERY,
     LOG_EVENT_MUTATION,
     REPOSITORY_EMBEDDING_EXISTS_QUERY,
-    CURRENT_USER_ID_VERIFICATION_STATUS_QUERY,
+    AUTH_STATUS_QUERY,
 } from './queries'
 
 interface APIResponse<T> {
@@ -90,13 +90,17 @@ export class SourcegraphGraphQLAPIClient {
         )
     }
 
-    public async getCurrentUserIDAndVerificationStatus(): Promise<{ id: string; hasVerifiedEmail: boolean } | Error> {
+    public async getAuthStatus(): Promise<
+        { id: string; hasVerifiedEmail: boolean; requiresVerifiedEmail: boolean } | Error
+    > {
         return this.fetchSourcegraphAPI<APIResponse<CurrentUserIdVerificationStatusResponse>>(
-            CURRENT_USER_ID_VERIFICATION_STATUS_QUERY,
+            AUTH_STATUS_QUERY,
             {}
         ).then(response =>
             extractDataOrError(response, data =>
-                data.currentUser ? data.currentUser : new Error('current user not found')
+                data.currentUser
+                    ? { ...data.currentUser, requiresVerifiedEmail: data.site.requiresVerifiedEmailForCody }
+                    : new Error('current user not found')
             )
         )
     }

diff --git a/client/cody-shared/src/sourcegraph-api/graphql/queries.ts b/client/cody-shared/src/sourcegraph-api/graphql/queries.ts
@@ -5,8 +5,12 @@ query CurrentUser {
     }
 }`
 
-export const CURRENT_USER_ID_VERIFICATION_STATUS_QUERY = `
+// TODO: We can only query these fields on dotcom ... instrospection query?
+export const AUTH_STATUS_QUERY = `
 query CurrentUser {
+	site {
+		requiresVerifiedEmailForCody
+	}
     currentUser {
         id
 		hasVerifiedEmail