openmagic 0.33.0 → 0.33.1

package/dist/cli.js

@@ -82,9 +82,12 @@ import {
   readdirSync,
   copyFileSync,
   mkdirSync as mkdirSync2,
-  realpathSync
+  realpathSync,
+  rmSync
 } from "fs";
 import { join as join2, resolve, relative, dirname, extname } from "path";
+import { tmpdir } from "os";
+import { createHash } from "crypto";
 var IGNORED_DIRS = /* @__PURE__ */ new Set([
   "node_modules",
   ".git",
@@ -145,6 +148,25 @@ function readFileSafe(filePath, roots) {
     return { error: `Failed to read file: ${e.message}` };
   }
 }
+var BACKUP_DIR = join2(tmpdir(), "openmagic-backups");
+var backupMap = /* @__PURE__ */ new Map();
+function getBackupPath(filePath) {
+  const hash = createHash("md5").update(resolve(filePath)).digest("hex").slice(0, 12);
+  const name = filePath.split(/[/\\]/).pop() || "file";
+  return join2(BACKUP_DIR, `${hash}_${name}`);
+}
+function getBackupForFile(filePath) {
+  return backupMap.get(resolve(filePath));
+}
+function cleanupBackups() {
+  try {
+    if (existsSync2(BACKUP_DIR)) {
+      rmSync(BACKUP_DIR, { recursive: true, force: true });
+    }
+  } catch {
+  }
+  backupMap.clear();
+}
 function writeFileSafe(filePath, content, roots) {
   if (!isPathSafe(filePath, roots)) {
     return { ok: false, error: "Path is outside allowed roots" };
@@ -152,8 +174,10 @@ function writeFileSafe(filePath, content, roots) {
   try {
     let backupPath;
     if (existsSync2(filePath)) {
-      backupPath = filePath + ".openmagic-backup";
+      if (!existsSync2(BACKUP_DIR)) mkdirSync2(BACKUP_DIR, { recursive: true });
+      backupPath = getBackupPath(filePath);
       copyFileSync(filePath, backupPath);
+      backupMap.set(resolve(filePath), backupPath);
     }
     const dir = dirname(filePath);
     if (!existsSync2(dir)) {
@@ -1706,6 +1730,30 @@ async function handleMessage(ws, msg, state, roots) {
       }
       break;
     }
+    case "fs.undo": {
+      const payload = msg.payload;
+      if (!payload?.path) {
+        sendError(ws, "invalid_payload", "Missing path", msg.id);
+        break;
+      }
+      const backupPath = getBackupForFile(payload.path);
+      if (!backupPath) {
+        sendError(ws, "fs_error", "No backup found", msg.id);
+        break;
+      }
+      try {
+        const backupContent = readFileSync3(backupPath, "utf-8");
+        const writeResult = writeFileSafe(payload.path, backupContent, roots);
+        if (!writeResult.ok) {
+          sendError(ws, "fs_error", writeResult.error || "Undo failed", msg.id);
+          break;
+        }
+        send(ws, { id: msg.id, type: "fs.undone", payload: { path: payload.path, ok: true } });
+      } catch (e) {
+        sendError(ws, "fs_error", `Backup read failed: ${e.message}`, msg.id);
+      }
+      break;
+    }
     case "fs.list": {
       const payload = msg.payload;
       const root = payload?.root || roots[0];
@@ -2392,6 +2440,7 @@ program.name("openmagic").description("AI-powered coding toolbar for any web app
   const shutdown = () => {
     console.log("");
     console.log(chalk.dim("  Shutting down OpenMagic..."));
+    cleanupBackups();
     proxyServer.close();
     process.exit(0);
   };