agent-relay-orchestrator 0.60.0 → 0.60.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-relay-orchestrator",
-  "version": "0.60.0",
+  "version": "0.60.1",
   "description": "Agent Relay orchestrator — manages agent lifecycle across hosts",
   "type": "module",
   "bin": {

package/src/control.ts

@@ -4,7 +4,7 @@ import type { ManagedAgentReport, RelayClient, RelayCommand } from "./relay";
 import { handleSelfUpgrade } from "./self-upgrade";
 import { spawnAgent, stopSession, type SpawnOptions } from "./spawn";
 import { cleanupWorkspace, mergeWorkspace, pruneWorktrees, reconcileWorkspace, refreshWorkspaceDeps, workspacesRoot } from "./workspace-probe";
-import { armWorkspacePrAutoMerge, mergeWorkspacePr } from "./workspace-pr";
+import { armWorkspacePrAutoMerge, mergeWorkspacePr, refreshWorkspacePrBranch } from "./workspace-pr";
 
 interface ControlHandler {
   handleCommand(command: RelayCommand): Promise<boolean>;
@@ -152,6 +152,17 @@ export function createControlHandler(
           prUrl: typeof command.params.prUrl === "string" ? command.params.prUrl : undefined,
         });
         await relay.updateCommand(command.id, result.relayMerged ? "succeeded" : "failed", result as unknown as Record<string, unknown>, result.error);
+      } else if (command.type === "workspace.pr-refresh") {
+        const rawPrNumber = command.params.prNumber;
+        const result = refreshWorkspacePrBranch({
+          id: typeof command.params.workspaceId === "string" ? command.params.workspaceId : undefined,
+          repoRoot: typeof command.params.repoRoot === "string" ? command.params.repoRoot : undefined,
+          worktreePath: typeof command.params.worktreePath === "string" ? command.params.worktreePath : undefined,
+          branch: typeof command.params.branch === "string" ? command.params.branch : undefined,
+          prNumber: typeof rawPrNumber === "number" && Number.isSafeInteger(rawPrNumber) ? rawPrNumber : undefined,
+          prUrl: typeof command.params.prUrl === "string" ? command.params.prUrl : undefined,
+        });
+        await relay.updateCommand(command.id, result.prRefreshed ? "succeeded" : "failed", result as unknown as Record<string, unknown>, result.error);
       } else if (command.type === "workspace.deps-refresh") {
         const result = refreshWorkspaceDeps(
           typeof command.params.repoRoot === "string" ? command.params.repoRoot : "",

package/src/workspace-pr.ts

@@ -127,3 +127,35 @@ export function mergeWorkspacePr(input: {
   }
   return { workspaceId: input.id, status: "merge_planned", relayMerged: true, prNumber: input.prNumber, prUrl: input.prUrl };
 }
+
+export function refreshWorkspacePrBranch(input: {
+  id?: string;
+  repoRoot?: string;
+  worktreePath?: string;
+  branch?: string;
+  prNumber?: number;
+  prUrl?: string;
+}): { workspaceId?: string; status: "merge_planned"; prRefreshed: boolean; prNumber?: number; prUrl?: string; error?: string } {
+  const cwd = input.worktreePath ? resolve(input.worktreePath) : input.repoRoot ? resolve(input.repoRoot) : undefined;
+  if (!cwd) return { workspaceId: input.id, status: "merge_planned", prRefreshed: false, error: "worktreePath or repoRoot required" };
+  const target = input.prNumber ? String(input.prNumber) : input.prUrl ?? input.branch;
+  if (!target) return { workspaceId: input.id, status: "merge_planned", prRefreshed: false, error: "prNumber, prUrl, or branch required" };
+  const proc = Bun.spawnSync(["gh", "pr", "update-branch", target], {
+    cwd,
+    stdin: "ignore",
+    stdout: "pipe",
+    stderr: "pipe",
+    env: process.env,
+  });
+  if (proc.exitCode !== 0) {
+    return {
+      workspaceId: input.id,
+      status: "merge_planned",
+      prRefreshed: false,
+      prNumber: input.prNumber,
+      prUrl: input.prUrl,
+      error: proc.stderr.toString().trim() || proc.stdout.toString().trim() || "gh pr update-branch failed",
+    };
+  }
+  return { workspaceId: input.id, status: "merge_planned", prRefreshed: true, prNumber: input.prNumber, prUrl: input.prUrl };
+}