inferoa 0.1.0

package/dist/src/skills/registry.js

@@ -0,0 +1,203 @@
+import { promises as fs } from "node:fs";
+import path from "node:path";
+import os from "node:os";
+import { fileURLToPath } from "node:url";
+import { pathExists } from "../util/fs.js";
+export class SkillRegistry {
+    workspace;
+    config;
+    constructor(workspace, config) {
+        this.workspace = workspace;
+        this.config = config;
+    }
+    async discover() {
+        const descriptors = [];
+        const roots = await this.discoveryRoots();
+        for (const root of roots) {
+            if (!(await pathExists(root.path))) {
+                continue;
+            }
+            if (root.importer) {
+                descriptors.push(...(await this.discoverImported(root.path, root.trust)));
+            }
+            else {
+                descriptors.push(...(await this.discoverNative(root.path, root.trust)));
+            }
+        }
+        return sortDescriptors(dedupe(descriptors));
+    }
+    async loadEnabled(descriptors) {
+        descriptors ??= await this.discover();
+        const enabled = new Set(this.config.skills.enabled);
+        const loaded = [];
+        for (const descriptor of descriptors) {
+            if (!enabled.has(descriptor.id) && !enabled.has(descriptor.name)) {
+                continue;
+            }
+            if (descriptor.path) {
+                loaded.push({ ...descriptor, body: await fs.readFile(descriptor.path, "utf8") });
+            }
+        }
+        return loaded;
+    }
+    async discoveryRoots() {
+        const home = os.homedir();
+        const packageRoot = path.resolve(path.dirname(fileURLToPath(import.meta.url)), "..", "..", "..");
+        const roots = [
+            { path: path.join(packageRoot, "skills"), trust: "package" },
+            { path: path.join(home, ".inferoa", "skills"), trust: "user" },
+            { path: path.join(this.workspace.root, ".inferoa", "skills"), trust: "workspace" },
+            { path: path.join(home, ".agents", "skills"), trust: "user" },
+            { path: path.join(this.workspace.root, "AGENTS.md"), trust: "imported", importer: true },
+        ];
+        roots.push(...configuredRoots("INFEROA_SKILL_ROOTS", "user"));
+        roots.push(...configuredRoots("INFEROA_INSTRUCTION_ROOTS", "imported", true));
+        roots.push(...(await findNestedSkillRoots(path.join(home, ".inferoa", "plugins", "cache"))));
+        return roots;
+    }
+    async discoverNative(root, trust) {
+        const entries = sortDirents(await fs.readdir(root, { withFileTypes: true }).catch(() => []));
+        const descriptors = [];
+        for (const entry of entries) {
+            const skillPath = entry.isDirectory() ? path.join(root, entry.name, "SKILL.md") : path.join(root, entry.name);
+            if (!skillPath.endsWith(".md") || !(await pathExists(skillPath))) {
+                continue;
+            }
+            const body = await fs.readFile(skillPath, "utf8");
+            const meta = parseFrontmatter(body);
+            const name = String(meta.name ?? path.basename(path.dirname(skillPath)));
+            descriptors.push({
+                id: slug(name),
+                name,
+                description: String(meta.description ?? firstParagraph(body) ?? ""),
+                source: root,
+                path: skillPath,
+                trust,
+                required_tools: arrayOfStrings(meta.required_tools),
+                activation: arrayOfStrings(meta.activation),
+            });
+        }
+        return descriptors;
+    }
+    async discoverImported(target, trust) {
+        const stat = await fs.stat(target).catch(() => undefined);
+        if (!stat) {
+            return [];
+        }
+        const files = stat.isDirectory()
+            ? sortDirents(await fs.readdir(target, { withFileTypes: true }))
+                .filter((entry) => entry.isFile() && /\.(md|mdc|txt)$/.test(entry.name))
+                .map((entry) => path.join(target, entry.name))
+            : [target];
+        return await Promise.all(files.map(async (file) => {
+            const body = await fs.readFile(file, "utf8");
+            const name = path.basename(file).replace(/\.(md|mdc|txt)$/, "");
+            return {
+                id: slug(name),
+                name,
+                description: firstParagraph(body) ?? "Imported local instruction",
+                source: file,
+                path: file,
+                trust,
+                required_tools: [],
+                activation: [],
+            };
+        }));
+    }
+}
+function parseFrontmatter(body) {
+    const match = /^---\n([\s\S]*?)\n---/.exec(body);
+    if (!match) {
+        return {};
+    }
+    const out = {};
+    for (const line of match[1].split(/\r?\n/)) {
+        const kv = /^([A-Za-z0-9_-]+):\s*(.*)$/.exec(line);
+        if (kv?.[1]) {
+            out[kv[1]] = kv[2]?.replace(/^["']|["']$/g, "") ?? "";
+        }
+    }
+    return out;
+}
+function firstParagraph(body) {
+    return body
+        .replace(/^---[\s\S]*?---/, "")
+        .split(/\n\s*\n/)
+        .map((part) => part.trim())
+        .find(Boolean);
+}
+function arrayOfStrings(value) {
+    return Array.isArray(value) ? value.map(String) : [];
+}
+function slug(value) {
+    return value.toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-|-$/g, "");
+}
+function dedupe(items) {
+    const seen = new Set();
+    const out = [];
+    for (const item of items) {
+        const key = `${item.id}:${item.path ?? item.source}`;
+        if (!seen.has(key)) {
+            seen.add(key);
+            out.push(item);
+        }
+    }
+    return out;
+}
+function sortDirents(entries) {
+    return entries.slice().sort((left, right) => left.name.localeCompare(right.name));
+}
+function sortDescriptors(items) {
+    return items.slice().sort((left, right) => {
+        const id = left.id.localeCompare(right.id);
+        if (id !== 0)
+            return id;
+        const source = left.source.localeCompare(right.source);
+        if (source !== 0)
+            return source;
+        return (left.path ?? "").localeCompare(right.path ?? "");
+    });
+}
+async function findNestedSkillRoots(root) {
+    const out = [];
+    async function visit(dir, depth) {
+        if (depth > 6) {
+            return;
+        }
+        const entries = sortDirents(await fs.readdir(dir, { withFileTypes: true }).catch(() => []));
+        for (const entry of entries) {
+            if (!entry.isDirectory()) {
+                continue;
+            }
+            const full = path.join(dir, entry.name);
+            if (entry.name === "skills") {
+                out.push({ path: full, trust: "user" });
+                continue;
+            }
+            await visit(full, depth + 1);
+        }
+    }
+    await visit(root, 0);
+    return out.sort((left, right) => left.path.localeCompare(right.path));
+}
+function configuredRoots(envName, trust, importer = false) {
+    return (process.env[envName] ?? "")
+        .split(path.delimiter)
+        .map((item) => item.trim())
+        .filter(Boolean)
+        .map((item) => ({
+        path: expandHome(item),
+        trust,
+        importer,
+    }));
+}
+function expandHome(value) {
+    if (value === "~") {
+        return os.homedir();
+    }
+    if (value.startsWith(`~${path.sep}`)) {
+        return path.join(os.homedir(), value.slice(2));
+    }
+    return value;
+}
+//# sourceMappingURL=registry.js.map

package/dist/src/skills/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../src/skills/registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAEzC,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAiB3C,MAAM,OAAO,aAAa;IAEL;IACA;IAFnB,YACmB,SAA4B,EAC5B,MAAuB;QADvB,cAAS,GAAT,SAAS,CAAmB;QAC5B,WAAM,GAAN,MAAM,CAAiB;IACvC,CAAC;IAEJ,KAAK,CAAC,QAAQ;QACZ,MAAM,WAAW,GAAsB,EAAE,CAAC;QAC1C,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,CAAC,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;gBACnC,SAAS;YACX,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC5E,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC;QACD,OAAO,eAAe,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,WAA+B;QAC/C,WAAW,KAAK,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACpD,MAAM,MAAM,GAAY,EAAE,CAAC;QAC3B,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;gBACjE,SAAS;YACX,CAAC;YACD,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gBACpB,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;YACnF,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAC1B,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QACjG,MAAM,KAAK,GAA4E;YACrF,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE;YAC5D,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE;YAC9D,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE;YAClF,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE;YAC7D,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE;SACzF,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC,CAAC;QAC9D,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,2BAA2B,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;QAC9E,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7F,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,IAAY,EAAE,KAA+B;QACxE,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7F,MAAM,WAAW,GAAsB,EAAE,CAAC;QAC1C,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9G,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;gBACjE,SAAS;YACX,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YAClD,MAAM,IAAI,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACzE,WAAW,CAAC,IAAI,CAAC;gBACf,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI;gBACJ,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,IAAI,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACnE,MAAM,EAAE,IAAI;gBACZ,IAAI,EAAE,SAAS;gBACf,KAAK;gBACL,cAAc,EAAE,cAAc,CAAC,IAAI,CAAC,cAAc,CAAC;gBACnD,UAAU,EAAE,cAAc,CAAC,IAAI,CAAC,UAAU,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,MAAc,EAAE,KAA+B;QAC5E,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE;YAC9B,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;iBAC3D,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;iBACvE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QACb,OAAO,MAAM,OAAO,CAAC,GAAG,CACtB,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YACvB,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;YAChE,OAAO;gBACL,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI;gBACJ,WAAW,EAAE,cAAc,CAAC,IAAI,CAAC,IAAI,4BAA4B;gBACjE,MAAM,EAAE,IAAI;gBACZ,IAAI,EAAE,IAAI;gBACV,KAAK;gBACL,cAAc,EAAE,EAAE;gBAClB,UAAU,EAAE,EAAE;aACf,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,IAAY;IACpC,MAAM,KAAK,GAAG,uBAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,MAAM,GAAG,GAAe,EAAE,CAAC;IAC3B,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,CAAC,CAAE,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5C,MAAM,EAAE,GAAG,4BAA4B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACZ,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QACxD,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO,IAAI;SACR,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC;SAC9B,KAAK,CAAC,SAAS,CAAC;SAChB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,IAAI,CAAC,OAAO,CAAC,CAAC;AACnB,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACvD,CAAC;AAED,SAAS,IAAI,CAAC,KAAa;IACzB,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;AAC/E,CAAC;AAED,SAAS,MAAM,CAAC,KAAwB;IACtC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,GAAG,GAAsB,EAAE,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QACrD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACnB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACd,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,WAAW,CAA6B,OAAY;IAC3D,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AACpF,CAAC;AAED,SAAS,eAAe,CAAC,KAAwB;IAC/C,OAAO,KAAK,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACxC,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC3C,IAAI,EAAE,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACvD,IAAI,MAAM,KAAK,CAAC;YAAE,OAAO,MAAM,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,oBAAoB,CAAC,IAAY;IAC9C,MAAM,GAAG,GAAwD,EAAE,CAAC;IACpE,KAAK,UAAU,KAAK,CAAC,GAAW,EAAE,KAAa;QAC7C,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACd,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5F,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;gBACzB,SAAS;YACX,CAAC;YACD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACxC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC5B,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;gBACxC,SAAS;YACX,CAAC;YACD,MAAM,KAAK,CAAC,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IACD,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACrB,OAAO,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AACxE,CAAC;AAED,SAAS,eAAe,CAAC,OAAe,EAAE,KAA+B,EAAE,QAAQ,GAAG,KAAK;IACzF,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;SAChC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;SACrB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,MAAM,CAAC,OAAO,CAAC;SACf,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACd,IAAI,EAAE,UAAU,CAAC,IAAI,CAAC;QACtB,KAAK;QACL,QAAQ;KACT,CAAC,CAAC,CAAC;AACR,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAC/B,IAAI,KAAK,KAAK,GAAG,EAAE,CAAC;QAClB,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC;IACtB,CAAC;IACD,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QACrC,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/src/tools/autoresearch-tools.d.ts

@@ -0,0 +1,6 @@
+import type { JsonObject, ToolResult } from "../types.js";
+import type { ToolExecutionContext } from "./context.js";
+export declare function initExperiment(args: JsonObject, context: ToolExecutionContext): Promise<ToolResult>;
+export declare function runExperiment(args: JsonObject, context: ToolExecutionContext): Promise<ToolResult>;
+export declare function logExperiment(args: JsonObject, context: ToolExecutionContext): Promise<ToolResult>;
+export declare function updateNotes(args: JsonObject, context: ToolExecutionContext): Promise<ToolResult>;

package/dist/src/tools/autoresearch-tools.js

@@ -0,0 +1,412 @@
+import { spawn } from "node:child_process";
+import path from "node:path";
+import { access } from "node:fs/promises";
+import { fail, ok, truncateText } from "../util/limit.js";
+import { createExperiment, logPendingRun, parseAsiLines, parseMetricLines, readAutoresearchState, recordRun, setAutoresearchMode, summarizeAutoresearchProgress, writeAutoresearchState, } from "../autoresearch/state.js";
+const HARNESS = "autoresearch.sh";
+const HARNESS_COMMAND = `bash ${HARNESS}`;
+const HARNESS_VALIDATION_TIMEOUT_MS = 60_000;
+const HARNESS_TIMEOUT_GRACE_MS = 500;
+const HARNESS_OUTPUT_LIMIT_BYTES = 2 * 1024 * 1024;
+export async function initExperiment(args, context) {
+    try {
+        await access(path.join(context.workspace.root, HARNESS));
+        const state = ensureAutoresearchEnabled(context);
+        const primaryMetric = requiredString(args.primary_metric, "primary_metric");
+        const shouldValidate = booleanArg(args.validate_harness) !== false;
+        let harnessStatus;
+        if (shouldValidate) {
+            harnessStatus = await validateHarness(context, primaryMetric);
+            if (!harnessStatus.ok) {
+                return fail("harness_validation_failed", harnessStatus.message, autoresearchFailureData(state, {
+                    harness_status: harnessStatus,
+                }));
+            }
+            if (harnessStatus.parsed_primary === null) {
+                return fail("harness_metric_missing", `Harness did not print required METRIC ${primaryMetric}=value.`, autoresearchFailureData(state, {
+                    harness_status: harnessStatus,
+                }));
+            }
+        }
+        const experiment = createExperiment({
+            name: requiredString(args.name, "name"),
+            goal: stringArg(args.goal) ?? state.goal,
+            primary_metric: primaryMetric,
+            metric_unit: stringArg(args.metric_unit),
+            direction: directionArg(args.direction),
+            scope_paths: stringArrayArg(args.scope_paths),
+            off_limits: stringArrayArg(args.off_limits),
+            constraints: stringArrayArg(args.constraints),
+            max_iterations: positiveIntArg(args.max_iterations),
+            harness_status: harnessStatus,
+        });
+        const next = writeAutoresearchState(context.store, context.session_id, {
+            enabled: true,
+            goal: experiment.goal ?? state.goal,
+            experiment,
+        }, context.run_id);
+        return ok(`Autoresearch initialized: ${experiment.name}\nMetric: ${experiment.primary_metric}\nBenchmark: ${HARNESS_COMMAND}`, {
+            autoresearch: next,
+            progress: summarizeAutoresearchProgress(experiment),
+            harness_status: harnessStatus,
+        });
+    }
+    catch (error) {
+        return fail("init_experiment_failed", error instanceof Error ? error.message : String(error));
+    }
+}
+export async function runExperiment(args, context) {
+    const state = readAutoresearchState(context.store, context.session_id);
+    if (!state.enabled || !state.experiment) {
+        return fail("autoresearch_not_initialized", "no active autoresearch experiment; call /autoresearch and init_experiment first", autoresearchFailureData(state));
+    }
+    if (state.experiment.pending_run) {
+        return fail("autoresearch_pending_run", `run ${state.experiment.pending_run.id} is still pending; call log_experiment before starting another run`, autoresearchFailureData(state, { pending_run: state.experiment.pending_run }));
+    }
+    let timeoutMs;
+    try {
+        timeoutMs = timeoutArg(args);
+    }
+    catch (error) {
+        return fail("autoresearch_timeout_invalid", error instanceof Error ? error.message : String(error), autoresearchFailureData(state));
+    }
+    const started = Date.now();
+    const result = await runHarness(context.workspace.root, timeoutMs);
+    const durationMs = Date.now() - started;
+    const resource = context.store.putResource(context.session_id, "autoresearch.run.output", result.output, {
+        command: HARNESS_COMMAND,
+        exit_code: result.exitCode,
+        duration_ms: durationMs,
+        timed_out: result.timedOut,
+        output_truncated: result.outputTruncated,
+    });
+    const parsedMetrics = result.parsedMetrics;
+    const parsedPrimary = parsedMetrics[state.experiment.primary_metric] ?? null;
+    const nextExperiment = recordRun(state.experiment, {
+        command: HARNESS_COMMAND,
+        exit_code: result.exitCode,
+        duration_ms: durationMs,
+        output_resource_uri: resource.uri,
+        timed_out: result.timedOut,
+        output_truncated: result.outputTruncated,
+        parsed_metrics: parsedMetrics,
+        parsed_primary: parsedPrimary,
+        asi: result.asi,
+        completed_at: new Date().toISOString(),
+    });
+    const next = writeAutoresearchState(context.store, context.session_id, { ...state, experiment: nextExperiment }, context.run_id);
+    const preview = truncateText(result.output, 4000).text;
+    const status = result.timedOut ? `timed out exit=${result.exitCode}` : result.exitCode === 0 ? "passed" : `failed exit=${result.exitCode}`;
+    return ok(`Run ${nextExperiment.pending_run?.id} ${status} in ${(durationMs / 1000).toFixed(1)}s`, {
+        command: HARNESS_COMMAND,
+        exit_code: result.exitCode,
+        duration_ms: durationMs,
+        timed_out: result.timedOut,
+        output_truncated: result.outputTruncated,
+        output_resource_uri: resource.uri,
+        parsed_metrics: parsedMetrics,
+        parsed_primary: parsedPrimary,
+        output_preview: preview,
+        autoresearch: next,
+        progress: summarizeAutoresearchProgress(nextExperiment),
+    });
+}
+async function validateHarness(context, primaryMetric) {
+    const started = Date.now();
+    const result = await runHarness(context.workspace.root, HARNESS_VALIDATION_TIMEOUT_MS);
+    const durationMs = Date.now() - started;
+    const parsedMetrics = result.parsedMetrics;
+    const parsedPrimary = parsedMetrics[primaryMetric] ?? null;
+    const resource = context.store.putResource(context.session_id, "autoresearch.harness.validation", result.output, {
+        command: HARNESS_COMMAND,
+        exit_code: result.exitCode,
+        duration_ms: durationMs,
+        primary_metric: primaryMetric,
+        timed_out: result.timedOut,
+        output_truncated: result.outputTruncated,
+    });
+    const ok = result.exitCode === 0;
+    const metricMessage = parsedPrimary === null ? `missing METRIC ${primaryMetric}=value` : `${primaryMetric}=${parsedPrimary}`;
+    return {
+        ok,
+        command: HARNESS_COMMAND,
+        exit_code: result.exitCode,
+        duration_ms: durationMs,
+        parsed_metrics: parsedMetrics,
+        parsed_primary: parsedPrimary,
+        output_resource_uri: resource.uri,
+        timed_out: result.timedOut,
+        output_truncated: result.outputTruncated,
+        message: result.timedOut ? `harness timed out; ${metricMessage}` : ok ? `validated ${metricMessage}` : `harness exited ${result.exitCode}; ${metricMessage}`,
+        validated_at: new Date().toISOString(),
+    };
+}
+export async function logExperiment(args, context) {
+    let state;
+    try {
+        state = readAutoresearchState(context.store, context.session_id);
+        if (!state.enabled || !state.experiment) {
+            return fail("autoresearch_not_initialized", "no active autoresearch experiment; call init_experiment first", autoresearchFailureData(state));
+        }
+        if (!state.experiment.pending_run) {
+            return fail("log_experiment_failed", "no pending autoresearch run; call run_experiment first", autoresearchFailureData(state));
+        }
+        const status = statusArg(args.status);
+        const nextExperiment = logPendingRun(state.experiment, {
+            status,
+            metric: metricArg(args.metric, state.experiment.pending_run.parsed_primary, status),
+            description: requiredString(args.description, "description"),
+            metrics: numericRecordArg(args.metrics),
+            asi: objectArg(args.asi),
+        });
+        const reachedLimit = nextExperiment.max_iterations !== undefined &&
+            nextExperiment.results.filter((result) => result.status === "keep").length >= nextExperiment.max_iterations;
+        const next = writeAutoresearchState(context.store, context.session_id, { ...state, enabled: reachedLimit ? false : state.enabled, experiment: nextExperiment }, context.run_id);
+        if (reachedLimit) {
+            setAutoresearchMode(context.store, context.session_id, { mode: "off", goal: state.goal }, context.run_id);
+        }
+        const latest = nextExperiment.results.at(-1);
+        return ok(`Logged run ${latest.run_id}: ${latest.status} ${nextExperiment.primary_metric}=${latest.metric === null ? "missing" : latest.metric}`, {
+            result: latest,
+            autoresearch: next,
+            progress: summarizeAutoresearchProgress(nextExperiment),
+        });
+    }
+    catch (error) {
+        return fail("log_experiment_failed", error instanceof Error ? error.message : String(error), state ? autoresearchFailureData(state) : undefined);
+    }
+}
+export async function updateNotes(args, context) {
+    const state = readAutoresearchState(context.store, context.session_id);
+    if (!state.enabled || !state.experiment) {
+        return fail("autoresearch_not_initialized", "no active autoresearch experiment; call init_experiment first", autoresearchFailureData(state));
+    }
+    const body = stringArg(args.body) ?? state.experiment.notes;
+    const appendIdea = stringArg(args.append_idea);
+    const notes = appendIdea ? appendIdeaToNotes(body, appendIdea) : body;
+    const experiment = { ...state.experiment, notes };
+    const next = writeAutoresearchState(context.store, context.session_id, { ...state, experiment }, context.run_id);
+    return ok(appendIdea ? "Autoresearch idea appended." : "Autoresearch notes updated.", {
+        notes,
+        autoresearch: next,
+    });
+}
+function autoresearchFailureData(state, extra = {}) {
+    return {
+        autoresearch: state,
+        ...(state.experiment ? { progress: summarizeAutoresearchProgress(state.experiment) } : {}),
+        ...extra,
+    };
+}
+function ensureAutoresearchEnabled(context) {
+    const state = readAutoresearchState(context.store, context.session_id);
+    if (state.enabled) {
+        return state;
+    }
+    return setAutoresearchMode(context.store, context.session_id, { mode: "on", goal: state.goal }, context.run_id);
+}
+function runHarness(cwd, timeoutMs) {
+    return new Promise((resolve) => {
+        const child = spawn("bash", [HARNESS], { cwd, detached: true, stdio: ["ignore", "pipe", "pipe"] });
+        let output = "";
+        let outputBytes = 0;
+        let outputTruncated = false;
+        let parseBuffer = "";
+        const parsedMetrics = {};
+        const asi = {};
+        let timedOut = false;
+        let settled = false;
+        let hardKillTimer;
+        const appendNotice = (message) => {
+            output += `${output.endsWith("\n") || output.length === 0 ? "" : "\n"}${message}\n`;
+        };
+        const markTruncated = () => {
+            if (!outputTruncated) {
+                outputTruncated = true;
+                appendNotice(`[autoresearch output truncated at ${HARNESS_OUTPUT_LIMIT_BYTES} bytes]`);
+            }
+        };
+        const appendOutput = (chunk) => {
+            const buffer = Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk, "utf8");
+            parseHarnessChunk(buffer.toString("utf8"));
+            if (outputBytes >= HARNESS_OUTPUT_LIMIT_BYTES) {
+                markTruncated();
+                return;
+            }
+            const remaining = HARNESS_OUTPUT_LIMIT_BYTES - outputBytes;
+            if (buffer.length <= remaining) {
+                output += buffer.toString("utf8");
+                outputBytes += buffer.length;
+                return;
+            }
+            output += buffer.subarray(0, remaining).toString("utf8");
+            outputBytes = HARNESS_OUTPUT_LIMIT_BYTES;
+            markTruncated();
+        };
+        const parseHarnessChunk = (text) => {
+            parseBuffer += text;
+            const lines = parseBuffer.split(/\r?\n/);
+            parseBuffer = lines.pop() ?? "";
+            for (const line of lines) {
+                Object.assign(parsedMetrics, parseMetricLines(line));
+                Object.assign(asi, parseAsiLines(line));
+            }
+        };
+        const flushHarnessParser = () => {
+            if (!parseBuffer.trim()) {
+                return;
+            }
+            Object.assign(parsedMetrics, parseMetricLines(parseBuffer));
+            Object.assign(asi, parseAsiLines(parseBuffer));
+            parseBuffer = "";
+        };
+        const signalHarness = (signal) => {
+            if (!child.pid) {
+                return;
+            }
+            try {
+                process.kill(-child.pid, signal);
+            }
+            catch {
+                try {
+                    child.kill(signal);
+                }
+                catch {
+                    // The process may already have exited between timeout and cleanup.
+                }
+            }
+        };
+        const finish = (exitCode) => {
+            if (settled) {
+                return;
+            }
+            settled = true;
+            clearTimeout(timeout);
+            if (hardKillTimer) {
+                clearTimeout(hardKillTimer);
+            }
+            flushHarnessParser();
+            resolve({ exitCode, output, timedOut, outputTruncated, parsedMetrics, asi });
+        };
+        const timeout = setTimeout(() => {
+            timedOut = true;
+            appendNotice(`[autoresearch timed out after ${timeoutMs}ms]`);
+            signalHarness("SIGTERM");
+            hardKillTimer = setTimeout(() => {
+                appendNotice(`[autoresearch hard kill after ${timeoutMs + HARNESS_TIMEOUT_GRACE_MS}ms]`);
+                signalHarness("SIGKILL");
+            }, HARNESS_TIMEOUT_GRACE_MS);
+            hardKillTimer.unref();
+        }, timeoutMs);
+        timeout.unref();
+        child.stdout.on("data", (chunk) => {
+            appendOutput(chunk);
+        });
+        child.stderr.on("data", (chunk) => {
+            appendOutput(chunk);
+        });
+        child.on("error", (error) => {
+            appendNotice(error.message);
+            finish(127);
+        });
+        child.on("close", (code) => {
+            finish(code);
+        });
+    });
+}
+function requiredString(value, name) {
+    const text = stringArg(value)?.trim();
+    if (!text) {
+        throw new Error(`${name} is required`);
+    }
+    return text;
+}
+function stringArg(value) {
+    return typeof value === "string" ? value : undefined;
+}
+function stringArrayArg(value) {
+    return Array.isArray(value) ? value.filter((item) => typeof item === "string") : undefined;
+}
+function directionArg(value) {
+    return value === "lower" || value === "higher" ? value : undefined;
+}
+function statusArg(value) {
+    if (value === "keep" || value === "discard" || value === "crash" || value === "checks_failed") {
+        return value;
+    }
+    throw new Error("status must be keep, discard, crash, or checks_failed");
+}
+function positiveIntArg(value) {
+    if (value === undefined) {
+        return undefined;
+    }
+    if (typeof value !== "number" || !Number.isInteger(value) || value <= 0) {
+        throw new Error("max_iterations must be a positive integer when provided");
+    }
+    return value;
+}
+function finiteNumberArg(value, name) {
+    if (typeof value !== "number" || !Number.isFinite(value)) {
+        throw new Error(`${name} must be a finite number`);
+    }
+    return value;
+}
+function metricArg(value, fallback, status) {
+    if (value !== undefined) {
+        return finiteNumberArg(value, "metric");
+    }
+    if (typeof fallback === "number" && Number.isFinite(fallback)) {
+        return fallback;
+    }
+    if (status !== "keep") {
+        return null;
+    }
+    throw new Error("metric is required when the pending run did not parse the primary metric");
+}
+function timeoutArg(args) {
+    const timeoutMs = args.timeout_ms;
+    if (timeoutMs !== undefined) {
+        if (typeof timeoutMs === "number" && Number.isFinite(timeoutMs) && timeoutMs > 0) {
+            return Math.trunc(timeoutMs);
+        }
+        throw new Error("timeout_ms must be a positive finite number when provided");
+    }
+    const timeoutSeconds = args.timeout_seconds;
+    if (timeoutSeconds !== undefined) {
+        if (typeof timeoutSeconds === "number" && Number.isFinite(timeoutSeconds) && timeoutSeconds > 0) {
+            return Math.trunc(timeoutSeconds * 1000);
+        }
+        throw new Error("timeout_seconds must be a positive finite number when provided");
+    }
+    return 600_000;
+}
+function numericRecordArg(value) {
+    if (!value || typeof value !== "object" || Array.isArray(value)) {
+        return undefined;
+    }
+    const out = {};
+    for (const [key, raw] of Object.entries(value)) {
+        if (typeof raw === "number" && Number.isFinite(raw)) {
+            out[key] = raw;
+        }
+    }
+    return out;
+}
+function objectArg(value) {
+    return value && typeof value === "object" && !Array.isArray(value) ? value : undefined;
+}
+function booleanArg(value) {
+    return typeof value === "boolean" ? value : undefined;
+}
+function appendIdeaToNotes(notes, idea) {
+    const trimmed = notes.trimEnd();
+    const bullet = `- ${idea.trim()}`;
+    if (!trimmed) {
+        return `## Ideas\n${bullet}\n`;
+    }
+    if (!trimmed.includes("## Ideas")) {
+        return `${trimmed}\n\n## Ideas\n${bullet}\n`;
+    }
+    return `${trimmed}\n${bullet}\n`;
+}
+//# sourceMappingURL=autoresearch-tools.js.map