fss-link 1.0.40 → 1.0.45

package/dist/ui/components/SearchEngineConfigDialog.tsx

@@ -0,0 +1,280 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import React, { useState, useEffect } from 'react';
+import { Box, Text, useInput } from 'ink';
+import { Colors } from '../colors.js';
+import { getFSSLinkDatabase } from '../../config/database.js';
+
+interface SearchEngineConfigDialogProps {
+  onSubmit: (config: SearchEngineConfigResult) => void;
+  onCancel: () => void;
+}
+
+export interface SearchEngineConfigResult {
+  braveApiKey?: string;
+  braveEnabled: boolean;
+  tavilyApiKey?: string;
+  tavilyEnabled: boolean;
+}
+
+type FieldType = 'braveApiKey' | 'tavilyApiKey' | 'submit';
+
+export function SearchEngineConfigDialog({
+  onSubmit,
+  onCancel,
+}: SearchEngineConfigDialogProps): React.JSX.Element {
+  const [braveApiKey, setBraveApiKey] = useState('');
+  const [tavilyApiKey, setTavilyApiKey] = useState('');
+  const [currentField, setCurrentField] = useState<FieldType>('braveApiKey');
+  const [isLoading, setIsLoading] = useState(true);
+
+  // Load existing configuration from database
+  useEffect(() => {
+    const loadConfig = async () => {
+      const db = await getFSSLinkDatabase();
+      const existingBrave = await db.getUserPreference('webscraper.brave_api_key');
+      const existingTavily = await db.getUserPreference('webscraper.tavily_api_key');
+      
+      if (existingBrave) {
+        setBraveApiKey(existingBrave);
+      }
+      if (existingTavily) {
+        setTavilyApiKey(existingTavily);
+      }
+      
+      setIsLoading(false);
+    };
+    
+    loadConfig().catch(console.error);
+  }, []);
+
+  useInput((input, key) => {
+    if (isLoading) return;
+
+    // Clean input similar to OpenAIKeyPrompt
+    let cleanInput = (input || '')
+      // eslint-disable-next-line no-control-regex
+      .replace(/\u001b\[[0-9;]*[a-zA-Z]/g, '') // Filter ESC sequences
+      .replace(/\[200~/g, '') // Filter paste start marker
+      .replace(/\[201~/g, '') // Filter paste end marker
+      .replace(/^\[|~$/g, ''); // Filter bracket/tilde residuals
+
+    // Filter non-printable characters (ASCII < 32)
+    cleanInput = cleanInput
+      .split('')
+      .filter((ch) => ch.charCodeAt(0) >= 32)
+      .join('');
+
+    if (cleanInput.length > 0) {
+      if (currentField === 'braveApiKey') {
+        setBraveApiKey((prev) => prev + cleanInput);
+      } else if (currentField === 'tavilyApiKey') {
+        setTavilyApiKey((prev) => prev + cleanInput);
+      }
+      return;
+    }
+
+    // Handle Enter key
+    if (input.includes('\n') || input.includes('\r')) {
+      if (currentField === 'braveApiKey') {
+        setCurrentField('tavilyApiKey');
+        return;
+      } else if (currentField === 'tavilyApiKey') {
+        setCurrentField('submit');
+        return;
+      } else if (currentField === 'submit') {
+        // Submit the configuration
+        const hasAtLeastOne = braveApiKey.trim() || tavilyApiKey.trim();
+        if (!hasAtLeastOne) {
+          // If no API keys provided, go back to brave field
+          setCurrentField('braveApiKey');
+          return;
+        }
+        
+        // Save to database and set environment variables
+        const saveConfig = async () => {
+          const db = await getFSSLinkDatabase();
+          if (braveApiKey.trim()) {
+            await db.setUserPreference('webscraper.brave_api_key', braveApiKey.trim());
+            process.env['BRAVE_SEARCH_API_KEY'] = braveApiKey.trim();
+          }
+          if (tavilyApiKey.trim()) {
+            await db.setUserPreference('webscraper.tavily_api_key', tavilyApiKey.trim());
+            process.env['TAVILY_API_KEY'] = tavilyApiKey.trim();
+          }
+        };
+        
+        saveConfig().catch(console.error);
+        
+        onSubmit({
+          braveApiKey: braveApiKey.trim() || undefined,
+          braveEnabled: !!braveApiKey.trim(),
+          tavilyApiKey: tavilyApiKey.trim() || undefined,
+          tavilyEnabled: !!tavilyApiKey.trim(),
+        });
+      }
+      return;
+    }
+
+    if (key.escape) {
+      onCancel();
+      return;
+    }
+
+    // Handle Tab key for field navigation
+    if (key.tab) {
+      if (currentField === 'braveApiKey') {
+        setCurrentField('tavilyApiKey');
+      } else if (currentField === 'tavilyApiKey') {
+        setCurrentField('submit');
+      } else if (currentField === 'submit') {
+        setCurrentField('braveApiKey');
+      }
+      return;
+    }
+
+    // Handle arrow keys for field navigation
+    if (key.upArrow) {
+      if (currentField === 'tavilyApiKey') {
+        setCurrentField('braveApiKey');
+      } else if (currentField === 'submit') {
+        setCurrentField('tavilyApiKey');
+      }
+      return;
+    }
+
+    if (key.downArrow) {
+      if (currentField === 'braveApiKey') {
+        setCurrentField('tavilyApiKey');
+      } else if (currentField === 'tavilyApiKey') {
+        setCurrentField('submit');
+      }
+      return;
+    }
+
+    // Handle backspace
+    if (key.backspace || key.delete) {
+      if (currentField === 'braveApiKey') {
+        setBraveApiKey((prev) => prev.slice(0, -1));
+      } else if (currentField === 'tavilyApiKey') {
+        setTavilyApiKey((prev) => prev.slice(0, -1));
+      }
+      return;
+    }
+  });
+
+  if (isLoading) {
+    return (
+      <Box
+        borderStyle="round"
+        borderColor={Colors.AccentBlue}
+        flexDirection="column"
+        padding={1}
+        width="100%"
+      >
+        <Text>Loading search engine configuration...</Text>
+      </Box>
+    );
+  }
+
+  return (
+    <Box
+      borderStyle="round"
+      borderColor={Colors.AccentPurple}
+      flexDirection="column"
+      padding={1}
+      width="100%"
+    >
+      <Text bold color={Colors.AccentPurple}>
+        Search Engine Configuration
+      </Text>
+      <Box marginTop={1}>
+        <Text>
+          Configure search engine APIs for web research. You need at least one API key.
+        </Text>
+      </Box>
+      
+      <Box marginTop={1}>
+        <Text bold color={Colors.AccentBlue}>Brave Search (General Web Search)</Text>
+      </Box>
+      <Box marginTop={1}>
+        <Text color={Colors.Gray}>
+          Get API key: <Text color={Colors.AccentBlue}>https://api.search.brave.com/</Text>
+        </Text>
+      </Box>
+      <Box marginTop={1} flexDirection="row">
+        <Box width={14}>
+          <Text
+            color={currentField === 'braveApiKey' ? Colors.AccentPurple : Colors.Gray}
+          >
+            Brave API Key:
+          </Text>
+        </Box>
+        <Box flexGrow={1}>
+          <Text>
+            {currentField === 'braveApiKey' ? '> ' : '  '}
+            {braveApiKey ? '•'.repeat(Math.min(braveApiKey.length, 20)) + (braveApiKey.length > 20 ? '...' : '') : 'optional'}
+          </Text>
+        </Box>
+      </Box>
+
+      <Box marginTop={1}>
+        <Text bold color={Colors.AccentBlue}>Tavily Search (Research Focus)</Text>
+      </Box>
+      <Box marginTop={1}>
+        <Text color={Colors.Gray}>
+          Get API key: <Text color={Colors.AccentBlue}>https://tavily.com/</Text>
+        </Text>
+      </Box>
+      <Box marginTop={1} flexDirection="row">
+        <Box width={14}>
+          <Text
+            color={currentField === 'tavilyApiKey' ? Colors.AccentPurple : Colors.Gray}
+          >
+            Tavily API Key:
+          </Text>
+        </Box>
+        <Box flexGrow={1}>
+          <Text>
+            {currentField === 'tavilyApiKey' ? '> ' : '  '}
+            {tavilyApiKey ? '•'.repeat(Math.min(tavilyApiKey.length, 20)) + (tavilyApiKey.length > 20 ? '...' : '') : 'optional'}
+          </Text>
+        </Box>
+      </Box>
+
+      <Box marginTop={1} flexDirection="row">
+        <Box width={14}>
+          <Text
+            color={currentField === 'submit' ? Colors.AccentGreen : Colors.Gray}
+          >
+            Ready:
+          </Text>
+        </Box>
+        <Box flexGrow={1}>
+          <Text color={currentField === 'submit' ? Colors.AccentGreen : Colors.Gray}>
+            {currentField === 'submit' ? '> ' : '  '}
+            Save & Continue
+          </Text>
+        </Box>
+      </Box>
+
+      <Box marginTop={1}>
+        <Text color={Colors.Gray}>
+          Press Enter to continue, Tab/↑↓ to navigate, Esc to cancel
+        </Text>
+      </Box>
+      
+      {!braveApiKey && !tavilyApiKey && (
+        <Box marginTop={1}>
+          <Text color={Colors.AccentYellow}>
+            ⚠️  At least one API key is required for search functionality
+          </Text>
+        </Box>
+      )}
+    </Box>
+  );
+}

package/dist/ui/components/SessionSummaryDisplay.test.tsx

@@ -0,0 +1,75 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { render } from 'ink-testing-library';
+import { describe, it, expect, vi } from 'vitest';
+import { SessionSummaryDisplay } from './SessionSummaryDisplay.js';
+import * as SessionContext from '../contexts/SessionContext.js';
+import { SessionMetrics } from '../contexts/SessionContext.js';
+
+vi.mock('../contexts/SessionContext.js', async (importOriginal) => {
+  const actual = await importOriginal<typeof SessionContext>();
+  return {
+    ...actual,
+    useSessionStats: vi.fn(),
+  };
+});
+
+const useSessionStatsMock = vi.mocked(SessionContext.useSessionStats);
+
+const renderWithMockedStats = (metrics: SessionMetrics) => {
+  useSessionStatsMock.mockReturnValue({
+    stats: {
+      sessionStartTime: new Date(),
+      metrics,
+      lastPromptTokenCount: 0,
+      promptCount: 5,
+    },
+
+    getPromptCount: () => 5,
+    startNewPrompt: vi.fn(),
+  });
+
+  return render(<SessionSummaryDisplay duration="1h 23m 45s" />);
+};
+
+describe('<SessionSummaryDisplay />', () => {
+  it('renders the summary display with a title', () => {
+    const metrics: SessionMetrics = {
+      models: {
+        'gemini-2.5-pro': {
+          api: { totalRequests: 10, totalErrors: 1, totalLatencyMs: 50234 },
+          tokens: {
+            prompt: 1000,
+            candidates: 2000,
+            total: 3500,
+            cached: 500,
+            thoughts: 300,
+            tool: 200,
+          },
+        },
+      },
+      tools: {
+        totalCalls: 0,
+        totalSuccess: 0,
+        totalFail: 0,
+        totalDurationMs: 0,
+        totalDecisions: { accept: 0, reject: 0, modify: 0 },
+        byName: {},
+      },
+      files: {
+        totalLinesAdded: 42,
+        totalLinesRemoved: 15,
+      },
+    };
+
+    const { lastFrame } = renderWithMockedStats(metrics);
+    const output = lastFrame();
+
+    expect(output).toContain('Agent powering down. Goodbye!');
+    expect(output).toMatchSnapshot();
+  });
+});

package/dist/ui/components/SessionSummaryDisplay.tsx

@@ -0,0 +1,18 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import React from 'react';
+import { StatsDisplay } from './StatsDisplay.js';
+
+interface SessionSummaryDisplayProps {
+  duration: string;
+}
+
+export const SessionSummaryDisplay: React.FC<SessionSummaryDisplayProps> = ({
+  duration,
+}) => (
+  <StatsDisplay title="Agent powering down. Goodbye!" duration={duration} />
+);