@wrongstack/core 0.5.0 → 0.5.2

package/dist/storage/index.js

@@ -186,7 +186,7 @@ var DefaultSessionStore = class {
       );
     }
     try {
-      return new FileSessionWriter(id, handle, startedAt, meta, { dir: this.dir, filePath: file });
+      return new FileSessionWriter(id, handle, startedAt, meta, this.events, { dir: this.dir, filePath: file });
     } catch (err) {
       await handle.close().catch(() => {
       });
@@ -214,6 +214,7 @@ var DefaultSessionStore = class {
         model: data.metadata.model,
         provider: data.metadata.provider
       },
+      this.events,
       { resumed: true, dir: this.dir, filePath: file }
     );
     return { writer, data };
@@ -377,11 +378,12 @@ var DefaultSessionStore = class {
   }
 };
 var FileSessionWriter = class {
-  constructor(id, handle, startedAt, meta, opts = {}) {
+  constructor(id, handle, startedAt, meta, events, opts = {}) {
     this.id = id;
     this.handle = handle;
     this.startedAt = startedAt;
     this.meta = meta;
+    this.events = events;
     this.resumed = opts.resumed ?? false;
     this.manifestFile = opts.dir ? path2.join(opts.dir, `${id}.summary.json`) : "";
     this.filePath = opts.filePath ?? "";
@@ -398,6 +400,7 @@ var FileSessionWriter = class {
   handle;
   startedAt;
   meta;
+  events;
   closed = false;
   closing = false;
   manifestFile;
@@ -415,6 +418,13 @@ var FileSessionWriter = class {
   resumed;
   appendFailCount = 0;
   lastAppendWarnAt = 0;
+  // Rewind support: track pending file changes and prompt index
+  promptIndex = 0;
+  pendingFileSnapshots = [];
+  // Register a pending file change. Call recordFileChange() from tools.
+  recordFileChange(input) {
+    this.pendingFileSnapshots.push(input);
+  }
   async writeSessionStartLazy() {
     const record = `${JSON.stringify({
       type: this.resumed ? "session_resumed" : "session_start",
@@ -490,6 +500,87 @@ var FileSessionWriter = class {
     } catch {
     }
   }
+  async writeCheckpoint(promptIndex, promptPreview) {
+    const fileCount = this.pendingFileSnapshots.length;
+    if (fileCount > 0) {
+      await this.writeFileSnapshot(promptIndex, [...this.pendingFileSnapshots]);
+      this.pendingFileSnapshots = [];
+    }
+    this.promptIndex = promptIndex + 1;
+    await this.append({
+      type: "checkpoint",
+      ts: (/* @__PURE__ */ new Date()).toISOString(),
+      promptIndex,
+      promptPreview
+    });
+    this.events?.emit("checkpoint.written", {
+      promptIndex,
+      promptPreview,
+      ts: (/* @__PURE__ */ new Date()).toISOString(),
+      fileCount
+    });
+  }
+  async writeFileSnapshot(promptIndex, files) {
+    await this.append({
+      type: "file_snapshot",
+      ts: (/* @__PURE__ */ new Date()).toISOString(),
+      promptIndex,
+      files
+    });
+  }
+  async truncateToCheckpoint(targetPromptIndex) {
+    if (!this.filePath) return 0;
+    const raw = await fsp.readFile(this.filePath, "utf8");
+    const lines = raw.split("\n");
+    const kept = [];
+    let removedCount = 0;
+    let targetCheckpointLine = -1;
+    let afterTarget = false;
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i];
+      if (!line.trim()) continue;
+      let event;
+      try {
+        event = JSON.parse(line);
+      } catch {
+        kept.push(line);
+        continue;
+      }
+      if (event.type === "checkpoint") {
+        if (event.promptIndex === targetPromptIndex) {
+          targetCheckpointLine = kept.length;
+          afterTarget = true;
+        } else if (event.promptIndex > targetPromptIndex) {
+          afterTarget = true;
+        }
+      }
+      if (event.promptIndex !== void 0 && event.promptIndex > targetPromptIndex) {
+        removedCount++;
+      } else if (event.promptIndex === void 0) {
+        if (!afterTarget || targetCheckpointLine === -1) {
+          kept.push(line);
+        } else {
+          removedCount++;
+        }
+      } else {
+        kept.push(line);
+      }
+    }
+    const truncated = kept.join("\n");
+    await fsp.writeFile(this.filePath, truncated + "\n", "utf8");
+    await this.append({
+      type: "rewound",
+      ts: (/* @__PURE__ */ new Date()).toISOString(),
+      toPromptIndex: targetPromptIndex,
+      revertedFiles: []
+    });
+    this.events?.emit("session.rewound", {
+      toPromptIndex: targetPromptIndex,
+      revertedFiles: [],
+      removedEvents: removedCount
+    });
+    return removedCount;
+  }
 };
 function userInputTitle(content) {
   if (typeof content === "string") return content.slice(0, 60);
@@ -1765,6 +1856,158 @@ var SessionAnalyzer = class {
     return last - first;
   }
 };
+var DefaultSessionRewinder = class {
+  constructor(sessionsDir) {
+    this.sessionsDir = sessionsDir;
+  }
+  sessionsDir;
+  async listCheckpoints(sessionId) {
+    const file = path2.join(this.sessionsDir, `${sessionId}.jsonl`);
+    const raw = await fsp.readFile(file, "utf8");
+    const events = parseEvents(raw);
+    const fileCountMap = /* @__PURE__ */ new Map();
+    for (const event of events) {
+      if (event.type === "file_snapshot") {
+        const e = event;
+        fileCountMap.set(e.promptIndex, (fileCountMap.get(e.promptIndex) ?? 0) + e.files.length);
+      }
+    }
+    const checkpoints = [];
+    for (const event of events) {
+      if (event.type === "checkpoint") {
+        const e = event;
+        checkpoints.push({
+          promptIndex: e.promptIndex,
+          promptPreview: e.promptPreview,
+          ts: e.ts,
+          fileCount: fileCountMap.get(e.promptIndex) ?? 0
+        });
+      }
+    }
+    return checkpoints;
+  }
+  async rewindToCheckpoint(sessionId, checkpointIndex) {
+    const file = path2.join(this.sessionsDir, `${sessionId}.jsonl`);
+    const raw = await fsp.readFile(file, "utf8");
+    const events = parseEvents(raw);
+    let targetIdx = -1;
+    for (let i = 0; i < events.length; i++) {
+      const event = events[i];
+      if (event.type === "checkpoint") {
+        const checkpointEvent = event;
+        if (checkpointEvent.promptIndex === checkpointIndex) {
+          targetIdx = i;
+          break;
+        }
+      }
+    }
+    if (targetIdx === -1) {
+      throw new Error(`Checkpoint ${checkpointIndex} not found`);
+    }
+    const snapshotsToRevert = [];
+    for (let i = targetIdx + 1; i < events.length; i++) {
+      const event = events[i];
+      if (event.type === "checkpoint") {
+        break;
+      }
+      if (event.type === "file_snapshot") {
+        const snapshotEvent = event;
+        if (snapshotEvent.promptIndex >= checkpointIndex) {
+          snapshotsToRevert.push({ promptIndex: snapshotEvent.promptIndex, files: snapshotEvent.files });
+        }
+      }
+    }
+    const result = await revertSnapshots(snapshotsToRevert);
+    const removedEvents = events.length - targetIdx - 1;
+    return { ...result, toPromptIndex: checkpointIndex, removedEvents };
+  }
+  async rewindLastN(sessionId, n) {
+    const file = path2.join(this.sessionsDir, `${sessionId}.jsonl`);
+    const raw = await fsp.readFile(file, "utf8");
+    const events = parseEvents(raw);
+    const checkpoints = [];
+    for (const event of events) {
+      if (event.type === "checkpoint") {
+        checkpoints.push({ promptIndex: event.promptIndex, ts: event.ts });
+      }
+    }
+    if (checkpoints.length === 0) {
+      return { revertedFiles: [], errors: [], toPromptIndex: 0, removedEvents: 0 };
+    }
+    checkpoints.sort((a, b) => b.promptIndex - a.promptIndex);
+    const targetIndex = checkpoints[n]?.promptIndex ?? 0;
+    const snapshotsToRevert = [];
+    let shouldRevert = false;
+    for (const event of events) {
+      if (event.type === "checkpoint" && event.promptIndex === targetIndex) {
+        shouldRevert = true;
+        continue;
+      }
+      if (shouldRevert && event.type === "file_snapshot") {
+        snapshotsToRevert.push({ promptIndex: event.promptIndex, files: event.files });
+      }
+    }
+    const result = await revertSnapshots(snapshotsToRevert.reverse());
+    return { ...result, toPromptIndex: targetIndex, removedEvents: snapshotsToRevert.length };
+  }
+  async rewindToStart(sessionId) {
+    const file = path2.join(this.sessionsDir, `${sessionId}.jsonl`);
+    const raw = await fsp.readFile(file, "utf8");
+    const events = parseEvents(raw);
+    const allSnapshots = [];
+    for (const event of events) {
+      if (event.type === "file_snapshot") {
+        allSnapshots.push({ promptIndex: event.promptIndex, files: event.files });
+      }
+    }
+    if (allSnapshots.length === 0) {
+      return { revertedFiles: [], errors: [], toPromptIndex: 0, removedEvents: 0 };
+    }
+    const result = await revertSnapshots(allSnapshots.reverse());
+    return { ...result, toPromptIndex: 0, removedEvents: allSnapshots.length };
+  }
+};
+function parseEvents(raw) {
+  const lines = raw.split("\n").filter((l) => l.trim());
+  const events = [];
+  for (const line of lines) {
+    try {
+      const parsed = JSON.parse(line);
+      if (parsed !== null && typeof parsed === "object" && typeof parsed.type === "string" && typeof parsed.ts === "string") {
+        events.push(parsed);
+      }
+    } catch {
+    }
+  }
+  return events;
+}
+async function revertSnapshots(snapshots) {
+  const revertedFiles = [];
+  const errors = [];
+  for (const snapshot of snapshots) {
+    for (const file of snapshot.files) {
+      try {
+        if (file.action === "deleted") {
+          if (file.before !== null) {
+            await fsp.writeFile(file.path, file.before, { mode: 420 });
+            revertedFiles.push(file.path);
+          }
+        } else if (file.action === "created") {
+          await fsp.unlink(file.path);
+          revertedFiles.push(file.path);
+        } else if (file.action === "modified") {
+          if (file.before !== null) {
+            await fsp.writeFile(file.path, file.before, { mode: 420 });
+            revertedFiles.push(file.path);
+          }
+        }
+      } catch (err) {
+        errors.push(`${file.path}: ${err instanceof Error ? err.message : String(err)}`);
+      }
+    }
+  }
+  return { revertedFiles, errors };
+}
 async function loadTodosCheckpoint(filePath) {
   let raw;
   try {
@@ -2047,6 +2290,6 @@ var DirectorStateCheckpoint = class {
   }
 };
 
-export { ConfigMigrationError, DEFAULT_CONFIG_MIGRATIONS, DefaultAttachmentStore, DefaultConfigLoader, DefaultConfigStore, DefaultMemoryStore, DefaultSessionReader, DefaultSessionStore, DirectorStateCheckpoint, QueueStore, RecoveryLock, SessionAnalyzer, addPlanItem, attachPlanCheckpoint, attachTodosCheckpoint, clearPlan, emptyPlan, formatPlan, loadDirectorState, loadPlan, loadTodosCheckpoint, removePlanItem, runConfigMigrations, savePlan, saveTodosCheckpoint, setPlanItemStatus };
+export { ConfigMigrationError, DEFAULT_CONFIG_MIGRATIONS, DefaultAttachmentStore, DefaultConfigLoader, DefaultConfigStore, DefaultMemoryStore, DefaultSessionReader, DefaultSessionRewinder, DefaultSessionStore, DirectorStateCheckpoint, QueueStore, RecoveryLock, SessionAnalyzer, addPlanItem, attachPlanCheckpoint, attachTodosCheckpoint, clearPlan, emptyPlan, formatPlan, loadDirectorState, loadPlan, loadTodosCheckpoint, removePlanItem, runConfigMigrations, savePlan, saveTodosCheckpoint, setPlanItemStatus };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map