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

package/dist/src/isolation/discovery.js

@@ -0,0 +1,477 @@
+// @bun
+// packages/isolation-plugin/src/isolation/discovery.ts
+import { existsSync as existsSync3 } from "fs";
+import { resolve as resolve4 } from "path";
+import { resolveRuntimeWorkspaceLayout } from "@rig/core/layout";
+
+// 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";
+
+// packages/isolation-plugin/src/native-extract.ts
+import { existsSync, mkdirSync, readFileSync, renameSync, statSync, writeFileSync } from "fs";
+import { tmpdir } from "os";
+import { resolve } from "path";
+
+// packages/isolation-plugin/src/embedded-native-assets.ts
+var embeddedNatives = null;
+
+// packages/isolation-plugin/src/native-extract.ts
+var sharedNativeOutputDir = resolve(tmpdir(), "rig-native");
+var extractionCache = {};
+function hasEmbeddedNatives() {
+  return embeddedNatives != null;
+}
+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;
+}
+
+// packages/isolation-plugin/src/sidecar-arg.ts
+var RIG_NATIVE_RUNTIME_SIDECAR_ARG = "__rig_native_runtime_sidecar";
+
+// packages/isolation-plugin/src/runtime-native.ts
+var sharedNativeRuntimeOutputDir = resolve2(tmpdir2(), "rig-native");
+var sharedNativeRuntimeOutputPath = resolve2(sharedNativeRuntimeOutputDir, `runtime-native-${process.platform}-${process.arch}.${suffix}`);
+var colocatedNativeRuntimeFileName = `runtime-native.${suffix}`;
+var nativeRuntimeLibrary = await loadNativeRuntimeLibrary();
+function runtimeScanWorktreesNative(worktreesRoot) {
+  const response = runNativeRuntimeSidecar({
+    op: "scan-worktrees",
+    input: { worktreesRoot }
+  });
+  if (!response.ok) {
+    throw new Error(response.error);
+  }
+  return response.entries ?? [];
+}
+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;
+}
+function resolveNativeRuntimeSidecarSourcePath() {
+  const envRoots = [
+    process.cwd()?.trim(),
+    process.env.RIG_HOST_PROJECT_ROOT?.trim(),
+    process.env.PROJECT_RIG_ROOT?.trim()
+  ].filter(Boolean);
+  for (const root of envRoots) {
+    for (const relative of [
+      "packages/isolation-plugin/src/runtime-native-sidecar.ts",
+      "packages/isolation-plugin/dist/src/runtime-native-sidecar.js"
+    ]) {
+      const candidate = resolve2(root, relative);
+      if (existsSync2(candidate)) {
+        return candidate;
+      }
+    }
+  }
+  for (const localCandidate of [
+    resolve2(import.meta.dir, "runtime-native-sidecar.js"),
+    resolve2(import.meta.dir, "runtime-native-sidecar.ts")
+  ]) {
+    if (existsSync2(localCandidate))
+      return localCandidate;
+  }
+  return 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 resolveNativeRuntimeSidecarCommand(request) {
+  if (hasEmbeddedNatives()) {
+    return {
+      argv: [process.execPath, RIG_NATIVE_RUNTIME_SIDECAR_ARG, JSON.stringify(request)],
+      env: {}
+    };
+  }
+  const hostBinary = process.env.RIG_NATIVE_HOST_BINARY?.trim();
+  if (hostBinary) {
+    return {
+      argv: [hostBinary, RIG_NATIVE_RUNTIME_SIDECAR_ARG, JSON.stringify(request)],
+      env: {}
+    };
+  }
+  const sidecarSourcePath = resolveNativeRuntimeSidecarSourcePath();
+  if (!sidecarSourcePath) {
+    throw new Error("runtime-native-sidecar.ts source file not found.");
+  }
+  const bunCli = resolveNativeRuntimeSidecarInvocation();
+  return {
+    argv: [bunCli.command, sidecarSourcePath, JSON.stringify(request)],
+    env: bunCli.env
+  };
+}
+function runNativeRuntimeSidecar(request) {
+  const { argv, env } = resolveNativeRuntimeSidecarCommand(request);
+  const proc = Bun.spawnSync(argv, {
+    cwd: process.env.RIG_HOST_PROJECT_ROOT?.trim() || process.cwd(),
+    stdout: "pipe",
+    stderr: "pipe",
+    env: {
+      ...process.env,
+      ...env,
+      RIG_NATIVE_RUNTIME_SIDECAR: "1"
+    }
+  });
+  if (proc.exitCode !== 0) {
+    throw new Error(proc.stderr.toString() || proc.stdout.toString() || `runtime native sidecar exited ${proc.exitCode}`);
+  }
+  const stdout = proc.stdout.toString().trim();
+  if (!stdout) {
+    throw new Error("runtime native sidecar returned empty output.");
+  }
+  const responseLine = stdout.split(`
+`).map((line) => line.trim()).filter((line) => line.startsWith("{")).at(-1);
+  if (!responseLine) {
+    throw new Error(`runtime native sidecar returned no JSON response: ${stdout}`);
+  }
+  return JSON.parse(responseLine);
+}
+function resolveNativeRuntimeSidecarInvocation() {
+  const bunPath = Bun.which("bun");
+  if (bunPath) {
+    return { command: bunPath, env: {} };
+  }
+  if (process.execPath?.trim()) {
+    return {
+      command: process.execPath,
+      env: { BUN_BE_BUN: "1" }
+    };
+  }
+  throw new Error("bun is required to run the runtime native sidecar.");
+}
+
+// packages/isolation-plugin/src/isolation/discovery.ts
+import { loadRuntimeContext } from "@rig/core/runtime-context";
+
+// packages/isolation-plugin/src/isolation/shared.ts
+import { agentId, safeGitRefComponent, taskRuntimeId } from "@rig/core/safe-identifiers";
+import { resolveCheckoutRoot } from "@rig/core/checkout-root";
+var generatedCredentialFiles = new Set;
+function resolveMonorepoRoot(projectRoot) {
+  return resolveCheckoutRoot(projectRoot);
+}
+
+// packages/isolation-plugin/src/isolation/worktree.ts
+import { assertPathInsideRoot, safeGitRefComponent as safeGitRefComponent2, safePathSegment } from "@rig/core/safe-identifiers";
+function runtimeWorktreeNameFromRuntimeId(runtimeId) {
+  return safePathSegment(runtimeId.replace(/^task-/, ""), { fallback: "runtime", maxLength: 72 });
+}
+
+// packages/isolation-plugin/src/isolation/toolchain.ts
+import { assertPathInsideRoot as assertPathInsideRoot2, safePathSegment as safePathSegment2 } from "@rig/core/safe-identifiers";
+
+// packages/isolation-plugin/src/isolation/runtime-binary-build.ts
+import { resolveRigLayout } from "@rig/core/layout";
+import { runtimeProvisioningEnv } from "@rig/core/runtime-provisioning-env";
+var runtimeBinaryBuildQueue = Promise.resolve();
+
+// packages/isolation-plugin/src/isolation/git-native.ts
+import { tmpdir as tmpdir3 } from "os";
+import { dirname as dirname2, isAbsolute, resolve as resolve3 } from "path";
+var sharedGitNativeOutputDir = resolve3(tmpdir3(), "rig-native");
+var sharedGitNativeOutputPath = resolve3(sharedGitNativeOutputDir, `rig-git-${process.platform}-${process.arch}${process.platform === "win32" ? ".exe" : ""}`);
+
+// packages/isolation-plugin/src/isolation/toolchain.ts
+import {
+  GUARD_TOOLCHAIN_SOURCES,
+  LIFECYCLE_TOOLCHAIN_SOURCES,
+  TOOL_MATERIALIZER
+} from "@rig/contracts";
+import { defineCapability } from "@rig/core/capability";
+import { buildProjectPluginHost, requireInstalledCapability } from "@rig/core/capability-loaders";
+import { resolveBunBinaryPath } from "@rig/core/runtime-paths";
+var ToolMaterializerCap = defineCapability(TOOL_MATERIALIZER);
+var GuardToolchainSourcesCap = defineCapability(GUARD_TOOLCHAIN_SOURCES);
+var LifecycleToolchainSourcesCap = defineCapability(LIFECYCLE_TOOLCHAIN_SOURCES);
+var SNAPSHOT_SIDECAR_SOURCE = ["packages", "isolation-plugin", "src", "snapshot-sidecar.ts"].join("/");
+function runtimeWorktreeId(workspaceDir) {
+  return workspaceDir.split("/").at(-1) || "runtime";
+}
+
+// packages/isolation-plugin/src/isolation/discovery.ts
+async function readRuntimeMetadata(metadataPath) {
+  if (!existsSync3(metadataPath)) {
+    return null;
+  }
+  try {
+    return await Bun.file(metadataPath).json();
+  } catch {
+    return null;
+  }
+}
+async function listAgentRuntimes(projectRoot) {
+  const runtimes = [];
+  const monorepoRoot = resolveMonorepoRoot(projectRoot);
+  const worktreesRoot = resolve4(monorepoRoot, ".worktrees");
+  if (!existsSync3(worktreesRoot)) {
+    return [];
+  }
+  const workspaces = runtimeScanWorktreesNative(worktreesRoot);
+  for (let scan of workspaces) {
+    let workspacePath = scan.workspaceDir;
+    const workspaceLayout = resolveRuntimeWorkspaceLayout(workspacePath);
+    const rootDir = scan.runtimeDir || workspaceLayout.runtimeDir;
+    const metadataPath = scan.metadataPath || resolve4(rootDir, "runtime.json");
+    const contextFile = scan.contextPath || workspaceLayout.contextPath;
+    let mode = "worktree";
+    let createdAt = new Date().toISOString();
+    let taskId = "";
+    let runtimeId = "";
+    let binDir = workspaceLayout.binDir;
+    let logsDir = workspaceLayout.logsDir;
+    let stateDir = workspaceLayout.stateDir;
+    let sessionDir = workspaceLayout.sessionDir;
+    const metadata = await readRuntimeMetadata(metadataPath);
+    if (metadata) {
+      runtimeId = metadata.id;
+      mode = metadata.mode;
+      createdAt = metadata.createdAt;
+    }
+    if (existsSync3(contextFile)) {
+      try {
+        const ctx = loadRuntimeContext(contextFile);
+        taskId = ctx.taskId;
+        workspacePath = ctx.workspaceDir;
+        binDir = ctx.binDir;
+        logsDir = ctx.logsDir;
+        stateDir = ctx.stateDir;
+        sessionDir = ctx.sessionDir;
+      } catch {}
+    }
+    if (!runtimeId) {
+      runtimeId = taskRuntimeId(taskId || runtimeWorktreeId(workspacePath));
+    }
+    runtimes.push({
+      id: runtimeId,
+      taskId,
+      mode,
+      rootDir,
+      workspaceDir: workspacePath,
+      homeDir: resolve4(rootDir, "home"),
+      tmpDir: resolve4(rootDir, "tmp"),
+      cacheDir: resolve4(rootDir, "cache"),
+      logsDir,
+      stateDir,
+      sessionDir,
+      claudeHomeDir: resolve4(rootDir, "home", ".claude"),
+      contextFile,
+      binDir,
+      createdAt
+    });
+  }
+  return runtimes;
+}
+function runtimeRootForCleanup(projectRoot, runtimeId) {
+  let monorepoRoot = null;
+  try {
+    monorepoRoot = resolveMonorepoRoot(projectRoot);
+  } catch {}
+  const workspaceRootCandidate = monorepoRoot ?? projectRoot;
+  const initialWorkspaceDir = resolve4(workspaceRootCandidate, ".worktrees", runtimeWorktreeNameFromRuntimeId(runtimeId));
+  const runtimeRoot = resolveRuntimeWorkspaceLayout(initialWorkspaceDir).runtimeDir;
+  return { monorepoRoot, initialWorkspaceDir, runtimeRoot };
+}
+export {
+  runtimeRootForCleanup,
+  listAgentRuntimes
+};

package/dist/src/isolation/git-native.d.ts

@@ -0,0 +1,28 @@
+export type PendingFile = {
+    path: string;
+    status: string;
+};
+export type TreeCommitUpdate = {
+    path: string;
+    content: string;
+    sourceFilePath?: never;
+} | {
+    path: string;
+    sourceFilePath: string;
+    content?: never;
+};
+export declare function runtimeRigGitFileName(): string;
+export declare function ensureRigGitBinaryPath(outputPath?: string): Promise<string>;
+export declare function ensureRigGitBinaryPathSync(outputPath?: string): string;
+export declare function materializeRigGitBinary(targetDir: string): Promise<string>;
+export declare function nativeBranchName(repoPath: string): string | null;
+export declare function nativeHeadOid(repoPath: string): string | null;
+export declare function nativeChangeCount(repoPath: string): number | null;
+export declare function nativePendingFiles(repoPath: string): PendingFile[] | null;
+export declare function nativeFileHasChanges(repoPath: string, filePath: string): boolean | null;
+export declare function nativeFetchRef(repoPath: string, remote: string, branch: string): string;
+export declare function nativeReadBlobAtRef(repoPath: string, ref: string, path: string): string;
+export declare function nativeReadBlobBytesAtRef(repoPath: string, ref: string, path: string): Uint8Array;
+export declare function buildTreeCommitUpdatesJson(updates: TreeCommitUpdate[]): string;
+export declare function nativeWriteTreeCommit(repoPath: string, baseRef: string, updates: TreeCommitUpdate[], message: string): string;
+export declare function nativePushRefWithLease(repoPath: string, localOid: string, remoteRef: string, expectedOldOid: string, remote?: string): string;