summaryrefslogtreecommitdiff
path: root/packages/cli/src/ui
diff options
context:
space:
mode:
Diffstat (limited to 'packages/cli/src/ui')
-rw-r--r--packages/cli/src/ui/commands/authCommand.test.ts36
-rw-r--r--packages/cli/src/ui/commands/authCommand.ts16
-rw-r--r--packages/cli/src/ui/commands/types.ts3
-rw-r--r--packages/cli/src/ui/hooks/slashCommandProcessor.test.ts27
-rw-r--r--packages/cli/src/ui/hooks/slashCommandProcessor.ts10
5 files changed, 84 insertions, 8 deletions
diff --git a/packages/cli/src/ui/commands/authCommand.test.ts b/packages/cli/src/ui/commands/authCommand.test.ts
new file mode 100644
index 00000000..d6d925db
--- /dev/null
+++ b/packages/cli/src/ui/commands/authCommand.test.ts
@@ -0,0 +1,36 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { describe, it, expect, beforeEach } from 'vitest';
+import { authCommand } from './authCommand.js';
+import { type CommandContext } from './types.js';
+import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
+
+describe('authCommand', () => {
+ let mockContext: CommandContext;
+
+ beforeEach(() => {
+ mockContext = createMockCommandContext();
+ });
+
+ it('should return a dialog action to open the auth dialog', () => {
+ if (!authCommand.action) {
+ throw new Error('The auth command must have an action.');
+ }
+
+ const result = authCommand.action(mockContext, '');
+
+ expect(result).toEqual({
+ type: 'dialog',
+ dialog: 'auth',
+ });
+ });
+
+ it('should have the correct name and description', () => {
+ expect(authCommand.name).toBe('auth');
+ expect(authCommand.description).toBe('change the auth method');
+ });
+});
diff --git a/packages/cli/src/ui/commands/authCommand.ts b/packages/cli/src/ui/commands/authCommand.ts
new file mode 100644
index 00000000..29bd2c9d
--- /dev/null
+++ b/packages/cli/src/ui/commands/authCommand.ts
@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { OpenDialogActionReturn, SlashCommand } from './types.js';
+
+export const authCommand: SlashCommand = {
+ name: 'auth',
+ description: 'change the auth method',
+ action: (_context, _args): OpenDialogActionReturn => ({
+ type: 'dialog',
+ dialog: 'auth',
+ }),
+};
diff --git a/packages/cli/src/ui/commands/types.ts b/packages/cli/src/ui/commands/types.ts
index 9aad3399..b35374a8 100644
--- a/packages/cli/src/ui/commands/types.ts
+++ b/packages/cli/src/ui/commands/types.ts
@@ -66,14 +66,13 @@ export interface MessageActionReturn {
export interface OpenDialogActionReturn {
type: 'dialog';
// TODO: Add 'theme' | 'auth' | 'editor' | 'privacy' as migration happens.
- dialog: 'help' | 'theme';
+ dialog: 'help' | 'auth' | 'theme';
}
export type SlashCommandActionReturn =
| ToolActionReturn
| MessageActionReturn
| OpenDialogActionReturn;
-
// The standardized contract for any command in the system.
export interface SlashCommand {
name: string;
diff --git a/packages/cli/src/ui/hooks/slashCommandProcessor.test.ts b/packages/cli/src/ui/hooks/slashCommandProcessor.test.ts
index 563092a2..d920117d 100644
--- a/packages/cli/src/ui/hooks/slashCommandProcessor.test.ts
+++ b/packages/cli/src/ui/hooks/slashCommandProcessor.test.ts
@@ -507,6 +507,33 @@ describe('useSlashCommandProcessor', () => {
expect(commandResult).toEqual({ type: 'handled' });
});
+ it('should open the auth dialog when a new command returns an auth dialog action', async () => {
+ const mockAction = vi.fn().mockResolvedValue({
+ type: 'dialog',
+ dialog: 'auth',
+ });
+ const newAuthCommand: SlashCommand = { name: 'auth', action: mockAction };
+
+ const mockLoader = async () => [newAuthCommand];
+ const commandServiceInstance = new ActualCommandService(mockLoader);
+ vi.mocked(CommandService).mockImplementation(
+ () => commandServiceInstance,
+ );
+
+ const { result } = getProcessorHook();
+ await vi.waitFor(() => {
+ expect(
+ result.current.slashCommands.some((c) => c.name === 'auth'),
+ ).toBe(true);
+ });
+
+ const commandResult = await result.current.handleSlashCommand('/auth');
+
+ expect(mockAction).toHaveBeenCalledTimes(1);
+ expect(mockOpenAuthDialog).toHaveBeenCalledWith();
+ expect(commandResult).toEqual({ type: 'handled' });
+ });
+
it('should open the theme dialog when a new command returns a theme dialog action', async () => {
const mockAction = vi.fn().mockResolvedValue({
type: 'dialog',
diff --git a/packages/cli/src/ui/hooks/slashCommandProcessor.ts b/packages/cli/src/ui/hooks/slashCommandProcessor.ts
index 59f748bf..3a071fab 100644
--- a/packages/cli/src/ui/hooks/slashCommandProcessor.ts
+++ b/packages/cli/src/ui/hooks/slashCommandProcessor.ts
@@ -243,11 +243,6 @@ export const useSlashCommandProcessor = (
},
},
{
- name: 'auth',
- description: 'change the auth method',
- action: (_mainCommand, _subCommand, _args) => openAuthDialog(),
- },
- {
name: 'editor',
description: 'set external editor preference',
action: (_mainCommand, _subCommand, _args) => openEditorDialog(),
@@ -1027,7 +1022,6 @@ export const useSlashCommandProcessor = (
return commands;
}, [
addMessage,
- openAuthDialog,
openEditorDialog,
openPrivacyNotice,
toggleCorgiMode,
@@ -1125,6 +1119,9 @@ export const useSlashCommandProcessor = (
case 'help':
setShowHelp(true);
return { type: 'handled' };
+ case 'auth':
+ openAuthDialog();
+ return { type: 'handled' };
case 'theme':
openThemeDialog();
return { type: 'handled' };
@@ -1205,6 +1202,7 @@ export const useSlashCommandProcessor = (
[
addItem,
setShowHelp,
+ openAuthDialog,
commands,
legacyCommands,
commandContext,