@h-rig/runtime 0.0.6-alpha.1 → 0.0.6-alpha.2

package/dist/bin/rig-agent-dispatch.js

@@ -9193,6 +9193,18 @@ async function runAgentWrapper(options = {}) {
     console.error(`[rig-agent] Failed to provision runtime: ${error}`);
     return 1;
   }
+  try {
+    await waitForDirtyBaselineReady(runtime, taskId);
+  } catch (error) {
+    emitWrapperEvent("runtime.baseline.failed", {
+      runtimeId: runtime.id,
+      taskId,
+      workspaceDir: runtime.workspaceDir,
+      error: error instanceof Error ? error.message : String(error)
+    });
+    console.error(`[rig-agent] Failed to apply selected baseline before provider launch: ${error}`);
+    return 1;
+  }
   const providerArgs = buildProviderArgs(provider, runtime, argv);
   const providerCommand = [providerBinary(provider), ...providerArgs];
   emitWrapperEvent("provider.launch", {
@@ -9413,6 +9425,34 @@ function providerBinary(provider) {
 function emitWrapperEvent(type, payload) {
   console.log(`__RIG_WRAPPER_EVENT__${JSON.stringify({ type, payload, at: new Date().toISOString() })}`);
 }
+function sleep(ms) {
+  return new Promise((resolveSleep) => setTimeout(resolveSleep, ms));
+}
+async function waitForDirtyBaselineReady(runtime, taskId) {
+  const readyFile = process.env.RIG_DIRTY_BASELINE_READY_FILE?.trim();
+  if (process.env.RIG_BASELINE_MODE !== "dirty-snapshot" || !readyFile)
+    return;
+  const timeoutMs = Number.parseInt(process.env.RIG_DIRTY_BASELINE_TIMEOUT_MS ?? "30000", 10);
+  const deadline = Date.now() + (Number.isFinite(timeoutMs) && timeoutMs > 0 ? timeoutMs : 30000);
+  emitWrapperEvent("runtime.baseline.waiting", {
+    runtimeId: runtime.id,
+    taskId,
+    workspaceDir: runtime.workspaceDir,
+    readyFile
+  });
+  while (!existsSync33(readyFile)) {
+    if (Date.now() >= deadline) {
+      throw new Error(`Timed out waiting for dirty baseline ready file: ${readyFile}`);
+    }
+    await sleep(50);
+  }
+  emitWrapperEvent("runtime.baseline.completed", {
+    runtimeId: runtime.id,
+    taskId,
+    workspaceDir: runtime.workspaceDir,
+    readyFile
+  });
+}
 async function resolveTaskId(projectRoot, monorepoRoot) {
   if (process.env.RIG_TASK_ID) {
     return process.env.RIG_TASK_ID;

package/dist/src/control-plane/agent-wrapper.js

@@ -9193,6 +9193,18 @@ async function runAgentWrapper(options = {}) {
     console.error(`[rig-agent] Failed to provision runtime: ${error}`);
     return 1;
   }
+  try {
+    await waitForDirtyBaselineReady(runtime, taskId);
+  } catch (error) {
+    emitWrapperEvent("runtime.baseline.failed", {
+      runtimeId: runtime.id,
+      taskId,
+      workspaceDir: runtime.workspaceDir,
+      error: error instanceof Error ? error.message : String(error)
+    });
+    console.error(`[rig-agent] Failed to apply selected baseline before provider launch: ${error}`);
+    return 1;
+  }
   const providerArgs = buildProviderArgs(provider, runtime, argv);
   const providerCommand = [providerBinary(provider), ...providerArgs];
   emitWrapperEvent("provider.launch", {
@@ -9413,6 +9425,34 @@ function providerBinary(provider) {
 function emitWrapperEvent(type, payload) {
   console.log(`__RIG_WRAPPER_EVENT__${JSON.stringify({ type, payload, at: new Date().toISOString() })}`);
 }
+function sleep(ms) {
+  return new Promise((resolveSleep) => setTimeout(resolveSleep, ms));
+}
+async function waitForDirtyBaselineReady(runtime, taskId) {
+  const readyFile = process.env.RIG_DIRTY_BASELINE_READY_FILE?.trim();
+  if (process.env.RIG_BASELINE_MODE !== "dirty-snapshot" || !readyFile)
+    return;
+  const timeoutMs = Number.parseInt(process.env.RIG_DIRTY_BASELINE_TIMEOUT_MS ?? "30000", 10);
+  const deadline = Date.now() + (Number.isFinite(timeoutMs) && timeoutMs > 0 ? timeoutMs : 30000);
+  emitWrapperEvent("runtime.baseline.waiting", {
+    runtimeId: runtime.id,
+    taskId,
+    workspaceDir: runtime.workspaceDir,
+    readyFile
+  });
+  while (!existsSync33(readyFile)) {
+    if (Date.now() >= deadline) {
+      throw new Error(`Timed out waiting for dirty baseline ready file: ${readyFile}`);
+    }
+    await sleep(50);
+  }
+  emitWrapperEvent("runtime.baseline.completed", {
+    runtimeId: runtime.id,
+    taskId,
+    workspaceDir: runtime.workspaceDir,
+    readyFile
+  });
+}
 async function resolveTaskId(projectRoot, monorepoRoot) {
   if (process.env.RIG_TASK_ID) {
     return process.env.RIG_TASK_ID;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@h-rig/runtime",
-  "version": "0.0.6-alpha.1",
+  "version": "0.0.6-alpha.2",
   "type": "module",
   "description": "Rig package",
   "license": "UNLICENSED",
@@ -64,11 +64,11 @@
   "module": "./dist/src/index.js",
   "dependencies": {
     "@libsql/client": "^0.17.2",
-    "@rig/contracts": "npm:@h-rig/contracts@0.0.6-alpha.1",
-    "@rig/core": "npm:@h-rig/core@0.0.6-alpha.1",
-    "@rig/hook-kit": "npm:@h-rig/hook-kit@0.0.6-alpha.1",
-    "@rig/shared": "npm:@h-rig/shared@0.0.6-alpha.1",
-    "@rig/validator-kit": "npm:@h-rig/validator-kit@0.0.6-alpha.1",
+    "@rig/contracts": "npm:@h-rig/contracts@0.0.6-alpha.2",
+    "@rig/core": "npm:@h-rig/core@0.0.6-alpha.2",
+    "@rig/hook-kit": "npm:@h-rig/hook-kit@0.0.6-alpha.2",
+    "@rig/shared": "npm:@h-rig/shared@0.0.6-alpha.2",
+    "@rig/validator-kit": "npm:@h-rig/validator-kit@0.0.6-alpha.2",
     "effect": "4.0.0-beta.78",
     "smol-toml": "^1.6.0"
   }