@osolmaz/pi-workflows 0.8.1 → 0.9.0

package/src/herdr/setup.ts

@@ -0,0 +1,93 @@
+import { spawnSync, type SpawnSyncReturns } from "node:child_process";
+import fs from "node:fs";
+import path from "node:path";
+import { HERDR_PLUGIN_ID } from "./constants.js";
+
+export type HerdrSetupResult = {
+  changed: boolean;
+  message: string;
+};
+
+type Spawn = (command: string, args: readonly string[]) => SpawnSyncReturns<string>;
+
+export function setupHerdrPlugin(packageRoot: string, spawn: Spawn = runCommand): HerdrSetupResult {
+  const root = path.resolve(packageRoot);
+  const manifest = path.join(root, "herdr-plugin.toml");
+  if (!fs.existsSync(manifest)) {
+    throw new Error(`Herdr plugin manifest not found: ${manifest}`);
+  }
+
+  const listed = spawn("herdr", ["plugin", "list", "--plugin", HERDR_PLUGIN_ID, "--json"]);
+  if (listed.error) throw new Error(`Could not run Herdr: ${listed.error.message}`);
+  if (listed.status !== 0) {
+    throw new Error(`Could not inspect Herdr plugins: ${bounded(listed.stderr || listed.stdout)}`);
+  }
+  const installed = installedPlugin(listed.stdout);
+  if (installed !== undefined) {
+    if (path.resolve(installed.root) !== root) {
+      throw new Error(
+        `Herdr plugin ${HERDR_PLUGIN_ID} is already registered from ${installed.root}. Unlink it before linking ${root}.`,
+      );
+    }
+    if (installed.enabled) {
+      return { changed: false, message: `Herdr plugin ${HERDR_PLUGIN_ID} is already linked.` };
+    }
+    const enabled = spawn("herdr", ["plugin", "enable", HERDR_PLUGIN_ID]);
+    if (enabled.error) throw new Error(`Could not run Herdr: ${enabled.error.message}`);
+    if (enabled.status !== 0) {
+      throw new Error(
+        `Could not enable the Herdr plugin: ${bounded(enabled.stderr || enabled.stdout)}`,
+      );
+    }
+    return { changed: true, message: `Enabled Herdr plugin ${HERDR_PLUGIN_ID}.` };
+  }
+
+  const linked = spawn("herdr", ["plugin", "link", root]);
+  if (linked.error) throw new Error(`Could not run Herdr: ${linked.error.message}`);
+  if (linked.status !== 0) {
+    throw new Error(`Could not link the Herdr plugin: ${bounded(linked.stderr || linked.stdout)}`);
+  }
+  return { changed: true, message: `Linked Herdr plugin ${HERDR_PLUGIN_ID} from ${root}.` };
+}
+
+function installedPlugin(stdout: string): { root: string; enabled: boolean } | undefined {
+  let value: unknown;
+  try {
+    value = JSON.parse(stdout) as unknown;
+  } catch {
+    throw new Error("Herdr returned invalid plugin JSON.");
+  }
+  if (!isRecord(value) || !isRecord(value.result) || !Array.isArray(value.result.plugins)) {
+    throw new Error("Herdr returned an invalid plugin list.");
+  }
+  for (const plugin of value.result.plugins) {
+    if (
+      isRecord(plugin) &&
+      plugin.plugin_id === HERDR_PLUGIN_ID &&
+      typeof plugin.plugin_root === "string" &&
+      typeof plugin.enabled === "boolean"
+    ) {
+      return { root: plugin.plugin_root, enabled: plugin.enabled };
+    }
+  }
+  return undefined;
+}
+
+function runCommand(command: string, args: readonly string[]): SpawnSyncReturns<string> {
+  return spawnSync(command, [...args], {
+    encoding: "utf8",
+    stdio: ["ignore", "pipe", "pipe"],
+  });
+}
+
+function bounded(value: string): string {
+  const compact = value
+    .replace(/[\r\n\t]+/gu, " ")
+    .replace(/ +/gu, " ")
+    .trim();
+  return compact.length <= 300 ? compact : `${compact.slice(0, 299)}…`;
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}

package/src/viewer/cli.ts

@@ -1,9 +1,10 @@
 #!/usr/bin/env node
 import fs, { realpathSync } from "node:fs";
 import path from "node:path";
-import { pathToFileURL } from "node:url";
+import { fileURLToPath, pathToFileURL } from "node:url";
 import { SqliteControllerStore } from "../controllers/sqlite.js";
 import { projectControllerStoreBaseDir } from "../controllers/store.js";
+import { setupHerdrPlugin } from "../herdr/setup.js";
 import { sanitizeText } from "../render/ansi.js";
 import { listRunBundles, readRunBundle, workflowRunsBaseDir } from "../workflows/store.js";
 import {
@@ -23,6 +24,7 @@ Usage:
   pi-workflows controllers [--controller-dir <dir>]
   pi-workflows controller <controller> <key> [--controller-dir <dir>]
   pi-workflows host [--project <dir>] [-- <extra pi args>]
+  pi-workflows herdr setup
 
 Commands:
   view          Open the live workflow TUI. With --once, print a snapshot.
@@ -30,6 +32,7 @@ Commands:
   controllers   List durable controller resources.
   controller    Show one resource, its effects, child workflows, and events.
   host          Run the always-on workflow host in the foreground.
+  herdr         Set up the bundled Herdr plugin.
 
 Options:
   --dir <runsDir>          Runs directory (default: ~/.pi/agent/workflows/runs)
@@ -43,6 +46,7 @@ export type CliArgs = {
   runId?: string;
   controllerName?: string;
   resourceKey?: string;
+  herdrAction?: string;
   dir: string;
   controllerDir: string;
   once: boolean;
@@ -98,6 +102,12 @@ export function parseCliArgs(argv: string[]): CliArgs {
       once,
     };
   }
+  if (command === "herdr") {
+    if (positionals.length !== 1 || positionals[0] !== "setup") {
+      throw new Error("herdr requires the setup action");
+    }
+    return { command, herdrAction: positionals[0], dir, controllerDir, once };
+  }
   if (positionals.length > 1) {
     throw new Error(`Unexpected argument: ${positionals[1]}`);
   }
@@ -229,6 +239,11 @@ export async function main(argv: string[] = process.argv.slice(2)): Promise<numb
     if (args.command === "host") {
       return await runHost(args.project ?? process.cwd(), args.piArgs);
     }
+    if (args.command === "herdr") {
+      const result = setupHerdrPlugin(packageRoot());
+      process.stdout.write(`${result.message}\n`);
+      return 0;
+    }
     if (args.command === "view") {
       if (args.once || !process.stdout.isTTY) {
         await printOnce(args.dir, args.runId);
@@ -271,6 +286,10 @@ function openControllerStore(controllerDir: string): SqliteControllerStore | und
   return new SqliteControllerStore(file, { readOnly: true });
 }
 
+function packageRoot(): string {
+  return path.resolve(path.dirname(fileURLToPath(import.meta.url)), "../..");
+}
+
 function requiredValue(args: string[], option: string): string {
   const value = args.shift();
   if (!value) {