Auth First Run (#1207)

Co-authored-by: Tommaso Sciortino <sciortino@gmail.com> Co-authored-by: N. Taylor Mullen <ntaylormullen@google.com>
2025-12-19 09:33:53 +00:00 · 2025-06-19 16:52:22 -07:00
parent c48fcaa8c3
commit 04518b52c0
37 changed files with 636 additions and 349 deletions
--- a/packages/cli/src/ui/components/AuthDialog.tsx
+++ b/packages/cli/src/ui/components/AuthDialog.tsx
@@ -0,0 +1,94 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import React, { useState } from 'react';
+import { Box, Text, useInput } from 'ink';
+import { Colors } from '../colors.js';
+import { RadioButtonSelect } from './shared/RadioButtonSelect.js';
+import { LoadedSettings, SettingScope } from '../../config/settings.js';
+import { AuthType } from '@gemini-cli/core';
+import { validateAuthMethod } from '../../config/auth.js';
+
+interface AuthDialogProps {
+  onSelect: (authMethod: string | undefined, scope: SettingScope) => void;
+  onHighlight: (authMethod: string | undefined) => void;
+  settings: LoadedSettings;
+  initialErrorMessage?: string | null;
+}
+
+export function AuthDialog({
+  onSelect,
+  onHighlight,
+  settings,
+  initialErrorMessage,
+}: AuthDialogProps): React.JSX.Element {
+  const [errorMessage, setErrorMessage] = useState<string | null>(
+    initialErrorMessage || null,
+  );
+  const authItems = [
+    {
+      label: 'Login with Google Personal Account',
+      value: AuthType.LOGIN_WITH_GOOGLE_PERSONAL,
+    },
+    { label: 'Gemini API Key', value: AuthType.USE_GEMINI },
+    {
+      label: 'Login with GCP Project and Google Work Account',
+      value: AuthType.LOGIN_WITH_GOOGLE_ENTERPRISE,
+    },
+    { label: 'Vertex AI', value: AuthType.USE_VERTEX_AI },
+  ];
+
+  let initialAuthIndex = authItems.findIndex(
+    (item) => item.value === settings.merged.selectedAuthType,
+  );
+
+  if (initialAuthIndex === -1) {
+    initialAuthIndex = 0;
+  }
+
+  const handleAuthSelect = (authMethod: string) => {
+    const error = validateAuthMethod(authMethod);
+    if (error) {
+      setErrorMessage(error);
+    } else {
+      setErrorMessage(null);
+      onSelect(authMethod, SettingScope.User);
+    }
+  };
+
+  useInput((_input, key) => {
+    if (key.escape) {
+      onSelect(undefined, SettingScope.User);
+    }
+  });
+
+  return (
+    <Box
+      borderStyle="round"
+      borderColor={Colors.Gray}
+      flexDirection="column"
+      padding={1}
+      width="100%"
+    >
+      <Text bold>Select Auth Method</Text>
+      <RadioButtonSelect
+        items={authItems}
+        initialIndex={initialAuthIndex}
+        onSelect={handleAuthSelect}
+        onHighlight={onHighlight}
+        isFocused={true}
+      />
+      {errorMessage && (
+        <Box marginTop={1}>
+          <Text color={Colors.AccentRed}>{errorMessage}</Text>
+        </Box>
+      )}
+      <Box marginTop={1}>
+        <Text color={Colors.Gray}>(Use Enter to select)</Text>
+      </Box>
+    </Box>
+  );
+}