@plumpslabs/kuma 2.3.8 → 2.3.9

package/dist/kumaCheckpoint-XIQWRMGV.js

@@ -0,0 +1,207 @@
+import {
+  sessionMemory
+} from "./chunk-R6B3VUSB.js";
+import {
+  getDb,
+  resetDbInstance
+} from "./chunk-3TX6Q37T.js";
+import {
+  getProjectRoot
+} from "./chunk-E2KFPEBT.js";
+
+// src/engine/kumaCheckpoint.ts
+import fs from "fs";
+import path from "path";
+var CHECKPOINT_DIR = ".kuma/checkpoints";
+async function createCheckpoint(label, description) {
+  try {
+    const root = getProjectRoot();
+    const cpDir = path.join(root, CHECKPOINT_DIR, sanitizeLabel(label));
+    if (fs.existsSync(cpDir)) {
+      return `\u26A0\uFE0F Checkpoint "${label}" already exists. Use a different label or remove it first.`;
+    }
+    fs.mkdirSync(cpDir, { recursive: true });
+    const db = await getDb();
+    const filesStmt = db.prepare(
+      "SELECT DISTINCT file_path FROM change_log ORDER BY id DESC LIMIT 100"
+    );
+    const files = [];
+    while (filesStmt.step()) {
+      const row = filesStmt.getAsObject();
+      const fp = row.file_path;
+      const fullPath = path.resolve(root, fp);
+      if (fs.existsSync(fullPath)) {
+        const content = fs.readFileSync(fullPath, "utf-8");
+        const hash = simpleHash(content);
+        const fileDir = path.dirname(path.join(cpDir, "files", fp));
+        if (!fs.existsSync(fileDir)) fs.mkdirSync(fileDir, { recursive: true });
+        fs.writeFileSync(path.join(cpDir, "files", fp), content, "utf-8");
+        files.push({ path: fp, hash });
+      }
+    }
+    filesStmt.free();
+    const dbData = db.export();
+    fs.writeFileSync(path.join(cpDir, "kuma.db"), Buffer.from(dbData));
+    const manifest = {
+      label,
+      timestamp: Date.now(),
+      files,
+      dbSnapshot: true,
+      description
+    };
+    fs.writeFileSync(
+      path.join(cpDir, "manifest.json"),
+      JSON.stringify(manifest, null, 2),
+      "utf-8"
+    );
+    sessionMemory.recordToolCall("kuma_checkpoint_create", {
+      label,
+      filesCount: files.length
+    });
+    return [
+      `\u2705 **Checkpoint Created**: "${label}"`,
+      `\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501`,
+      ``,
+      `\u{1F4E6} **${files.length} file(s)** snapshotted`,
+      `\u{1F5C4}\uFE0F  **Database** snapshotted`,
+      `\u{1F4CD} ${cpDir}`,
+      ``,
+      `\u{1F4A1} To restore: kuma_safety({ action: 'rollback', label: '${label}' })`,
+      description ? `\u{1F4DD} ${description}` : ""
+    ].filter(Boolean).join("\n");
+  } catch (err) {
+    return `\u274C Checkpoint failed: ${err}`;
+  }
+}
+async function rollbackToCheckpoint(label) {
+  try {
+    const root = getProjectRoot();
+    const cpDir = path.join(root, CHECKPOINT_DIR, sanitizeLabel(label));
+    const manifestPath = path.join(cpDir, "manifest.json");
+    if (!fs.existsSync(manifestPath)) {
+      return `\u274C Checkpoint "${label}" not found. Use kuma_safety({ action: 'checkpoint_list' }) to see available checkpoints.`;
+    }
+    const manifest = JSON.parse(
+      fs.readFileSync(manifestPath, "utf-8")
+    );
+    let restored = 0;
+    let failed = 0;
+    for (const f of manifest.files) {
+      const snapshotPath = path.join(cpDir, "files", f.path);
+      if (fs.existsSync(snapshotPath)) {
+        try {
+          const content = fs.readFileSync(snapshotPath, "utf-8");
+          const targetPath = path.resolve(root, f.path);
+          const targetDir = path.dirname(targetPath);
+          if (!fs.existsSync(targetDir)) fs.mkdirSync(targetDir, { recursive: true });
+          fs.writeFileSync(targetPath, content, "utf-8");
+          restored++;
+        } catch {
+          failed++;
+        }
+      }
+    }
+    const dbSnapshotPath = path.join(cpDir, "kuma.db");
+    if (fs.existsSync(dbSnapshotPath)) {
+      const kumaDir = path.join(root, ".kuma");
+      const dbPath = path.join(kumaDir, "kuma.db");
+      const snapshotData = fs.readFileSync(dbSnapshotPath);
+      fs.writeFileSync(dbPath, snapshotData);
+      resetDbInstance();
+    }
+    sessionMemory.recordToolCall("kuma_checkpoint_rollback", {
+      label,
+      restored,
+      failed
+    });
+    return [
+      `\u{1F504} **Rollback Complete**: "${label}"`,
+      `\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501`,
+      ``,
+      `\u2705 Restored **${restored} file(s)**`,
+      failed > 0 ? `\u26A0\uFE0F ${failed} file(s) failed to restore` : "",
+      `\u{1F5C4}\uFE0F  Database restored to checkpoint state`,
+      ``,
+      `\u{1F4A1} You may need to restart your language server for full effect.`
+    ].filter(Boolean).join("\n");
+  } catch (err) {
+    return `\u274C Rollback failed: ${err}`;
+  }
+}
+function listCheckpoints() {
+  try {
+    const root = getProjectRoot();
+    const cpDir = path.join(root, CHECKPOINT_DIR);
+    if (!fs.existsSync(cpDir)) {
+      return "\u{1F4ED} No checkpoints found. Use kuma_safety({ action: 'checkpoint', label: 'pre-feature-x' }) to create one.";
+    }
+    const entries = fs.readdirSync(cpDir);
+    const checkpoints = [];
+    for (const entry of entries) {
+      const manifestPath = path.join(cpDir, entry, "manifest.json");
+      if (fs.existsSync(manifestPath)) {
+        try {
+          const manifest = JSON.parse(
+            fs.readFileSync(manifestPath, "utf-8")
+          );
+          checkpoints.push({ label: entry, manifest });
+        } catch {
+        }
+      }
+    }
+    if (checkpoints.length === 0) {
+      return "\u{1F4ED} No valid checkpoints found.";
+    }
+    const lines = [
+      "\u{1F4E6} **Checkpoints**",
+      "\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501",
+      ""
+    ];
+    for (const cp of checkpoints) {
+      const time = new Date(cp.manifest.timestamp).toLocaleString();
+      lines.push(`  \u{1F4CC} **${cp.manifest.label}**`);
+      lines.push(`     \u{1F550} ${time} | \u{1F4C1} ${cp.manifest.files.length} files | \u{1F4BE} ${cp.manifest.dbSnapshot ? "DB included" : "no DB"}`);
+      if (cp.manifest.description) {
+        lines.push(`     \u{1F4DD} ${cp.manifest.description}`);
+      }
+      lines.push("");
+    }
+    return lines.join("\n");
+  } catch (err) {
+    return `Error: ${err}`;
+  }
+}
+function pruneCheckpoints(keep = 5) {
+  try {
+    const root = getProjectRoot();
+    const cpDir = path.join(root, CHECKPOINT_DIR);
+    if (!fs.existsSync(cpDir)) return "\u{1F4ED} No checkpoints to prune.";
+    const entries = fs.readdirSync(cpDir).map((e) => ({ name: e, time: fs.statSync(path.join(cpDir, e)).mtimeMs })).sort((a, b) => b.time - a.time);
+    let removed = 0;
+    for (let i = keep; i < entries.length; i++) {
+      fs.rmSync(path.join(cpDir, entries[i].name), { recursive: true, force: true });
+      removed++;
+    }
+    return removed > 0 ? `\u{1F9F9} Pruned **${removed} old checkpoint(s)**. Kept ${Math.min(keep, entries.length)} most recent.` : "\u2705 No checkpoints needed pruning.";
+  } catch (err) {
+    return `Error: ${err}`;
+  }
+}
+function sanitizeLabel(label) {
+  return label.replace(/[^a-zA-Z0-9_-]/g, "_").substring(0, 64);
+}
+function simpleHash(str) {
+  let hash = 0;
+  for (let i = 0; i < str.length; i++) {
+    const char = str.charCodeAt(i);
+    hash = (hash << 5) - hash + char;
+    hash = hash & hash;
+  }
+  return Math.abs(hash).toString(36);
+}
+export {
+  createCheckpoint,
+  listCheckpoints,
+  pruneCheckpoints,
+  rollbackToCheckpoint
+};