@threadbase-sh/streamer 1.70.5 → 1.70.6

package/dist/index.js

@@ -14671,6 +14671,7 @@ function pruneAgentConversations(cache) {
 }
 
 // src/services/host-pressure/hostPressure.ts
+import { execFile as execFile4 } from "child_process";
 import { cpus, freemem, loadavg, totalmem } from "os";
 import { monitorEventLoopDelay } from "perf_hooks";
 var HOST_PRESSURE_SAMPLE_MS = 5e3;
@@ -14705,6 +14706,59 @@ var HOST_PRESSURE_BARS = {
 };
 var REASON_ORDER = ["memory", "event_loop", "load", "agents"];
 var RANK = { ok: 0, elevated: 1, critical: 2 };
+function parseMacMemoryPressureFreeRatio(output) {
+  const match = output.match(/System-wide memory free percentage:\s*(\d+(?:\.\d+)?)%/);
+  if (!match) return void 0;
+  const percentage = Number(match[1]);
+  return percentage <= 100 ? percentage / 100 : void 0;
+}
+function runMacMemoryPressure() {
+  return new Promise((resolve2, reject) => {
+    execFile4(
+      "/usr/bin/memory_pressure",
+      ["-Q"],
+      { encoding: "utf8", timeout: 1e3 },
+      (error, stdout) => {
+        if (error) {
+          reject(error);
+          return;
+        }
+        resolve2(stdout);
+      }
+    );
+  });
+}
+async function readMacMemoryPressureFreeRatio(run2 = runMacMemoryPressure) {
+  try {
+    return parseMacMemoryPressureFreeRatio(await run2());
+  } catch {
+    return void 0;
+  }
+}
+var MacMemoryPressureProbe = class {
+  constructor(read) {
+    this.read = read;
+  }
+  read;
+  current;
+  pending = false;
+  refresh() {
+    if (this.pending) return;
+    this.pending = true;
+    void this.read().then((ratio) => {
+      this.current = ratio;
+    }).finally(() => {
+      this.pending = false;
+    });
+  }
+  value() {
+    return this.current;
+  }
+};
+function hostMemoryFreeRatio(platform3, totalBytes, freeBytes, macPressureFreeRatio) {
+  if (platform3 === "darwin") return macPressureFreeRatio;
+  return totalBytes > 0 ? freeBytes / totalBytes : 1;
+}
 function worst(levels) {
   return levels.reduce(
     (acc, level) => RANK[level] > RANK[acc] ? level : acc,
@@ -14764,7 +14818,7 @@ function cpuBusyRatio(previous, next) {
   return 1 - idle / total;
 }
 function classifyHostPressure(sample, previous, platform3) {
-  const memRaw = schmittLowIsWorse(
+  const mem = sample.memFreeRatio === void 0 ? "ok" : schmittLowIsWorse(
     sample.memFreeRatio,
     previous,
     HOST_PRESSURE_BARS.memFreeRatio.enterElevated,
@@ -14772,7 +14826,6 @@ function classifyHostPressure(sample, previous, platform3) {
     HOST_PRESSURE_BARS.memFreeRatio.enterCritical,
     HOST_PRESSURE_BARS.memFreeRatio.leaveCritical
   );
-  const mem = platform3 === "darwin" && memRaw === "critical" ? "elevated" : memRaw;
   const eventLoop = schmittHighIsWorse(
     sample.eventLoopP99Ms,
     previous,
@@ -14856,23 +14909,28 @@ var HostPressureMonitor = class {
 };
 function createHostPressureMonitor(wsHub, liveAgents) {
   const histogram = monitorEventLoopDelay({ resolution: 20 });
-  const windows = process.platform === "win32";
+  const platform3 = process.platform;
+  const windows = platform3 === "win32";
+  const macMemoryPressure = platform3 === "darwin" ? new MacMemoryPressureProbe(readMacMemoryPressureFreeRatio) : null;
   let prevCpuTimes = null;
+  macMemoryPressure?.refresh();
   const monitor = new HostPressureMonitor({
     wsHub,
     histogram,
     readSample: () => {
       const eventLoopP99Ms = histogram.percentile(99) / 1e6;
       histogram.reset();
-      const total = totalmem();
+      const total = platform3 === "darwin" ? 0 : totalmem();
+      const free = platform3 === "darwin" ? 0 : freemem();
       const cpuList = cpus();
       const sample = {
         liveAgents: liveAgents(),
-        memFreeRatio: total > 0 ? freemem() / total : 1,
+        memFreeRatio: hostMemoryFreeRatio(platform3, total, free, macMemoryPressure?.value()),
         eventLoopP99Ms,
         load1: loadavg()[0],
         ncpu: cpuList.length
       };
+      macMemoryPressure?.refresh();
       if (windows) {
         const cpuTimes = cpuList.map((cpu) => cpu.times);
         sample.cpuBusyRatio = cpuBusyRatio(prevCpuTimes, cpuTimes);