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/statsCommand.test.ts78
-rw-r--r--packages/cli/src/ui/commands/statsCommand.ts63
-rw-r--r--packages/cli/src/ui/hooks/slashCommandProcessor.test.ts81
-rw-r--r--packages/cli/src/ui/hooks/slashCommandProcessor.ts30
4 files changed, 151 insertions, 101 deletions
diff --git a/packages/cli/src/ui/commands/statsCommand.test.ts b/packages/cli/src/ui/commands/statsCommand.test.ts
new file mode 100644
index 00000000..485fcf69
--- /dev/null
+++ b/packages/cli/src/ui/commands/statsCommand.test.ts
@@ -0,0 +1,78 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { vi, describe, it, expect, beforeEach } from 'vitest';
+import { statsCommand } from './statsCommand.js';
+import { type CommandContext } from './types.js';
+import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
+import { MessageType } from '../types.js';
+import { formatDuration } from '../utils/formatters.js';
+
+describe('statsCommand', () => {
+ let mockContext: CommandContext;
+ const startTime = new Date('2025-07-14T10:00:00.000Z');
+ const endTime = new Date('2025-07-14T10:00:30.000Z');
+
+ beforeEach(() => {
+ vi.useFakeTimers();
+ vi.setSystemTime(endTime);
+
+ // 1. Create the mock context with all default values
+ mockContext = createMockCommandContext();
+
+ // 2. Directly set the property on the created mock context
+ mockContext.session.stats.sessionStartTime = startTime;
+ });
+
+ it('should display general session stats when run with no subcommand', () => {
+ if (!statsCommand.action) throw new Error('Command has no action');
+
+ statsCommand.action(mockContext, '');
+
+ const expectedDuration = formatDuration(
+ endTime.getTime() - startTime.getTime(),
+ );
+ expect(mockContext.ui.addItem).toHaveBeenCalledWith(
+ {
+ type: MessageType.STATS,
+ duration: expectedDuration,
+ },
+ expect.any(Number),
+ );
+ });
+
+ it('should display model stats when using the "model" subcommand', () => {
+ const modelSubCommand = statsCommand.subCommands?.find(
+ (sc) => sc.name === 'model',
+ );
+ if (!modelSubCommand?.action) throw new Error('Subcommand has no action');
+
+ modelSubCommand.action(mockContext, '');
+
+ expect(mockContext.ui.addItem).toHaveBeenCalledWith(
+ {
+ type: MessageType.MODEL_STATS,
+ },
+ expect.any(Number),
+ );
+ });
+
+ it('should display tool stats when using the "tools" subcommand', () => {
+ const toolsSubCommand = statsCommand.subCommands?.find(
+ (sc) => sc.name === 'tools',
+ );
+ if (!toolsSubCommand?.action) throw new Error('Subcommand has no action');
+
+ toolsSubCommand.action(mockContext, '');
+
+ expect(mockContext.ui.addItem).toHaveBeenCalledWith(
+ {
+ type: MessageType.TOOL_STATS,
+ },
+ expect.any(Number),
+ );
+ });
+});
diff --git a/packages/cli/src/ui/commands/statsCommand.ts b/packages/cli/src/ui/commands/statsCommand.ts
new file mode 100644
index 00000000..87e902d4
--- /dev/null
+++ b/packages/cli/src/ui/commands/statsCommand.ts
@@ -0,0 +1,63 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { MessageType, HistoryItemStats } from '../types.js';
+import { formatDuration } from '../utils/formatters.js';
+import { type CommandContext, type SlashCommand } from './types.js';
+
+export const statsCommand: SlashCommand = {
+ name: 'stats',
+ altName: 'usage',
+ description: 'check session stats. Usage: /stats [model|tools]',
+ action: (context: CommandContext) => {
+ const now = new Date();
+ const { sessionStartTime } = context.session.stats;
+ if (!sessionStartTime) {
+ context.ui.addItem(
+ {
+ type: MessageType.ERROR,
+ text: 'Session start time is unavailable, cannot calculate stats.',
+ },
+ Date.now(),
+ );
+ return;
+ }
+ const wallDuration = now.getTime() - sessionStartTime.getTime();
+
+ const statsItem: HistoryItemStats = {
+ type: MessageType.STATS,
+ duration: formatDuration(wallDuration),
+ };
+
+ context.ui.addItem(statsItem, Date.now());
+ },
+ subCommands: [
+ {
+ name: 'model',
+ description: 'Show model-specific usage statistics.',
+ action: (context: CommandContext) => {
+ context.ui.addItem(
+ {
+ type: MessageType.MODEL_STATS,
+ },
+ Date.now(),
+ );
+ },
+ },
+ {
+ name: 'tools',
+ description: 'Show tool-specific usage statistics.',
+ action: (context: CommandContext) => {
+ context.ui.addItem(
+ {
+ type: MessageType.TOOL_STATS,
+ },
+ Date.now(),
+ );
+ },
+ },
+ ],
+};
diff --git a/packages/cli/src/ui/hooks/slashCommandProcessor.test.ts b/packages/cli/src/ui/hooks/slashCommandProcessor.test.ts
index 6946bde0..f39795c0 100644
--- a/packages/cli/src/ui/hooks/slashCommandProcessor.test.ts
+++ b/packages/cli/src/ui/hooks/slashCommandProcessor.test.ts
@@ -54,7 +54,16 @@ vi.mock('../../utils/version.js', () => ({
}));
import { act, renderHook } from '@testing-library/react';
-import { vi, describe, it, expect, beforeEach, afterEach, Mock } from 'vitest';
+import {
+ vi,
+ describe,
+ it,
+ expect,
+ beforeEach,
+ afterEach,
+ beforeAll,
+ Mock,
+} from 'vitest';
import open from 'open';
import { useSlashCommandProcessor } from './slashCommandProcessor.js';
import { MessageType, SlashCommandProcessorResult } from '../types.js';
@@ -207,76 +216,6 @@ describe('useSlashCommandProcessor', () => {
const getProcessor = (showToolDescriptions: boolean = false) =>
getProcessorHook(showToolDescriptions).result.current;
- describe('/stats command', () => {
- it('should show detailed session statistics', async () => {
- // Arrange
- mockUseSessionStats.mockReturnValue({
- stats: {
- sessionStartTime: new Date('2025-01-01T00:00:00.000Z'),
- },
- });
-
- const { handleSlashCommand } = getProcessor();
- const mockDate = new Date('2025-01-01T01:02:03.000Z'); // 1h 2m 3s duration
- vi.setSystemTime(mockDate);
-
- // Act
- await act(async () => {
- handleSlashCommand('/stats');
- });
-
- // Assert
- expect(mockAddItem).toHaveBeenNthCalledWith(
- 2, // Called after the user message
- expect.objectContaining({
- type: MessageType.STATS,
- duration: '1h 2m 3s',
- }),
- expect.any(Number),
- );
-
- vi.useRealTimers();
- });
-
- it('should show model-specific statistics when using /stats model', async () => {
- // Arrange
- const { handleSlashCommand } = getProcessor();
-
- // Act
- await act(async () => {
- handleSlashCommand('/stats model');
- });
-
- // Assert
- expect(mockAddItem).toHaveBeenNthCalledWith(
- 2, // Called after the user message
- expect.objectContaining({
- type: MessageType.MODEL_STATS,
- }),
- expect.any(Number),
- );
- });
-
- it('should show tool-specific statistics when using /stats tools', async () => {
- // Arrange
- const { handleSlashCommand } = getProcessor();
-
- // Act
- await act(async () => {
- handleSlashCommand('/stats tools');
- });
-
- // Assert
- expect(mockAddItem).toHaveBeenNthCalledWith(
- 2, // Called after the user message
- expect.objectContaining({
- type: MessageType.TOOL_STATS,
- }),
- expect.any(Number),
- );
- });
- });
-
describe('Other commands', () => {
it('/editor should open editor dialog and return handled', async () => {
const { handleSlashCommand } = getProcessor();
diff --git a/packages/cli/src/ui/hooks/slashCommandProcessor.ts b/packages/cli/src/ui/hooks/slashCommandProcessor.ts
index e8d773b4..31397af5 100644
--- a/packages/cli/src/ui/hooks/slashCommandProcessor.ts
+++ b/packages/cli/src/ui/hooks/slashCommandProcessor.ts
@@ -248,36 +248,6 @@ export const useSlashCommandProcessor = (
action: (_mainCommand, _subCommand, _args) => openEditorDialog(),
},
{
- name: 'stats',
- altName: 'usage',
- description: 'check session stats. Usage: /stats [model|tools]',
- action: (_mainCommand, subCommand, _args) => {
- if (subCommand === 'model') {
- addMessage({
- type: MessageType.MODEL_STATS,
- timestamp: new Date(),
- });
- return;
- } else if (subCommand === 'tools') {
- addMessage({
- type: MessageType.TOOL_STATS,
- timestamp: new Date(),
- });
- return;
- }
-
- const now = new Date();
- const { sessionStartTime } = session.stats;
- const wallDuration = now.getTime() - sessionStartTime.getTime();
-
- addMessage({
- type: MessageType.STATS,
- duration: formatDuration(wallDuration),
- timestamp: new Date(),
- });
- },
- },
- {
name: 'mcp',
description: 'list configured MCP servers and tools',
action: async (_mainCommand, _subCommand, _args) => {
generated by cgit v1.2.3 (git 2.51.0) at 2025-11-08 17:54:48 +0000