@duckmind/dm-windows-x64 0.60.4 → 0.60.8

package/extensions/dm-subagents/src/runs/background/control-channel.js

@@ -1 +1,190 @@
-import{randomUUID as I}from"node:crypto";import*as J from"node:fs";import*as H from"node:path";import*as w from"node:fs";import*as B from"node:path";var S=[10,25,50,100,200,500,1000,2000,4000],x=new Set(["EACCES","EBUSY","EPERM"]),k=typeof SharedArrayBuffer<"u"?new SharedArrayBuffer(4):void 0,_=k?new Int32Array(k):void 0;function v(K){if(K<=0)return;if(_)try{Atomics.wait(_,0,0,K);return}catch{}let Z=Date.now()+K;while(Date.now()<Z);}function R(K){let Z=K?.code;return typeof Z==="string"&&x.has(Z)}function g(K,Z,$,C,j){for(let X=0;;X++)try{K.renameSync(Z,$);return}catch(Q){let G=C[X];if(G===void 0||!R(Q))throw Q;j(G)}}function q(K={}){let Z=K.fs??w,$=K.now??Date.now,C=K.pid??process.pid,j=K.random??Math.random,Q=K.retryRenameErrors??process.platform==="win32"?K.retryDelaysMs??S:[],G=K.wait??v;return(Y,W)=>{Z.mkdirSync(B.dirname(Y),{recursive:!0});let L=B.join(B.dirname(Y),`.${B.basename(Y)}.${C}.${$()}.${j().toString(36).slice(2)}.tmp`);try{Z.writeFileSync(L,JSON.stringify(W,null,2),"utf-8"),g(Z,L,Y,Q,G)}finally{Z.rmSync(L,{force:!0})}}}var F=q();import*as O from"node:os";import*as V from"node:path";function N(K){return K.trim().replace(/[^A-Za-z0-9._-]+/g,"-").replace(/^-+|-+$/g,"")||"unknown"}function P(K){let Z=K?.env??process.env,$=K&&Object.hasOwn(K,"getuid")?K.getuid:process.getuid?.bind(process);if(typeof $==="function")return`uid-${$()}`;for(let Q of["USERNAME","USER","LOGNAME"]){let G=Z[Q];if(G)return`user-${N(G)}`}let C=K&&Object.hasOwn(K,"userInfo")?K.userInfo:O.userInfo;try{let Q=C?.().username;if(Q)return`user-${N(Q)}`}catch{}let j=Z.USERPROFILE??Z.HOME;if(j)return`home-${N(j)}`;let X=K&&Object.hasOwn(K,"homedir")?K.homedir:O.homedir;try{let Q=X?.();if(Q)return`home-${N(Q)}`}catch{}return"shared"}var U=V.join(O.tmpdir(),`dm-subagents-${P()}`),i=V.join(U,"async-subagent-results"),s=V.join(U,"async-subagent-runs"),a=V.join(U,"chain-runs"),t=V.join(U,"artifacts");var A=250;var D=process.platform==="win32"?"SIGBREAK":"SIGUSR2",h="steer-requests",f="steer-targets";function z(K){return H.join(K,"control")}function T(K){return H.join(z(K),"interrupt.json")}function M(K){return H.join(z(K),"timeout.json")}function b(K){return H.join(z(K),h)}function m(K,Z){return H.join(z(K),f,String(Z))}function y(K){return`${String(K.ts).padStart(13,"0")}-${Buffer.from(K.id).toString("base64url")}.json`}function E(K,Z){let $=H.join(K,y(Z));return F($,Z),$}function c(K,Z={},$={}){let C=T(K),j={...Z,ts:Z.ts??$.now?.()??Date.now(),type:"interrupt"};return F(C,j),C}function d(K,Z={},$={}){let C=M(K),j={...Z,ts:Z.ts??$.now?.()??Date.now(),type:"timeout"};return F(C,j),C}function jK(K,Z,$={}){let C=Z.message.trim();if(!C)throw Error("steer message must not be empty.");if(Z.targetIndex!==void 0&&(!Number.isInteger(Z.targetIndex)||Z.targetIndex<0))throw Error("steer targetIndex must be a non-negative integer.");let j={type:"steer",id:Z.id??$.randomId?.()??I(),ts:Z.ts??$.now?.()??Date.now(),message:C,...Z.targetIndex!==void 0?{targetIndex:Z.targetIndex}:{},...Z.source?{source:Z.source}:{}};return E(b(K),j)}function CK(K,Z,$){if(!Number.isInteger(Z)||Z<0)throw Error("steer child index must be a non-negative integer.");return E(m(K,Z),{...$,targetIndex:Z,type:"steer"})}function l(K){if(!K||typeof K!=="object"||Array.isArray(K))return;let Z=K;if(Z.type!=="steer")return;if(typeof Z.id!=="string"||!Z.id.trim())return;if(typeof Z.ts!=="number"||!Number.isFinite(Z.ts))return;if(typeof Z.message!=="string"||!Z.message.trim())return;if(Z.targetIndex!==void 0&&(!Number.isInteger(Z.targetIndex)||Z.targetIndex<0))return;return{type:"steer",id:Z.id.trim(),ts:Z.ts,message:Z.message.trim(),...Z.targetIndex!==void 0?{targetIndex:Z.targetIndex}:{},...typeof Z.source==="string"&&Z.source.trim()?{source:Z.source}:{}}}function u(K,Z=J){if(!Z.existsSync(K))return[];let $=[];for(let C of Z.readdirSync(K).filter((j)=>j.endsWith(".json")).sort()){let j=H.join(K,C),X;try{X=l(JSON.parse(Z.readFileSync(j,"utf-8")))}catch{X=void 0}try{Z.rmSync(j,{recursive:!0})}catch{continue}if(X)$.push(X)}return $.sort((C,j)=>C.ts-j.ts||C.id.localeCompare(j.id))}function o(K,Z=J){return u(b(K),Z)}function n(K,Z=J){let $=T(K);if(!Z.existsSync($))return!1;try{Z.rmSync($,{force:!0,recursive:!0})}catch{}return!0}function p(K,Z=J){let $=M(K);if(!Z.existsSync($))return!1;try{Z.rmSync($,{force:!0,recursive:!0})}catch{}return!0}function QK(K){let Z=c(K.asyncDir,K.source?{source:K.source}:{},{now:K.now});if(typeof K.pid==="number"&&K.pid>0)try{(K.kill??process.kill)(K.pid,K.signal??D)}catch($){if($?.code==="ENOSYS")return;try{J.rmSync(Z,{force:!0})}catch{}throw $}}function XK(K){d(K.asyncDir,K.source?{source:K.source}:{},{now:K.now})}function GK(K,Z){let $=Z.fs??J,C=Z.timers??{setInterval,clearInterval},j=z(K);try{$.mkdirSync(j,{recursive:!0})}catch{}let X=!1,Q=()=>{if(X)return;try{if(p(K,$))Z.onTimeout?.();if(n(K,$))Z.onInterrupt();for(let W of o(K,$))Z.onSteer?.(W)}catch{}};Q();let G;try{G=$.watch(j,()=>Q()),G.on?.("error",()=>{})}catch{G=void 0}let Y=C.setInterval(Q,Z.pollIntervalMs??A);return Y.unref?.(),()=>{if(X)return;X=!0;try{G?.close()}catch{}C.clearInterval(Y)}}export{E as writeSteerRequestToDir,GK as watchAsyncControlInbox,M as timeoutRequestPath,m as stepSteerInboxDir,b as steerRequestsDir,d as requestAsyncTimeout,jK as requestAsyncSteer,c as requestAsyncInterrupt,T as interruptRequestPath,CK as enqueueStepSteer,XK as deliverTimeoutRequest,QK as deliverInterruptRequest,z as controlInboxDir,p as consumeTimeoutRequest,u as consumeSteerRequestsFromDir,o as consumeSteerRequests,n as consumeInterruptRequest,D as INTERRUPT_SIGNAL};
+import { randomUUID } from "node:crypto";
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { writeAtomicJson } from "../../shared/atomic-json.js";
+import { POLL_INTERVAL_MS } from "../../shared/types.js";
+export const INTERRUPT_SIGNAL = process.platform === "win32" ? "SIGBREAK" : "SIGUSR2";
+const STEER_REQUESTS_DIR = "steer-requests";
+const STEER_TARGETS_DIR = "steer-targets";
+export function controlInboxDir(asyncDir) {
+  return path.join(asyncDir, "control");
+}
+export function interruptRequestPath(asyncDir) {
+  return path.join(controlInboxDir(asyncDir), "interrupt.json");
+}
+export function timeoutRequestPath(asyncDir) {
+  return path.join(controlInboxDir(asyncDir), "timeout.json");
+}
+export function steerRequestsDir(asyncDir) {
+  return path.join(controlInboxDir(asyncDir), STEER_REQUESTS_DIR);
+}
+export function stepSteerInboxDir(asyncDir, index) {
+  return path.join(controlInboxDir(asyncDir), STEER_TARGETS_DIR, String(index));
+}
+function steerRequestFileName(request) {
+  return `${String(request.ts).padStart(13, "0")}-${Buffer.from(request.id).toString("base64url")}.json`;
+}
+export function writeSteerRequestToDir(dir, request) {
+  const requestPath = path.join(dir, steerRequestFileName(request));
+  writeAtomicJson(requestPath, request);
+  return requestPath;
+}
+export function requestAsyncInterrupt(asyncDir, payload = {}, deps = {}) {
+  const requestPath = interruptRequestPath(asyncDir);
+  const request = { ...payload, ts: payload.ts ?? deps.now?.() ?? Date.now(), type: "interrupt" };
+  writeAtomicJson(requestPath, request);
+  return requestPath;
+}
+export function requestAsyncTimeout(asyncDir, payload = {}, deps = {}) {
+  const requestPath = timeoutRequestPath(asyncDir);
+  const request = { ...payload, ts: payload.ts ?? deps.now?.() ?? Date.now(), type: "timeout" };
+  writeAtomicJson(requestPath, request);
+  return requestPath;
+}
+export function requestAsyncSteer(asyncDir, payload, deps = {}) {
+  const message = payload.message.trim();
+  if (!message)
+    throw new Error("steer message must not be empty.");
+  if (payload.targetIndex !== undefined && (!Number.isInteger(payload.targetIndex) || payload.targetIndex < 0)) {
+    throw new Error("steer targetIndex must be a non-negative integer.");
+  }
+  const request = {
+    type: "steer",
+    id: payload.id ?? deps.randomId?.() ?? randomUUID(),
+    ts: payload.ts ?? deps.now?.() ?? Date.now(),
+    message,
+    ...payload.targetIndex !== undefined ? { targetIndex: payload.targetIndex } : {},
+    ...payload.source ? { source: payload.source } : {}
+  };
+  return writeSteerRequestToDir(steerRequestsDir(asyncDir), request);
+}
+export function enqueueStepSteer(asyncDir, index, request) {
+  if (!Number.isInteger(index) || index < 0)
+    throw new Error("steer child index must be a non-negative integer.");
+  return writeSteerRequestToDir(stepSteerInboxDir(asyncDir, index), { ...request, targetIndex: index, type: "steer" });
+}
+function parseSteerRequest(raw) {
+  if (!raw || typeof raw !== "object" || Array.isArray(raw))
+    return;
+  const input = raw;
+  if (input.type !== "steer")
+    return;
+  if (typeof input.id !== "string" || !input.id.trim())
+    return;
+  if (typeof input.ts !== "number" || !Number.isFinite(input.ts))
+    return;
+  if (typeof input.message !== "string" || !input.message.trim())
+    return;
+  if (input.targetIndex !== undefined && (!Number.isInteger(input.targetIndex) || input.targetIndex < 0))
+    return;
+  return {
+    type: "steer",
+    id: input.id.trim(),
+    ts: input.ts,
+    message: input.message.trim(),
+    ...input.targetIndex !== undefined ? { targetIndex: input.targetIndex } : {},
+    ...typeof input.source === "string" && input.source.trim() ? { source: input.source } : {}
+  };
+}
+export function consumeSteerRequestsFromDir(dir, fsImpl = fs) {
+  if (!fsImpl.existsSync(dir))
+    return [];
+  const requests = [];
+  for (const entry of fsImpl.readdirSync(dir).filter((name) => name.endsWith(".json")).sort()) {
+    const requestPath = path.join(dir, entry);
+    let parsed;
+    try {
+      parsed = parseSteerRequest(JSON.parse(fsImpl.readFileSync(requestPath, "utf-8")));
+    } catch {
+      parsed = undefined;
+    }
+    try {
+      fsImpl.rmSync(requestPath, { recursive: true });
+    } catch {
+      continue;
+    }
+    if (parsed)
+      requests.push(parsed);
+  }
+  return requests.sort((left, right) => left.ts - right.ts || left.id.localeCompare(right.id));
+}
+export function consumeSteerRequests(asyncDir, fsImpl = fs) {
+  return consumeSteerRequestsFromDir(steerRequestsDir(asyncDir), fsImpl);
+}
+export function consumeInterruptRequest(asyncDir, fsImpl = fs) {
+  const requestPath = interruptRequestPath(asyncDir);
+  if (!fsImpl.existsSync(requestPath))
+    return false;
+  try {
+    fsImpl.rmSync(requestPath, { force: true, recursive: true });
+  } catch {}
+  return true;
+}
+export function consumeTimeoutRequest(asyncDir, fsImpl = fs) {
+  const requestPath = timeoutRequestPath(asyncDir);
+  if (!fsImpl.existsSync(requestPath))
+    return false;
+  try {
+    fsImpl.rmSync(requestPath, { force: true, recursive: true });
+  } catch {}
+  return true;
+}
+export function deliverInterruptRequest(input) {
+  const requestPath = requestAsyncInterrupt(input.asyncDir, input.source ? { source: input.source } : {}, { now: input.now });
+  if (typeof input.pid === "number" && input.pid > 0) {
+    try {
+      (input.kill ?? process.kill)(input.pid, input.signal ?? INTERRUPT_SIGNAL);
+    } catch (error) {
+      if (error?.code === "ENOSYS") {
+        return;
+      }
+      try {
+        fs.rmSync(requestPath, { force: true });
+      } catch {}
+      throw error;
+    }
+  }
+}
+export function deliverTimeoutRequest(input) {
+  requestAsyncTimeout(input.asyncDir, input.source ? { source: input.source } : {}, { now: input.now });
+}
+export function watchAsyncControlInbox(asyncDir, opts) {
+  const fsImpl = opts.fs ?? fs;
+  const timers = opts.timers ?? { setInterval, clearInterval };
+  const dir = controlInboxDir(asyncDir);
+  try {
+    fsImpl.mkdirSync(dir, { recursive: true });
+  } catch {}
+  let disposed = false;
+  const check = () => {
+    if (disposed)
+      return;
+    try {
+      if (consumeTimeoutRequest(asyncDir, fsImpl))
+        opts.onTimeout?.();
+      if (consumeInterruptRequest(asyncDir, fsImpl))
+        opts.onInterrupt();
+      for (const request of consumeSteerRequests(asyncDir, fsImpl))
+        opts.onSteer?.(request);
+    } catch {}
+  };
+  check();
+  let watcher;
+  try {
+    watcher = fsImpl.watch(dir, () => check());
+    watcher.on?.("error", () => {});
+  } catch {
+    watcher = undefined;
+  }
+  const interval = timers.setInterval(check, opts.pollIntervalMs ?? POLL_INTERVAL_MS);
+  interval.unref?.();
+  return () => {
+    if (disposed)
+      return;
+    disposed = true;
+    try {
+      watcher?.close();
+    } catch {}
+    timers.clearInterval(interval);
+  };
+}