@h-rig/isolation-plugin 0.0.6-alpha.156 → 0.0.6-alpha.158

package/dist/src/sidecar-arg.d.ts

@@ -0,0 +1,7 @@
+/**
+ * argv[2] sentinel that routes a self-exec of the compiled `rig` binary to the
+ * isolation-owned runtime-native sidecar handler. Kept side-effect-free so seed
+ * entrypoints can match it without importing `runtime-native.ts` and dlopening the
+ * Zig library on every invocation.
+ */
+export declare const RIG_NATIVE_RUNTIME_SIDECAR_ARG = "__rig_native_runtime_sidecar";

package/dist/src/sidecar-arg.js

@@ -0,0 +1,6 @@
+// @bun
+// packages/isolation-plugin/src/sidecar-arg.ts
+var RIG_NATIVE_RUNTIME_SIDECAR_ARG = "__rig_native_runtime_sidecar";
+export {
+  RIG_NATIVE_RUNTIME_SIDECAR_ARG
+};

package/dist/src/sidecar-entrypoint.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Shared compiled-binary sentinel router. Every self-exec role entrypoint must
+ * check this before parsing its own args because the runtime native sidecar
+ * re-execs process.execPath, which can be `rig`, `rig-run`, or a hardlinked role.
+ *
+ * Lives in @rig/isolation-plugin (the fs-snapshot/provisioning domain) because it
+ * routes to the snapshot runtime-native sidecar handler and owns the sentinel.
+ */
+export declare function runRuntimeNativeSidecarEntrypoint(arg?: string | undefined, rawRequest?: string | undefined): Promise<boolean>;

package/dist/src/sidecar-entrypoint.js

@@ -0,0 +1,401 @@
+// @bun
+var __defProp = Object.defineProperty;
+var __returnValue = (v) => v;
+function __exportSetter(name, newValue) {
+  this[name] = __returnValue.bind(null, newValue);
+}
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, {
+      get: all[name],
+      enumerable: true,
+      configurable: true,
+      set: __exportSetter.bind(all, name)
+    });
+};
+var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
+
+// packages/isolation-plugin/src/sidecar-arg.ts
+var RIG_NATIVE_RUNTIME_SIDECAR_ARG = "__rig_native_runtime_sidecar";
+
+// packages/isolation-plugin/src/embedded-native-assets.ts
+var embeddedNatives = null;
+
+// packages/isolation-plugin/src/native-extract.ts
+import { existsSync, mkdirSync, readFileSync, renameSync, statSync, writeFileSync } from "fs";
+import { tmpdir } from "os";
+import { resolve } from "path";
+function extractEmbeddedNative(name) {
+  if (name in extractionCache) {
+    return extractionCache[name] ?? null;
+  }
+  const entry = embeddedNatives?.[name];
+  if (!entry) {
+    extractionCache[name] = null;
+    return null;
+  }
+  try {
+    const targetPath = resolve(sharedNativeOutputDir, entry.fileName);
+    mkdirSync(sharedNativeOutputDir, { recursive: true });
+    const upToDate = existsSync(targetPath) && statSync(targetPath).size === entry.size;
+    if (!upToDate) {
+      const bytes = readFileSync(entry.filePath);
+      const tempPath = `${targetPath}.${process.pid}.${Date.now()}.tmp`;
+      writeFileSync(tempPath, bytes, { mode: 493 });
+      renameSync(tempPath, targetPath);
+    }
+    extractionCache[name] = targetPath;
+  } catch {
+    extractionCache[name] = null;
+  }
+  return extractionCache[name] ?? null;
+}
+var sharedNativeOutputDir, extractionCache;
+var init_native_extract = __esm(() => {
+  sharedNativeOutputDir = resolve(tmpdir(), "rig-native");
+  extractionCache = {};
+});
+
+// packages/isolation-plugin/src/runtime-native.ts
+import { dlopen, ptr, suffix, toBuffer } from "bun:ffi";
+import { copyFileSync, existsSync as existsSync2, mkdirSync as mkdirSync2, renameSync as renameSync2, rmSync, statSync as statSync2 } from "fs";
+import { tmpdir as tmpdir2 } from "os";
+import { dirname, resolve as resolve2 } from "path";
+function requireNativeRuntimeLibrary(feature) {
+  if (!nativeRuntimeLibrary) {
+    throw new Error(`Native Zig runtime is required for ${feature}`);
+  }
+  return nativeRuntimeLibrary;
+}
+function runtimePrepareTrackedPathsInProcess(input) {
+  const runtimeLibrary = requireNativeRuntimeLibrary("runtime dependency layer linking");
+  const logsDir = Buffer.from(input.logsDir, "utf8");
+  const stateDir = Buffer.from(input.stateDir, "utf8");
+  const sessionDir = Buffer.from(input.sessionDir, "utf8");
+  const eventsFile = Buffer.from(input.eventsFile, "utf8");
+  const controlledBashLogFile = Buffer.from(input.controlledBashLogFile, "utf8");
+  const resultPtr = runtimeLibrary.symbols.runtime_prepare_paths(Number(ptr(logsDir)), logsDir.byteLength, Number(ptr(stateDir)), stateDir.byteLength, Number(ptr(sessionDir)), sessionDir.byteLength, Number(ptr(eventsFile)), eventsFile.byteLength, Number(ptr(controlledBashLogFile)), controlledBashLogFile.byteLength);
+  if (!resultPtr) {
+    throw new Error("runtime_prepare_paths returned null");
+  }
+  try {
+    const view = viewAt(resultPtr, OPERATION_RESULT_SIZE);
+    throwIfOperationError(view, "runtime_prepare_paths");
+  } finally {
+    runtimeLibrary.symbols.snapshot_release(resultPtr);
+  }
+}
+function runtimeLinkDependencyLayerInProcess(sourceDir, targetDir) {
+  const runtimeLibrary = requireNativeRuntimeLibrary("runtime worktree discovery");
+  const sourceBuffer = Buffer.from(sourceDir, "utf8");
+  const targetBuffer = Buffer.from(targetDir, "utf8");
+  const resultPtr = runtimeLibrary.symbols.runtime_link_dependency_layer(Number(ptr(sourceBuffer)), sourceBuffer.byteLength, Number(ptr(targetBuffer)), targetBuffer.byteLength);
+  if (!resultPtr) {
+    throw new Error("runtime_link_dependency_layer returned null");
+  }
+  try {
+    const view = viewAt(resultPtr, OPERATION_RESULT_SIZE);
+    throwIfOperationError(view, "runtime_link_dependency_layer");
+  } finally {
+    runtimeLibrary.symbols.snapshot_release(resultPtr);
+  }
+}
+function runtimeScanWorktreesInProcess(worktreesRoot) {
+  const runtimeLibrary = requireNativeRuntimeLibrary("runtime worktree discovery");
+  const rootBuffer = Buffer.from(worktreesRoot, "utf8");
+  const resultPtr = runtimeLibrary.symbols.runtime_scan_worktrees(Number(ptr(rootBuffer)), rootBuffer.byteLength);
+  if (!resultPtr) {
+    throw new Error(`runtime_scan_worktrees returned null for ${worktreesRoot}`);
+  }
+  try {
+    const view = viewAt(resultPtr, RUNTIME_SCAN_RESULT_SIZE);
+    throwIfScanError(view, "runtime_scan_worktrees");
+    const entriesPtr = readU64(view, 8);
+    const entriesLen = readU64(view, 16);
+    const entries = [];
+    for (let index = 0;index < entriesLen; index += 1) {
+      const entryView = viewAt(entriesPtr + index * RUNTIME_SCAN_ENTRY_SIZE, RUNTIME_SCAN_ENTRY_SIZE);
+      entries.push({
+        workspaceDir: readString(readU64(entryView, 0), readU64(entryView, 8)),
+        runtimeDir: readString(readU64(entryView, 16), readU64(entryView, 24)),
+        contextPath: readString(readU64(entryView, 32), readU64(entryView, 40)),
+        metadataPath: readString(readU64(entryView, 48), readU64(entryView, 56))
+      });
+    }
+    return entries;
+  } finally {
+    runtimeLibrary.symbols.snapshot_release(resultPtr);
+  }
+}
+async function ensureNativeRuntimeLibraryPath(outputPath = sharedNativeRuntimeOutputPath, options = {}) {
+  const explicitLib = process.env.RIG_NATIVE_RUNTIME_LIB?.trim();
+  if (explicitLib && existsSync2(explicitLib)) {
+    return explicitLib;
+  }
+  const embeddedPath = extractEmbeddedNative("snapshot");
+  if (embeddedPath) {
+    return embeddedPath;
+  }
+  if (await buildNativeRuntimeLibrary(outputPath, options)) {
+    return outputPath;
+  }
+  return !options.force && existsSync2(outputPath) ? outputPath : null;
+}
+async function loadNativeRuntimeLibrary() {
+  if (process.env.RIG_DISABLE_ZIG_NATIVE === "1") {
+    return null;
+  }
+  const explicitLib = process.env.RIG_NATIVE_RUNTIME_LIB?.trim();
+  if (explicitLib && existsSync2(explicitLib)) {
+    const loaded = tryDlopenNativeRuntimeLibrary(explicitLib);
+    if (loaded) {
+      return loaded;
+    }
+  }
+  const embeddedPath = extractEmbeddedNative("snapshot");
+  if (embeddedPath) {
+    const loaded = tryDlopenNativeRuntimeLibrary(embeddedPath);
+    if (loaded) {
+      return loaded;
+    }
+  }
+  for (const candidate of nativeRuntimeLibraryCandidates()) {
+    if (!candidate || !existsSync2(candidate)) {
+      continue;
+    }
+    const loaded = tryDlopenNativeRuntimeLibrary(candidate);
+    if (loaded) {
+      return loaded;
+    }
+  }
+  const builtLibraryPath = await ensureNativeRuntimeLibraryPath(sharedNativeRuntimeOutputPath, { force: true });
+  if (!builtLibraryPath) {
+    return null;
+  }
+  return tryDlopenNativeRuntimeLibrary(builtLibraryPath);
+}
+function nativePackageLibraryCandidates(fromDir, names) {
+  const candidates = [];
+  let cursor = resolve2(fromDir);
+  for (let index = 0;index < 8; index += 1) {
+    for (const name of names) {
+      candidates.push(resolve2(cursor, "native", `${process.platform}-${process.arch}`, name), resolve2(cursor, "native", `${process.platform}-${process.arch}`, "lib", name), resolve2(cursor, "native", name), resolve2(cursor, "native", "lib", name));
+    }
+    const parent = dirname(cursor);
+    if (parent === cursor)
+      break;
+    cursor = parent;
+  }
+  return candidates;
+}
+function nativeRuntimeLibraryCandidates() {
+  const explicit = process.env.RIG_NATIVE_RUNTIME_LIB?.trim() || "";
+  const execDir = process.execPath?.trim() ? dirname(process.execPath.trim()) : "";
+  const platformSpecific = `runtime-native-${process.platform}-${process.arch}.${suffix}`;
+  return [...new Set([
+    explicit,
+    ...nativePackageLibraryCandidates(import.meta.dir, [colocatedNativeRuntimeFileName, platformSpecific]),
+    execDir ? resolve2(execDir, colocatedNativeRuntimeFileName) : "",
+    execDir ? resolve2(execDir, platformSpecific) : "",
+    execDir ? resolve2(execDir, "..", colocatedNativeRuntimeFileName) : "",
+    execDir ? resolve2(execDir, "..", platformSpecific) : "",
+    execDir ? resolve2(execDir, "lib", colocatedNativeRuntimeFileName) : "",
+    execDir ? resolve2(execDir, "..", "lib", colocatedNativeRuntimeFileName) : "",
+    sharedNativeRuntimeOutputPath
+  ].filter(Boolean))];
+}
+function resolveNativeRuntimeSourcePath() {
+  const explicit = process.env.RIG_NATIVE_RUNTIME_SOURCE?.trim();
+  if (explicit && existsSync2(explicit)) {
+    return explicit;
+  }
+  const bundled = resolve2(import.meta.dir, "../native/snapshot.zig");
+  return existsSync2(bundled) ? bundled : null;
+}
+async function buildNativeRuntimeLibrary(outputPath, options = {}) {
+  if (process.env.RIG_DISABLE_ZIG_NATIVE === "1") {
+    return false;
+  }
+  const zigBinary = Bun.which("zig");
+  const sourcePath = resolveNativeRuntimeSourcePath();
+  if (!zigBinary || !sourcePath) {
+    return false;
+  }
+  const tempOutputPath = `${outputPath}.${process.pid}.${Date.now()}.${Math.random().toString(36).slice(2)}.tmp`;
+  try {
+    mkdirSync2(dirname(outputPath), { recursive: true });
+    const needsBuild = options.force === true || !existsSync2(outputPath) || statSync2(sourcePath).mtimeMs > statSync2(outputPath).mtimeMs;
+    if (!needsBuild) {
+      return true;
+    }
+    const build = Bun.spawn([
+      zigBinary,
+      "build-lib",
+      sourcePath,
+      "-dynamic",
+      "-O",
+      "ReleaseFast",
+      `-femit-bin=${tempOutputPath}`
+    ], {
+      cwd: import.meta.dir,
+      stdout: "pipe",
+      stderr: "pipe"
+    });
+    const exitCode = await build.exited;
+    if (exitCode !== 0 || !existsSync2(tempOutputPath)) {
+      rmSync(tempOutputPath, { force: true });
+      return false;
+    }
+    renameSync2(tempOutputPath, outputPath);
+    return true;
+  } catch {
+    rmSync(tempOutputPath, { force: true });
+    return false;
+  }
+}
+function tryDlopenNativeRuntimeLibrary(outputPath) {
+  try {
+    return dlopen(outputPath, {
+      rig_scope_match: {
+        args: ["ptr", "ptr"],
+        returns: "u8"
+      },
+      snapshot_capture: {
+        args: ["ptr", "u64", "ptr", "u64"],
+        returns: "ptr"
+      },
+      snapshot_delta: {
+        args: ["ptr", "ptr"],
+        returns: "ptr"
+      },
+      snapshot_store_delta: {
+        args: ["ptr", "ptr", "ptr", "u64", "ptr", "u64", "ptr", "u64", "ptr", "u64"],
+        returns: "ptr"
+      },
+      snapshot_inspect_delta: {
+        args: ["ptr", "u64"],
+        returns: "ptr"
+      },
+      snapshot_apply_delta: {
+        args: ["ptr", "u64", "ptr", "u64"],
+        returns: "ptr"
+      },
+      snapshot_release: {
+        args: ["ptr"],
+        returns: "void"
+      },
+      runtime_hash_file: {
+        args: ["ptr", "u64"],
+        returns: "ptr"
+      },
+      runtime_hash_tree: {
+        args: ["ptr", "u64"],
+        returns: "ptr"
+      },
+      runtime_prepare_paths: {
+        args: ["ptr", "u64", "ptr", "u64", "ptr", "u64", "ptr", "u64", "ptr", "u64"],
+        returns: "ptr"
+      },
+      runtime_link_dependency_layer: {
+        args: ["ptr", "u64", "ptr", "u64"],
+        returns: "ptr"
+      },
+      runtime_scan_worktrees: {
+        args: ["ptr", "u64"],
+        returns: "ptr"
+      }
+    });
+  } catch {
+    return null;
+  }
+}
+function viewAt(address, size) {
+  return Buffer.from(toBuffer(address, 0, size));
+}
+function readU64(view, offset) {
+  return Number(view.readBigUInt64LE(offset));
+}
+function readString(address, length) {
+  if (!address || !length) {
+    return "";
+  }
+  return Buffer.from(toBuffer(address, 0, length)).toString("utf8");
+}
+function throwIfOperationError(view, opName) {
+  const errorPtr = readU64(view, 8);
+  const errorLen = readU64(view, 16);
+  if (!errorPtr || errorLen === 0) {
+    return;
+  }
+  throw new Error(`${opName} failed: ${readString(errorPtr, errorLen)}`);
+}
+function throwIfScanError(view, opName) {
+  const errorPtr = readU64(view, 24);
+  const errorLen = readU64(view, 32);
+  if (!errorPtr || errorLen === 0) {
+    return;
+  }
+  throw new Error(`${opName} failed: ${readString(errorPtr, errorLen)}`);
+}
+var sharedNativeRuntimeOutputDir, sharedNativeRuntimeOutputPath, colocatedNativeRuntimeFileName, OPERATION_RESULT_SIZE = 24, RUNTIME_SCAN_ENTRY_SIZE = 64, RUNTIME_SCAN_RESULT_SIZE = 40, nativeRuntimeLibrary;
+var init_runtime_native = __esm(async () => {
+  init_native_extract();
+  sharedNativeRuntimeOutputDir = resolve2(tmpdir2(), "rig-native");
+  sharedNativeRuntimeOutputPath = resolve2(sharedNativeRuntimeOutputDir, `runtime-native-${process.platform}-${process.arch}.${suffix}`);
+  colocatedNativeRuntimeFileName = `runtime-native.${suffix}`;
+  nativeRuntimeLibrary = await loadNativeRuntimeLibrary();
+});
+
+// packages/isolation-plugin/src/runtime-native-sidecar.ts
+var exports_runtime_native_sidecar = {};
+__export(exports_runtime_native_sidecar, {
+  runRuntimeNativeSidecarCli: () => runRuntimeNativeSidecarCli
+});
+function runRuntimeNativeSidecarCli(rawRequest) {
+  if (!rawRequest) {
+    console.error("usage: runtime-native-sidecar <request-json>");
+    process.exit(2);
+  }
+  try {
+    const request = JSON.parse(rawRequest);
+    const response = handleRequest(request);
+    process.stdout.write(`${JSON.stringify(response)}
+`);
+    process.exit(0);
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    process.stdout.write(`${JSON.stringify({ ok: false, error: message })}
+`);
+    process.exit(1);
+  }
+}
+function handleRequest(request) {
+  switch (request.op) {
+    case "prepare-paths":
+      runtimePrepareTrackedPathsInProcess(request.input);
+      return { ok: true };
+    case "link-dependency-layer":
+      runtimeLinkDependencyLayerInProcess(request.input.sourceDir, request.input.targetDir);
+      return { ok: true };
+    case "scan-worktrees":
+      return { ok: true, entries: runtimeScanWorktreesInProcess(request.input.worktreesRoot) };
+    default:
+      return { ok: false, error: `Unsupported runtime native sidecar op: ${request.op ?? "<missing>"}` };
+  }
+}
+var init_runtime_native_sidecar = __esm(async () => {
+  await init_runtime_native();
+  if (false) {}
+});
+
+// packages/isolation-plugin/src/sidecar-entrypoint.ts
+async function runRuntimeNativeSidecarEntrypoint(arg = process.argv[2], rawRequest = process.argv[3]) {
+  if (arg !== RIG_NATIVE_RUNTIME_SIDECAR_ARG)
+    return false;
+  const { runRuntimeNativeSidecarCli: runRuntimeNativeSidecarCli2 } = await init_runtime_native_sidecar().then(() => exports_runtime_native_sidecar);
+  runRuntimeNativeSidecarCli2(rawRequest);
+  return true;
+}
+export {
+  runRuntimeNativeSidecarEntrypoint
+};

package/dist/src/snapshot-sidecar.d.ts

@@ -0,0 +1,2 @@
+export declare function resolveRuntimeSnapshotArtifactDir(workspaceDir: string, taskId: string): string;
+export declare function runRuntimeSnapshotSidecar(argv?: string[]): Promise<void>;