@kb-labs/agent-cli 0.5.0

package/dist/cli/commands/rollback.js

@@ -0,0 +1,109 @@
+import { defineCommand, useLogger } from '@kb-labs/sdk';
+import { ChangeStore } from '@kb-labs/agent-history';
+import { promises } from 'fs';
+import path from 'path';
+
+// src/cli/commands/rollback.ts
+var rollback_default = defineCommand({
+  id: "rollback",
+  description: "Rollback file changes made by agents in a session or run",
+  handler: {
+    async execute(ctx, input) {
+      const logger = useLogger();
+      const flags = input.flags ?? input;
+      const sessionId = flags["session-id"] ?? flags["sessionId"];
+      const runId = flags["run-id"] ?? flags["runId"];
+      const filePath = flags["file"];
+      const dryRun = Boolean(flags["dry-run"] ?? flags["dryRun"]);
+      const asJson = Boolean(flags["json"]);
+      if (!sessionId) {
+        return output(ctx, asJson, { success: false, error: "Missing --session-id" }, 1);
+      }
+      const workingDir = process.cwd();
+      const store = new ChangeStore(workingDir);
+      try {
+        let changes;
+        if (runId && !filePath) {
+          changes = await store.listRun(sessionId, runId);
+        } else if (filePath) {
+          changes = await store.listFile(sessionId, filePath);
+          if (runId) {
+            changes = changes.filter((c) => c.runId === runId);
+          }
+        } else {
+          changes = await store.listSession(sessionId);
+        }
+        if (changes.length === 0) {
+          return output(ctx, asJson, { success: true, rolledBack: 0, message: "No changes to rollback" }, 0);
+        }
+        const ordered = [...changes].sort(
+          (a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime()
+        );
+        if (dryRun) {
+          const preview = ordered.map((c) => ({
+            filePath: c.filePath,
+            operation: c.operation,
+            action: !c.before ? "delete (was new file)" : "restore to before state"
+          }));
+          return output(ctx, asJson, { success: true, dryRun: true, wouldRollback: preview }, 0);
+        }
+        let rolledBack = 0;
+        const errors = [];
+        for (const change of ordered) {
+          const fullPath = path.join(workingDir, change.filePath);
+          try {
+            if (!change.before) {
+              await promises.unlink(fullPath).catch((e) => {
+                if (e.code !== "ENOENT") {
+                  throw e;
+                }
+              });
+            } else {
+              await promises.mkdir(path.dirname(fullPath), { recursive: true });
+              await promises.writeFile(fullPath, change.before.content, "utf-8");
+            }
+            rolledBack++;
+          } catch (e) {
+            errors.push(`${change.filePath}: ${String(e)}`);
+          }
+        }
+        return output(ctx, asJson, { success: errors.length === 0, rolledBack, errors }, errors.length > 0 ? 1 : 0);
+      } catch (e) {
+        logger.error("agent:rollback error", e instanceof Error ? e : void 0);
+        return output(ctx, asJson, { success: false, error: String(e) }, 1);
+      }
+    }
+  }
+});
+function output(ctx, asJson, data, exitCode) {
+  if (asJson) {
+    ctx.ui.write(JSON.stringify(data, null, 2) + "\n");
+  } else {
+    const d = data;
+    if (d["success"]) {
+      if (d["dryRun"]) {
+        ctx.ui.write(`[DRY RUN] Would rollback:
+`);
+        for (const p of d["wouldRollback"] ?? []) {
+          ctx.ui.write(`  ${p["operation"]?.toUpperCase()} ${p["filePath"]} \u2192 ${p["action"]}
+`);
+        }
+      } else {
+        ctx.ui.write(`\u2705 Rolled back ${d["rolledBack"]} change(s)
+`);
+        for (const e of d["errors"] ?? []) {
+          ctx.ui.write(`  \u26A0\uFE0F  ${e}
+`);
+        }
+      }
+    } else {
+      ctx.ui.write(`\u274C ${d["error"]}
+`);
+    }
+  }
+  return { exitCode };
+}
+
+export { rollback_default as default };
+//# sourceMappingURL=rollback.js.map
+//# sourceMappingURL=rollback.js.map

package/dist/cli/commands/rollback.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/cli/commands/rollback.ts"],"names":["fsp"],"mappings":";;;;;;AA+BA,IAAO,mBAAQ,aAAA,CAAc;AAAA,EAC3B,EAAA,EAAI,UAAA;AAAA,EACJ,WAAA,EAAa,0DAAA;AAAA,EAEb,OAAA,EAAS;AAAA,IACP,MAAM,OAAA,CAAQ,GAAA,EAAsB,KAAA,EAAqD;AACvF,MAAA,MAAM,SAAS,SAAA,EAAU;AACzB,MAAA,MAAM,KAAA,GAAU,MAAkC,KAAA,IAAS,KAAA;AAE3D,MAAA,MAAM,SAAA,GAAa,KAAA,CAAM,YAAY,CAAA,IAAK,MAAM,WAAW,CAAA;AAC3D,MAAA,MAAM,KAAA,GAAS,KAAA,CAAM,QAAQ,CAAA,IAAK,MAAM,OAAO,CAAA;AAC/C,MAAA,MAAM,QAAA,GAAW,MAAM,MAAM,CAAA;AAC7B,MAAA,MAAM,SAAS,OAAA,CAAQ,KAAA,CAAM,SAAS,CAAA,IAAK,KAAA,CAAM,QAAQ,CAAC,CAAA;AAC1D,MAAA,MAAM,MAAA,GAAS,OAAA,CAAQ,KAAA,CAAM,MAAM,CAAC,CAAA;AAEpC,MAAA,IAAI,CAAC,SAAA,EAAW;AACd,QAAA,OAAO,MAAA,CAAO,KAAK,MAAA,EAAQ,EAAE,SAAS,KAAA,EAAO,KAAA,EAAO,sBAAA,EAAuB,EAAG,CAAC,CAAA;AAAA,MACjF;AAEA,MAAA,MAAM,UAAA,GAAa,QAAQ,GAAA,EAAI;AAC/B,MAAA,MAAM,KAAA,GAAQ,IAAI,WAAA,CAAY,UAAU,CAAA;AAExC,MAAA,IAAI;AACF,QAAA,IAAI,OAAA;AAEJ,QAAA,IAAI,KAAA,IAAS,CAAC,QAAA,EAAU;AACtB,UAAA,OAAA,GAAU,MAAM,KAAA,CAAM,OAAA,CAAQ,SAAA,EAAW,KAAK,CAAA;AAAA,QAChD,WAAW,QAAA,EAAU;AACnB,UAAA,OAAA,GAAU,MAAM,KAAA,CAAM,QAAA,CAAS,SAAA,EAAW,QAAQ,CAAA;AAClD,UAAA,IAAI,KAAA,EAAO;AACT,YAAA,OAAA,GAAU,QAAQ,MAAA,CAAO,CAAC,CAAA,KAAM,CAAA,CAAE,UAAU,KAAK,CAAA;AAAA,UACnD;AAAA,QACF,CAAA,MAAO;AACL,UAAA,OAAA,GAAU,MAAM,KAAA,CAAM,WAAA,CAAY,SAAS,CAAA;AAAA,QAC7C;AAEA,QAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,UAAA,OAAO,MAAA,CAAO,GAAA,EAAK,MAAA,EAAQ,EAAE,OAAA,EAAS,IAAA,EAAM,UAAA,EAAY,CAAA,EAAG,OAAA,EAAS,wBAAA,EAAyB,EAAG,CAAC,CAAA;AAAA,QACnG;AAGA,QAAA,MAAM,OAAA,GAAU,CAAC,GAAG,OAAO,CAAA,CAAE,IAAA;AAAA,UAC3B,CAAC,CAAA,EAAG,CAAA,KAAM,IAAI,KAAK,CAAA,CAAE,SAAS,CAAA,CAAE,OAAA,KAAY,IAAI,IAAA,CAAK,CAAA,CAAE,SAAS,EAAE,OAAA;AAAQ,SAC5E;AAEA,QAAA,IAAI,MAAA,EAAQ;AACV,UAAA,MAAM,OAAA,GAAU,OAAA,CAAQ,GAAA,CAAI,CAAC,CAAA,MAAO;AAAA,YAClC,UAAU,CAAA,CAAE,QAAA;AAAA,YACZ,WAAW,CAAA,CAAE,SAAA;AAAA,YACb,MAAA,EAAQ,CAAC,CAAA,CAAE,MAAA,GAAS,uBAAA,GAA0B;AAAA,WAChD,CAAE,CAAA;AACF,UAAA,OAAO,MAAA,CAAO,GAAA,EAAK,MAAA,EAAQ,EAAE,OAAA,EAAS,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,aAAA,EAAe,OAAA,EAAQ,EAAG,CAAC,CAAA;AAAA,QACvF;AAEA,QAAA,IAAI,UAAA,GAAa,CAAA;AACjB,QAAA,MAAM,SAAmB,EAAC;AAE1B,QAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC5B,UAAA,MAAM,QAAA,GAAW,IAAA,CAAK,IAAA,CAAK,UAAA,EAAY,OAAO,QAAQ,CAAA;AACtD,UAAA,IAAI;AACF,YAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAElB,cAAA,MAAMA,SAAI,MAAA,CAAO,QAAQ,CAAA,CAAE,KAAA,CAAM,CAAC,CAAA,KAA6B;AAC7D,gBAAA,IAAI,CAAA,CAAE,SAAS,QAAA,EAAU;AAAC,kBAAA,MAAM,CAAA;AAAA,gBAAE;AAAA,cACpC,CAAC,CAAA;AAAA,YACH,CAAA,MAAO;AAEL,cAAA,MAAMA,QAAA,CAAI,MAAM,IAAA,CAAK,OAAA,CAAQ,QAAQ,CAAA,EAAG,EAAE,SAAA,EAAW,IAAA,EAAM,CAAA;AAC3D,cAAA,MAAMA,SAAI,SAAA,CAAU,QAAA,EAAU,MAAA,CAAO,MAAA,CAAO,SAAS,OAAO,CAAA;AAAA,YAC9D;AACA,YAAA,UAAA,EAAA;AAAA,UACF,SAAS,CAAA,EAAG;AACV,YAAA,MAAA,CAAO,IAAA,CAAK,GAAG,MAAA,CAAO,QAAQ,KAAK,MAAA,CAAO,CAAC,CAAC,CAAA,CAAE,CAAA;AAAA,UAChD;AAAA,QACF;AAEA,QAAA,OAAO,MAAA,CAAO,GAAA,EAAK,MAAA,EAAQ,EAAE,SAAS,MAAA,CAAO,MAAA,KAAW,CAAA,EAAG,UAAA,EAAY,QAAO,EAAG,MAAA,CAAO,MAAA,GAAS,CAAA,GAAI,IAAI,CAAC,CAAA;AAAA,MAC5G,SAAS,CAAA,EAAG;AACV,QAAA,MAAA,CAAO,KAAA,CAAM,sBAAA,EAAwB,CAAA,YAAa,KAAA,GAAQ,IAAI,MAAS,CAAA;AACvE,QAAA,OAAO,MAAA,CAAO,GAAA,EAAK,MAAA,EAAQ,EAAE,OAAA,EAAS,KAAA,EAAO,KAAA,EAAO,MAAA,CAAO,CAAC,CAAA,EAAE,EAAG,CAAC,CAAA;AAAA,MACpE;AAAA,IACF;AAAA;AAEJ,CAAC;AAED,SAAS,MAAA,CAAO,GAAA,EAAsB,MAAA,EAAiB,IAAA,EAAe,QAAA,EAAwC;AAC5G,EAAA,IAAI,MAAA,EAAQ;AACV,IAAA,GAAA,CAAI,EAAA,CAAG,MAAM,IAAA,CAAK,SAAA,CAAU,MAAM,IAAA,EAAM,CAAC,IAAI,IAAI,CAAA;AAAA,EACnD,CAAA,MAAO;AACL,IAAA,MAAM,CAAA,GAAI,IAAA;AACV,IAAA,IAAI,CAAA,CAAE,SAAS,CAAA,EAAG;AAChB,MAAA,IAAI,CAAA,CAAE,QAAQ,CAAA,EAAG;AACf,QAAA,GAAA,CAAI,GAAG,KAAA,CAAM,CAAA;AAAA,CAA6B,CAAA;AAC1C,QAAA,KAAA,MAAW,CAAA,IAAM,CAAA,CAAE,eAAe,CAAA,IAAuC,EAAC,EAAG;AAC3E,UAAA,GAAA,CAAI,EAAA,CAAG,KAAA,CAAM,CAAA,EAAA,EAAK,CAAA,CAAE,WAAW,CAAA,EAAG,WAAA,EAAa,CAAA,CAAA,EAAI,EAAE,UAAU,CAAC,CAAA,QAAA,EAAM,CAAA,CAAE,QAAQ,CAAC;AAAA,CAAI,CAAA;AAAA,QACvF;AAAA,MACF,CAAA,MAAO;AACL,QAAA,GAAA,CAAI,EAAA,CAAG,KAAA,CAAM,CAAA,mBAAA,EAAiB,CAAA,CAAE,YAAY,CAAC,CAAA;AAAA,CAAc,CAAA;AAC3D,QAAA,KAAA,MAAW,CAAA,IAAM,CAAA,CAAE,QAAQ,CAAA,IAAkB,EAAC,EAAG;AAC/C,UAAA,GAAA,CAAI,EAAA,CAAG,KAAA,CAAM,CAAA,gBAAA,EAAS,CAAC;AAAA,CAAI,CAAA;AAAA,QAC7B;AAAA,MACF;AAAA,IACF,CAAA,MAAO;AACL,MAAA,GAAA,CAAI,EAAA,CAAG,KAAA,CAAM,CAAA,OAAA,EAAK,CAAA,CAAE,OAAO,CAAC;AAAA,CAAI,CAAA;AAAA,IAClC;AAAA,EACF;AACA,EAAA,OAAO,EAAE,QAAA,EAAS;AACpB","file":"rollback.js","sourcesContent":["/**\n * agent:rollback — Rollback file changes made by agents.\n *\n * Uses ChangeStore from @kb-labs/agent-history.\n * All rollback logic is in ChangeTrackingMiddleware.rollbackRun() or handled here via ChangeStore.\n *\n * Usage:\n *   pnpm kb agent:rollback --run-id={id} --session-id={id}\n *   pnpm kb agent:rollback --file=src/index.ts --session-id={id}\n *   pnpm kb agent:rollback --session-id={id}\n *   pnpm kb agent:rollback --run-id={id} --session-id={id} --dry-run\n */\n\nimport { defineCommand, useLogger } from '@kb-labs/sdk';\nimport type { PluginContextV3 } from '@kb-labs/sdk';\nimport { ChangeStore } from '@kb-labs/agent-history';\nimport type { FileChange } from '@kb-labs/agent-history';\nimport { promises as fsp } from 'node:fs';\nimport path from 'node:path';\n\ntype RollbackInput = {\n  'run-id'?: string;\n  runId?: string;\n  'session-id'?: string;\n  sessionId?: string;\n  file?: string;\n  'dry-run'?: boolean;\n  dryRun?: boolean;\n  json?: boolean;\n};\n\nexport default defineCommand({\n  id: 'rollback',\n  description: 'Rollback file changes made by agents in a session or run',\n\n  handler: {\n    async execute(ctx: PluginContextV3, input: RollbackInput): Promise<{ exitCode: number }> {\n      const logger = useLogger();\n      const flags = ((input as Record<string, unknown>).flags ?? input) as Record<string, unknown>;\n\n      const sessionId = (flags['session-id'] ?? flags['sessionId']) as string | undefined;\n      const runId = (flags['run-id'] ?? flags['runId']) as string | undefined;\n      const filePath = flags['file'] as string | undefined;\n      const dryRun = Boolean(flags['dry-run'] ?? flags['dryRun']);\n      const asJson = Boolean(flags['json']);\n\n      if (!sessionId) {\n        return output(ctx, asJson, { success: false, error: 'Missing --session-id' }, 1);\n      }\n\n      const workingDir = process.cwd();\n      const store = new ChangeStore(workingDir);\n\n      try {\n        let changes: FileChange[];\n\n        if (runId && !filePath) {\n          changes = await store.listRun(sessionId, runId);\n        } else if (filePath) {\n          changes = await store.listFile(sessionId, filePath);\n          if (runId) {\n            changes = changes.filter((c) => c.runId === runId);\n          }\n        } else {\n          changes = await store.listSession(sessionId);\n        }\n\n        if (changes.length === 0) {\n          return output(ctx, asJson, { success: true, rolledBack: 0, message: 'No changes to rollback' }, 0);\n        }\n\n        // Process in reverse chronological order\n        const ordered = [...changes].sort(\n          (a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime(),\n        );\n\n        if (dryRun) {\n          const preview = ordered.map((c) => ({\n            filePath: c.filePath,\n            operation: c.operation,\n            action: !c.before ? 'delete (was new file)' : 'restore to before state',\n          }));\n          return output(ctx, asJson, { success: true, dryRun: true, wouldRollback: preview }, 0);\n        }\n\n        let rolledBack = 0;\n        const errors: string[] = [];\n\n        for (const change of ordered) {\n          const fullPath = path.join(workingDir, change.filePath);\n          try {\n            if (!change.before) {\n              // File was created by agent — delete it\n              await fsp.unlink(fullPath).catch((e: NodeJS.ErrnoException) => {\n                if (e.code !== 'ENOENT') {throw e;}\n              });\n            } else {\n              // Restore to before state\n              await fsp.mkdir(path.dirname(fullPath), { recursive: true });\n              await fsp.writeFile(fullPath, change.before.content, 'utf-8');\n            }\n            rolledBack++;\n          } catch (e) {\n            errors.push(`${change.filePath}: ${String(e)}`);\n          }\n        }\n\n        return output(ctx, asJson, { success: errors.length === 0, rolledBack, errors }, errors.length > 0 ? 1 : 0);\n      } catch (e) {\n        logger.error('agent:rollback error', e instanceof Error ? e : undefined);\n        return output(ctx, asJson, { success: false, error: String(e) }, 1);\n      }\n    },\n  },\n});\n\nfunction output(ctx: PluginContextV3, asJson: boolean, data: unknown, exitCode: number): { exitCode: number } {\n  if (asJson) {\n    ctx.ui.write(JSON.stringify(data, null, 2) + '\\n');\n  } else {\n    const d = data as Record<string, unknown>;\n    if (d['success']) {\n      if (d['dryRun']) {\n        ctx.ui.write(`[DRY RUN] Would rollback:\\n`);\n        for (const p of (d['wouldRollback'] as Array<Record<string, string>>) ?? []) {\n          ctx.ui.write(`  ${p['operation']?.toUpperCase()} ${p['filePath']} → ${p['action']}\\n`);\n        }\n      } else {\n        ctx.ui.write(`✅ Rolled back ${d['rolledBack']} change(s)\\n`);\n        for (const e of (d['errors'] as string[]) ?? []) {\n          ctx.ui.write(`  ⚠️  ${e}\\n`);\n        }\n      }\n    } else {\n      ctx.ui.write(`❌ ${d['error']}\\n`);\n    }\n  }\n  return { exitCode };\n}\n"]}

package/dist/cli/commands/run.d.ts

@@ -0,0 +1,42 @@
+import * as _kb_labs_shared_command_kit from '@kb-labs/shared-command-kit';
+import { AgentMode } from '@kb-labs/agent-contracts';
+
+type RunInput = {
+    task: string;
+    workingDir?: string;
+    maxIterations?: number;
+    temperature?: number;
+    verbose?: boolean;
+    quiet?: boolean;
+    detailed?: boolean;
+    sessionId?: string;
+    tier?: 'small' | 'medium' | 'large';
+    escalate?: boolean;
+    mode?: AgentMode;
+    complexity?: 'simple' | 'medium' | 'complex';
+    files?: string[];
+    trace?: string;
+    approve?: boolean;
+    execute?: boolean;
+    spec?: boolean;
+    'dry-run'?: boolean;
+    debug?: boolean;
+    /** Abort execution after this many seconds (0 or undefined = no timeout). */
+    timeout?: number;
+    /** Override token budget (e.g. 300000 for heavy tasks). Overrides config value. */
+    budget?: number;
+    /** Output structured JSON instead of human-readable output. */
+    json?: boolean;
+    argv?: string[];
+};
+declare const _default: _kb_labs_shared_command_kit.CommandHandlerV3<unknown, RunInput, {
+    success: boolean;
+    summary: string;
+    filesCreated: string[];
+    filesModified: string[];
+    filesRead: string[];
+    iterations: number;
+    tokensUsed: number;
+}>;
+
+export { _default as default };