opencode-ultra 0.3.0 → 0.4.1

package/dist/index.js

@@ -14369,6 +14369,49 @@ function log(message, data) {
     console.error(`${PREFIX} ${message}`);
   }
 }
+// src/shared/ttl-map.ts
+class TtlMap {
+  map = new Map;
+  maxSize;
+  ttlMs;
+  constructor(opts = {}) {
+    this.maxSize = opts.maxSize ?? 1000;
+    this.ttlMs = opts.ttlMs ?? 0;
+  }
+  get(key) {
+    const entry = this.map.get(key);
+    if (!entry)
+      return;
+    if (this.ttlMs > 0 && Date.now() - entry.createdAt > this.ttlMs) {
+      this.map.delete(key);
+      return;
+    }
+    return entry.value;
+  }
+  set(key, value) {
+    if (this.map.has(key)) {
+      this.map.delete(key);
+    }
+    if (this.map.size >= this.maxSize) {
+      const oldest = this.map.keys().next().value;
+      if (oldest !== undefined)
+        this.map.delete(oldest);
+    }
+    this.map.set(key, { value, createdAt: Date.now() });
+  }
+  has(key) {
+    return this.get(key) !== undefined;
+  }
+  delete(key) {
+    return this.map.delete(key);
+  }
+  get size() {
+    return this.map.size;
+  }
+  clear() {
+    this.map.clear();
+  }
+}
 // src/config.ts
 var AgentOverrideSchema = exports_external.object({
   model: exports_external.string().optional(),
@@ -14396,6 +14439,11 @@ var CategoryConfigSchema = exports_external.object({
 var PluginConfigSchema = exports_external.object({
   agents: AgentOverridesSchema.optional(),
   categories: exports_external.record(exports_external.string(), CategoryConfigSchema).optional(),
+  fragments: exports_external.record(exports_external.string(), exports_external.array(exports_external.string())).optional(),
+  prompt_renderer: exports_external.object({
+    default: exports_external.enum(["markdown", "xml", "json"]).optional(),
+    model_overrides: exports_external.record(exports_external.string(), exports_external.enum(["markdown", "xml", "json"])).optional()
+  }).optional(),
   disabled_agents: exports_external.array(exports_external.string()).optional(),
   disabled_hooks: exports_external.array(exports_external.string()).optional(),
   disabled_tools: exports_external.array(exports_external.string()).optional(),
@@ -14819,7 +14867,7 @@ var THINK_MESSAGE = `Extended thinking enabled. Take your time to reason thoroug
 // src/hooks/rules-injector.ts
 import * as fs2 from "fs";
 import * as path2 from "path";
-var cache = new Map;
+var cache = new TtlMap({ maxSize: 50, ttlMs: 10 * 60 * 1000 });
 function loadCachedFile(filePath) {
   try {
     if (!fs2.existsSync(filePath))
@@ -14874,6 +14922,94 @@ function loadCodeStyle(projectDir) {
   return hit.content;
 }
 
+// src/hooks/fragment-injector.ts
+import * as fs3 from "fs";
+import * as path3 from "path";
+var cache2 = new TtlMap({ maxSize: 100, ttlMs: 10 * 60 * 1000 });
+async function readCached(absPath) {
+  try {
+    const st = await fs3.promises.stat(absPath);
+    if (!st.isFile())
+      return null;
+    const prev = cache2.get(absPath);
+    if (prev && prev.mtimeMs === st.mtimeMs)
+      return prev.content;
+    const content = await fs3.promises.readFile(absPath, "utf-8");
+    cache2.set(absPath, { mtimeMs: st.mtimeMs, content });
+    return content;
+  } catch (err) {
+    if (err?.code !== "ENOENT") {
+      log("fragment read failed", { path: absPath, error: String(err) });
+    }
+    return null;
+  }
+}
+function createFragmentInjector(ctx, internalSessions, fragments) {
+  return async (input, output) => {
+    if (input.sessionID && internalSessions.has(input.sessionID))
+      return;
+    const agent = input.agent;
+    if (!agent || !fragments)
+      return;
+    const fragmentPaths = fragments[agent];
+    if (!Array.isArray(fragmentPaths) || fragmentPaths.length === 0)
+      return;
+    for (const rel of fragmentPaths) {
+      const abs = path3.resolve(ctx.directory, rel);
+      const content = await readCached(abs);
+      if (content === null) {
+        log("fragment file not found; skipping", { agent, path: rel });
+        continue;
+      }
+      output.system.push(content);
+    }
+  };
+}
+
+// src/hooks/prompt-renderer.ts
+function escapeXml(s) {
+  return s.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/\"/g, "&quot;").replace(/'/g, "&apos;");
+}
+function resolvePromptFormat(inputModel, config2) {
+  const def = config2?.default ?? "markdown";
+  const overrides = config2?.model_overrides ?? {};
+  const providerID = inputModel?.providerID;
+  const modelID = inputModel?.modelID;
+  const key = typeof providerID === "string" && typeof modelID === "string" ? `${providerID}/${modelID}` : undefined;
+  if (key && overrides[key])
+    return overrides[key];
+  if (typeof modelID === "string" && overrides[modelID])
+    return overrides[modelID];
+  return def;
+}
+function formatSystemSection(format2, content) {
+  if (format2 === "markdown")
+    return content;
+  if (format2 === "xml") {
+    return `<section name="Rules">${escapeXml(content)}</section>`;
+  }
+  if (format2 === "json") {
+    return JSON.stringify({ section: "Rules", content });
+  }
+  return content;
+}
+function createPromptRendererHook(internalSessions, config2) {
+  return (input, output) => {
+    if (input.sessionID && internalSessions.has(input.sessionID))
+      return;
+    if (!Array.isArray(output.system) || output.system.length === 0)
+      return;
+    const format2 = resolvePromptFormat(input.model, config2);
+    if (format2 === "markdown")
+      return;
+    try {
+      output.system = output.system.map((s) => formatSystemSection(format2, s));
+    } catch (err) {
+      log("prompt renderer failed", { error: String(err) });
+    }
+  };
+}
+
 // node_modules/@opencode-ai/plugin/node_modules/zod/v4/classic/external.js
 var exports_external2 = {};
 __export(exports_external2, {
@@ -15603,15 +15739,15 @@ function mergeDefs2(...defs) {
 function cloneDef2(schema) {
   return mergeDefs2(schema._zod.def);
 }
-function getElementAtPath2(obj, path3) {
-  if (!path3)
+function getElementAtPath2(obj, path4) {
+  if (!path4)
     return obj;
-  return path3.reduce((acc, key) => acc?.[key], obj);
+  return path4.reduce((acc, key) => acc?.[key], obj);
 }
 function promiseAllObject2(promisesObj) {
   const keys = Object.keys(promisesObj);
-  const promises = keys.map((key) => promisesObj[key]);
-  return Promise.all(promises).then((results) => {
+  const promises2 = keys.map((key) => promisesObj[key]);
+  return Promise.all(promises2).then((results) => {
     const resolvedObj = {};
     for (let i = 0;i < keys.length; i++) {
       resolvedObj[keys[i]] = results[i];
@@ -15965,11 +16101,11 @@ function aborted2(x, startIndex = 0) {
   }
   return false;
 }
-function prefixIssues2(path3, issues) {
+function prefixIssues2(path4, issues) {
   return issues.map((iss) => {
     var _a2;
     (_a2 = iss).path ?? (_a2.path = []);
-    iss.path.unshift(path3);
+    iss.path.unshift(path4);
     return iss;
   });
 }
@@ -16137,7 +16273,7 @@ function treeifyError2(error48, _mapper) {
     return issue3.message;
   };
   const result = { errors: [] };
-  const processError = (error49, path3 = []) => {
+  const processError = (error49, path4 = []) => {
     var _a2, _b;
     for (const issue3 of error49.issues) {
       if (issue3.code === "invalid_union" && issue3.errors.length) {
@@ -16147,7 +16283,7 @@ function treeifyError2(error48, _mapper) {
       } else if (issue3.code === "invalid_element") {
         processError({ issues: issue3.issues }, issue3.path);
       } else {
-        const fullpath = [...path3, ...issue3.path];
+        const fullpath = [...path4, ...issue3.path];
         if (fullpath.length === 0) {
           result.errors.push(mapper(issue3));
           continue;
@@ -16179,8 +16315,8 @@ function treeifyError2(error48, _mapper) {
 }
 function toDotPath2(_path) {
   const segs = [];
-  const path3 = _path.map((seg) => typeof seg === "object" ? seg.key : seg);
-  for (const seg of path3) {
+  const path4 = _path.map((seg) => typeof seg === "object" ? seg.key : seg);
+  for (const seg of path4) {
     if (typeof seg === "number")
       segs.push(`[${seg}]`);
     else if (typeof seg === "symbol")
@@ -27229,7 +27365,8 @@ function resolveCategory(categoryName, configCategories) {
 
 // src/tools/spawn-agent.ts
 function showToast(ctx, title, message, variant = "info") {
-  ctx.client?.tui?.showToast?.({
+  const client = ctx.client;
+  client.tui?.showToast?.({
     body: { title, message, variant, duration: 2000 }
   })?.catch?.(() => {});
 }
@@ -27334,6 +27471,18 @@ spawn_agent({
       if (!agents || agents.length === 0) {
         return "No agents specified.";
       }
+      for (let i = 0;i < agents.length; i++) {
+        const t = agents[i];
+        if (!t.agent || typeof t.agent !== "string" || t.agent.trim().length === 0) {
+          return `Error: agents[${i}].agent is required (non-empty string)`;
+        }
+        if (!t.prompt || typeof t.prompt !== "string" || t.prompt.trim().length === 0) {
+          return `Error: agents[${i}].prompt is required (non-empty string)`;
+        }
+        if (!t.description || typeof t.description !== "string") {
+          return `Error: agents[${i}].description is required`;
+        }
+      }
       const agentNames = agents.map((a) => a.agent);
       log("spawn_agent", { count: agents.length, agents: agentNames });
       showToast(ctx, "spawn_agent", `${agents.length} agents: ${agentNames.join(", ")}`);
@@ -27519,6 +27668,243 @@ Original task:
 ${original}`;
 }
 
+// src/tools/batch-read.ts
+import * as fs4 from "fs";
+import * as path4 from "path";
+var MAX_PATHS = 20;
+function resolvePath(projectDir, inputPath) {
+  return path4.isAbsolute(inputPath) ? inputPath : path4.join(projectDir, inputPath);
+}
+function isProbablyBinary(buf) {
+  const sample = buf.subarray(0, Math.min(buf.length, 8192));
+  for (const b of sample) {
+    if (b === 0)
+      return true;
+  }
+  return false;
+}
+async function readOne(projectDir, inputPath) {
+  const abs = resolvePath(projectDir, inputPath);
+  try {
+    const stat = await fs4.promises.stat(abs);
+    if (!stat.isFile()) {
+      return `=== ${inputPath} ===
+[File not found]
+`;
+    }
+    const buf = await fs4.promises.readFile(abs);
+    if (isProbablyBinary(buf)) {
+      return `=== ${inputPath} ===
+[Binary file]
+`;
+    }
+    const content = buf.toString("utf-8");
+    return `=== ${inputPath} ===
+${content}
+`;
+  } catch (err) {
+    if (err?.code === "ENOENT") {
+      return `=== ${inputPath} ===
+[File not found]
+`;
+    }
+    log("batch_read failed to read file", { path: abs, error: String(err) });
+    return `=== ${inputPath} ===
+[File not found]
+`;
+  }
+}
+function createBatchReadTool(ctx) {
+  return tool({
+    description: "Read multiple files in parallel and concatenate contents.",
+    args: {
+      paths: tool.schema.array(tool.schema.string()).describe("File paths (max 20).")
+    },
+    execute: async ({ paths }) => {
+      if (!Array.isArray(paths)) {
+        return "Error: paths must be an array of strings";
+      }
+      if (paths.length > MAX_PATHS) {
+        return `Error: maximum ${MAX_PATHS} paths allowed`;
+      }
+      const results = await Promise.all(paths.map((p) => readOne(ctx.directory, p)));
+      return results.join("");
+    }
+  });
+}
+
+// src/tools/continuity-ledger.ts
+import * as fs5 from "fs";
+import * as path5 from "path";
+function normalizeLedgerName(name) {
+  const trimmed = (name ?? "").trim().replace(/\.md$/i, "");
+  const spaced = trimmed.replace(/\s+/g, "-");
+  const cleaned = spaced.replace(/[^A-Za-z0-9._-]/g, "");
+  return cleaned.length > 0 ? cleaned : "ledger";
+}
+function ledgerDir(projectDir) {
+  return path5.join(projectDir, ".opencode", "ledgers");
+}
+function ledgerPath(projectDir, name) {
+  return path5.join(ledgerDir(projectDir), `${normalizeLedgerName(name)}.md`);
+}
+function createLedgerSaveTool(ctx) {
+  return tool({
+    description: "Save a continuity ledger markdown document to .opencode/ledgers/.",
+    args: {
+      name: tool.schema.string().describe('Ledger name (e.g. "auth-refactor").'),
+      content: tool.schema.string().describe("Markdown content.")
+    },
+    execute: async ({ name, content }) => {
+      const dir = ledgerDir(ctx.directory);
+      const outPath = ledgerPath(ctx.directory, name);
+      try {
+        await fs5.promises.mkdir(dir, { recursive: true });
+        await fs5.promises.writeFile(outPath, content, "utf-8");
+        return `Saved to .opencode/ledgers/${path5.basename(outPath)}`;
+      } catch (err) {
+        log("ledger_save failed", { error: String(err) });
+        return `Error: failed to save ledger`;
+      }
+    }
+  });
+}
+function createLedgerLoadTool(ctx) {
+  return tool({
+    description: "Load a continuity ledger markdown document from .opencode/ledgers/.",
+    args: {
+      name: tool.schema.string().optional().describe("Ledger name (defaults to most recent).")
+    },
+    execute: async ({ name }) => {
+      const dir = ledgerDir(ctx.directory);
+      try {
+        if (name && name.trim().length > 0) {
+          const p = ledgerPath(ctx.directory, name);
+          if (!fs5.existsSync(p))
+            return "No ledgers found";
+          return await fs5.promises.readFile(p, "utf-8");
+        }
+        if (!fs5.existsSync(dir))
+          return "No ledgers found";
+        const entries = await fs5.promises.readdir(dir);
+        const files = entries.filter((e) => e.toLowerCase().endsWith(".md")).map((e) => path5.join(dir, e));
+        if (files.length === 0)
+          return "No ledgers found";
+        let latest = files[0];
+        let latestMtime = -1;
+        for (const f of files) {
+          try {
+            const st = await fs5.promises.stat(f);
+            if (st.mtimeMs > latestMtime) {
+              latestMtime = st.mtimeMs;
+              latest = f;
+            }
+          } catch {}
+        }
+        return await fs5.promises.readFile(latest, "utf-8");
+      } catch (err) {
+        log("ledger_load failed", { error: String(err) });
+        return "No ledgers found";
+      }
+    }
+  });
+}
+
+// src/tools/ast-search.ts
+import * as fs6 from "fs";
+import * as path6 from "path";
+import { spawnSync } from "child_process";
+var MAX_MATCHES = 50;
+function stripAnsi(s) {
+  return s.replace(/\u001b\[[0-9;]*m/g, "");
+}
+function resolveSearchPath(projectDir, p) {
+  return path6.isAbsolute(p) ? p : path6.join(projectDir, p);
+}
+function candidatesFor(bin) {
+  const out = [bin];
+  if (process.platform === "win32") {
+    out.push(`${bin}.exe`, `${bin}.cmd`, `${bin}.bat`);
+  }
+  return out;
+}
+function findAstGrepBinary() {
+  const pathEnv = process.env.PATH ?? "";
+  const dirs = pathEnv.split(path6.delimiter).filter(Boolean);
+  const bins = ["sg", "ast-grep"];
+  for (const dir of dirs) {
+    for (const b of bins) {
+      for (const cand of candidatesFor(b)) {
+        const full = path6.join(dir, cand);
+        try {
+          const st = fs6.statSync(full);
+          if (!st.isFile())
+            continue;
+          fs6.accessSync(full, fs6.constants.X_OK);
+          return full;
+        } catch {}
+      }
+    }
+  }
+  return null;
+}
+function normalizeOutput(stdout) {
+  const lines = stripAnsi(stdout).split(`
+`).map((l) => l.trimEnd()).filter((l) => l.length > 0);
+  const out = [];
+  for (const line of lines) {
+    const m3 = line.match(/^(.*?):(\d+):(\d+):(.*)$/);
+    if (m3) {
+      out.push(`${m3[1]}:${m3[2]}: ${m3[4].trim()}`);
+      continue;
+    }
+    const m2 = line.match(/^(.*?):(\d+):(.*)$/);
+    if (m2) {
+      out.push(`${m2[1]}:${m2[2]}: ${m2[3].trim()}`);
+      continue;
+    }
+    out.push(line);
+  }
+  return out.slice(0, MAX_MATCHES).join(`
+`);
+}
+function createAstSearchTool(ctx, binaryPath) {
+  return tool({
+    description: "AST-aware code search using ast-grep (sg).",
+    args: {
+      pattern: tool.schema.string().describe('ast-grep pattern (e.g. "console.log($$$)").'),
+      path: tool.schema.string().optional().describe("Directory to search (default: project root)."),
+      lang: tool.schema.string().optional().describe('Language (e.g. "typescript", "python").')
+    },
+    execute: async ({ pattern, path: searchPath, lang }) => {
+      const target = resolveSearchPath(ctx.directory, searchPath ?? ".");
+      const args = ["--pattern", pattern, target];
+      if (lang && lang.trim().length > 0) {
+        args.push("--lang", lang);
+      }
+      const res = spawnSync(binaryPath, args, {
+        cwd: ctx.directory,
+        encoding: "utf-8"
+      });
+      if (res.error) {
+        log("ast_search spawn failed", { error: String(res.error) });
+        return `Error: ${String(res.error)}`;
+      }
+      const stdout = res.stdout ?? "";
+      const stderr = res.stderr ?? "";
+      if ((res.status ?? 0) === 1 && stdout.trim().length === 0) {
+        return "No matches";
+      }
+      if ((res.status ?? 0) !== 0 && (res.status ?? 0) !== 1) {
+        log("ast_search failed", { status: res.status, stderr });
+        return stderr.trim().length > 0 ? stderr.trim() : "Error: ast_search failed";
+      }
+      const normalized = normalizeOutput(stdout);
+      return normalized.length > 0 ? normalized : "No matches";
+    }
+  });
+}
+
 // src/hooks/todo-enforcer.ts
 var DEFAULT_MAX_ENFORCEMENTS = 5;
 var sessionState = new Map;
@@ -27601,7 +27987,7 @@ ${unfinished.map((t) => `- [ ] ${t.text}`).join(`
 }
 
 // src/hooks/comment-checker.ts
-import * as fs3 from "fs";
+import * as fs7 from "fs";
 var AI_SLOP_PATTERNS = [
   /\/\/ .{80,}/,
   /\/\/ (This|The|We|Here|Note:)/i,
@@ -27680,7 +28066,7 @@ function createCommentCheckerHook(internalSessions, maxRatio = 0.3, slopThreshol
     if (!filePath || !isCodeFile(filePath))
       return;
     try {
-      const content = fs3.readFileSync(filePath, "utf-8");
+      const content = fs7.readFileSync(filePath, "utf-8");
       const result = checkComments(content, filePath, maxRatio, slopThreshold);
       if (result.shouldWarn) {
         output.output += `
@@ -27852,14 +28238,16 @@ class Semaphore {
       this.active++;
       return;
     }
-    return new Promise((resolve) => {
+    return new Promise((resolve2) => {
       this.queue.push(() => {
         this.active++;
-        resolve();
+        resolve2();
       });
     });
   }
   release() {
+    if (this.active <= 0)
+      return;
     this.active--;
     const next = this.queue.shift();
     if (next)
@@ -27949,7 +28337,8 @@ async function registerMcps(ctx, disabled, apiKeys) {
       if (config3.env) {
         body.env = config3.env;
       }
-      await ctx.client.mcp?.add?.({ body });
+      const client = ctx.client;
+      await client.mcp?.add?.({ body });
       log(`MCP ${name} registered${apiKey ? " (with API key)" : ""}`);
     } catch (err) {
       log(`MCP ${name} registration failed: ${err}`);
@@ -27975,11 +28364,18 @@ var OpenCodeUltra = async (ctx) => {
     resolveAgentModel
   });
   const ralphTools = createRalphLoopTools(ctx, internalSessions);
+  const batchRead = createBatchReadTool(ctx);
+  const ledgerSave = createLedgerSaveTool(ctx);
+  const ledgerLoad = createLedgerLoadTool(ctx);
+  const astGrepBin = findAstGrepBinary();
+  const astSearch = astGrepBin ? createAstSearchTool(ctx, astGrepBin) : null;
   const todoEnforcer = createTodoEnforcer(ctx, internalSessions, pluginConfig.todo_enforcer?.maxEnforcements);
   const commentCheckerHook = createCommentCheckerHook(internalSessions, pluginConfig.comment_checker?.maxRatio, pluginConfig.comment_checker?.slopThreshold);
   const tokenTruncationHook = createTokenTruncationHook(internalSessions, pluginConfig.token_truncation?.maxChars);
-  const sessionCompactionHook = createSessionCompactionHook(ctx, internalSessions);
-  const pendingKeywords = new Map;
+  const sessionCompactionHook = createSessionCompactionHook({ client: ctx.client }, internalSessions);
+  const fragmentInjector = createFragmentInjector(ctx, internalSessions, pluginConfig.fragments);
+  const promptRendererHook = createPromptRendererHook(internalSessions, pluginConfig.prompt_renderer);
+  const pendingKeywords = new TtlMap({ maxSize: 200, ttlMs: 5 * 60 * 1000 });
   log("Config loaded", {
     agentCount: Object.keys(agents).length,
     disabledHooks: [...disabledHooks],
@@ -27999,6 +28395,18 @@ var OpenCodeUltra = async (ctx) => {
   if (!disabledTools.has("cancel_ralph")) {
     toolRegistry.cancel_ralph = ralphTools.cancel_ralph;
   }
+  if (!disabledTools.has("batch_read")) {
+    toolRegistry.batch_read = batchRead;
+  }
+  if (!disabledTools.has("ledger_save")) {
+    toolRegistry.ledger_save = ledgerSave;
+  }
+  if (!disabledTools.has("ledger_load")) {
+    toolRegistry.ledger_load = ledgerLoad;
+  }
+  if (!disabledTools.has("ast_search") && astSearch) {
+    toolRegistry.ast_search = astSearch;
+  }
   return {
     tool: toolRegistry,
     config: async (config3) => {
@@ -28048,7 +28456,8 @@ var OpenCodeUltra = async (ctx) => {
             if (output.message.variant === undefined) {
               output.message.variant = "max";
             }
-            ctx.client?.tui?.showToast?.({
+            const client = ctx.client;
+            client.tui?.showToast?.({
               body: {
                 title: "ULTRAWORK MODE",
                 message: "Maximum precision engaged. All agents at your disposal.",
@@ -28107,6 +28516,12 @@ ${arch}`);
 ${style}`);
         }
       }
+      if (!disabledHooks.has("fragment-injector")) {
+        await fragmentInjector(input, output);
+      }
+      if (!disabledHooks.has("prompt-renderer")) {
+        promptRendererHook(input, output);
+      }
     },
     "experimental.session.compacting": async (input, output) => {
       if (disabledHooks.has("session-compaction"))

package/dist/shared/index.d.ts

@@ -1,3 +1,5 @@
 export { getConfigDir, getCacheDir, getDataDir } from "./paths";
 export { parseJsonc } from "./jsonc";
 export { log } from "./log";
+export type { ExtendedClient, ModelRef, ToastOptions, SystemTransformInput } from "./types";
+export { TtlMap } from "./ttl-map";

package/dist/shared/ttl-map.d.ts

@@ -0,0 +1,19 @@
+/**
+ * Simple Map with max size eviction (LRU-like: evicts oldest entry when full).
+ * Optionally supports TTL-based expiry.
+ */
+export declare class TtlMap<K, V> {
+    private map;
+    private maxSize;
+    private ttlMs;
+    constructor(opts?: {
+        maxSize?: number;
+        ttlMs?: number;
+    });
+    get(key: K): V | undefined;
+    set(key: K, value: V): void;
+    has(key: K): boolean;
+    delete(key: K): boolean;
+    get size(): number;
+    clear(): void;
+}