openmagic 0.33.0 → 0.33.2

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)) {
@@ -165,13 +189,14 @@ function writeFileSafe(filePath, content, roots) {
     return { ok: false, error: `Failed to write file: ${e.message}` };
   }
 }
+var MAX_LIST_ENTRIES = 2e3;
 function listFiles(rootPath, roots, maxDepth = 4) {
   if (!isPathSafe(rootPath, roots)) {
     return [];
   }
   const entries = [];
   function walk(dir, depth) {
-    if (depth > maxDepth) return;
+    if (depth > maxDepth || entries.length >= MAX_LIST_ENTRIES) return;
     let items;
     try {
       items = readdirSync(dir);
@@ -179,6 +204,7 @@ function listFiles(rootPath, roots, maxDepth = 4) {
       return;
     }
     for (const item of items) {
+      if (entries.length >= MAX_LIST_ENTRIES) return;
       if (IGNORED_DIRS.has(item)) continue;
       if (item.startsWith(".") && item !== ".env.example") continue;
       const fullPath = join2(dir, item);
@@ -227,12 +253,15 @@ var GREP_EXTENSIONS = /* @__PURE__ */ new Set([
   ".py",
   ".rb"
 ]);
+var MAX_GREP_FILE_SIZE = 256 * 1024;
+var MAX_GREP_FILES_SCANNED = 500;
 function grepFiles(pattern, searchRoot, roots, maxResults = 30) {
   if (!isPathSafe(searchRoot, roots)) return [];
   const results = [];
   const lowerPattern = pattern.toLowerCase();
+  let filesScanned = 0;
   function walk(dir, depth) {
-    if (depth > 5 || results.length >= maxResults) return;
+    if (depth > 6 || results.length >= maxResults || filesScanned >= MAX_GREP_FILES_SCANNED) return;
     let items;
     try {
       items = readdirSync(dir);
@@ -240,7 +269,7 @@ function grepFiles(pattern, searchRoot, roots, maxResults = 30) {
       return;
     }
     for (const item of items) {
-      if (results.length >= maxResults) return;
+      if (results.length >= maxResults || filesScanned >= MAX_GREP_FILES_SCANNED) return;
       if (IGNORED_DIRS.has(item) || item.startsWith(".") && item !== ".env.example") continue;
       const fullPath = join2(dir, item);
       let stat;
@@ -255,6 +284,8 @@ function grepFiles(pattern, searchRoot, roots, maxResults = 30) {
       } else if (stat.isFile()) {
         const ext = extname(item).toLowerCase();
         if (!GREP_EXTENSIONS.has(ext)) continue;
+        if (stat.size > MAX_GREP_FILE_SIZE) continue;
+        filesScanned++;
         try {
           const content = readFileSync2(fullPath, "utf-8");
           const lines = content.split("\n");
@@ -1706,6 +1737,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 +2447,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);
   };