@ricsam/r5d-worker 0.0.46 → 0.0.48

package/dist/mjs/main.mjs

@@ -8,6 +8,7 @@ import { installCliUpdate, readInstalledCliVersion } from "./cli-update.mjs";
 import { configureVisibleGitIdentity } from "./git-identity.mjs";
 import { hasWorkerHeartbeatTimedOut, WORKER_HEARTBEAT_INTERVAL_MS } from "./heartbeat.mjs";
 import { terminateProcessTree } from "./process-tree.mjs";
+import { openWorkerPortForwardRelay } from "./port-forward-client.mjs";
 import {
   isRetryableWorkerServerStatus,
   superviseWorkerRuntime,
@@ -21,7 +22,8 @@ import {
   WORKSPACE_PERIODIC_SCAN_INTERVAL_MS,
   WorkspaceSyncSingleFlight,
   calculateWorkspaceDiffFingerprint,
-  synchronizeWorkspace
+  synchronizeWorkspace,
+  workspaceProjectsForSync
 } from "./workspace-sync.mjs";
 class WorkerServerUnavailableError extends Error {
   name = "WorkerServerUnavailableError";
@@ -578,6 +580,14 @@ function validatePlanId(planId) {
     throw new Error(`Invalid plan id from server: ${planId}`);
   }
 }
+const NON_RECURSIVE_GIT_CONFIG_ARGS = [
+  "-c",
+  "submodule.recurse=false",
+  "-c",
+  "fetch.recurseSubmodules=false",
+  "-c",
+  "push.recurseSubmodules=false"
+];
 function gitExtraHeaderConfigKey(extraHeaderUrl) {
   return `http.${normalizeBaseUrl(extraHeaderUrl)}/.extraHeader`;
 }
@@ -585,10 +595,23 @@ function gitAuthArgs(auth) {
   if (!auth) {
     return [];
   }
-  return ["-c", `${gitExtraHeaderConfigKey(auth.extraHeaderUrl)}=${auth.header}`];
+  const scopedKey = gitExtraHeaderConfigKey(auth.extraHeaderUrl);
+  return ["-c", "http.extraHeader=", "-c", `${scopedKey}=`, "-c", `${scopedKey}=${auth.header}`];
+}
+const visibleCheckoutGitTestHarness = {
+  gitAuthArgs,
+  commandArgs: (args, auth) => ["git", ...NON_RECURSIVE_GIT_CONFIG_ARGS, ...gitAuthArgs(auth), ...args],
+  cloneArgs: (remoteUrl, branchPath) => ["clone", "--no-recurse-submodules", "--origin", "origin", remoteUrl, branchPath],
+  fetchArgs: (...args) => ["fetch", "--no-recurse-submodules", ...args]
+};
+function visibleCheckoutCloneArgs(remoteUrl, branchPath) {
+  return visibleCheckoutGitTestHarness.cloneArgs(remoteUrl, branchPath);
+}
+function visibleCheckoutFetchArgs(...args) {
+  return visibleCheckoutGitTestHarness.fetchArgs(...args);
 }
 function runGit(args, options = {}) {
-  const command = ["git", ...gitAuthArgs(options.auth), ...args];
+  const command = visibleCheckoutGitTestHarness.commandArgs(args, options.auth);
   const result = Bun.spawnSync(command, {
     cwd: options.cwd,
     stdout: "pipe",
@@ -732,8 +755,28 @@ function resolveWorkerFilePath(branchPath, inputPath, builtInPaths) {
 function fallbackInternalRemoteUrl(baseUrl, projectId) {
   return new URL(`/git/${projectId}.git`, baseUrl).toString();
 }
-function githubRemoteUrlFor(baseUrl, projectId, manifest) {
-  return manifest?.repoHttpUrl ?? fallbackInternalRemoteUrl(baseUrl, projectId);
+function visibleCheckoutRemoteFor(input) {
+  const canonicalCheckout = input.manifest?.canonicalCheckouts?.find((checkout) => checkout.branchName === input.branchName);
+  const isCanonicalManifestBranch = Boolean(canonicalCheckout);
+  const usesCanonicalRemote = isCanonicalManifestBranch || !input.manifest?.repoHttpUrl;
+  if (usesCanonicalRemote) {
+    return {
+      remoteUrl: fallbackInternalRemoteUrl(input.baseUrl, input.projectId),
+      authHeader: `Authorization: Bearer ${input.token}`,
+      persistAuth: false,
+      reconcileExistingOrigin: isCanonicalManifestBranch,
+      requireRemoteBranch: isCanonicalManifestBranch,
+      requiredBaseCommit: canonicalCheckout?.scaffoldCommitHash ?? null
+    };
+  }
+  return {
+    remoteUrl: input.manifest?.repoHttpUrl ?? fallbackInternalRemoteUrl(input.baseUrl, input.projectId),
+    authHeader: input.manifest?.repoAuthHeader ?? null,
+    persistAuth: true,
+    reconcileExistingOrigin: Boolean(input.manifest?.repoHttpUrl),
+    requireRemoteBranch: false,
+    requiredBaseCommit: null
+  };
 }
 function gitExtraHeaderUrlForRemote(remoteUrl) {
   try {
@@ -762,6 +805,10 @@ function ensureOriginRemote(branchPath, remoteUrl) {
     runGit(["remote", "add", "origin", remoteUrl], { cwd: branchPath });
   }
 }
+function originRemoteUrl(branchPath) {
+  if (!tryGit(["remote", "get-url", "origin"], { cwd: branchPath })) return null;
+  return runGit(["remote", "get-url", "origin"], { cwd: branchPath });
+}
 function shellQuote(value) {
   return `'${value.replace(/'/g, "'\\''")}'`;
 }
@@ -830,6 +877,18 @@ function configureVisibleGitHubAuth(branchPath, remoteUrl, authHeader) {
   }
   runGit(["config", `${gitExtraHeaderConfigKey(gitExtraHeaderUrlForRemote(remoteUrl))}`, authHeader], { cwd: branchPath });
 }
+function clearVisibleGitAuthentication(branchPath) {
+  let extraHeaderKeys = [];
+  try {
+    extraHeaderKeys = runGit(["config", "--local", "--name-only", "--get-regexp", "^http\\..*extraheader$"], { cwd: branchPath }).split(/\r?\n/).filter(Boolean);
+  } catch {
+  }
+  for (const key of extraHeaderKeys) {
+    tryGit(["config", "--local", "--unset-all", key], { cwd: branchPath });
+  }
+  runGit(["config", "--local", "--replace-all", "credential.helper", ""], { cwd: branchPath });
+  tryGit(["config", "--local", "--unset-all", "credential.useHttpPath"], { cwd: branchPath });
+}
 function dockerConfigPath() {
   return path.join(os.homedir(), ".docker", "config.json");
 }
@@ -918,18 +977,58 @@ function checkoutVisibleBranch(input) {
   });
   const clean = !hasWorktreeChanges(input.branchPath);
   if (remoteBranchExists && clean) {
-    runGit(["checkout", "-B", input.branchName, `origin/${input.branchName}`], { cwd: input.branchPath });
+    runGit(["checkout", "--no-recurse-submodules", "-B", input.branchName, `origin/${input.branchName}`], {
+      cwd: input.branchPath
+    });
     return;
   }
   if (branchExists) {
-    runGit(["checkout", input.branchName], { cwd: input.branchPath });
+    runGit(["checkout", "--no-recurse-submodules", input.branchName], { cwd: input.branchPath });
     return;
   }
   if (defaultRemoteExists) {
-    runGit(["checkout", "-B", input.branchName, `origin/${input.defaultBranch}`], { cwd: input.branchPath });
+    runGit(["checkout", "--no-recurse-submodules", "-B", input.branchName, `origin/${input.defaultBranch}`], {
+      cwd: input.branchPath
+    });
     return;
   }
-  runGit(["checkout", "-B", input.branchName], { cwd: input.branchPath });
+  runGit(["checkout", "--no-recurse-submodules", "-B", input.branchName], { cwd: input.branchPath });
+}
+function migrateExistingCheckoutToRequiredRemote(input) {
+  if (originRemoteUrl(input.branchPath) === input.remoteUrl) return;
+  if (hasWorktreeChanges(input.branchPath)) {
+    throw new Error(`Cannot migrate dirty checkout for ${input.branchName} to its canonical remote`);
+  }
+  runGit(
+    visibleCheckoutFetchArgs("--no-tags", input.remoteUrl, `+refs/heads/${input.branchName}:refs/remotes/origin/${input.branchName}`),
+    {
+      cwd: input.branchPath,
+      auth: input.remoteAuth
+    }
+  );
+  const canonicalRemoteRef = `refs/remotes/origin/${input.branchName}`;
+  if (!tryGit(["show-ref", "--verify", "--quiet", canonicalRemoteRef], { cwd: input.branchPath })) {
+    throw new Error(`Canonical branch ${input.branchName} is unavailable during checkout migration`);
+  }
+  if (input.requiredBaseCommit && (!tryGit(["cat-file", "-e", `${input.requiredBaseCommit}^{commit}`], { cwd: input.branchPath }) || !tryGit(["merge-base", "--is-ancestor", input.requiredBaseCommit, canonicalRemoteRef], { cwd: input.branchPath }))) {
+    throw new Error(`Canonical branch ${input.branchName} is not derived from required scaffold ${input.requiredBaseCommit}`);
+  }
+  if (tryGit(["rev-parse", "--verify", "HEAD"], { cwd: input.branchPath })) {
+    const previousHead = runGit(["rev-parse", "HEAD"], { cwd: input.branchPath });
+    runGit(["update-ref", `refs/r5d/pre-canonical-checkout/${previousHead}`, previousHead], { cwd: input.branchPath });
+  }
+  ensureOriginRemote(input.branchPath, input.remoteUrl);
+  runGit(["checkout", "--no-recurse-submodules", "-B", input.branchName, `origin/${input.branchName}`], {
+    cwd: input.branchPath
+  });
+}
+function assertCheckoutDerivedFromRequiredBase(input) {
+  const remoteRevision = `refs/remotes/origin/${input.branchName}`;
+  for (const revision of [remoteRevision, "HEAD"]) {
+    if (!tryGit(["cat-file", "-e", `${input.requiredBaseCommit}^{commit}`], { cwd: input.branchPath }) || !tryGit(["merge-base", "--is-ancestor", input.requiredBaseCommit, revision], { cwd: input.branchPath })) {
+      throw new Error(`Canonical checkout ${input.branchName} is not derived from required scaffold ${input.requiredBaseCommit}`);
+    }
+  }
 }
 function ensureVisibleGitCheckout(input) {
   validateBranchName(input.branchName);
@@ -939,40 +1038,114 @@ function ensureVisibleGitCheckout(input) {
   if (createdCheckout) {
     fs.rmSync(branchPath, { recursive: true, force: true });
     fs.mkdirSync(path.dirname(branchPath), { recursive: true });
-    if (!tryGit(["clone", "--origin", "origin", input.githubRemoteUrl, branchPath], { auth: input.githubAuth })) {
+    const cloned = tryGit(visibleCheckoutCloneArgs(input.remoteUrl, branchPath), {
+      auth: input.remoteAuth
+    });
+    if (!cloned && input.requireRemoteBranch) {
+      fs.rmSync(branchPath, { recursive: true, force: true });
+      throw new Error(`Failed to clone canonical checkout for branch ${input.branchName}`);
+    }
+    if (!cloned) {
       fs.mkdirSync(branchPath, { recursive: true });
       runGit(["init"], { cwd: branchPath });
     }
-    ensureOriginRemote(branchPath, input.githubRemoteUrl);
-    configureVisibleGitHubAuth(branchPath, input.githubRemoteUrl, input.githubAuthHeader);
+    ensureOriginRemote(branchPath, input.remoteUrl);
+    if (input.clearPersistentAuth) clearVisibleGitAuthentication(branchPath);
+    else configureVisibleGitHubAuth(branchPath, input.remoteUrl, input.persistentAuthHeader);
     configureVisibleGitIdentity(branchPath, visibleGitIdentity, runGit);
-    tryGit(["fetch", "origin", "--prune"], { cwd: branchPath, auth: input.githubAuth });
+    tryGit(visibleCheckoutFetchArgs("origin", "--prune"), { cwd: branchPath, auth: input.remoteAuth });
+    if (input.requireRemoteBranch && !tryGit(["show-ref", "--verify", "--quiet", `refs/remotes/origin/${input.branchName}`], { cwd: branchPath })) {
+      fs.rmSync(branchPath, { recursive: true, force: true });
+      throw new Error(`Canonical branch ${input.branchName} is unavailable after clone`);
+    }
     checkoutVisibleBranch({ branchPath, branchName: input.branchName, defaultBranch: input.defaultBranch });
+    if (input.requiredBaseCommit) {
+      try {
+        assertCheckoutDerivedFromRequiredBase({
+          branchPath,
+          branchName: input.branchName,
+          requiredBaseCommit: input.requiredBaseCommit
+        });
+      } catch (error) {
+        fs.rmSync(branchPath, { recursive: true, force: true });
+        throw error;
+      }
+    }
     return branchPath;
   }
+  if (input.requireRemoteBranch) {
+    if (input.allowRequiredRemoteMigration === false && originRemoteUrl(branchPath) !== input.remoteUrl) {
+      throw new Error(`Canonical checkout ${input.branchName} no longer uses its required remote`);
+    }
+    migrateExistingCheckoutToRequiredRemote({
+      branchPath,
+      branchName: input.branchName,
+      remoteUrl: input.remoteUrl,
+      remoteAuth: input.remoteAuth,
+      requiredBaseCommit: input.requiredBaseCommit
+    });
+    runGit(
+      visibleCheckoutFetchArgs("--no-tags", input.remoteUrl, `+refs/heads/${input.branchName}:refs/remotes/origin/${input.branchName}`),
+      {
+        cwd: branchPath,
+        auth: input.remoteAuth
+      }
+    );
+  }
   if (input.reconcileExistingOrigin) {
-    ensureOriginRemote(branchPath, input.githubRemoteUrl);
+    ensureOriginRemote(branchPath, input.remoteUrl);
   }
-  configureVisibleGitHubAuth(branchPath, input.githubRemoteUrl, input.githubAuthHeader);
+  if (input.requireRemoteBranch && !tryGit(["show-ref", "--verify", "--quiet", `refs/remotes/origin/${input.branchName}`], { cwd: branchPath })) {
+    throw new Error(`Existing checkout for ${input.branchName} does not contain its canonical remote branch`);
+  }
+  if (input.requiredBaseCommit) {
+    assertCheckoutDerivedFromRequiredBase({
+      branchPath,
+      branchName: input.branchName,
+      requiredBaseCommit: input.requiredBaseCommit
+    });
+  }
+  if (input.clearPersistentAuth) clearVisibleGitAuthentication(branchPath);
+  else configureVisibleGitHubAuth(branchPath, input.remoteUrl, input.persistentAuthHeader);
   configureVisibleGitIdentity(branchPath, visibleGitIdentity, runGit);
   return branchPath;
 }
 function ensureBranchWorkspace(input) {
   const defaultBranch = input.manifest?.defaultBranch || "main";
-  const githubRemoteUrl = githubRemoteUrlFor(input.baseUrl, input.projectId, input.manifest);
-  const githubAuth = gitAuthForRemote(githubRemoteUrl, input.manifest?.repoAuthHeader);
+  const remote = visibleCheckoutRemoteFor(input);
   return {
     branchPath: ensureVisibleGitCheckout({
       projectRoot: input.projectRoot,
       branchName: input.branchName,
-      githubRemoteUrl,
-      githubAuth,
-      githubAuthHeader: input.manifest?.repoAuthHeader ?? null,
+      remoteUrl: remote.remoteUrl,
+      remoteAuth: gitAuthForRemote(remote.remoteUrl, remote.authHeader),
+      persistentAuthHeader: remote.persistAuth ? remote.authHeader : null,
       defaultBranch,
-      reconcileExistingOrigin: Boolean(input.manifest?.repoHttpUrl)
+      reconcileExistingOrigin: remote.reconcileExistingOrigin,
+      requireRemoteBranch: remote.requireRemoteBranch,
+      requiredBaseCommit: remote.requiredBaseCommit,
+      clearPersistentAuth: !remote.persistAuth
     })
   };
 }
+function workspaceSyncCheckoutTargets(input) {
+  const selected = /* @__PURE__ */ new Map();
+  const add = (target) => {
+    selected.set(`${target.projectId}\0${target.branchName}`, target);
+  };
+  if (input.canonicalCheckoutOnly) {
+    add(input.canonicalCheckoutOnly);
+  } else if (input.includeAllManifestCheckouts) {
+    for (const project of input.projects) {
+      for (const branchName of project.branches) add({ projectId: project.projectId, branchName });
+    }
+  } else {
+    for (const target of input.pendingTargets) add(target);
+  }
+  return [...selected.values()].sort(
+    (left, right) => left.projectId.localeCompare(right.projectId) || left.branchName.localeCompare(right.branchName)
+  );
+}
 function resolveCommandCwd(branchPath, cwd) {
   if (!cwd) {
     return branchPath;
@@ -2196,6 +2369,7 @@ async function startWorker(options) {
   let reloadAfterClose = false;
   const workspaceSyncInput = (trigger, overrides = {}) => {
     if (!workspaceRemoteUrl) throw new Error("Worker workspace manifest has not been received");
+    const projects = workspaceProjectsForSync([...manifestByProjectId.values()], trigger);
     return {
       workerLabel: label,
       remoteUrl: workspaceRemoteUrl,
@@ -2203,7 +2377,7 @@ async function startWorker(options) {
       projectsRoot,
       plansRoot: planRoot,
       shadowRoot: workspaceShadowRoot,
-      projects: [...manifestByProjectId.values()],
+      projects,
       trigger,
       ...overrides
     };
@@ -2215,7 +2389,7 @@ async function startWorker(options) {
   const allManifestCheckouts = () => [...manifestByProjectId.values()].flatMap(
     (manifest) => manifest.branches.map((branchName) => ({ projectId: manifest.projectId, branchName }))
   );
-  const ensureVisibleWorkspaceCheckouts = (targets) => {
+  const ensureVisibleWorkspaceCheckouts = (targets, options2 = {}) => {
     const created = [];
     for (const target of targets) {
       const manifest = manifestByProjectId.get(target.projectId);
@@ -2227,15 +2401,25 @@ async function startWorker(options) {
       const projectRoot = projectRootFor(projectsRoot, manifest.projectId, manifestByProjectId);
       const branchPath = path.join(projectRoot, target.branchName);
       const existed = hasNormalVisibleGitDir(branchPath);
-      const githubRemoteUrl = githubRemoteUrlFor(baseUrl, manifest.projectId, manifest);
+      const remote = visibleCheckoutRemoteFor({
+        baseUrl,
+        token,
+        projectId: manifest.projectId,
+        branchName: target.branchName,
+        manifest
+      });
       ensureVisibleGitCheckout({
         projectRoot,
         branchName: target.branchName,
-        githubRemoteUrl,
-        githubAuth: gitAuthForRemote(githubRemoteUrl, manifest.repoAuthHeader),
-        githubAuthHeader: manifest.repoAuthHeader,
+        remoteUrl: remote.remoteUrl,
+        remoteAuth: gitAuthForRemote(remote.remoteUrl, remote.authHeader),
+        persistentAuthHeader: remote.persistAuth ? remote.authHeader : null,
         defaultBranch: manifest.defaultBranch || "main",
-        reconcileExistingOrigin: Boolean(manifest.repoHttpUrl)
+        reconcileExistingOrigin: remote.reconcileExistingOrigin,
+        requireRemoteBranch: remote.requireRemoteBranch,
+        requiredBaseCommit: remote.requiredBaseCommit,
+        allowRequiredRemoteMigration: !(options2.strictCanonicalRevalidation && remote.requireRemoteBranch),
+        clearPersistentAuth: !remote.persistAuth
       });
       if (!existed) created.push(target);
     }
@@ -2247,15 +2431,30 @@ async function startWorker(options) {
       let pendingTargets = [];
       const result = await workspaceSyncSingleFlight.runPrepared(() => {
         const shouldEnsureCheckouts = !overrides.skipVisibleMirror && !overrides.resetToCanonical;
+        const syncProjects = workspaceProjectsForSync([...manifestByProjectId.values()], trigger);
+        const allowedCheckoutKeys = new Set(
+          syncProjects.flatMap((project) => project.branches.map((branchName) => manifestCheckoutKey(project.projectId, branchName)))
+        );
+        const scopedPendingTargets = [...pendingManifestCheckouts.values()].filter(
+          (target) => allowedCheckoutKeys.has(manifestCheckoutKey(target.projectId, target.branchName))
+        );
         let checkoutTargets = [];
         if (shouldEnsureCheckouts) {
-          checkoutTargets = trigger.type === "connect" ? allManifestCheckouts() : [...pendingManifestCheckouts.values()];
+          checkoutTargets = workspaceSyncCheckoutTargets({
+            projects: syncProjects,
+            pendingTargets: scopedPendingTargets,
+            includeAllManifestCheckouts: trigger.type === "connect",
+            ...trigger.canonicalCheckoutOnly && trigger.projectId && trigger.branchName ? { canonicalCheckoutOnly: { projectId: trigger.projectId, branchName: trigger.branchName } } : {}
+          });
         }
-        pendingTargets = shouldEnsureCheckouts ? [...pendingManifestCheckouts.values()] : [];
-        const createdCheckouts = ensureVisibleWorkspaceCheckouts(checkoutTargets);
+        pendingTargets = shouldEnsureCheckouts ? scopedPendingTargets : [];
+        const createdCheckouts = ensureVisibleWorkspaceCheckouts(checkoutTargets, {
+          strictCanonicalRevalidation: true
+        });
         const newVisibleCheckoutByKey = /* @__PURE__ */ new Map();
         for (const target of [...overrides.newVisibleCheckouts ?? [], ...createdCheckouts, ...pendingTargets]) {
-          newVisibleCheckoutByKey.set(manifestCheckoutKey(target.projectId, target.branchName), target);
+          const key = manifestCheckoutKey(target.projectId, target.branchName);
+          if (allowedCheckoutKeys.has(key)) newVisibleCheckoutByKey.set(key, target);
         }
         const newVisibleCheckouts = [...newVisibleCheckoutByKey.values()];
         return workspaceSyncInput(trigger, {
@@ -2331,7 +2530,9 @@ async function startWorker(options) {
         version: getWorkerVersion(),
         r5dctlVersion: readInstalledCliVersion("r5dctl"),
         capabilities: {
-          updateClis: true
+          updateClis: true,
+          canonicalResolverCheckout: true,
+          browserPortForwarding: true
         },
         projectRoot: projectsRoot,
         artifactRoot,
@@ -2391,7 +2592,15 @@ async function startWorker(options) {
             if (!workspaceRemoteUrl || activeWorkspaceIncidentId || cliUpdateInProgress || periodicWorkspaceScanInFlight) return;
             periodicWorkspaceScanInFlight = true;
             void (async () => {
-              if (pendingManifestCheckouts.size > 0) {
+              const genericCheckoutKeys = new Set(
+                workspaceProjectsForSync([...manifestByProjectId.values()], { type: "periodic" }).flatMap(
+                  (project) => project.branches.map((branchName) => manifestCheckoutKey(project.projectId, branchName))
+                )
+              );
+              const hasPendingGenericCheckout = [...pendingManifestCheckouts.values()].some(
+                (target) => genericCheckoutKeys.has(manifestCheckoutKey(target.projectId, target.branchName))
+              );
+              if (hasPendingGenericCheckout) {
                 previousPeriodicFingerprint = null;
                 await runRequestedWorkspaceSync(crypto.randomUUID(), {
                   type: "periodic",
@@ -2476,6 +2685,17 @@ async function startWorker(options) {
         pendingProcessTerminals.delete(message.runId);
         return;
       }
+      if (message.type === "port_forward_connect") {
+        openWorkerPortForwardRelay({
+          baseUrl,
+          token,
+          label,
+          forwardId: message.forwardId,
+          relayConnectionId: message.relayConnectionId,
+          workerPort: message.workerPort
+        });
+        return;
+      }
       if (message.type === "update_clis") {
         if (cliUpdateInProgress) {
           sendWorkerMessage(ws, {
@@ -2808,5 +3028,8 @@ export {
   resolveHostShell,
   resolveWorkerFilePath,
   syncSessionArtifacts,
+  visibleCheckoutGitTestHarness,
+  visibleCheckoutRemoteFor,
+  workspaceSyncCheckoutTargets,
   writeWorkerTextFile
 };

package/dist/mjs/package.json

@@ -1,5 +1,5 @@
 {
   "name": "@ricsam/r5d-worker",
-  "version": "0.0.46",
+  "version": "0.0.48",
   "type": "module"
 }

package/dist/mjs/port-forward-client.mjs

@@ -0,0 +1,114 @@
+import net from "node:net";
+import WebSocket, {} from "ws";
+const RELAY_FRAME_BYTES = 64 * 1024;
+const RELAY_PAUSE_BYTES = 1024 * 1024;
+const RELAY_RESUME_BYTES = 256 * 1024;
+function relayUrl(baseUrl, label, forwardId, relayConnectionId) {
+  const url = new URL("/worker/port-forward/ws", baseUrl);
+  url.protocol = url.protocol === "https:" ? "wss:" : "ws:";
+  url.searchParams.set("label", label);
+  url.searchParams.set("forward", forwardId);
+  url.searchParams.set("connection", relayConnectionId);
+  return url.toString();
+}
+function rawDataBuffer(data) {
+  if (Buffer.isBuffer(data)) return data;
+  if (Array.isArray(data)) return Buffer.concat(data);
+  return Buffer.from(data);
+}
+function openWorkerPortForwardRelay(options) {
+  const socket = net.createConnection({ host: "127.0.0.1", port: options.workerPort, allowHalfOpen: true });
+  socket.pause();
+  socket.setNoDelay(true);
+  const relay = new WebSocket(relayUrl(options.baseUrl, options.label, options.forwardId, options.relayConnectionId), {
+    headers: { Authorization: `Bearer ${options.token}` },
+    perMessageDeflate: false
+  });
+  let relayReady = false;
+  let targetReady = false;
+  const pendingChunks = [];
+  let pendingBytes = 0;
+  let closing = false;
+  let resumeTimer;
+  const maybeResume = () => {
+    if (relayReady && targetReady && !resumeTimer) socket.resume();
+  };
+  const cleanup = () => {
+    if (closing) return;
+    closing = true;
+    if (resumeTimer) clearInterval(resumeTimer);
+    socket.destroy();
+    if (relay.readyState === WebSocket.OPEN || relay.readyState === WebSocket.CONNECTING) relay.close();
+  };
+  const sendError = (message) => {
+    if (relay.readyState === WebSocket.OPEN) relay.send(JSON.stringify({ type: "error", error: message }));
+    cleanup();
+  };
+  const waitForRelayDrain = () => {
+    if (resumeTimer || relay.bufferedAmount <= RELAY_PAUSE_BYTES) return;
+    socket.pause();
+    resumeTimer = setInterval(() => {
+      if (relay.readyState !== WebSocket.OPEN) return cleanup();
+      if (relay.bufferedAmount > RELAY_RESUME_BYTES) return;
+      clearInterval(resumeTimer);
+      resumeTimer = void 0;
+      maybeResume();
+    }, 10);
+  };
+  const sendChunk = (chunk) => {
+    for (let offset = 0; offset < chunk.byteLength; offset += RELAY_FRAME_BYTES) {
+      relay.send(chunk.subarray(offset, Math.min(chunk.byteLength, offset + RELAY_FRAME_BYTES)), { binary: true });
+    }
+    waitForRelayDrain();
+  };
+  socket.once("connect", () => {
+    targetReady = true;
+    maybeResume();
+  });
+  socket.on("data", (chunk) => {
+    if (!relayReady || relay.readyState !== WebSocket.OPEN) {
+      pendingBytes += chunk.byteLength;
+      if (pendingBytes > RELAY_PAUSE_BYTES) return cleanup();
+      pendingChunks.push(Buffer.from(chunk));
+      socket.pause();
+      return;
+    }
+    sendChunk(chunk);
+  });
+  socket.on("end", () => {
+    if (relay.readyState === WebSocket.OPEN) relay.send(JSON.stringify({ type: "end" }));
+  });
+  socket.on("error", (error) => sendError(`Could not connect to 127.0.0.1:${options.workerPort}: ${error.message}`));
+  socket.on("close", cleanup);
+  relay.on("message", (data, isBinary) => {
+    if (!isBinary) {
+      let control;
+      try {
+        control = JSON.parse(rawDataBuffer(data).toString("utf8"));
+      } catch {
+        return cleanup();
+      }
+      if (control.type === "ready") {
+        relayReady = true;
+        for (const chunk of pendingChunks.splice(0)) sendChunk(chunk);
+        pendingBytes = 0;
+        maybeResume();
+      } else if (control.type === "end") {
+        socket.end();
+      } else if (control.type === "error") {
+        socket.destroy(new Error(control.error || "Port-forward relay failed."));
+      }
+      return;
+    }
+    if (!socket.write(rawDataBuffer(data))) {
+      relay.pause();
+      socket.once("drain", () => relay.resume());
+    }
+  });
+  relay.once("error", cleanup);
+  relay.once("close", cleanup);
+  socket.pause();
+}
+export {
+  openWorkerPortForwardRelay
+};