grepmax 0.26.59 → 0.26.61

package/dist/lib/utils/resource-snapshot.js

@@ -3,18 +3,5 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.buildResourceSnapshot = buildResourceSnapshot;
 function buildResourceSnapshot(counters, memory = process.memoryUsage(), at = Date.now()) {
     const mb = (bytes) => Math.round(bytes / 1024 / 1024);
-    return {
-        at,
-        reason: counters.reason,
-        footprintMb: counters.footprintMb,
-        rssMb: mb(memory.rss),
-        heapUsedMb: mb(memory.heapUsed),
-        externalMb: mb(memory.external),
-        arrayBuffersMb: mb(memory.arrayBuffers),
-        lanceCacheMb: counters.lanceCacheBytes === null ? null : mb(counters.lanceCacheBytes),
-        workers: counters.workers,
-        pendingFiles: counters.pendingFiles,
-        operations: counters.operations,
-        maintenance: counters.maintenance,
-    };
+    return Object.assign({ at, reason: counters.reason, footprintMb: counters.footprintMb, rssMb: mb(memory.rss), heapUsedMb: mb(memory.heapUsed), externalMb: mb(memory.external), arrayBuffersMb: mb(memory.arrayBuffers), lanceCacheMb: counters.lanceCacheBytes === null ? null : mb(counters.lanceCacheBytes), workers: counters.workers, pendingFiles: counters.pendingFiles, operations: counters.operations, maintenance: counters.maintenance }, (counters.host ? { host: counters.host } : {}));
 }

package/dist/lib/workers/pool.js

@@ -58,8 +58,10 @@ const config_1 = require("../../config");
 const embedding_generation_1 = require("../index/embedding-generation");
 const index_config_1 = require("../index/index-config");
 const autostart_1 = require("../utils/autostart");
+const host_guard_policy_1 = require("../utils/host-guard-policy");
 const kernel_zone_1 = require("../utils/kernel-zone");
 const logger_1 = require("../utils/logger");
+const resource_budget_1 = require("../utils/resource-budget");
 const safety_latch_1 = require("../utils/safety-latch");
 const TASK_TIMEOUT_MS = (() => {
     var _a;
@@ -135,11 +137,37 @@ class ProcessWorker {
         // fire for the same crash.
         this.cleanedUp = false;
         const memArgs = maxMemoryMb ? [`--max-old-space-size=${maxMemoryMb}`] : [];
-        this.child = childProcess.fork(modulePath, {
-            execArgv: [...memArgs, ...execArgv],
-            env: Object.assign(Object.assign({}, process.env), embeddingEnvironment),
-            serialization: "advanced",
+        this.resourceReservation = resource_budget_1.resourceBudget.reserve(resource_budget_1.WORKER_RESOURCE_RESERVE_MB, "worker");
+        try {
+            const denied = (0, autostart_1.daemonStartDeniedReason)();
+            if (denied)
+                throw new Error(denied);
+            this.child = childProcess.fork(modulePath, {
+                execArgv: [...memArgs, ...execArgv],
+                env: Object.assign(Object.assign({}, process.env), embeddingEnvironment),
+                serialization: "advanced",
+            });
+        }
+        catch (error) {
+            this.resourceReservation.release();
+            throw error;
+        }
+        // Keep the reservation until close, including SIGKILL/timeout paths.
+        this.child.once("close", () => {
+            try {
+                this.resourceReservation.release();
+            }
+            catch (_a) { }
         });
+        try {
+            if (!this.child.pid)
+                throw new Error("worker PID unavailable");
+            this.resourceReservation.attach(this.child.pid);
+        }
+        catch (error) {
+            this.child.kill("SIGKILL");
+            throw error;
+        }
     }
 }
 function resolveProcessWorker() {
@@ -365,7 +393,7 @@ class WorkerPool {
         var _a;
         return Object.assign(new Error((_a = this.spawnDeniedReason) !== null && _a !== void 0 ? _a : "Worker spawn denied"), { code: "HOST_SAFETY" });
     }
-    denySpawn(reason, persist = true) {
+    denySpawn(reason, persist = false) {
         // A pool cannot automatically recover after a failed admission. In
         // particular, timer/exit callbacks must not retry forks or strand queues.
         this.spawnDeniedReason = `Worker spawn denied: ${reason}`;
@@ -396,22 +424,31 @@ class WorkerPool {
             const quarantine = (0, autostart_1.daemonStartDeniedReason)();
             if (quarantine)
                 return this.denySpawn(quarantine, false);
+            const strictHostGuard = (0, host_guard_policy_1.hostGuardPolicy)() === "strict";
             if (process.platform === "darwin") {
                 const memory = (0, kernel_zone_1.probeMemoryPressure)();
-                if (memory.status !== "known" || memory.pressure !== "normal") {
-                    return this.denySpawn(`OS memory pressure ${memory.status === "known" ? memory.pressure : memory.status}; ${(0, kernel_zone_1.formatPressureProbe)(memory)}`);
+                if ((memory.status === "known" && memory.pressure === "critical") ||
+                    (strictHostGuard &&
+                        (memory.status !== "known" || memory.pressure !== "normal"))) {
+                    return this.denySpawn(`OS memory pressure ${memory.status === "known" ? memory.pressure : memory.status}; ${(0, kernel_zone_1.formatPressureProbe)(memory)}`, memory.status === "known" && memory.pressure === "critical");
                 }
                 const kernel = (0, kernel_zone_1.probeKernelZoneUsage)();
-                if (kernel.status !== "known" || kernel.usage.pressure !== "ok") {
-                    return this.denySpawn(`kernel pressure ${kernel.status === "known" ? kernel.usage.pressure : kernel.status}; ${(0, kernel_zone_1.formatPressureProbe)(kernel)}`);
+                if ((kernel.status === "known" && kernel.usage.pressure === "critical") ||
+                    (strictHostGuard &&
+                        (kernel.status !== "known" || kernel.usage.pressure !== "ok"))) {
+                    return this.denySpawn(`kernel pressure ${kernel.status === "known" ? kernel.usage.pressure : kernel.status}; ${(0, kernel_zone_1.formatPressureProbe)(kernel)}`, kernel.status === "known" && kernel.usage.pressure === "critical");
                 }
             }
             if (process.platform === "darwin") {
                 // Refresh the cheap OS sample after the potentially slow kernel probe.
                 const freshMemory = (0, kernel_zone_1.probeMemoryPressure)();
-                if (freshMemory.status !== "known" ||
-                    freshMemory.pressure !== "normal") {
-                    return this.denySpawn(`OS memory pressure ${freshMemory.status === "known" ? freshMemory.pressure : freshMemory.status}; ${(0, kernel_zone_1.formatPressureProbe)(freshMemory)}`);
+                if ((freshMemory.status === "known" &&
+                    freshMemory.pressure === "critical") ||
+                    (strictHostGuard &&
+                        (freshMemory.status !== "known" ||
+                            freshMemory.pressure !== "normal"))) {
+                    return this.denySpawn(`OS memory pressure ${freshMemory.status === "known" ? freshMemory.pressure : freshMemory.status}; ${(0, kernel_zone_1.formatPressureProbe)(freshMemory)}`, freshMemory.status === "known" &&
+                        freshMemory.pressure === "critical");
                 }
             }
             // Probes are synchronous but another process can quarantine the host
@@ -421,8 +458,10 @@ class WorkerPool {
                 return this.denySpawn(finalQuarantine, false);
             worker = new ProcessWorker(this.modulePath, this.execArgv, this.embeddingEnvironment, config_1.MAX_WORKER_MEMORY_MB);
         }
-        catch (_a) {
-            return this.denySpawn("host admission or fork failed");
+        catch (error) {
+            return this.denySpawn(error instanceof Error
+                ? error.message
+                : "host admission or fork failed");
         }
         (0, logger_1.log)("pool", `spawn PID:${worker.child.pid} (${this.workers.length + 1}/${Math.max(1, config_1.CONFIG.WORKER_THREADS)})`);
         const onMessage = (msg) => {
@@ -625,6 +664,15 @@ class WorkerPool {
             this.dispatch();
             return;
         }
+        try {
+            resource_budget_1.resourceBudget.check();
+        }
+        catch (error) {
+            this.denySpawn(error instanceof Error
+                ? error.message
+                : "aggregate resource admission unavailable");
+            return;
+        }
         idle.busy = true;
         idle.pendingTaskId = task.id;
         idle.busySince = Date.now();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "grepmax",
-  "version": "0.26.59",
+  "version": "0.26.61",
   "author": "Robert Owens <78518764+reowens@users.noreply.github.com>",
   "homepage": "https://github.com/reowens/grepmax",
   "bugs": {

package/plugins/grepmax/.claude-plugin/plugin.json

@@ -1,6 +1,6 @@
 {
   "name": "grepmax",
-  "version": "0.26.59",
+  "version": "0.26.61",
   "description": "Semantic code search for Claude Code. Automatically indexes your project and provides intelligent search capabilities.",
   "author": {
     "name": "Robert Owens",