codemini-cli 0.4.0 → 0.4.2

package/src/core/tool-args.js

@@ -0,0 +1,181 @@
+import path from 'node:path';
+
+export function parseInlineRangePath(value) {
+  const text = String(value || '').trim();
+  if (!text) return null;
+  const match = text.match(/^(.*?):(\d+)(?:-(\d+))?$/);
+  if (!match) return null;
+  const [, maybePath, startRaw, endRaw] = match;
+  if (!maybePath || /^(?:[A-Za-z])$/.test(maybePath)) return null;
+  const startLine = Number(startRaw);
+  const endLine = Number(endRaw || startRaw);
+  if (!Number.isFinite(startLine) || startLine <= 0) return null;
+  if (!Number.isFinite(endLine) || endLine < startLine) return null;
+  return {
+    path: maybePath,
+    start_line: startLine,
+    end_line: endLine
+  };
+}
+
+export function normalizeReadArgs(rawArgs) {
+  const source =
+    rawArgs && typeof rawArgs === 'object' && !Array.isArray(rawArgs)
+      ? { ...rawArgs }
+      : { path: typeof rawArgs === 'string' ? rawArgs : '' };
+
+  const normalized = { ...source };
+  const aliasPath = String(source.path || source.file_path || source.file || source.target || '').trim();
+  if (aliasPath) normalized.path = aliasPath;
+
+  if (!Number.isFinite(Number(normalized.start_line)) && Number.isFinite(Number(source.offset))) {
+    normalized.start_line = Number(source.offset);
+  }
+
+  if (!Number.isFinite(Number(normalized.end_line)) && Number.isFinite(Number(source.limit))) {
+    const startLine = Number(normalized.start_line);
+    const limit = Number(source.limit);
+    if (startLine > 0 && limit > 0) {
+      normalized.end_line = startLine + limit - 1;
+    }
+  }
+
+  const inlineRange = parseInlineRangePath(normalized.path);
+  if (inlineRange) {
+    normalized.path = inlineRange.path;
+    if (!Number.isFinite(Number(normalized.start_line))) normalized.start_line = inlineRange.start_line;
+    if (!Number.isFinite(Number(normalized.end_line))) normalized.end_line = inlineRange.end_line;
+  }
+
+  return normalized;
+}
+
+export function normalizePathArgs(rawArgs, aliases = []) {
+  const source =
+    rawArgs && typeof rawArgs === 'object' && !Array.isArray(rawArgs)
+      ? { ...rawArgs }
+      : { path: typeof rawArgs === 'string' ? rawArgs : '' };
+  const normalized = { ...source };
+  const keys = ['path', ...aliases];
+  for (const key of keys) {
+    const value = String(source?.[key] || '').trim();
+    if (value) {
+      normalized.path = value;
+      break;
+    }
+  }
+  return normalized;
+}
+
+export function normalizePatternArgs(rawArgs, aliases = [], defaultPathAliases = []) {
+  const source =
+    rawArgs && typeof rawArgs === 'object' && !Array.isArray(rawArgs)
+      ? { ...rawArgs }
+      : { pattern: typeof rawArgs === 'string' ? rawArgs : '' };
+  const normalized = { ...source };
+  for (const key of ['pattern', ...aliases]) {
+    const value = String(source?.[key] || '').trim();
+    if (value) {
+      normalized.pattern = value;
+      break;
+    }
+  }
+  for (const key of ['path', ...defaultPathAliases]) {
+    const value = String(source?.[key] || '').trim();
+    if (value) {
+      normalized.path = value;
+      break;
+    }
+  }
+  return normalized;
+}
+
+export function normalizeWriteArgs(rawArgs) {
+  const source =
+    rawArgs && typeof rawArgs === 'object' && !Array.isArray(rawArgs)
+      ? { ...rawArgs }
+      : { path: typeof rawArgs === 'string' ? rawArgs : '' };
+  const normalized = { ...source };
+  const filePath = String(source.path || source.file_path || source.file || '').trim();
+  if (filePath) normalized.path = filePath;
+  if (normalized.content == null) {
+    if (source.text != null) normalized.content = source.text;
+    if (source.new_content != null) normalized.content = source.new_content;
+  }
+  return normalized;
+}
+
+export function normalizeWebFetchArgs(rawArgs) {
+  const normalized = normalizePathArgs(rawArgs, ['url', 'href', 'link', 'target']);
+  const url = String(normalized.url || normalized.path || '').trim();
+  return { ...normalized, url };
+}
+
+export function normalizeWebSearchArgs(rawArgs) {
+  const normalized = normalizePatternArgs(rawArgs, ['query', 'q', 'keyword']);
+  const query = String(normalized.query || normalized.pattern || '').trim();
+  return { ...normalized, query };
+}
+
+function buildDeleteApprovalDetails(source, rawPath) {
+  const existing =
+    source?.approval && typeof source.approval === 'object' && !Array.isArray(source.approval)
+      ? source.approval
+      : {};
+  const approvalPath = String(existing.path || rawPath || '').trim();
+  const approvalName = String(existing.name || (approvalPath ? path.basename(approvalPath) : '') || '').trim();
+  const approvalType = String(existing.type || '').trim();
+
+  const approval = {};
+  if (approvalPath) approval.path = approvalPath;
+  if (approvalName) approval.name = approvalName;
+  if (approvalType) approval.type = approvalType;
+  return Object.keys(approval).length > 0 ? approval : undefined;
+}
+
+export function normalizeToolArguments(toolName, args, rawArguments) {
+  const rawText = typeof rawArguments === 'string' ? rawArguments.trim() : '';
+  const primitive =
+    args == null || Array.isArray(args) || typeof args !== 'object'
+      ? args
+      : null;
+  const source =
+    args && typeof args === 'object' && !Array.isArray(args)
+      ? { ...args }
+      : {};
+
+  if (primitive != null && typeof primitive !== 'object') {
+    source._raw = rawText || String(primitive);
+  } else if (!source._raw && rawText && source._invalid_json) {
+    source._raw = rawText;
+  }
+
+  const stringValue =
+    typeof primitive === 'string'
+      ? primitive.trim()
+      : String(source._raw || '').trim();
+
+  if (toolName === 'read') return normalizeReadArgs({ ...source, ...(stringValue && !source.path ? { path: stringValue } : {}) });
+  if (toolName === 'list') return normalizePathArgs({ ...source, ...(stringValue && !source.path ? { path: stringValue } : {}) }, ['dir', 'directory']);
+  if (toolName === 'glob') return normalizePatternArgs({ ...source, ...(stringValue && !source.pattern ? { pattern: stringValue } : {}) }, ['glob', 'query'], ['directory']);
+  if (toolName === 'grep') return normalizePatternArgs({ ...source, ...(stringValue && !source.pattern ? { pattern: stringValue } : {}) }, ['query', 'symbol', 'q'], ['directory', 'dir', 'cwd']);
+  if (toolName === 'write') return normalizeWriteArgs({ ...source, ...(stringValue && !source.path ? { path: stringValue } : {}) });
+
+  if (toolName === 'edit') {
+    const value = String(source.path || source.file || source.file_path || '').trim();
+    if (value && !source.path) source.path = value;
+    return source;
+  }
+
+  if (toolName === 'delete') {
+    const normalized = normalizePathArgs(
+      { ...source, ...(stringValue && !source.path ? { path: stringValue } : {}) },
+      ['file_path', 'file', 'target', 'directory', 'dir']
+    );
+    const approval = buildDeleteApprovalDetails(normalized, normalized.path);
+    if (approval) normalized.approval = approval;
+    return normalized;
+  }
+
+  return source;
+}

package/src/core/tool-result-store.js

@@ -0,0 +1,206 @@
+import fs from 'node:fs/promises';
+import os from 'node:os';
+import path from 'node:path';
+import { BoundedCache } from './bounded-cache.js';
+import { trimInline } from './string-utils.js';
+
+const TOOL_RESULT_DISK_THRESHOLD = 6000;
+const PREVIEW_SIZE_BYTES = 2000;
+const TOOL_RESULTS_SUBDIR = 'tool-results';
+
+let currentResultDir = null;
+let resultDirReady = false;
+
+const storedResults = new BoundedCache({
+  maxSize: 64,
+  ttlMs: 30 * 60 * 1000,
+  onEvict(_key, value) {
+    if (value?.filePath) {
+      fs.unlink(value.filePath).catch(() => {});
+    }
+  }
+});
+
+const readCache = new BoundedCache({ maxSize: 128, ttlMs: 10 * 60 * 1000 });
+
+function generatePreview(content) {
+  if (content.length <= PREVIEW_SIZE_BYTES) {
+    return { preview: content, hasMore: false };
+  }
+  const truncated = content.slice(0, PREVIEW_SIZE_BYTES);
+  const lastNewline = truncated.lastIndexOf('\n');
+  const cutPoint = lastNewline > PREVIEW_SIZE_BYTES * 0.5 ? lastNewline : PREVIEW_SIZE_BYTES;
+  return { preview: content.slice(0, cutPoint), hasMore: true };
+}
+
+function formatFileSize(chars) {
+  if (chars < 1024) return `${chars} B`;
+  return `${(chars / 1024).toFixed(1)} KB`;
+}
+
+export function setResultDir(dir) {
+  currentResultDir = dir ? path.join(dir, TOOL_RESULTS_SUBDIR) : null;
+  resultDirReady = false;
+}
+
+async function ensureResultDir() {
+  if (!currentResultDir) return false;
+  if (!resultDirReady) {
+    await fs.mkdir(currentResultDir, { recursive: true });
+    resultDirReady = true;
+  }
+  return true;
+}
+
+export async function storeResultIfNeeded(callId, formattedContent, rawResult) {
+  if (formattedContent.length <= TOOL_RESULT_DISK_THRESHOLD) {
+    return formattedContent;
+  }
+  try {
+    const ready = await ensureResultDir();
+    const dir = ready ? currentResultDir : path.join(os.tmpdir(), 'codemini-results');
+    if (!resultDirReady && dir === currentResultDir) {
+      await fs.mkdir(dir, { recursive: true });
+    } else if (!resultDirReady) {
+      await fs.mkdir(dir, { recursive: true });
+    }
+    const filePath = path.join(dir, `${callId}.txt`);
+    const payload = typeof rawResult === 'string' ? rawResult : JSON.stringify(rawResult, null, 2);
+    await fs.writeFile(filePath, payload, 'utf-8');
+    const summary = summarizeToolResult(rawResult);
+    const { preview, hasMore } = generatePreview(payload);
+    storedResults.set(callId, { filePath, summary });
+
+    return `<persisted-output>
+Output too large (${formatFileSize(payload.length)}). Full output saved to: ${filePath}
+
+Preview (first ${formatFileSize(PREVIEW_SIZE_BYTES)}):
+${preview}${hasMore ? '\n...' : ''}
+
+Summary: ${summary}
+</persisted-output>`;
+  } catch {
+    return formattedContent;
+  }
+}
+
+export function clearResultStore() {
+  const files = [];
+  for (const [, val] of storedResults.entries()) {
+    files.push(val.filePath);
+  }
+  storedResults.clear();
+  readCache.clear();
+  return Promise.allSettled(files.map((filePath) => fs.unlink(filePath).catch(() => {})));
+}
+
+export function checkReadDedup(filePath, startLine, endLine, mtimeMs) {
+  const key = `${filePath}:${startLine || 0}:${endLine || 0}:${mtimeMs}`;
+  if (readCache.has(key)) {
+    return true;
+  }
+  readCache.set(key, true);
+  return false;
+}
+
+export function summarizeToolResult(result) {
+  if (result === null || result === undefined) return 'no output';
+  if (typeof result === 'string') {
+    const oneLine = result.replace(/\s+/g, ' ').trim();
+    return oneLine.length > 90 ? `${oneLine.slice(0, 87)}...` : oneLine || 'empty string';
+  }
+  if (typeof result === 'object') {
+    const obj = result;
+    if (Array.isArray(obj)) return `array(${obj.length})`;
+    if ('deleted' in obj && 'path' in obj) {
+      const kind = trimInline(obj.type || 'item', 16);
+      const target = trimInline(obj.path || '', 96);
+      if (obj.deleted) return target ? `deleted ${kind} ${target}` : `deleted ${kind}`;
+      if (obj.cancelled) return target ? `cancelled delete ${target}` : 'cancelled delete';
+    }
+    if ('path' in obj && 'action' in obj) {
+      const p = String(obj.path || '');
+      const action = String(obj.action || 'write');
+      const line = Number(obj.changed_line || 1);
+      const suffix =
+        action === 'delete'
+          ? 'deleted'
+          : action === 'create'
+            ? 'created'
+            : action === 'patch'
+              ? 'patched'
+              : action === 'replace_block' || action === 'replace_text'
+                ? 'edited'
+                : action === 'append'
+                  ? 'appended'
+                  : 'updated';
+      return p ? `${suffix} ${p}${line > 0 ? ` @L${line}` : ''}` : suffix;
+    }
+    if ('path' in obj && 'phase' in obj) {
+      const phase = String(obj.phase || '');
+      const p = String(obj.path || '');
+      const total = Number(obj.total_lines);
+      const start =
+        Number(obj.suggested_start_line || obj.start_line) > 0
+          ? Number(obj.suggested_start_line || obj.start_line)
+          : 1;
+      const end =
+        Number(obj.suggested_end_line || obj.end_line) >= start
+          ? Number(obj.suggested_end_line || obj.end_line)
+          : start;
+      const rangeText = start > 0 && end >= start ? ` lines ${start}-${end}` : '';
+      const totalText = total > 0 ? ` of ${total}` : '';
+      const enclosingText = obj.enclosing_symbol ? ` in ${obj.enclosing_symbol}` : '';
+      const errorText = obj.error ? ` (${trimInline(obj.error, 64)})` : '';
+      const truncatedText = obj.truncated ? ' [truncated]' : '';
+      return phase === 'metadata'
+        ? `metadata for ${p}${rangeText}${totalText}${errorText}`
+        : `content from ${p}${rangeText}${totalText}${enclosingText}${truncatedText}`;
+    }
+    if ('stdout' in obj || 'stderr' in obj || 'code' in obj) {
+      const stdout = trimInline(obj.stdout || '', 96);
+      const stderr = trimInline(obj.stderr || '', 96);
+      const command = trimInline(obj.command || '', 72);
+      const lead = command ? `${command} -> ` : '';
+      if (stdout) return `${lead}exit ${obj.code ?? 0}\nstdout: ${stdout}`;
+      if (stderr) return `${lead}exit ${obj.code ?? 0}\nstderr: ${stderr}`;
+      return `${lead}exit ${obj.code ?? 0}`;
+    }
+    if ('task_id' in obj && 'startup_confirmed' in obj) {
+      const status = trimInline(obj.status || 'unknown', 32);
+      const taskId = trimInline(obj.task_id || '', 24);
+      const source = trimInline(obj.startup_source || '', 24);
+      const outputFile = trimInline(obj.output_file || '', 72);
+      const output = Array.isArray(obj.recent_output) ? trimInline(obj.recent_output.slice(-1)[0] || '', 96) : '';
+      return `${taskId || 'task'} ${status}${source ? ` (${source})` : ''}${outputFile ? ` -> ${outputFile}` : ''}${output ? `\n${output}` : ''}`;
+    }
+    if ('tasks' in obj && Array.isArray(obj.tasks)) {
+      const count = obj.tasks.length;
+      const first = obj.tasks[0];
+      const lead = first?.task_id ? `${trimInline(first.task_id, 24)} ${trimInline(first.status || 'unknown', 24)}` : '';
+      return `tasks(${count})${lead ? `\n${lead}` : ''}`;
+    }
+    if ('files' in obj && Array.isArray(obj.files)) {
+      return `patched ${obj.files.length} file(s)`;
+    }
+    if ('diff' in obj && 'new_hash' in obj && 'path' in obj) {
+      const p = String(obj.path || '');
+      return p ? `diff preview for ${p}` : 'diff preview';
+    }
+    if ('created' in obj && Array.isArray(obj.created)) {
+      return `created ${obj.created.length} task(s)`;
+    }
+    if ('tasks' in obj && Array.isArray(obj.tasks)) {
+      return `${obj.tasks.length} task(s)`;
+    }
+    if ('newTodos' in obj && Array.isArray(obj.newTodos)) {
+      return obj.newTodos.length > 0 ? `updated ${obj.newTodos.length} todo item(s)` : 'cleared todo list';
+    }
+    if ('newPlan' in obj) {
+      return obj.newPlan ? `updated plan state (${String(obj.newPlan.status || 'draft')})` : 'cleared plan state';
+    }
+    const keys = Object.keys(obj);
+    return keys.length > 0 ? `keys: ${keys.slice(0, 5).join(',')}` : 'object';
+  }
+  return String(result);
+}