@kb-labs/agent-entry 2.93.0 → 2.96.0

package/dist/rest/handlers/rollback-handler.js

@@ -1,4 +1,4 @@
-import { defineHandler } from '@kb-labs/sdk';
+import { defineHandler, rethrowForRest } from '@kb-labs/sdk';
 import { SessionManager } from '@kb-labs/agent-core';
 import { ChangeStore } from '@kb-labs/agent-history';
 import { promises } from 'fs';
@@ -7,82 +7,86 @@ import path from 'path';
 // src/rest/handlers/rollback-handler.ts
 var rollback_handler_default = defineHandler({
   async execute(ctx, input) {
-    const params = input.params;
-    const sessionId = params?.sessionId;
-    if (!sessionId) {
-      throw new Error("Session ID is required");
-    }
-    const body = input.body ?? {};
-    const sessionManager = new SessionManager(ctx.cwd);
-    const session = await sessionManager.loadSession(sessionId);
-    if (!session) {
-      throw new Error(`Session not found: ${sessionId}`);
-    }
-    const workingDir = session.workingDir || ctx.cwd;
-    const store = new ChangeStore(workingDir);
-    let changes;
-    if (body.runId) {
-      changes = await store.listRun(sessionId, body.runId);
-    } else if (body.changeIds?.length) {
-      const ids = new Set(body.changeIds);
-      const all = await store.listSession(sessionId);
-      changes = all.filter((c) => ids.has(c.id));
-    } else {
-      changes = await store.listSession(sessionId);
-    }
-    if (changes.length === 0) {
-      return { rolledBack: 0, skipped: 0, conflicts: [], success: true };
-    }
-    const sorted = [...changes].sort(
-      (a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime()
-    );
-    const fullHistory = await store.listSession(sessionId);
-    const rollbackIds = new Set(sorted.map((c) => c.id));
-    const conflicts = [];
-    let rolledBack = 0;
-    let skipped = 0;
-    for (const change of sorted) {
-      if (!body.skipConflicts) {
-        const laterChanges = fullHistory.filter(
-          (c) => c.filePath === change.filePath && new Date(c.timestamp).getTime() > new Date(change.timestamp).getTime() && !rollbackIds.has(c.id)
-        );
-        if (laterChanges.length > 0) {
-          const laterAgents = [...new Set(laterChanges.map((c) => c.agentId))].join(", ");
+    try {
+      const params = input.params;
+      const sessionId = params?.sessionId;
+      if (!sessionId) {
+        throw new Error("Session ID is required");
+      }
+      const body = input.body ?? {};
+      const sessionManager = new SessionManager(ctx.cwd);
+      const session = await sessionManager.loadSession(sessionId);
+      if (!session) {
+        throw new Error(`Session not found: ${sessionId}`);
+      }
+      const workingDir = session.workingDir || ctx.cwd;
+      const store = new ChangeStore(workingDir);
+      let changes;
+      if (body.runId) {
+        changes = await store.listRun(sessionId, body.runId);
+      } else if (body.changeIds?.length) {
+        const ids = new Set(body.changeIds);
+        const all = await store.listSession(sessionId);
+        changes = all.filter((c) => ids.has(c.id));
+      } else {
+        changes = await store.listSession(sessionId);
+      }
+      if (changes.length === 0) {
+        return { rolledBack: 0, skipped: 0, conflicts: [], success: true };
+      }
+      const sorted = [...changes].sort(
+        (a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime()
+      );
+      const fullHistory = await store.listSession(sessionId);
+      const rollbackIds = new Set(sorted.map((c) => c.id));
+      const conflicts = [];
+      let rolledBack = 0;
+      let skipped = 0;
+      for (const change of sorted) {
+        if (!body.skipConflicts) {
+          const laterChanges = fullHistory.filter(
+            (c) => c.filePath === change.filePath && new Date(c.timestamp).getTime() > new Date(change.timestamp).getTime() && !rollbackIds.has(c.id)
+          );
+          if (laterChanges.length > 0) {
+            const laterAgents = [...new Set(laterChanges.map((c) => c.agentId))].join(", ");
+            conflicts.push({
+              filePath: change.filePath,
+              changeId: change.id,
+              reason: `File was modified after this change by: ${laterAgents}. Use skipConflicts: true to force.`
+            });
+            skipped++;
+            continue;
+          }
+        }
+        const fullPath = path.join(workingDir, change.filePath);
+        try {
+          if (change.before) {
+            await promises.mkdir(path.dirname(fullPath), { recursive: true });
+            await promises.writeFile(fullPath, change.before.content, "utf-8");
+          } else {
+            await promises.unlink(fullPath).catch((e) => {
+              if (e.code !== "ENOENT") {
+                throw e;
+              }
+            });
+          }
+          rolledBack++;
+        } catch (e) {
           conflicts.push({
             filePath: change.filePath,
             changeId: change.id,
-            reason: `File was modified after this change by: ${laterAgents}. Use skipConflicts: true to force.`
+            reason: `Write failed: ${e instanceof Error ? e.message : String(e)}`
           });
           skipped++;
-          continue;
-        }
-      }
-      const fullPath = path.join(workingDir, change.filePath);
-      try {
-        if (change.before) {
-          await promises.mkdir(path.dirname(fullPath), { recursive: true });
-          await promises.writeFile(fullPath, change.before.content, "utf-8");
-        } else {
-          await promises.unlink(fullPath).catch((e) => {
-            if (e.code !== "ENOENT") {
-              throw e;
-            }
-          });
         }
-        rolledBack++;
-      } catch (e) {
-        conflicts.push({
-          filePath: change.filePath,
-          changeId: change.id,
-          reason: `Write failed: ${e instanceof Error ? e.message : String(e)}`
-        });
-        skipped++;
       }
+      ctx.platform.logger.info(
+        `[rollback-handler] Session ${sessionId}: rolled back ${rolledBack}, skipped ${skipped}`
+      );
+      return { rolledBack, skipped, conflicts, success: skipped === 0 };
+    } catch (err) {
+      rethrowForRest(err);
     }
-    ctx.platform.logger.info(
-      `[rollback-handler] Session ${sessionId}: rolled back ${rolledBack}, skipped ${skipped}`
-    );
-    return { rolledBack, skipped, conflicts, success: skipped === 0 };
   }
 });
 

package/dist/rest/handlers/rollback-handler.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/rest/handlers/rollback-handler.ts"],"names":["fsp"],"mappings":";;;;;;;AA+BA,IAAO,2BAAQ,aAAA,CAAc;AAAA,EAC3B,MAAM,OAAA,CAAQ,GAAA,EAAsB,KAAA,EAA8D;AAChG,IAAA,MAAM,SAAS,KAAA,CAAM,MAAA;AACrB,IAAA,MAAM,YAAY,MAAA,EAAQ,SAAA;AAC1B,IAAA,IAAI,CAAC,SAAA,EAAW;AAAC,MAAA,MAAM,IAAI,MAAM,wBAAwB,CAAA;AAAA,IAAE;AAE3D,IAAA,MAAM,IAAA,GAAQ,KAAA,CAAM,IAAA,IAAQ,EAAC;AAE7B,IAAA,MAAM,cAAA,GAAiB,IAAI,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AACjD,IAAA,MAAM,OAAA,GAAU,MAAM,cAAA,CAAe,WAAA,CAAY,SAAS,CAAA;AAC1D,IAAA,IAAI,CAAC,OAAA,EAAS;AAAC,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,mBAAA,EAAsB,SAAS,CAAA,CAAE,CAAA;AAAA,IAAE;AAElE,IAAA,MAAM,UAAA,GAAa,OAAA,CAAQ,UAAA,IAAc,GAAA,CAAI,GAAA;AAC7C,IAAA,MAAM,KAAA,GAAQ,IAAI,WAAA,CAAY,UAAU,CAAA;AAExC,IAAA,IAAI,OAAA;AACJ,IAAA,IAAI,KAAK,KAAA,EAAO;AACd,MAAA,OAAA,GAAU,MAAM,KAAA,CAAM,OAAA,CAAQ,SAAA,EAAW,KAAK,KAAK,CAAA;AAAA,IACrD,CAAA,MAAA,IAAW,IAAA,CAAK,SAAA,EAAW,MAAA,EAAQ;AACjC,MAAA,MAAM,GAAA,GAAM,IAAI,GAAA,CAAI,IAAA,CAAK,SAAS,CAAA;AAClC,MAAA,MAAM,GAAA,GAAM,MAAM,KAAA,CAAM,WAAA,CAAY,SAAS,CAAA;AAC7C,MAAA,OAAA,GAAU,GAAA,CAAI,OAAO,CAAC,CAAA,KAAM,IAAI,GAAA,CAAI,CAAA,CAAE,EAAE,CAAC,CAAA;AAAA,IAC3C,CAAA,MAAO;AACL,MAAA,OAAA,GAAU,MAAM,KAAA,CAAM,WAAA,CAAY,SAAS,CAAA;AAAA,IAC7C;AAEA,IAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,MAAA,OAAO,EAAE,YAAY,CAAA,EAAG,OAAA,EAAS,GAAG,SAAA,EAAW,EAAC,EAAG,OAAA,EAAS,IAAA,EAAK;AAAA,IACnE;AAGA,IAAA,MAAM,MAAA,GAAS,CAAC,GAAG,OAAO,CAAA,CAAE,IAAA;AAAA,MAC1B,CAAC,CAAA,EAAG,CAAA,KAAM,IAAI,KAAK,CAAA,CAAE,SAAS,CAAA,CAAE,OAAA,KAAY,IAAI,IAAA,CAAK,CAAA,CAAE,SAAS,EAAE,OAAA;AAAQ,KAC5E;AAGA,IAAA,MAAM,WAAA,GAAc,MAAM,KAAA,CAAM,WAAA,CAAY,SAAS,CAAA;AACrD,IAAA,MAAM,WAAA,GAAc,IAAI,GAAA,CAAI,MAAA,CAAO,IAAI,CAAC,CAAA,KAAM,CAAA,CAAE,EAAE,CAAC,CAAA;AAEnD,IAAA,MAAM,YAA6B,EAAC;AACpC,IAAA,IAAI,UAAA,GAAa,CAAA;AACjB,IAAA,IAAI,OAAA,GAAU,CAAA;AAEd,IAAA,KAAA,MAAW,UAAU,MAAA,EAAQ;AAE3B,MAAA,IAAI,CAAC,KAAK,aAAA,EAAe;AACvB,QAAA,MAAM,eAAe,WAAA,CAAY,MAAA;AAAA,UAC/B,CAAC,CAAA,KACC,CAAA,CAAE,QAAA,KAAa,MAAA,CAAO,YACtB,IAAI,IAAA,CAAK,CAAA,CAAE,SAAS,CAAA,CAAE,OAAA,KAAY,IAAI,IAAA,CAAK,MAAA,CAAO,SAAS,CAAA,CAAE,OAAA,MAC7D,CAAC,WAAA,CAAY,GAAA,CAAI,CAAA,CAAE,EAAE;AAAA,SACzB;AACA,QAAA,IAAI,YAAA,CAAa,SAAS,CAAA,EAAG;AAC3B,UAAA,MAAM,WAAA,GAAc,CAAC,GAAG,IAAI,IAAI,YAAA,CAAa,GAAA,CAAI,CAAC,CAAA,KAAM,EAAE,OAAO,CAAC,CAAC,CAAA,CAAE,KAAK,IAAI,CAAA;AAC9E,UAAA,SAAA,CAAU,IAAA,CAAK;AAAA,YACb,UAAU,MAAA,CAAO,QAAA;AAAA,YACjB,UAAU,MAAA,CAAO,EAAA;AAAA,YACjB,MAAA,EAAQ,2CAA2C,WAAW,CAAA,mCAAA;AAAA,WAC/D,CAAA;AACD,UAAA,OAAA,EAAA;AACA,UAAA;AAAA,QACF;AAAA,MACF;AAEA,MAAA,MAAM,QAAA,GAAW,IAAA,CAAK,IAAA,CAAK,UAAA,EAAY,OAAO,QAAQ,CAAA;AACtD,MAAA,IAAI;AACF,QAAA,IAAI,OAAO,MAAA,EAAQ;AACjB,UAAA,MAAMA,QAAA,CAAI,MAAM,IAAA,CAAK,OAAA,CAAQ,QAAQ,CAAA,EAAG,EAAE,SAAA,EAAW,IAAA,EAAM,CAAA;AAC3D,UAAA,MAAMA,SAAI,SAAA,CAAU,QAAA,EAAU,MAAA,CAAO,MAAA,CAAO,SAAS,OAAO,CAAA;AAAA,QAC9D,CAAA,MAAO;AACL,UAAA,MAAMA,SAAI,MAAA,CAAO,QAAQ,CAAA,CAAE,KAAA,CAAM,CAAC,CAAA,KAA6B;AAC7D,YAAA,IAAI,CAAA,CAAE,SAAS,QAAA,EAAU;AAAC,cAAA,MAAM,CAAA;AAAA,YAAE;AAAA,UACpC,CAAC,CAAA;AAAA,QACH;AACA,QAAA,UAAA,EAAA;AAAA,MACF,SAAS,CAAA,EAAG;AACV,QAAA,SAAA,CAAU,IAAA,CAAK;AAAA,UACb,UAAU,MAAA,CAAO,QAAA;AAAA,UACjB,UAAU,MAAA,CAAO,EAAA;AAAA,UACjB,MAAA,EAAQ,iBAAiB,CAAA,YAAa,KAAA,GAAQ,EAAE,OAAA,GAAU,MAAA,CAAO,CAAC,CAAC,CAAA;AAAA,SACpE,CAAA;AACD,QAAA,OAAA,EAAA;AAAA,MACF;AAAA,IACF;AAEA,IAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA;AAAA,MAClB,CAAA,2BAAA,EAA8B,SAAS,CAAA,cAAA,EAAiB,UAAU,aAAa,OAAO,CAAA;AAAA,KACxF;AAEA,IAAA,OAAO,EAAE,UAAA,EAAY,OAAA,EAAS,SAAA,EAAW,OAAA,EAAS,YAAY,CAAA,EAAE;AAAA,EAClE;AACF,CAAC","file":"rollback-handler.js","sourcesContent":["/**\n * POST /sessions/:sessionId/rollback — Rollback file changes.\n * Body: { runId?: string; changeIds?: string[]; skipConflicts?: boolean }\n */\n\nimport { defineHandler, type RestInput, type PluginContextV3 } from '@kb-labs/sdk';\nimport { SessionManager } from '@kb-labs/agent-core';\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\ninterface RollbackRequest {\n  runId?: string;\n  changeIds?: string[];\n  skipConflicts?: boolean;\n}\n\ninterface ConflictEntry {\n  filePath: string;\n  changeId: string;\n  reason: string;\n}\n\ninterface RollbackResponse {\n  rolledBack: number;\n  skipped: number;\n  conflicts: ConflictEntry[];\n  success: boolean;\n}\n\nexport default defineHandler({\n  async execute(ctx: PluginContextV3, input: RestInput<RollbackRequest>): Promise<RollbackResponse> {\n    const params = input.params as Record<string, string> | undefined;\n    const sessionId = params?.sessionId;\n    if (!sessionId) {throw new Error('Session ID is required');}\n\n    const body = (input.body ?? {}) as RollbackRequest;\n\n    const sessionManager = new SessionManager(ctx.cwd);\n    const session = await sessionManager.loadSession(sessionId);\n    if (!session) {throw new Error(`Session not found: ${sessionId}`);}\n\n    const workingDir = session.workingDir || ctx.cwd;\n    const store = new ChangeStore(workingDir);\n\n    let changes: FileChange[];\n    if (body.runId) {\n      changes = await store.listRun(sessionId, body.runId);\n    } else if (body.changeIds?.length) {\n      const ids = new Set(body.changeIds);\n      const all = await store.listSession(sessionId);\n      changes = all.filter((c) => ids.has(c.id));\n    } else {\n      changes = await store.listSession(sessionId);\n    }\n\n    if (changes.length === 0) {\n      return { rolledBack: 0, skipped: 0, conflicts: [], success: true };\n    }\n\n    // Reverse-chronological: undo latest first\n    const sorted = [...changes].sort(\n      (a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime(),\n    );\n\n    // For conflict detection: full session history\n    const fullHistory = await store.listSession(sessionId);\n    const rollbackIds = new Set(sorted.map((c) => c.id));\n\n    const conflicts: ConflictEntry[] = [];\n    let rolledBack = 0;\n    let skipped = 0;\n\n    for (const change of sorted) {\n      // Conflict: newer changes to same file NOT in our rollback set\n      if (!body.skipConflicts) {\n        const laterChanges = fullHistory.filter(\n          (c) =>\n            c.filePath === change.filePath &&\n            new Date(c.timestamp).getTime() > new Date(change.timestamp).getTime() &&\n            !rollbackIds.has(c.id),\n        );\n        if (laterChanges.length > 0) {\n          const laterAgents = [...new Set(laterChanges.map((c) => c.agentId))].join(', ');\n          conflicts.push({\n            filePath: change.filePath,\n            changeId: change.id,\n            reason: `File was modified after this change by: ${laterAgents}. Use skipConflicts: true to force.`,\n          });\n          skipped++;\n          continue;\n        }\n      }\n\n      const fullPath = path.join(workingDir, change.filePath);\n      try {\n        if (change.before) {\n          await fsp.mkdir(path.dirname(fullPath), { recursive: true });\n          await fsp.writeFile(fullPath, change.before.content, 'utf-8');\n        } else {\n          await fsp.unlink(fullPath).catch((e: NodeJS.ErrnoException) => {\n            if (e.code !== 'ENOENT') {throw e;}\n          });\n        }\n        rolledBack++;\n      } catch (e) {\n        conflicts.push({\n          filePath: change.filePath,\n          changeId: change.id,\n          reason: `Write failed: ${e instanceof Error ? e.message : String(e)}`,\n        });\n        skipped++;\n      }\n    }\n\n    ctx.platform.logger.info(\n      `[rollback-handler] Session ${sessionId}: rolled back ${rolledBack}, skipped ${skipped}`,\n    );\n\n    return { rolledBack, skipped, conflicts, success: skipped === 0 };\n  },\n});\n"]}
+{"version":3,"sources":["../../../src/rest/handlers/rollback-handler.ts"],"names":["fsp"],"mappings":";;;;;;;AA+BA,IAAO,2BAAQ,aAAA,CAAc;AAAA,EAC3B,MAAM,OAAA,CAAQ,GAAA,EAAsB,KAAA,EAA8D;AAChG,IAAA,IAAI;AACF,MAAA,MAAM,SAAS,KAAA,CAAM,MAAA;AACrB,MAAA,MAAM,YAAY,MAAA,EAAQ,SAAA;AAC1B,MAAA,IAAI,CAAC,SAAA,EAAW;AAAC,QAAA,MAAM,IAAI,MAAM,wBAAwB,CAAA;AAAA,MAAE;AAE3D,MAAA,MAAM,IAAA,GAAQ,KAAA,CAAM,IAAA,IAAQ,EAAC;AAE7B,MAAA,MAAM,cAAA,GAAiB,IAAI,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AACjD,MAAA,MAAM,OAAA,GAAU,MAAM,cAAA,CAAe,WAAA,CAAY,SAAS,CAAA;AAC1D,MAAA,IAAI,CAAC,OAAA,EAAS;AAAC,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,mBAAA,EAAsB,SAAS,CAAA,CAAE,CAAA;AAAA,MAAE;AAElE,MAAA,MAAM,UAAA,GAAa,OAAA,CAAQ,UAAA,IAAc,GAAA,CAAI,GAAA;AAC7C,MAAA,MAAM,KAAA,GAAQ,IAAI,WAAA,CAAY,UAAU,CAAA;AAExC,MAAA,IAAI,OAAA;AACJ,MAAA,IAAI,KAAK,KAAA,EAAO;AACd,QAAA,OAAA,GAAU,MAAM,KAAA,CAAM,OAAA,CAAQ,SAAA,EAAW,KAAK,KAAK,CAAA;AAAA,MACrD,CAAA,MAAA,IAAW,IAAA,CAAK,SAAA,EAAW,MAAA,EAAQ;AACjC,QAAA,MAAM,GAAA,GAAM,IAAI,GAAA,CAAI,IAAA,CAAK,SAAS,CAAA;AAClC,QAAA,MAAM,GAAA,GAAM,MAAM,KAAA,CAAM,WAAA,CAAY,SAAS,CAAA;AAC7C,QAAA,OAAA,GAAU,GAAA,CAAI,OAAO,CAAC,CAAA,KAAM,IAAI,GAAA,CAAI,CAAA,CAAE,EAAE,CAAC,CAAA;AAAA,MAC3C,CAAA,MAAO;AACL,QAAA,OAAA,GAAU,MAAM,KAAA,CAAM,WAAA,CAAY,SAAS,CAAA;AAAA,MAC7C;AAEA,MAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,QAAA,OAAO,EAAE,YAAY,CAAA,EAAG,OAAA,EAAS,GAAG,SAAA,EAAW,EAAC,EAAG,OAAA,EAAS,IAAA,EAAK;AAAA,MACnE;AAGA,MAAA,MAAM,MAAA,GAAS,CAAC,GAAG,OAAO,CAAA,CAAE,IAAA;AAAA,QAC1B,CAAC,CAAA,EAAG,CAAA,KAAM,IAAI,KAAK,CAAA,CAAE,SAAS,CAAA,CAAE,OAAA,KAAY,IAAI,IAAA,CAAK,CAAA,CAAE,SAAS,EAAE,OAAA;AAAQ,OAC5E;AAGA,MAAA,MAAM,WAAA,GAAc,MAAM,KAAA,CAAM,WAAA,CAAY,SAAS,CAAA;AACrD,MAAA,MAAM,WAAA,GAAc,IAAI,GAAA,CAAI,MAAA,CAAO,IAAI,CAAC,CAAA,KAAM,CAAA,CAAE,EAAE,CAAC,CAAA;AAEnD,MAAA,MAAM,YAA6B,EAAC;AACpC,MAAA,IAAI,UAAA,GAAa,CAAA;AACjB,MAAA,IAAI,OAAA,GAAU,CAAA;AAEd,MAAA,KAAA,MAAW,UAAU,MAAA,EAAQ;AAE3B,QAAA,IAAI,CAAC,KAAK,aAAA,EAAe;AACvB,UAAA,MAAM,eAAe,WAAA,CAAY,MAAA;AAAA,YAC/B,CAAC,CAAA,KACC,CAAA,CAAE,QAAA,KAAa,MAAA,CAAO,YACtB,IAAI,IAAA,CAAK,CAAA,CAAE,SAAS,CAAA,CAAE,OAAA,KAAY,IAAI,IAAA,CAAK,MAAA,CAAO,SAAS,CAAA,CAAE,OAAA,MAC7D,CAAC,WAAA,CAAY,GAAA,CAAI,CAAA,CAAE,EAAE;AAAA,WACzB;AACA,UAAA,IAAI,YAAA,CAAa,SAAS,CAAA,EAAG;AAC3B,YAAA,MAAM,WAAA,GAAc,CAAC,GAAG,IAAI,IAAI,YAAA,CAAa,GAAA,CAAI,CAAC,CAAA,KAAM,EAAE,OAAO,CAAC,CAAC,CAAA,CAAE,KAAK,IAAI,CAAA;AAC9E,YAAA,SAAA,CAAU,IAAA,CAAK;AAAA,cACb,UAAU,MAAA,CAAO,QAAA;AAAA,cACjB,UAAU,MAAA,CAAO,EAAA;AAAA,cACjB,MAAA,EAAQ,2CAA2C,WAAW,CAAA,mCAAA;AAAA,aAC/D,CAAA;AACD,YAAA,OAAA,EAAA;AACA,YAAA;AAAA,UACF;AAAA,QACF;AAEA,QAAA,MAAM,QAAA,GAAW,IAAA,CAAK,IAAA,CAAK,UAAA,EAAY,OAAO,QAAQ,CAAA;AACtD,QAAA,IAAI;AACF,UAAA,IAAI,OAAO,MAAA,EAAQ;AACjB,YAAA,MAAMA,QAAA,CAAI,MAAM,IAAA,CAAK,OAAA,CAAQ,QAAQ,CAAA,EAAG,EAAE,SAAA,EAAW,IAAA,EAAM,CAAA;AAC3D,YAAA,MAAMA,SAAI,SAAA,CAAU,QAAA,EAAU,MAAA,CAAO,MAAA,CAAO,SAAS,OAAO,CAAA;AAAA,UAC9D,CAAA,MAAO;AACL,YAAA,MAAMA,SAAI,MAAA,CAAO,QAAQ,CAAA,CAAE,KAAA,CAAM,CAAC,CAAA,KAA6B;AAC7D,cAAA,IAAI,CAAA,CAAE,SAAS,QAAA,EAAU;AAAC,gBAAA,MAAM,CAAA;AAAA,cAAE;AAAA,YACpC,CAAC,CAAA;AAAA,UACH;AACA,UAAA,UAAA,EAAA;AAAA,QACF,SAAS,CAAA,EAAG;AACV,UAAA,SAAA,CAAU,IAAA,CAAK;AAAA,YACb,UAAU,MAAA,CAAO,QAAA;AAAA,YACjB,UAAU,MAAA,CAAO,EAAA;AAAA,YACjB,MAAA,EAAQ,iBAAiB,CAAA,YAAa,KAAA,GAAQ,EAAE,OAAA,GAAU,MAAA,CAAO,CAAC,CAAC,CAAA;AAAA,WACpE,CAAA;AACD,UAAA,OAAA,EAAA;AAAA,QACF;AAAA,MACF;AAEA,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA;AAAA,QAClB,CAAA,2BAAA,EAA8B,SAAS,CAAA,cAAA,EAAiB,UAAU,aAAa,OAAO,CAAA;AAAA,OACxF;AAEA,MAAA,OAAO,EAAE,UAAA,EAAY,OAAA,EAAS,SAAA,EAAW,OAAA,EAAS,YAAY,CAAA,EAAE;AAAA,IAClE,SAAS,GAAA,EAAK;AACZ,MAAA,cAAA,CAAe,GAAG,CAAA;AAAA,IACpB;AAAA,EACF;AACF,CAAC","file":"rollback-handler.js","sourcesContent":["/**\n * POST /sessions/:sessionId/rollback — Rollback file changes.\n * Body: { runId?: string; changeIds?: string[]; skipConflicts?: boolean }\n */\n\nimport { defineHandler, rethrowForRest, type RestInput, type PluginContextV3 } from '@kb-labs/sdk';\nimport { SessionManager } from '@kb-labs/agent-core';\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\ninterface RollbackRequest {\n  runId?: string;\n  changeIds?: string[];\n  skipConflicts?: boolean;\n}\n\ninterface ConflictEntry {\n  filePath: string;\n  changeId: string;\n  reason: string;\n}\n\ninterface RollbackResponse {\n  rolledBack: number;\n  skipped: number;\n  conflicts: ConflictEntry[];\n  success: boolean;\n}\n\nexport default defineHandler({\n  async execute(ctx: PluginContextV3, input: RestInput<RollbackRequest>): Promise<RollbackResponse> {\n    try {\n      const params = input.params as Record<string, string> | undefined;\n      const sessionId = params?.sessionId;\n      if (!sessionId) {throw new Error('Session ID is required');}\n\n      const body = (input.body ?? {}) as RollbackRequest;\n\n      const sessionManager = new SessionManager(ctx.cwd);\n      const session = await sessionManager.loadSession(sessionId);\n      if (!session) {throw new Error(`Session not found: ${sessionId}`);}\n\n      const workingDir = session.workingDir || ctx.cwd;\n      const store = new ChangeStore(workingDir);\n\n      let changes: FileChange[];\n      if (body.runId) {\n        changes = await store.listRun(sessionId, body.runId);\n      } else if (body.changeIds?.length) {\n        const ids = new Set(body.changeIds);\n        const all = await store.listSession(sessionId);\n        changes = all.filter((c) => ids.has(c.id));\n      } else {\n        changes = await store.listSession(sessionId);\n      }\n\n      if (changes.length === 0) {\n        return { rolledBack: 0, skipped: 0, conflicts: [], success: true };\n      }\n\n      // Reverse-chronological: undo latest first\n      const sorted = [...changes].sort(\n        (a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime(),\n      );\n\n      // For conflict detection: full session history\n      const fullHistory = await store.listSession(sessionId);\n      const rollbackIds = new Set(sorted.map((c) => c.id));\n\n      const conflicts: ConflictEntry[] = [];\n      let rolledBack = 0;\n      let skipped = 0;\n\n      for (const change of sorted) {\n        // Conflict: newer changes to same file NOT in our rollback set\n        if (!body.skipConflicts) {\n          const laterChanges = fullHistory.filter(\n            (c) =>\n              c.filePath === change.filePath &&\n              new Date(c.timestamp).getTime() > new Date(change.timestamp).getTime() &&\n              !rollbackIds.has(c.id),\n          );\n          if (laterChanges.length > 0) {\n            const laterAgents = [...new Set(laterChanges.map((c) => c.agentId))].join(', ');\n            conflicts.push({\n              filePath: change.filePath,\n              changeId: change.id,\n              reason: `File was modified after this change by: ${laterAgents}. Use skipConflicts: true to force.`,\n            });\n            skipped++;\n            continue;\n          }\n        }\n\n        const fullPath = path.join(workingDir, change.filePath);\n        try {\n          if (change.before) {\n            await fsp.mkdir(path.dirname(fullPath), { recursive: true });\n            await fsp.writeFile(fullPath, change.before.content, 'utf-8');\n          } else {\n            await fsp.unlink(fullPath).catch((e: NodeJS.ErrnoException) => {\n              if (e.code !== 'ENOENT') {throw e;}\n            });\n          }\n          rolledBack++;\n        } catch (e) {\n          conflicts.push({\n            filePath: change.filePath,\n            changeId: change.id,\n            reason: `Write failed: ${e instanceof Error ? e.message : String(e)}`,\n          });\n          skipped++;\n        }\n      }\n\n      ctx.platform.logger.info(\n        `[rollback-handler] Session ${sessionId}: rolled back ${rolledBack}, skipped ${skipped}`,\n      );\n\n      return { rolledBack, skipped, conflicts, success: skipped === 0 };\n    } catch (err) {\n      rethrowForRest(err);\n    }\n  },\n});\n"]}

package/dist/rest/handlers/run-handler.js

@@ -1,4 +1,4 @@
-import { defineHandler, useAnalytics, useCache, useConfig, usePlatform } from '@kb-labs/sdk';
+import { defineHandler, useAnalytics, useCache, useConfig, rethrowForRest, usePlatform } from '@kb-labs/sdk';
 import { bootstrapAgentSDK, SessionManager, createSessionMemoryBridge, createCoreToolPack } from '@kb-labs/agent-core';
 import { createDefaultResponseRequirementsSelector } from '@kb-labs/agent-runtime';
 import { IncrementalTraceWriter } from '@kb-labs/agent-tracing';
@@ -353,161 +353,165 @@ async function inferFollowUpWorkingDir(sessionManager, sessionId, baseWorkingDir
 }
 var run_handler_default = defineHandler({
   async execute(ctx, input) {
-    const body = input.body;
-    if (!body?.task) {
-      throw new Error("Task is required");
-    }
-    const analytics = useAnalytics();
-    const runId = RunManager.generateRunId();
-    const startTime = Date.now();
-    let sessionId = body.sessionId;
-    let workingDir = body.workingDir || ctx.cwd;
-    let sessionManager = new SessionManager(workingDir);
-    if (!sessionId) {
-      const session = await sessionManager.createSession({
-        mode: "execute",
-        task: body.task,
-        agentId: body.agentId ?? "orchestrator"
-      });
-      sessionId = session.id;
-      ctx.platform.logger.info(`[run-handler] Created new session ${sessionId}`);
-    } else {
-      const existingSession = await sessionManager.loadSession(sessionId);
-      let resolvedSession = existingSession;
-      if (!resolvedSession && pathExists(ctx.cwd)) {
-        const fallbackManager = new SessionManager(ctx.cwd);
-        const fallbackSession = await fallbackManager.loadSession(sessionId);
-        if (fallbackSession) {
-          sessionManager = fallbackManager;
-          resolvedSession = fallbackSession;
-        }
+    try {
+      const body = input.body;
+      if (!body?.task) {
+        throw new Error("Task is required");
       }
-      if (!resolvedSession) {
-        throw new Error(`Session not found: ${sessionId} (cwd=${ctx.cwd})`);
-      }
-      workingDir = body.workingDir || resolvedSession.workingDir || workingDir;
-      if (!body.workingDir && isLikelyFollowUpScopeTask(body.task)) {
-        const inferred = await inferFollowUpWorkingDir(sessionManager, sessionId, resolvedSession.workingDir || ctx.cwd);
-        if (inferred && inferred !== workingDir) {
-          ctx.platform.logger.info(`[run-handler] Follow-up scope inferred: ${workingDir} -> ${inferred}`);
-          workingDir = inferred;
+      const analytics = useAnalytics();
+      const runId = RunManager.generateRunId();
+      const startTime = Date.now();
+      let sessionId = body.sessionId;
+      let workingDir = body.workingDir || ctx.cwd;
+      let sessionManager = new SessionManager(workingDir);
+      if (!sessionId) {
+        const session = await sessionManager.createSession({
+          mode: "execute",
+          task: body.task,
+          agentId: body.agentId ?? "orchestrator"
+        });
+        sessionId = session.id;
+        ctx.platform.logger.info(`[run-handler] Created new session ${sessionId}`);
+      } else {
+        const existingSession = await sessionManager.loadSession(sessionId);
+        let resolvedSession = existingSession;
+        if (!resolvedSession && pathExists(ctx.cwd)) {
+          const fallbackManager = new SessionManager(ctx.cwd);
+          const fallbackSession = await fallbackManager.loadSession(sessionId);
+          if (fallbackSession) {
+            sessionManager = fallbackManager;
+            resolvedSession = fallbackSession;
+          }
+        }
+        if (!resolvedSession) {
+          throw new Error(`Session not found: ${sessionId} (cwd=${ctx.cwd})`);
+        }
+        workingDir = body.workingDir || resolvedSession.workingDir || workingDir;
+        if (!body.workingDir && isLikelyFollowUpScopeTask(body.task)) {
+          const inferred = await inferFollowUpWorkingDir(sessionManager, sessionId, resolvedSession.workingDir || ctx.cwd);
+          if (inferred && inferred !== workingDir) {
+            ctx.platform.logger.info(`[run-handler] Follow-up scope inferred: ${workingDir} -> ${inferred}`);
+            workingDir = inferred;
+          }
         }
+        sessionManager = new SessionManager(workingDir);
+        ctx.platform.logger.info(`[run-handler] Continuing session ${sessionId} (workingDir=${workingDir})`);
       }
-      sessionManager = new SessionManager(workingDir);
-      ctx.platform.logger.info(`[run-handler] Continuing session ${sessionId} (workingDir=${workingDir})`);
-    }
-    ctx.platform.logger.info(`[run-handler] Starting run ${runId} for task: ${body.task}`);
-    await sessionManager.createUserTurn(sessionId, body.task, runId);
-    await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_STARTED, {
-      runId,
-      sessionId,
-      taskLength: body.task.length,
-      tier: body.tier ?? "medium",
-      enableEscalation: body.enableEscalation ?? true,
-      responseMode: body.responseMode ?? "auto",
-      verbose: body.verbose ?? false
-    });
-    const sessionMemory = createSessionMemoryBridge(workingDir, sessionId);
-    const responseRequirementsSelector = createDefaultResponseRequirementsSelector();
-    const toolRegistry = createToolRegistry({
-      workingDir,
-      currentTask: body.task,
-      sessionId,
-      verbose: body.verbose,
-      cache: useCache(),
-      sessionMemory,
-      responseRequirementsResolver: async ({ task, kernel }) => responseRequirementsSelector.select({
-        state: kernel,
-        messages: [],
-        task: task ?? body.task
-      })
-    });
-    const agentsConfig = await useConfig();
-    const finalSessionId = sessionId;
-    const traceDir = path.join(workingDir, ".kb", "traces", "incremental");
-    const traceWriter = new IncrementalTraceWriter(runId, {}, traceDir);
-    const agent = new AgentSDK().register(createCoreToolPack(toolRegistry)).createRunner({
-      sessionId: finalSessionId,
-      workingDir,
-      maxIterations: 50,
-      temperature: 0.1,
-      tier: body.tier ?? "medium",
-      tokenBudget: agentsConfig?.tokenBudget,
-      onEvent: (event) => {
-        const seqEvent = RunManager.broadcast(runId, event);
-        void sessionManager.addEvent(finalSessionId, {
-          ...seqEvent,
-          sessionId: finalSessionId,
-          runId,
-          metadata: {
-            ...seqEvent.metadata,
+      ctx.platform.logger.info(`[run-handler] Starting run ${runId} for task: ${body.task}`);
+      await sessionManager.createUserTurn(sessionId, body.task, runId);
+      await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_STARTED, {
+        runId,
+        sessionId,
+        taskLength: body.task.length,
+        tier: body.tier ?? "medium",
+        enableEscalation: body.enableEscalation ?? true,
+        responseMode: body.responseMode ?? "auto",
+        verbose: body.verbose ?? false
+      });
+      const sessionMemory = createSessionMemoryBridge(workingDir, sessionId);
+      const responseRequirementsSelector = createDefaultResponseRequirementsSelector();
+      const toolRegistry = createToolRegistry({
+        workingDir,
+        currentTask: body.task,
+        sessionId,
+        verbose: body.verbose,
+        cache: useCache(),
+        sessionMemory,
+        responseRequirementsResolver: async ({ task, kernel }) => responseRequirementsSelector.select({
+          state: kernel,
+          messages: [],
+          task: task ?? body.task
+        })
+      });
+      const agentsConfig = await useConfig();
+      const finalSessionId = sessionId;
+      const traceDir = path.join(workingDir, ".kb", "traces", "incremental");
+      const traceWriter = new IncrementalTraceWriter(runId, {}, traceDir);
+      const agent = new AgentSDK().register(createCoreToolPack(toolRegistry)).createRunner({
+        sessionId: finalSessionId,
+        workingDir,
+        maxIterations: 50,
+        temperature: 0.1,
+        tier: body.tier ?? "medium",
+        tokenBudget: agentsConfig?.tokenBudget,
+        onEvent: (event) => {
+          const seqEvent = RunManager.broadcast(runId, event);
+          void sessionManager.addEvent(finalSessionId, {
+            ...seqEvent,
             sessionId: finalSessionId,
             runId,
-            workingDir
-          }
-        });
-      }
-    });
-    const run = await RunManager.register(runId, body.task, agent, sessionManager, finalSessionId);
-    await RunManager.updateStatus(runId, "running");
-    void (async () => {
-      try {
-        const result = await agent.execute(body.task);
-        const durationMs = Date.now() - startTime;
-        const detailedTrace = traceWriter.getEntries();
-        await traceWriter.finalize?.();
-        if (detailedTrace.length > 0) {
-          await sessionManager.storeTraceArtifacts(finalSessionId, runId, detailedTrace);
-        }
-        if (result.fileChanges && result.fileChanges.length > 0) {
-          await sessionManager.attachFileChangesToTurn(finalSessionId, runId, result.fileChanges);
+            metadata: {
+              ...seqEvent.metadata,
+              sessionId: finalSessionId,
+              runId,
+              workingDir
+            }
+          });
         }
-        await RunManager.updateStatus(runId, result.success ? "completed" : "failed", {
-          completedAt: (/* @__PURE__ */ new Date()).toISOString(),
-          durationMs,
-          summary: result.summary,
-          error: result.error
-        });
-        await analytics?.track(
-          result.success ? AGENT_ANALYTICS_EVENTS.RUN_COMPLETED : AGENT_ANALYTICS_EVENTS.RUN_FAILED,
-          {
-            runId,
+      });
+      const run = await RunManager.register(runId, body.task, agent, sessionManager, finalSessionId);
+      await RunManager.updateStatus(runId, "running");
+      void (async () => {
+        try {
+          const result = await agent.execute(body.task);
+          const durationMs = Date.now() - startTime;
+          const detailedTrace = traceWriter.getEntries();
+          await traceWriter.finalize?.();
+          if (detailedTrace.length > 0) {
+            await sessionManager.storeTraceArtifacts(finalSessionId, runId, detailedTrace);
+          }
+          if (result.fileChanges && result.fileChanges.length > 0) {
+            await sessionManager.attachFileChangesToTurn(finalSessionId, runId, result.fileChanges);
+          }
+          await RunManager.updateStatus(runId, result.success ? "completed" : "failed", {
+            completedAt: (/* @__PURE__ */ new Date()).toISOString(),
             durationMs,
-            success: result.success,
-            summary: result.summary?.slice(0, 200)
+            summary: result.summary,
+            error: result.error
+          });
+          await analytics?.track(
+            result.success ? AGENT_ANALYTICS_EVENTS.RUN_COMPLETED : AGENT_ANALYTICS_EVENTS.RUN_FAILED,
+            {
+              runId,
+              durationMs,
+              success: result.success,
+              summary: result.summary?.slice(0, 200)
+            }
+          );
+          ctx.platform.logger.info(`[run-handler] Run ${runId} completed: ${result.success}`);
+        } catch (error) {
+          const errorMsg = error instanceof Error ? error.message : String(error);
+          const durationMs = Date.now() - startTime;
+          const detailedTrace = traceWriter.getEntries();
+          await traceWriter.finalize?.();
+          if (detailedTrace.length > 0) {
+            await sessionManager.storeTraceArtifacts(finalSessionId, runId, detailedTrace);
           }
-        );
-        ctx.platform.logger.info(`[run-handler] Run ${runId} completed: ${result.success}`);
-      } catch (error) {
-        const errorMsg = error instanceof Error ? error.message : String(error);
-        const durationMs = Date.now() - startTime;
-        const detailedTrace = traceWriter.getEntries();
-        await traceWriter.finalize?.();
-        if (detailedTrace.length > 0) {
-          await sessionManager.storeTraceArtifacts(finalSessionId, runId, detailedTrace);
+          await RunManager.updateStatus(runId, "failed", {
+            completedAt: (/* @__PURE__ */ new Date()).toISOString(),
+            durationMs,
+            error: errorMsg
+          });
+          await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_FAILED, {
+            runId,
+            durationMs,
+            error: errorMsg.slice(0, 200)
+          });
+          ctx.platform.logger.error(`[run-handler] Run ${runId} failed: ${errorMsg}`);
         }
-        await RunManager.updateStatus(runId, "failed", {
-          completedAt: (/* @__PURE__ */ new Date()).toISOString(),
-          durationMs,
-          error: errorMsg
-        });
-        await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_FAILED, {
-          runId,
-          durationMs,
-          error: errorMsg.slice(0, 200)
-        });
-        ctx.platform.logger.error(`[run-handler] Run ${runId} failed: ${errorMsg}`);
-      }
-    })();
-    const wsPath = AGENTS_WS_CHANNELS.SESSION_STREAM.replace(":sessionId", finalSessionId);
-    const eventsPath = `${AGENTS_WS_BASE_PATH}${wsPath}`;
-    return {
-      runId,
-      sessionId: finalSessionId,
-      eventsPath,
-      status: "started",
-      startedAt: run.startedAt
-    };
+      })();
+      const wsPath = AGENTS_WS_CHANNELS.SESSION_STREAM.replace(":sessionId", finalSessionId);
+      const eventsPath = `${AGENTS_WS_BASE_PATH}${wsPath}`;
+      return {
+        runId,
+        sessionId: finalSessionId,
+        eventsPath,
+        status: "started",
+        startedAt: run.startedAt
+      };
+    } catch (err) {
+      rethrowForRest(err);
+    }
   }
 });