index.mjs

import { spawnSync } from "node:child_process";
import path from "node:path";
import { fileURLToPath } from "node:url";

import { z } from "zod";

const HERE = path.dirname(fileURLToPath(import.meta.url));
const COMPANION = path.resolve(HERE, "./scripts/polycli-companion.bundle.mjs");

function tool(input) {
  return input;
}

tool.schema = z;

function runCompanion(argv) {
  const result = spawnSync(process.execPath, [COMPANION, ...argv], {
    cwd: process.cwd(),
    encoding: "utf8",
  });
  if (result.error) {
    throw result.error;
  }
  if (result.status !== 0 && !result.stdout) {
    const detail = String(result.stderr || "").trim() || `polycli companion exited with status ${result.status}`;
    const error = new Error(detail);
    error.status = result.status;
    error.stderr = result.stderr;
    throw error;
  }
  return result.stdout;
}

export const PolycliPlugin = async () => ({
  tool: {
    polycli_run: tool({
      description: "Run a polycli companion subcommand such as setup, health, ask, rescue, review, adversarial-review, status, result, cancel, timing, debug, or sessions.",
      args: {
        argv: tool.schema.array(tool.schema.string()),
      },
      async execute(args) {
        return runCompanion(args.argv);
      },
    }),
    polycli_timing: tool({
      description: "Read stored polycli timing history and aggregate summaries.",
      args: {
        provider: tool.schema.string().optional(),
        history: tool.schema.number().optional(),
        json: tool.schema.boolean().optional(),
      },
      async execute(args) {
        const argv = ["timing"];
        if (args.provider) argv.push("--provider", args.provider);
        if (args.history != null) argv.push("--history", String(args.history));
        if (args.json) argv.push("--json");
        return runCompanion(argv);
      },
    }),
  },
});