agentlytics 0.1.13 → 0.1.15

package/editors/kiro.js

@@ -0,0 +1,296 @@
+const path = require('path');
+const fs = require('fs');
+const os = require('os');
+const { getAppDataPath } = require('./base');
+
+// ============================================================
+// Kiro editor adapter
+// ============================================================
+
+const name = 'kiro';
+
+const KIRO_AGENT_DIR = path.join(
+  getAppDataPath('Kiro'), 'User', 'globalStorage', 'kiro.kiroagent'
+);
+const WORKSPACE_SESSIONS_DIR = path.join(KIRO_AGENT_DIR, 'workspace-sessions');
+
+function getChats() {
+  const chats = [];
+  if (!fs.existsSync(KIRO_AGENT_DIR)) return chats;
+
+  // Strategy 1: workspace-sessions (structured, has workspace info)
+  if (fs.existsSync(WORKSPACE_SESSIONS_DIR)) {
+    try {
+      for (const folder of fs.readdirSync(WORKSPACE_SESSIONS_DIR)) {
+        const wsDir = path.join(WORKSPACE_SESSIONS_DIR, folder);
+        if (!fs.statSync(wsDir).isDirectory()) continue;
+
+        // Decode base64 folder name to get workspace path
+        let workspacePath = null;
+        try {
+          workspacePath = Buffer.from(folder, 'base64').toString('utf-8');
+        } catch {}
+
+        const indexPath = path.join(wsDir, 'sessions.json');
+        let sessions = [];
+        try {
+          sessions = JSON.parse(fs.readFileSync(indexPath, 'utf-8'));
+        } catch { continue; }
+
+        for (const session of sessions) {
+          const sessionFile = path.join(wsDir, `${session.sessionId}.json`);
+          const exists = fs.existsSync(sessionFile);
+
+          chats.push({
+            source: 'kiro',
+            composerId: session.sessionId,
+            name: cleanTitle(session.title),
+            createdAt: parseInt(session.dateCreated) || null,
+            lastUpdatedAt: exists ? getFileMtime(sessionFile) : parseInt(session.dateCreated) || null,
+            mode: 'kiro',
+            folder: session.workspaceDirectory || workspacePath || null,
+            encrypted: false,
+            bubbleCount: 0,
+            _fullPath: exists ? sessionFile : null,
+            _type: 'workspace-session',
+          });
+        }
+      }
+    } catch {}
+  }
+
+  // Strategy 2: .chat files in hash directories (individual agent executions)
+  // Kiro saves a snapshot of the conversation after each API call, so multiple
+  // .chat files can share the same executionId. We group by executionId and
+  // keep only the latest snapshot (highest message count) per conversation.
+  const seenIds = new Set(chats.map(c => c.composerId));
+  const executionMap = new Map(); // executionId -> best candidate
+  try {
+    for (const dir of fs.readdirSync(KIRO_AGENT_DIR)) {
+      // Skip known non-workspace directories
+      if (['default', 'dev_data', 'index', 'sessions', 'workspace-sessions'].includes(dir)) continue;
+      const fullDir = path.join(KIRO_AGENT_DIR, dir);
+      if (!fs.statSync(fullDir).isDirectory()) continue;
+
+      let files;
+      try { files = fs.readdirSync(fullDir).filter(f => f.endsWith('.chat')); } catch { continue; }
+
+      for (const file of files) {
+        const fullPath = path.join(fullDir, file);
+        try {
+          const stat = fs.statSync(fullPath);
+          const meta = peekChatMeta(fullPath);
+          const chatId = meta.executionId || `${dir}/${file.replace('.chat', '')}`;
+          if (seenIds.has(chatId)) continue;
+
+          const candidate = {
+            source: 'kiro',
+            composerId: chatId,
+            name: meta.title || null,
+            createdAt: meta.startTime || stat.birthtime.getTime(),
+            lastUpdatedAt: meta.endTime || stat.mtime.getTime(),
+            mode: meta.workflow || 'kiro',
+            folder: meta.folder || null,
+            encrypted: false,
+            bubbleCount: meta.messageCount || 0,
+            _fullPath: fullPath,
+            _type: 'chat-file',
+          };
+
+          // Keep the snapshot with the most messages per executionId
+          if (meta.executionId) {
+            const existing = executionMap.get(meta.executionId);
+            if (!existing || meta.messageCount > existing.bubbleCount) {
+              // Update createdAt to the earliest startTime seen
+              if (existing && existing.createdAt < candidate.createdAt) {
+                candidate.createdAt = existing.createdAt;
+              }
+              executionMap.set(meta.executionId, candidate);
+            } else if (existing && meta.startTime && meta.startTime < existing.createdAt) {
+              existing.createdAt = meta.startTime;
+            }
+          } else {
+            chats.push(candidate);
+          }
+        } catch {}
+      }
+    }
+  } catch {}
+
+  // Add the deduplicated execution sessions
+  for (const chat of executionMap.values()) {
+    chats.push(chat);
+  }
+
+  return chats;
+}
+
+function peekChatMeta(filePath) {
+  const meta = { title: null, folder: null, startTime: null, endTime: null, workflow: null, messageCount: 0, executionId: null };
+  try {
+    const raw = fs.readFileSync(filePath, 'utf-8');
+    const data = JSON.parse(raw);
+
+    meta.executionId = data.executionId || null;
+
+    if (data.metadata) {
+      meta.startTime = data.metadata.startTime || null;
+      meta.endTime = data.metadata.endTime || null;
+      meta.workflow = data.metadata.workflow || null;
+    }
+
+    const chat = data.chat || [];
+    for (const msg of chat) {
+      if (msg.role === 'human') {
+        // Try to extract user request from rules block
+        const userReq = extractUserRequest(msg.content);
+        if (userReq && !meta.title) {
+          meta.title = cleanTitle(userReq);
+        }
+      }
+      if (msg.role === 'bot' || msg.role === 'human') meta.messageCount++;
+    }
+
+    // Try to extract folder from context
+    for (const ctx of data.context || []) {
+      if (ctx.type === 'steering' && ctx.id) {
+        // Extract workspace from steering file path
+        const match = ctx.id.match(/file:\/\/(.*?)\/.kiro\//);
+        if (match) meta.folder = match[1];
+      }
+    }
+  } catch {}
+  return meta;
+}
+
+function isSystemPrompt(content) {
+  if (typeof content !== 'string') return false;
+  return content.startsWith('<identity>') || content.startsWith('# ');
+}
+
+function extractUserRequest(content) {
+  if (typeof content !== 'string') return null;
+  // "## Included Rules" messages contain the actual user request after </user-rule>
+  const ruleEnd = content.lastIndexOf('</user-rule>');
+  if (ruleEnd >= 0) {
+    let userPart = content.substring(ruleEnd + '</user-rule>'.length).trim();
+    // Strip trailing EnvironmentContext block
+    const envIdx = userPart.indexOf('<EnvironmentContext>');
+    if (envIdx >= 0) userPart = userPart.substring(0, envIdx).trim();
+    // Strip steering-reminder blocks
+    const steerIdx = userPart.indexOf('<steering-reminder>');
+    if (steerIdx >= 0) userPart = userPart.substring(0, steerIdx).trim();
+    if (userPart) return userPart;
+  }
+  return null;
+}
+
+function getMessages(chat) {
+  if (!chat._fullPath || !fs.existsSync(chat._fullPath)) return [];
+
+  if (chat._type === 'workspace-session') {
+    return getWorkspaceSessionMessages(chat._fullPath);
+  }
+  return getChatFileMessages(chat._fullPath);
+}
+
+function getWorkspaceSessionMessages(filePath) {
+  const messages = [];
+  try {
+    const data = JSON.parse(fs.readFileSync(filePath, 'utf-8'));
+    const history = data.history || [];
+
+    for (const entry of history) {
+      const msg = entry.message;
+      if (!msg) continue;
+
+      const role = msg.role === 'user' ? 'user' : msg.role === 'assistant' ? 'assistant' : null;
+      if (!role) continue;
+
+      const content = extractContentFromMessage(msg.content);
+      if (!content) continue;
+
+      const result = { role, content };
+
+      // Extract model info from promptLogs
+      if (role === 'assistant' && entry.promptLogs && entry.promptLogs.length > 0) {
+        const log = entry.promptLogs[0];
+        result._model = log.modelTitle || null;
+      }
+
+      messages.push(result);
+    }
+  } catch {}
+  return messages;
+}
+
+function getChatFileMessages(filePath) {
+  const messages = [];
+  try {
+    const data = JSON.parse(fs.readFileSync(filePath, 'utf-8'));
+    const chat = data.chat || [];
+    const model = data.metadata?.modelId || null;
+
+    for (const msg of chat) {
+      if (msg.role === 'human') {
+        if (isSystemPrompt(msg.content)) continue;
+        // Try extracting user request from rules block first
+        const userReq = extractUserRequest(msg.content);
+        const content = userReq || extractUserText(msg.content);
+        if (content) messages.push({ role: 'user', content });
+      } else if (msg.role === 'bot') {
+        const content = typeof msg.content === 'string' ? msg.content : '';
+        if (content) messages.push({ role: 'assistant', content, _model: model });
+      } else if (msg.role === 'tool') {
+        const content = typeof msg.content === 'string' ? msg.content : '';
+        if (content) messages.push({ role: 'tool', content: content.substring(0, 2000) });
+      }
+    }
+  } catch {}
+  return messages;
+}
+
+function extractContentFromMessage(content) {
+  if (typeof content === 'string') return content;
+  if (!Array.isArray(content)) return '';
+  return content
+    .filter(c => c.type === 'text' || c.type === 'mention')
+    .map(c => c.text)
+    .join('\n') || '';
+}
+
+function extractUserText(content) {
+  if (typeof content === 'string') {
+    // Skip system prompt content
+    if (isSystemPrompt(content)) return null;
+    // Strip XML tags and rules blocks
+    return cleanTitle(content);
+  }
+  if (Array.isArray(content)) {
+    return content
+      .filter(c => c.type === 'text' || c.type === 'mention')
+      .map(c => c.text)
+      .join('\n') || '';
+  }
+  return '';
+}
+
+function cleanTitle(title) {
+  if (!title) return null;
+  let clean = title
+    .replace(/<[^>]+>[\s\S]*?<\/[^>]+>/g, '')
+    .replace(/<[^>]+>/g, '')
+    .replace(/## Included Rules[\s\S]*$/m, '')
+    .replace(/\s+/g, ' ')
+    .trim()
+    .substring(0, 120);
+  return clean || null;
+}
+
+function getFileMtime(filePath) {
+  try { return fs.statSync(filePath).mtime.getTime(); } catch { return null; }
+}
+
+const labels = { 'kiro': 'Kiro' };
+
+module.exports = { name, labels, getChats, getMessages };

package/editors/opencode.js

@@ -1,103 +1,184 @@
 const path = require('path');
 const fs = require('fs');
 const os = require('os');
-const { execSync } = require('child_process');
 
-const DB_PATH = path.join(os.homedir(), '.local', 'share', 'opencode', 'opencode.db');
+// OpenCode stores data in different locations depending on the platform
+// - Windows: %LOCALAPPDATA%\opencode\storage (not Roaming)
+// - macOS/Linux: ~/.local/share/opencode/storage (XDG path)
+function getOpenCodeStoragePath() {
+  const home = os.homedir();
+  switch (process.platform) {
+    case 'win32':
+      return path.join(home, 'AppData', 'Local', 'opencode', 'storage');
+    case 'darwin':
+    case 'linux':
+    default:
+      return path.join(home, '.local', 'share', 'opencode', 'storage');
+  }
+}
+
+const STORAGE_DIR = getOpenCodeStoragePath();
+const SESSION_DIR = path.join(STORAGE_DIR, 'session');
+const MESSAGE_DIR = path.join(STORAGE_DIR, 'message');
+const PART_DIR = path.join(STORAGE_DIR, 'part');
 
 // ============================================================
-// Query SQLite via CLI
+// Scan JSON files from OpenCode storage
 // ============================================================
 
-function queryDb(sql) {
-  if (!fs.existsSync(DB_PATH)) return [];
+function readJson(filePath) {
   try {
-    const raw = execSync(
-      `sqlite3 -json ${JSON.stringify(DB_PATH)} ${JSON.stringify(sql)}`,
-      { encoding: 'utf-8', maxBuffer: 10 * 1024 * 1024, stdio: ['pipe', 'pipe', 'pipe'] }
-    );
-    return JSON.parse(raw);
-  } catch { return []; }
+    return JSON.parse(fs.readFileSync(filePath, 'utf-8'));
+  } catch { return null; }
 }
 
-// ============================================================
-// Adapter interface
-// ============================================================
+function getAllSessions() {
+  const sessions = [];
+  if (!fs.existsSync(SESSION_DIR)) return sessions;
 
-const name = 'opencode';
+  for (const projectHash of fs.readdirSync(SESSION_DIR)) {
+    const projectDir = path.join(SESSION_DIR, projectHash);
+    if (!fs.statSync(projectDir).isDirectory()) continue;
 
-function getChats() {
-  const rows = queryDb('SELECT s.id, s.title, s.directory, s.time_created, s.time_updated, p.worktree, p.name as project_name, (SELECT count(*) FROM message m WHERE m.session_id = s.id) as msg_count FROM session s LEFT JOIN project p ON s.project_id = p.id ORDER BY s.time_updated DESC');
+    let files;
+    try { files = fs.readdirSync(projectDir).filter(f => f.startsWith('ses_') && f.endsWith('.json')); } catch { continue; }
 
-  return rows.map(row => ({
-    source: 'opencode',
-    composerId: row.id,
-    name: cleanTitle(row.title),
-    createdAt: row.time_created,
-    lastUpdatedAt: row.time_updated,
-    mode: 'opencode',
-    folder: row.worktree || row.directory || null,
-    encrypted: false,
-    bubbleCount: row.msg_count || 0,
-  }));
+    for (const file of files) {
+      const filePath = path.join(projectDir, file);
+      const data = readJson(filePath);
+      if (data && data.id) {
+        sessions.push({ ...data, _filePath: filePath });
+      }
+    }
+  }
+  return sessions;
 }
 
-function cleanTitle(title) {
-  if (!title) return null;
-  // Strip default "New session - <date>" titles
-  if (title.startsWith('New session - ')) return null;
-  return title.substring(0, 120) || null;
+function getMessageCount(sessionId) {
+  const sessionMsgDir = path.join(MESSAGE_DIR, sessionId);
+  if (!fs.existsSync(sessionMsgDir)) return 0;
+
+  try {
+    return fs.readdirSync(sessionMsgDir).filter(f => f.startsWith('msg_') && f.endsWith('.json')).length;
+  } catch { return 0; }
 }
 
-function getMessages(chat) {
-  // Get messages with their parts joined
-  const messages = queryDb(`SELECT m.id as msg_id, m.data as msg_data, m.time_created FROM message m WHERE m.session_id = '${chat.composerId}' ORDER BY m.time_created ASC`);
+function getMessagesForSession(sessionId) {
+  const sessionMsgDir = path.join(MESSAGE_DIR, sessionId);
+  if (!fs.existsSync(sessionMsgDir)) return [];
 
-  const result = [];
-  for (const msg of messages) {
-    let msgData;
-    try { msgData = JSON.parse(msg.msg_data); } catch { continue; }
+  let files;
+  try { files = fs.readdirSync(sessionMsgDir).filter(f => f.startsWith('msg_') && f.endsWith('.json')); } catch { return []; }
 
-    const role = msgData.role;
-    if (!role) continue;
+  const messages = [];
+  for (const file of files) {
+    const msgPath = path.join(sessionMsgDir, file);
+    const msg = readJson(msgPath);
+    if (!msg || !msg.id) continue;
 
     // Get parts for this message
-    const parts = queryDb(`SELECT data FROM part WHERE message_id = '${msg.msg_id}' ORDER BY time_created ASC`);
+    const msgPartDir = path.join(PART_DIR, msg.id);
+    const parts = [];
+    if (fs.existsSync(msgPartDir)) {
+      try {
+        const partFiles = fs.readdirSync(msgPartDir).filter(f => f.startsWith('prt_') && f.endsWith('.json'));
+        for (const partFile of partFiles) {
+          const part = readJson(path.join(msgPartDir, partFile));
+          if (part) parts.push(part);
+        }
+      } catch { /* skip */ }
+    }
 
+    // Build content from parts
     const contentParts = [];
     for (const part of parts) {
-      let partData;
-      try { partData = JSON.parse(part.data); } catch { continue; }
-
-      if (partData.type === 'text' && partData.text) {
-        contentParts.push(partData.text);
-      } else if (partData.type === 'tool-use' || partData.type === 'tool_use') {
-        const toolName = partData.name || partData.toolName || 'tool';
-        let argKeys = '';
-        try {
-          const input = typeof partData.input === 'string' ? JSON.parse(partData.input) : (partData.input || {});
-          argKeys = Object.keys(input).join(', ');
-        } catch {}
+      const type = part.type;
+
+      if (type === 'text' && part.text) {
+        contentParts.push(part.text);
+      } else if (type === 'thinking' || type === 'reasoning') {
+        if (part.text) contentParts.push(`[thinking] ${part.text}`);
+      } else if (type === 'tool-call' || type === 'tool_use' || type === 'tool') {
+        const toolName = part.name || part.toolName || part.tool || 'tool';
+        const args = part.args || part.arguments || part.state?.input || {};
+        const argKeys = typeof args === 'object' ? Object.keys(args).join(', ') : '';
         contentParts.push(`[tool-call: ${toolName}(${argKeys})]`);
-      } else if (partData.type === 'tool-result' || partData.type === 'tool_result') {
-        const preview = (partData.text || partData.output || '').substring(0, 500);
+      } else if (type === 'tool-result' || type === 'tool_result') {
+        const preview = (part.text || part.output || part.state?.output || '').substring(0, 500);
         contentParts.push(`[tool-result] ${preview}`);
+      } else if (type === 'step-start' || type === 'step-finish') {
+        // Skip metadata parts
       }
-      // Skip step-start, step-finish (metadata only)
+    }
+
+    // If no parts with content, check if message itself has content
+    if (contentParts.length === 0 && msg.role) {
+      contentParts.push(`[${msg.role}]`);
     }
 
     const content = contentParts.join('\n');
-    if (!content) continue;
-
-    const mappedRole = role === 'user' ? 'user' : role === 'assistant' ? 'assistant' : role;
-    result.push({
-      role: mappedRole,
-      content,
-      _model: msgData.model?.modelID,
-    });
+    if (content) {
+      // Extract model value - handle both string and object formats
+      let modelValue = null;
+      if (typeof msg.modelID === 'string') {
+        modelValue = msg.modelID;
+      } else if (msg.model && typeof msg.model === 'object' && msg.model.modelID) {
+        modelValue = msg.model.modelID;
+      } else if (typeof msg.model === 'string') {
+        modelValue = msg.model;
+      }
+
+      messages.push({
+        role: msg.role || 'assistant',
+        content,
+        _model: modelValue,
+        _inputTokens: msg.tokens?.input,
+        _outputTokens: msg.tokens?.output,
+        _cacheRead: msg.tokens?.cache?.read,
+        _cacheWrite: msg.tokens?.cache?.write,
+        _finish: msg.finish,
+      });
+    }
   }
 
-  return result;
+  // Sort by creation time
+  return messages.sort((a, b) => {
+    const aTime = a.time?.created || 0;
+    const bTime = b.time?.created || 0;
+    return aTime - bTime;
+  });
 }
 
-module.exports = { name, getChats, getMessages };
+// ============================================================
+// Adapter interface
+// ============================================================
+
+const name = 'opencode';
+
+function getChats() {
+  const sessions = getAllSessions();
+
+  return sessions.map(s => ({
+    source: 'opencode',
+    composerId: s.id,
+    name: s.title || null,
+    createdAt: s.time?.created || null,
+    lastUpdatedAt: s.time?.updated || null,
+    mode: s.mode || 'opencode',
+    folder: s.directory || null,
+    encrypted: false,
+    bubbleCount: getMessageCount(s.id),
+    _agent: s.agent,
+    _model: s.modelID,
+    _provider: s.providerID,
+    _sessionData: s,
+  })).sort((a, b) => (b.lastUpdatedAt || 0) - (a.lastUpdatedAt || 0));
+}
+
+function getMessages(chat) {
+  return getMessagesForSession(chat.composerId);
+}
+
+const labels = { 'opencode': 'OpenCode' };
+
+module.exports = { name, labels, getChats, getMessages };

package/editors/vscode.js

@@ -315,4 +315,6 @@ function getMessages(chat) {
   return messages;
 }
 
-module.exports = { name, getChats, getMessages };
+const labels = { 'vscode': 'VS Code', 'vscode-insiders': 'VS Code Insiders' };
+
+module.exports = { name, labels, getChats, getMessages };