cognitive-core 0.2.1 → 0.2.2

package/references/sessionlog/src/agent/agents/opencode.ts

@@ -1,540 +0,0 @@
-/**
- * OpenCode Agent
- *
- * Implementation of the Sessionlog agent interface for OpenCode.
- * Handles JSON export format with messages array, OpenCode-specific
- * plugin-based hook installation, and session lifecycle management.
- *
- * OpenCode uses `opencode export` CLI command for transcript access.
- * Sessions are stored in JSON format: { info: {...}, messages: [...] }
- */
-
-import * as fs from 'node:fs';
-import * as path from 'node:path';
-import * as os from 'node:os';
-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 OPENCODE_DIR = '.opencode';
-
-const HOOK_NAMES = [
-  'session-start',
-  'session-end',
-  'turn-start',
-  'turn-end',
-  'compaction',
-] as const;
-
-/** Tools that modify files in OpenCode */
-const FILE_MODIFICATION_TOOLS = new Set(['edit', 'write', 'patch']);
-
-const SESSIONLOG_PLUGIN_MARKER = 'Auto-generated by `sessionlog enable --agent opencode`';
-
-// ============================================================================
-// OpenCode Export JSON Types
-// ============================================================================
-
-export interface ExportSession {
-  info: SessionInfo;
-  messages: ExportMessage[];
-}
-
-export interface SessionInfo {
-  id: string;
-  title?: string;
-  createdAt?: number;
-  updatedAt?: number;
-}
-
-export interface ExportMessage {
-  info: MessageInfo;
-  parts: Part[];
-}
-
-export interface MessageInfo {
-  id: string;
-  sessionID?: string;
-  role: string;
-  time: { created: number; completed?: number };
-  tokens?: {
-    input: number;
-    output: number;
-    reasoning: number;
-    cache: { read: number; write: number };
-  };
-  cost?: number;
-}
-
-export interface Part {
-  type: string;
-  text?: string;
-  tool?: string;
-  callID?: string;
-  state?: ToolState;
-}
-
-export interface ToolState {
-  status: string;
-  input?: Record<string, unknown>;
-  output?: string;
-}
-
-// ============================================================================
-// OpenCode Agent Implementation
-// ============================================================================
-
-class OpenCodeAgent
-  implements Agent, HookSupport, TranscriptAnalyzer, TokenCalculator, TranscriptChunker
-{
-  readonly name = AGENT_NAMES.OPENCODE;
-  readonly type = AGENT_TYPES.OPENCODE;
-  readonly description = 'OpenCode - AI-powered terminal coding agent';
-  readonly isPreview = true;
-  readonly protectedDirs = [OPENCODE_DIR];
-
-  async detectPresence(cwd?: string): Promise<boolean> {
-    const repoRoot = cwd ?? process.cwd();
-    try {
-      await fs.promises.stat(path.join(repoRoot, OPENCODE_DIR));
-      return true;
-    } catch {
-      try {
-        await fs.promises.stat(path.join(repoRoot, 'opencode.json'));
-        return true;
-      } catch {
-        return false;
-      }
-    }
-  }
-
-  async getSessionDir(repoPath: string): Promise<string> {
-    if (process.env.SESSIONLOG_TEST_OPENCODE_PROJECT_DIR) {
-      return process.env.SESSIONLOG_TEST_OPENCODE_PROJECT_DIR;
-    }
-    const projectDir = sanitizePathForOpenCode(repoPath);
-    return path.join(os.tmpdir(), 'sessionlog-opencode', projectDir);
-  }
-
-  getSessionID(input: HookInput): string {
-    return input.sessionID;
-  }
-
-  resolveSessionFile(sessionDir: string, agentSessionID: string): string {
-    return path.join(sessionDir, `${agentSessionID}.json`);
-  }
-
-  async readTranscript(sessionRef: string): Promise<Buffer> {
-    return fs.promises.readFile(sessionRef);
-  }
-
-  formatResumeCommand(sessionID: string): string {
-    if (!sessionID.trim()) return 'opencode';
-    return `opencode -s ${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 ?? '');
-
-      switch (hookName) {
-        case 'session-start':
-          return {
-            type: EventType.SessionStart,
-            sessionID,
-            sessionRef: '',
-            timestamp: new Date(),
-          };
-
-        case 'turn-start':
-          return {
-            type: EventType.TurnStart,
-            sessionID,
-            sessionRef: '',
-            prompt: String(data.prompt ?? ''),
-            timestamp: new Date(),
-          };
-
-        case 'turn-end':
-          return {
-            type: EventType.TurnEnd,
-            sessionID,
-            sessionRef: '',
-            timestamp: new Date(),
-          };
-
-        case 'compaction':
-          return {
-            type: EventType.Compaction,
-            sessionID,
-            sessionRef: '',
-            timestamp: new Date(),
-          };
-
-        case 'session-end':
-          return {
-            type: EventType.SessionEnd,
-            sessionID,
-            sessionRef: '',
-            timestamp: new Date(),
-          };
-
-        default:
-          return null;
-      }
-    } catch {
-      return null;
-    }
-  }
-
-  async installHooks(repoPath: string, force = false): Promise<number> {
-    const pluginPath = path.join(repoPath, OPENCODE_DIR, 'plugins', 'sessionlog.ts');
-
-    // Check if already installed
-    if (!force) {
-      try {
-        const content = await fs.promises.readFile(pluginPath, 'utf-8');
-        if (content.includes(SESSIONLOG_PLUGIN_MARKER)) return 0;
-      } catch {
-        // File doesn't exist, proceed with install
-      }
-    }
-
-    // Create plugin directory
-    await fs.promises.mkdir(path.dirname(pluginPath), { recursive: true });
-
-    // Write the plugin file
-    const pluginContent = generateOpenCodePlugin();
-    await fs.promises.writeFile(pluginPath, pluginContent, 'utf-8');
-
-    return 5; // 5 lifecycle hooks
-  }
-
-  async uninstallHooks(repoPath: string): Promise<void> {
-    const pluginPath = path.join(repoPath, OPENCODE_DIR, 'plugins', 'sessionlog.ts');
-    try {
-      const content = await fs.promises.readFile(pluginPath, 'utf-8');
-      if (content.includes(SESSIONLOG_PLUGIN_MARKER)) {
-        await fs.promises.unlink(pluginPath);
-      }
-    } catch {
-      // File doesn't exist
-    }
-  }
-
-  async areHooksInstalled(repoPath: string): Promise<boolean> {
-    const pluginPath = path.join(repoPath, OPENCODE_DIR, 'plugins', 'sessionlog.ts');
-    try {
-      const content = await fs.promises.readFile(pluginPath, 'utf-8');
-      return content.includes(SESSIONLOG_PLUGIN_MARKER);
-    } catch {
-      return false;
-    }
-  }
-
-  // ===========================================================================
-  // TranscriptAnalyzer
-  // ===========================================================================
-
-  async getTranscriptPosition(transcriptPath: string): Promise<number> {
-    try {
-      const data = await fs.promises.readFile(transcriptPath, 'utf-8');
-      const session = JSON.parse(data) as ExportSession;
-      return session.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 session = JSON.parse(data) as ExportSession;
-    const files = new Set<string>();
-
-    for (let i = startOffset; i < session.messages.length; i++) {
-      const msg = session.messages[i];
-      if (msg.info.role !== 'assistant') continue;
-
-      for (const part of msg.parts) {
-        if (part.type !== 'tool' || !part.state) continue;
-        if (!FILE_MODIFICATION_TOOLS.has(part.tool ?? '')) continue;
-
-        const filePath = extractOpenCodeFilePath(part.state.input ?? {});
-        if (filePath) files.add(filePath);
-      }
-    }
-
-    return { files: Array.from(files), currentPosition: session.messages.length };
-  }
-
-  async extractPrompts(sessionRef: string, fromOffset: number): Promise<string[]> {
-    const data = await fs.promises.readFile(sessionRef, 'utf-8');
-    const session = JSON.parse(data) as ExportSession;
-    const prompts: string[] = [];
-
-    for (let i = fromOffset; i < session.messages.length; i++) {
-      const msg = session.messages[i];
-      if (msg.info.role !== 'user') continue;
-      const content = extractTextFromParts(msg.parts);
-      if (content) prompts.push(content);
-    }
-
-    return prompts;
-  }
-
-  async extractSummary(sessionRef: string): Promise<string> {
-    const data = await fs.promises.readFile(sessionRef, 'utf-8');
-    const session = JSON.parse(data) as ExportSession;
-
-    for (let i = session.messages.length - 1; i >= 0; i--) {
-      const msg = session.messages[i];
-      if (msg.info.role === 'assistant') {
-        const content = extractTextFromParts(msg.parts);
-        if (content) return content;
-      }
-    }
-
-    return '';
-  }
-
-  // ===========================================================================
-  // TokenCalculator
-  // ===========================================================================
-
-  async calculateTokenUsage(transcriptData: Buffer, fromOffset: number): Promise<TokenUsage> {
-    const usage = emptyTokenUsage();
-
-    try {
-      const session = JSON.parse(transcriptData.toString('utf-8')) as ExportSession;
-
-      for (let i = fromOffset; i < session.messages.length; i++) {
-        const msg = session.messages[i];
-        if (msg.info.role !== 'assistant' || !msg.info.tokens) continue;
-
-        usage.inputTokens += msg.info.tokens.input ?? 0;
-        usage.outputTokens += msg.info.tokens.output ?? 0;
-        usage.cacheReadTokens += msg.info.tokens.cache?.read ?? 0;
-        usage.cacheCreationTokens += msg.info.tokens.cache?.write ?? 0;
-        usage.apiCallCount++;
-      }
-    } catch {
-      // Invalid transcript data
-    }
-
-    return usage;
-  }
-
-  // ===========================================================================
-  // TranscriptChunker
-  // ===========================================================================
-
-  async chunkTranscript(content: Buffer, maxSize: number): Promise<Buffer[]> {
-    try {
-      const session = JSON.parse(content.toString('utf-8')) as ExportSession;
-      if (!session.messages?.length) return [content];
-
-      const infoStr = JSON.stringify(session.info);
-      const baseSize = Buffer.byteLength(`{"info":${infoStr},"messages":[]}`);
-
-      const chunks: Buffer[] = [];
-      let currentMessages: ExportMessage[] = [];
-      let currentSize = baseSize;
-
-      for (const msg of session.messages) {
-        const msgStr = JSON.stringify(msg);
-        const msgSize = Buffer.byteLength(msgStr) + 1; // +1 for comma
-
-        if (currentSize + msgSize > maxSize && currentMessages.length > 0) {
-          chunks.push(
-            Buffer.from(JSON.stringify({ info: session.info, messages: currentMessages })),
-          );
-          currentMessages = [];
-          currentSize = baseSize;
-        }
-
-        currentMessages.push(msg);
-        currentSize += msgSize;
-      }
-
-      if (currentMessages.length > 0) {
-        chunks.push(Buffer.from(JSON.stringify({ info: session.info, messages: currentMessages })));
-      }
-
-      return chunks.length > 0 ? chunks : [content];
-    } catch {
-      return [content];
-    }
-  }
-
-  async reassembleTranscript(chunks: Buffer[]): Promise<Buffer> {
-    const allMessages: ExportMessage[] = [];
-    let sessionInfo: SessionInfo | undefined;
-
-    for (const chunk of chunks) {
-      const session = JSON.parse(chunk.toString('utf-8')) as ExportSession;
-      if (!sessionInfo) sessionInfo = session.info;
-      allMessages.push(...(session.messages ?? []));
-    }
-
-    return Buffer.from(JSON.stringify({ info: sessionInfo ?? { id: '' }, messages: allMessages }));
-  }
-}
-
-// ============================================================================
-// Helpers
-// ============================================================================
-
-const nonAlphanumericRegex = /[^a-zA-Z0-9]/g;
-
-function sanitizePathForOpenCode(repoPath: string): string {
-  return repoPath.replace(nonAlphanumericRegex, '-');
-}
-
-function extractOpenCodeFilePath(input: Record<string, unknown>): string {
-  for (const key of ['filePath', 'path']) {
-    if (typeof input[key] === 'string' && input[key]) return input[key] as string;
-  }
-  return '';
-}
-
-/**
- * Extract text content from OpenCode message parts.
- */
-export function extractTextFromParts(parts: Part[]): string {
-  const texts: string[] = [];
-  for (const part of parts) {
-    if (part.type === 'text' && part.text) {
-      texts.push(part.text);
-    }
-  }
-  return texts.join('\n');
-}
-
-/**
- * Parse an OpenCode export session from raw bytes.
- */
-export function parseExportSession(data: Buffer | string): ExportSession | null {
-  const str = typeof data === 'string' ? data : data.toString('utf-8');
-  if (!str.trim()) return null;
-  return JSON.parse(str) as ExportSession;
-}
-
-/**
- * Extract all user prompts from raw export JSON.
- */
-export function extractAllUserPrompts(data: Buffer | string): string[] {
-  const session = parseExportSession(data);
-  if (!session) return [];
-
-  const prompts: string[] = [];
-  for (const msg of session.messages) {
-    if (msg.info.role !== 'user') continue;
-    const content = extractTextFromParts(msg.parts);
-    if (content) prompts.push(content);
-  }
-  return prompts;
-}
-
-/**
- * Generate the OpenCode plugin TypeScript source.
- */
-function generateOpenCodePlugin(): string {
-  return `// ${SESSIONLOG_PLUGIN_MARKER}
-// This plugin integrates OpenCode with Sessionlog for session tracking.
-// Do not edit manually.
-
-import type { Plugin } from "opencode";
-
-const plugin: Plugin = {
-  name: "sessionlog",
-  hooks: {
-    "session.start": async ({ session }) => {
-      const payload = JSON.stringify({ session_id: session.id });
-      const proc = Bun.spawn(["sessionlog", "hooks", "opencode", "session-start"], {
-        stdin: new Blob([payload]),
-        stdout: "ignore",
-        stderr: "ignore",
-      });
-      await proc.exited;
-    },
-    "session.end": async ({ session }) => {
-      const payload = JSON.stringify({ session_id: session.id });
-      const proc = Bun.spawn(["sessionlog", "hooks", "opencode", "session-end"], {
-        stdin: new Blob([payload]),
-        stdout: "ignore",
-        stderr: "ignore",
-      });
-      await proc.exited;
-    },
-    "message.create": async ({ message, session }) => {
-      if (message.role === "user") {
-        const text = message.parts
-          .filter((p: { type: string }) => p.type === "text")
-          .map((p: { text: string }) => p.text)
-          .join("\\n");
-        const payload = JSON.stringify({ session_id: session.id, prompt: text });
-        const proc = Bun.spawn(["sessionlog", "hooks", "opencode", "turn-start"], {
-          stdin: new Blob([payload]),
-          stdout: "ignore",
-          stderr: "ignore",
-        });
-        await proc.exited;
-      }
-    },
-    "message.complete": async ({ message, session }) => {
-      if (message.role === "assistant") {
-        const payload = JSON.stringify({ session_id: session.id });
-        const proc = Bun.spawn(["sessionlog", "hooks", "opencode", "turn-end"], {
-          stdin: new Blob([payload]),
-          stdout: "ignore",
-          stderr: "ignore",
-        });
-        await proc.exited;
-      }
-    },
-  },
-};
-
-export default plugin;
-`;
-}
-
-// ============================================================================
-// Registration
-// ============================================================================
-
-export function createOpenCodeAgent(): OpenCodeAgent {
-  return new OpenCodeAgent();
-}
-
-registerAgent(AGENT_NAMES.OPENCODE, () => new OpenCodeAgent());

package/references/sessionlog/src/agent/registry.ts

@@ -1,143 +0,0 @@
-/**
- * Agent Registry
- *
- * Central registry for AI agent implementations. Agents register
- * themselves via factory functions and are discovered at runtime.
- */
-
-import type { AgentName, AgentType } from '../types.js';
-import { DEFAULT_AGENT_NAME } from '../types.js';
-import type { Agent } from './types.js';
-
-// ============================================================================
-// Types
-// ============================================================================
-
-export type AgentFactory = () => Agent;
-
-// ============================================================================
-// Registry
-// ============================================================================
-
-const agents = new Map<AgentName, AgentFactory>();
-let cachedInstances = new Map<AgentName, Agent>();
-
-/**
- * Register an agent factory
- */
-export function registerAgent(name: AgentName, factory: AgentFactory): void {
-  agents.set(name, factory);
-  cachedInstances.delete(name);
-}
-
-/**
- * Get an agent by name
- */
-export function getAgent(name: AgentName): Agent | null {
-  const cached = cachedInstances.get(name);
-  if (cached) return cached;
-
-  const factory = agents.get(name);
-  if (!factory) return null;
-
-  const instance = factory();
-  cachedInstances.set(name, instance);
-  return instance;
-}
-
-/**
- * List all registered agent names (sorted)
- */
-export function listAgentNames(): AgentName[] {
-  return Array.from(agents.keys()).sort();
-}
-
-/**
- * Get all registered agents
- */
-export function listAgents(): Agent[] {
-  return listAgentNames().map((name) => getAgent(name)!);
-}
-
-/**
- * Detect which agents are present in the current environment
- */
-export async function detectAgents(cwd?: string): Promise<Agent[]> {
-  const detected: Agent[] = [];
-
-  for (const name of listAgentNames()) {
-    const agent = getAgent(name);
-    if (!agent) continue;
-
-    try {
-      const present = await agent.detectPresence(cwd);
-      if (present) detected.push(agent);
-    } catch {
-      // Skip agents that fail detection
-    }
-  }
-
-  return detected;
-}
-
-/**
- * Auto-detect the first present agent
- */
-export async function detectAgent(cwd?: string): Promise<Agent | null> {
-  const agents = await detectAgents(cwd);
-  return agents[0] ?? null;
-}
-
-/**
- * Get an agent by its human-readable type
- */
-export function getAgentByType(agentType: AgentType): Agent | null {
-  for (const name of listAgentNames()) {
-    const agent = getAgent(name);
-    if (agent && agent.type === agentType) return agent;
-  }
-  return null;
-}
-
-/**
- * Get the default agent
- */
-export function getDefaultAgent(): Agent | null {
-  return getAgent(DEFAULT_AGENT_NAME);
-}
-
-/**
- * Get the union of all protected directories across all agents
- */
-export function allProtectedDirs(): string[] {
-  const dirs = new Set<string>();
-  for (const name of listAgentNames()) {
-    const agent = getAgent(name);
-    if (agent) {
-      for (const dir of agent.protectedDirs) {
-        dirs.add(dir);
-      }
-    }
-  }
-  return Array.from(dirs);
-}
-
-/**
- * Resolve an agent name, trying exact match then type match
- */
-export function resolveAgent(nameOrType: string): Agent | null {
-  // Try exact name match
-  const byName = getAgent(nameOrType);
-  if (byName) return byName;
-
-  // Try type match
-  return getAgentByType(nameOrType);
-}
-
-/**
- * Reset the registry (for testing)
- */
-export function resetRegistry(): void {
-  agents.clear();
-  cachedInstances = new Map();
-}