openmagic 0.33.1 → 0.33.3

package/dist/cli.js

@@ -189,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);
@@ -203,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);
@@ -251,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);
@@ -264,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;
@@ -279,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");
@@ -1565,7 +1572,9 @@ async function handleLlmChat(params, onChunk, onDone, onError) {
 }
 
 // src/server.ts
-var VERSION = "0.31.4";
+import { createRequire } from "module";
+var _require = createRequire(import.meta.url);
+var VERSION = _require("../package.json").version;
 var __dirname = dirname2(fileURLToPath(import.meta.url));
 var MAX_SERVER_LOGS = 200;
 var serverLogs = [];
@@ -1922,6 +1931,12 @@ function createProxyServer(targetHost, targetPort, roots) {
     const status = proxyRes.statusCode || 200;
     if (!isHtml && status < 400) {
       res.writeHead(status, proxyRes.headers);
+      proxyRes.on("error", () => {
+        try {
+          res.end();
+        } catch {
+        }
+      });
       proxyRes.pipe(res);
       return;
     }
@@ -1937,6 +1952,12 @@ function createProxyServer(targetHost, targetPort, roots) {
       delete headers["last-modified"];
       headers["cache-control"] = "no-store";
       res.writeHead(status, headers);
+      proxyRes.on("error", () => {
+        try {
+          res.end(buildInjectionScript(token));
+        } catch {
+        }
+      });
       proxyRes.pipe(res, { end: false });
       proxyRes.on("end", () => {
         res.end(buildInjectionScript(token));
@@ -1944,7 +1965,19 @@ function createProxyServer(targetHost, targetPort, roots) {
       return;
     }
     const chunks = [];
-    proxyRes.on("data", (c) => chunks.push(c));
+    let totalSize = 0;
+    proxyRes.on("data", (c) => {
+      if (totalSize < 16384) {
+        chunks.push(c);
+        totalSize += c.length;
+      }
+    });
+    proxyRes.on("error", () => {
+      try {
+        res.end();
+      } catch {
+      }
+    });
     proxyRes.on("end", () => {
       const body = Buffer.concat(chunks).toString("utf-8").slice(0, 2e3);
       const toolbarScript = buildInjectionScript(token);
@@ -2223,6 +2256,7 @@ function checkDependenciesInstalled(cwd = process.cwd()) {
 }
 
 // src/cli.ts
+import { createRequire as createRequire2 } from "module";
 var origEmitWarning = process.emitWarning;
 process.emitWarning = function(warning, ...args) {
   if (typeof warning === "string" && warning.includes("util._extend")) return;
@@ -2239,7 +2273,8 @@ process.on("uncaughtException", (err) => {
 });
 var childProcesses = [];
 var lastDetectedPort = null;
-var VERSION2 = "0.31.1";
+var _require2 = createRequire2(import.meta.url);
+var VERSION2 = _require2("../package.json").version;
 function ask(question) {
   const rl = createInterface({ input: process.stdin, output: process.stdout });
   return new Promise((resolve4) => {
@@ -2670,9 +2705,9 @@ async function offerToStartDevServer(expectedPort) {
       }
     }, 3e3);
   };
-  process.on("exit", cleanup);
-  process.on("SIGINT", cleanup);
-  process.on("SIGTERM", cleanup);
+  process.once("exit", cleanup);
+  process.once("SIGINT", cleanup);
+  process.once("SIGTERM", cleanup);
   console.log(
     chalk.dim(`  Waiting for dev server on port ${port}...`)
   );