cognitive-core 0.2.1 → 0.2.2

package/references/sessionlog/src/agent/agents/gemini-cli.ts

@@ -1,632 +0,0 @@
-/**
- * Gemini CLI Agent
- *
- * Implementation of the Sessionlog agent interface for Google's Gemini CLI.
- * Handles JSON transcript format with messages array, Gemini-specific
- * hook installation, and session lifecycle management.
- */
-
-import * as fs from 'node:fs';
-import * as path from 'node:path';
-import * as os from 'node:os';
-import * as crypto from 'node:crypto';
-import {
-  AGENT_NAMES,
-  AGENT_TYPES,
-  type HookInput,
-  type Event,
-  type TokenUsage,
-  EventType,
-  emptyTokenUsage,
-} from '../../types.js';
-import type {
-  Agent,
-  HookSupport,
-  TranscriptAnalyzer,
-  TokenCalculator,
-  TranscriptChunker,
-} from '../types.js';
-import { registerAgent } from '../registry.js';
-
-// ============================================================================
-// Constants
-// ============================================================================
-
-const GEMINI_DIR = '.gemini';
-const SETTINGS_FILE_NAME = 'settings.json';
-
-const HOOK_NAMES = [
-  'session-start',
-  'session-end',
-  'before-agent',
-  'after-agent',
-  'before-model',
-  'after-model',
-  'before-tool-selection',
-  'before-tool',
-  'after-tool',
-  'pre-compress',
-  'notification',
-] as const;
-
-const SESSIONLOG_HOOK_PREFIX = 'sessionlog ';
-
-/** Tools that modify files in Gemini CLI */
-const FILE_MODIFICATION_TOOLS = new Set(['write_file', 'edit_file', 'save_file', 'replace']);
-
-// ============================================================================
-// Gemini Transcript Types (JSON)
-// ============================================================================
-
-export interface GeminiTranscript {
-  messages: GeminiMessage[];
-}
-
-export interface GeminiMessage {
-  id?: string;
-  type: string;
-  content: string;
-  toolCalls?: GeminiToolCall[];
-}
-
-export interface GeminiToolCall {
-  id: string;
-  name: string;
-  args: Record<string, unknown>;
-  status?: string;
-}
-
-// ============================================================================
-// Gemini Settings Types
-// ============================================================================
-
-interface GeminiHookEntry {
-  name: string;
-  type: string;
-  command: string;
-}
-
-interface GeminiHookMatcher {
-  matcher?: string;
-  hooks: GeminiHookEntry[];
-}
-
-// ============================================================================
-// Gemini CLI Agent Implementation
-// ============================================================================
-
-class GeminiCLIAgent
-  implements Agent, HookSupport, TranscriptAnalyzer, TokenCalculator, TranscriptChunker
-{
-  readonly name = AGENT_NAMES.GEMINI;
-  readonly type = AGENT_TYPES.GEMINI;
-  readonly description = "Gemini CLI - Google's AI coding assistant";
-  readonly isPreview = true;
-  readonly protectedDirs = [GEMINI_DIR];
-
-  async detectPresence(cwd?: string): Promise<boolean> {
-    const repoRoot = cwd ?? process.cwd();
-    const geminiDir = path.join(repoRoot, GEMINI_DIR);
-    try {
-      const stat = await fs.promises.stat(geminiDir);
-      return stat.isDirectory();
-    } catch {
-      return false;
-    }
-  }
-
-  async getSessionDir(repoPath: string): Promise<string> {
-    if (process.env.SESSIONLOG_TEST_GEMINI_PROJECT_DIR) {
-      return process.env.SESSIONLOG_TEST_GEMINI_PROJECT_DIR;
-    }
-    const projectDir = getProjectHash(repoPath);
-    return path.join(os.homedir(), '.gemini', 'tmp', projectDir, 'chats');
-  }
-
-  getSessionID(input: HookInput): string {
-    return input.sessionID;
-  }
-
-  resolveSessionFile(sessionDir: string, agentSessionID: string): string {
-    // Gemini names files as session-<date>-<shortid>.json
-    const shortID = agentSessionID.length > 8 ? agentSessionID.slice(0, 8) : agentSessionID;
-
-    // Try to find existing file
-    try {
-      const files = fs.readdirSync(sessionDir);
-      const pattern = `session-.*-${shortID}\\.json`;
-      const regex = new RegExp(pattern);
-      const matches = files.filter((f) => regex.test(f)).sort();
-      if (matches.length > 0) {
-        return path.join(sessionDir, matches[matches.length - 1]);
-      }
-    } catch {
-      // Directory doesn't exist yet
-    }
-
-    // Fallback: construct filename matching Gemini's convention
-    const timestamp = new Date().toISOString().slice(0, 16).replace(/:/g, '-');
-    return path.join(sessionDir, `session-${timestamp}-${shortID}.json`);
-  }
-
-  async readTranscript(sessionRef: string): Promise<Buffer> {
-    return fs.promises.readFile(sessionRef);
-  }
-
-  formatResumeCommand(sessionID: string): string {
-    return `gemini --resume ${sessionID}`;
-  }
-
-  // ===========================================================================
-  // HookSupport
-  // ===========================================================================
-
-  hookNames(): string[] {
-    return [...HOOK_NAMES];
-  }
-
-  parseHookEvent(hookName: string, stdin: string): Event | null {
-    try {
-      const data = JSON.parse(stdin) as Record<string, unknown>;
-
-      const sessionID = String(data.session_id ?? data.sessionID ?? '');
-      const sessionRef = String(data.transcript_path ?? data.transcriptPath ?? '');
-
-      switch (hookName) {
-        case 'session-start':
-          return {
-            type: EventType.SessionStart,
-            sessionID,
-            sessionRef,
-            timestamp: new Date(),
-          };
-
-        case 'before-agent':
-          return {
-            type: EventType.TurnStart,
-            sessionID,
-            sessionRef,
-            prompt: String(data.prompt ?? ''),
-            timestamp: new Date(),
-          };
-
-        case 'after-agent':
-          return {
-            type: EventType.TurnEnd,
-            sessionID,
-            sessionRef,
-            timestamp: new Date(),
-          };
-
-        case 'session-end':
-          return {
-            type: EventType.SessionEnd,
-            sessionID,
-            sessionRef,
-            timestamp: new Date(),
-          };
-
-        case 'pre-compress':
-          return {
-            type: EventType.Compaction,
-            sessionID,
-            sessionRef,
-            timestamp: new Date(),
-          };
-
-        // Pass-through hooks with no lifecycle action
-        case 'before-tool':
-        case 'after-tool':
-        case 'before-model':
-        case 'after-model':
-        case 'before-tool-selection':
-        case 'notification':
-          return null;
-
-        default:
-          return null;
-      }
-    } catch {
-      return null;
-    }
-  }
-
-  async installHooks(repoPath: string, force = false): Promise<number> {
-    const settingsPath = path.join(repoPath, GEMINI_DIR, SETTINGS_FILE_NAME);
-    let rawSettings: Record<string, unknown> = {};
-
-    try {
-      const content = await fs.promises.readFile(settingsPath, 'utf-8');
-      rawSettings = JSON.parse(content) as Record<string, unknown>;
-    } catch {
-      // No existing settings
-    }
-
-    // Ensure hooksConfig.enabled is true
-    const hooksConfig = (rawSettings.hooksConfig ?? {}) as Record<string, unknown>;
-    hooksConfig.enabled = true;
-    rawSettings.hooksConfig = hooksConfig;
-
-    const hooks = (rawSettings.hooks ?? {}) as Record<string, GeminiHookMatcher[]>;
-
-    const hookDefs = [
-      {
-        key: 'SessionStart',
-        hookName: 'session-start',
-        matcher: '',
-        name: 'sessionlog-session-start',
-      },
-      {
-        key: 'SessionEnd',
-        hookName: 'session-end',
-        matcher: 'exit',
-        name: 'sessionlog-session-end-exit',
-      },
-      {
-        key: 'SessionEnd',
-        hookName: 'session-end',
-        matcher: 'logout',
-        name: 'sessionlog-session-end-logout',
-      },
-      {
-        key: 'BeforeAgent',
-        hookName: 'before-agent',
-        matcher: '',
-        name: 'sessionlog-before-agent',
-      },
-      { key: 'AfterAgent', hookName: 'after-agent', matcher: '', name: 'sessionlog-after-agent' },
-      {
-        key: 'BeforeModel',
-        hookName: 'before-model',
-        matcher: '',
-        name: 'sessionlog-before-model',
-      },
-      { key: 'AfterModel', hookName: 'after-model', matcher: '', name: 'sessionlog-after-model' },
-      {
-        key: 'BeforeToolSelection',
-        hookName: 'before-tool-selection',
-        matcher: '',
-        name: 'sessionlog-before-tool-selection',
-      },
-      { key: 'BeforeTool', hookName: 'before-tool', matcher: '*', name: 'sessionlog-before-tool' },
-      { key: 'AfterTool', hookName: 'after-tool', matcher: '*', name: 'sessionlog-after-tool' },
-      {
-        key: 'PreCompress',
-        hookName: 'pre-compress',
-        matcher: '',
-        name: 'sessionlog-pre-compress',
-      },
-      {
-        key: 'Notification',
-        hookName: 'notification',
-        matcher: '',
-        name: 'sessionlog-notification',
-      },
-    ];
-
-    // If not force, check idempotency
-    if (!force) {
-      const sessionStartMatchers = hooks['SessionStart'] ?? [];
-      for (const m of sessionStartMatchers) {
-        for (const h of m.hooks) {
-          if (h.command === `sessionlog hooks gemini session-start`) {
-            return 0; // Already installed
-          }
-        }
-      }
-    }
-
-    // Remove existing sessionlog hooks
-    for (const key of Object.keys(hooks)) {
-      hooks[key] = removeSessionlogGeminiHooks(hooks[key] ?? []);
-      if (hooks[key].length === 0) delete hooks[key];
-    }
-
-    // Install all hooks
-    for (const def of hookDefs) {
-      const matchers = hooks[def.key] ?? [];
-      const cmd = `sessionlog hooks gemini ${def.hookName}`;
-      const entry: GeminiHookEntry = { name: def.name, type: 'command', command: cmd };
-
-      // Find or create matcher
-      let found = false;
-      for (const m of matchers) {
-        if ((m.matcher ?? '') === def.matcher) {
-          m.hooks.push(entry);
-          found = true;
-          break;
-        }
-      }
-      if (!found) {
-        const newMatcher: GeminiHookMatcher = { hooks: [entry] };
-        if (def.matcher) newMatcher.matcher = def.matcher;
-        matchers.push(newMatcher);
-      }
-      hooks[def.key] = matchers;
-    }
-
-    rawSettings.hooks = hooks;
-
-    const dir = path.dirname(settingsPath);
-    await fs.promises.mkdir(dir, { recursive: true });
-    await fs.promises.writeFile(settingsPath, JSON.stringify(rawSettings, null, 2));
-
-    return 12; // Total hooks installed
-  }
-
-  async uninstallHooks(repoPath: string): Promise<void> {
-    const settingsPath = path.join(repoPath, GEMINI_DIR, SETTINGS_FILE_NAME);
-
-    try {
-      const content = await fs.promises.readFile(settingsPath, 'utf-8');
-      const rawSettings = JSON.parse(content) as Record<string, unknown>;
-      const hooks = (rawSettings.hooks ?? {}) as Record<string, GeminiHookMatcher[]>;
-
-      for (const key of Object.keys(hooks)) {
-        hooks[key] = removeSessionlogGeminiHooks(hooks[key] ?? []);
-        if (hooks[key].length === 0) delete hooks[key];
-      }
-
-      if (Object.keys(hooks).length === 0) {
-        delete rawSettings.hooks;
-      } else {
-        rawSettings.hooks = hooks;
-      }
-
-      await fs.promises.writeFile(settingsPath, JSON.stringify(rawSettings, null, 2));
-    } catch {
-      // No file to modify
-    }
-  }
-
-  async areHooksInstalled(repoPath: string): Promise<boolean> {
-    const settingsPath = path.join(repoPath, GEMINI_DIR, SETTINGS_FILE_NAME);
-
-    try {
-      const content = await fs.promises.readFile(settingsPath, 'utf-8');
-      const rawSettings = JSON.parse(content) as Record<string, unknown>;
-      const hooks = (rawSettings.hooks ?? {}) as Record<string, GeminiHookMatcher[]>;
-
-      for (const key of Object.keys(hooks)) {
-        for (const m of hooks[key] ?? []) {
-          for (const h of m.hooks) {
-            if (h.command.startsWith(SESSIONLOG_HOOK_PREFIX)) return true;
-          }
-        }
-      }
-    } catch {
-      // No file
-    }
-
-    return false;
-  }
-
-  // ===========================================================================
-  // TranscriptAnalyzer
-  // ===========================================================================
-
-  async getTranscriptPosition(transcriptPath: string): Promise<number> {
-    try {
-      const data = await fs.promises.readFile(transcriptPath, 'utf-8');
-      const transcript = JSON.parse(data) as GeminiTranscript;
-      return transcript.messages?.length ?? 0;
-    } catch {
-      return 0;
-    }
-  }
-
-  async extractModifiedFilesFromOffset(
-    transcriptPath: string,
-    startOffset: number,
-  ): Promise<{ files: string[]; currentPosition: number }> {
-    const data = await fs.promises.readFile(transcriptPath, 'utf-8');
-    const transcript = parseGeminiTranscript(data);
-    const files = new Set<string>();
-
-    for (let i = startOffset; i < transcript.messages.length; i++) {
-      const msg = transcript.messages[i];
-      if (msg.type !== 'gemini') continue;
-
-      for (const tc of msg.toolCalls ?? []) {
-        if (!FILE_MODIFICATION_TOOLS.has(tc.name)) continue;
-        const filePath = extractGeminiFilePath(tc.args);
-        if (filePath) files.add(filePath);
-      }
-    }
-
-    return { files: Array.from(files), currentPosition: transcript.messages.length };
-  }
-
-  async extractPrompts(sessionRef: string, fromOffset: number): Promise<string[]> {
-    const data = await fs.promises.readFile(sessionRef, 'utf-8');
-    const transcript = parseGeminiTranscript(data);
-    const prompts: string[] = [];
-
-    for (let i = fromOffset; i < transcript.messages.length; i++) {
-      const msg = transcript.messages[i];
-      if (msg.type === 'user' && msg.content) {
-        prompts.push(msg.content);
-      }
-    }
-
-    return prompts;
-  }
-
-  async extractSummary(sessionRef: string): Promise<string> {
-    const data = await fs.promises.readFile(sessionRef, 'utf-8');
-    const transcript = parseGeminiTranscript(data);
-
-    for (let i = transcript.messages.length - 1; i >= 0; i--) {
-      if (transcript.messages[i].type === 'gemini' && transcript.messages[i].content) {
-        return transcript.messages[i].content;
-      }
-    }
-
-    return '';
-  }
-
-  // ===========================================================================
-  // TokenCalculator
-  // ===========================================================================
-
-  async calculateTokenUsage(transcriptData: Buffer, fromOffset: number): Promise<TokenUsage> {
-    const data = transcriptData.toString('utf-8');
-    const usage = emptyTokenUsage();
-
-    try {
-      const parsed = JSON.parse(data) as {
-        messages: Array<{
-          type: string;
-          tokens?: { input: number; output: number; cached: number };
-        }>;
-      };
-
-      for (let i = fromOffset; i < parsed.messages.length; i++) {
-        const msg = parsed.messages[i];
-        if (msg.type !== 'gemini' || !msg.tokens) continue;
-
-        usage.apiCallCount++;
-        usage.inputTokens += msg.tokens.input ?? 0;
-        usage.outputTokens += msg.tokens.output ?? 0;
-        usage.cacheReadTokens += msg.tokens.cached ?? 0;
-      }
-    } catch {
-      // Invalid transcript data
-    }
-
-    return usage;
-  }
-
-  // ===========================================================================
-  // TranscriptChunker
-  // ===========================================================================
-
-  async chunkTranscript(content: Buffer, maxSize: number): Promise<Buffer[]> {
-    try {
-      const transcript = JSON.parse(content.toString('utf-8')) as GeminiTranscript;
-      if (!transcript.messages?.length) return [content];
-
-      const chunks: Buffer[] = [];
-      let currentMessages: GeminiMessage[] = [];
-      const baseSize = Buffer.byteLength('{"messages":[]}');
-      let currentSize = baseSize;
-
-      for (const msg of transcript.messages) {
-        const msgBytes = Buffer.byteLength(JSON.stringify(msg));
-        const msgSize = msgBytes + 1; // +1 for comma
-
-        if (currentSize + msgSize > maxSize && currentMessages.length > 0) {
-          chunks.push(Buffer.from(JSON.stringify({ messages: currentMessages })));
-          currentMessages = [];
-          currentSize = baseSize;
-        }
-
-        currentMessages.push(msg);
-        currentSize += msgSize;
-      }
-
-      if (currentMessages.length > 0) {
-        chunks.push(Buffer.from(JSON.stringify({ messages: currentMessages })));
-      }
-
-      return chunks.length > 0 ? chunks : [content];
-    } catch {
-      // Fall back to raw content
-      return [content];
-    }
-  }
-
-  async reassembleTranscript(chunks: Buffer[]): Promise<Buffer> {
-    const allMessages: GeminiMessage[] = [];
-
-    for (const chunk of chunks) {
-      const transcript = JSON.parse(chunk.toString('utf-8')) as GeminiTranscript;
-      allMessages.push(...(transcript.messages ?? []));
-    }
-
-    return Buffer.from(JSON.stringify({ messages: allMessages }));
-  }
-}
-
-// ============================================================================
-// Helpers
-// ============================================================================
-
-function getProjectHash(projectRoot: string): string {
-  return crypto.createHash('sha256').update(projectRoot).digest('hex');
-}
-
-function removeSessionlogGeminiHooks(matchers: GeminiHookMatcher[]): GeminiHookMatcher[] {
-  const result: GeminiHookMatcher[] = [];
-  for (const m of matchers) {
-    const filteredHooks = m.hooks.filter((h) => !h.command.startsWith(SESSIONLOG_HOOK_PREFIX));
-    if (filteredHooks.length > 0) {
-      result.push({ ...m, hooks: filteredHooks });
-    }
-  }
-  return result;
-}
-
-function extractGeminiFilePath(args: Record<string, unknown>): string {
-  if (typeof args.file_path === 'string' && args.file_path) return args.file_path;
-  if (typeof args.path === 'string' && args.path) return args.path;
-  if (typeof args.filename === 'string' && args.filename) return args.filename;
-  return '';
-}
-
-/**
- * Parse Gemini JSON transcript, handling both string and array content formats.
- */
-function parseGeminiTranscript(data: string): GeminiTranscript {
-  const raw = JSON.parse(data) as { messages: Array<Record<string, unknown>> };
-  const messages: GeminiMessage[] = [];
-
-  for (const rawMsg of raw.messages ?? []) {
-    const msg: GeminiMessage = {
-      id: rawMsg.id as string | undefined,
-      type: rawMsg.type as string,
-      content: '',
-      toolCalls: rawMsg.toolCalls as GeminiToolCall[] | undefined,
-    };
-
-    // Handle both string and array content formats
-    const content = rawMsg.content;
-    if (typeof content === 'string') {
-      msg.content = content;
-    } else if (Array.isArray(content)) {
-      const texts: string[] = [];
-      for (const part of content) {
-        if (
-          part &&
-          typeof part === 'object' &&
-          typeof (part as Record<string, unknown>).text === 'string'
-        ) {
-          texts.push((part as Record<string, unknown>).text as string);
-        }
-      }
-      msg.content = texts.join('\n');
-    }
-
-    messages.push(msg);
-  }
-
-  return { messages };
-}
-
-// ============================================================================
-// Exported Helpers (for summarize module)
-// ============================================================================
-
-export { parseGeminiTranscript, FILE_MODIFICATION_TOOLS as GEMINI_FILE_MODIFICATION_TOOLS };
-
-export type { GeminiTranscript as ParsedGeminiTranscript };
-
-// ============================================================================
-// Registration
-// ============================================================================
-
-export function createGeminiCLIAgent(): GeminiCLIAgent {
-  return new GeminiCLIAgent();
-}
-
-registerAgent(AGENT_NAMES.GEMINI, () => new GeminiCLIAgent());