Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

apps/desktop/electron/connection-apply.ts

@@ -0,0 +1,111 @@
+async function applyConnectionChange({
+  cancelAndWait,
+  isPrimary,
+  rehomePrimary = null,
+  scope,
+  sendApplied,
+  stopPool,
+  teardownPrimary,
+  teardownSsh
+}) {
+  await cancelAndWait(scope)
+  await teardownSsh(scope)
+
+  if (!isPrimary) {
+    stopPool(scope)
+
+    return
+  }
+
+  if (rehomePrimary) {
+    await rehomePrimary()
+
+    return
+  }
+
+  await teardownPrimary()
+  sendApplied()
+}
+
+function commitConnectionFailure(current, starting, commit) {
+  if (current !== starting) {
+    return false
+  }
+
+  commit()
+
+  return true
+}
+
+async function resolveTerminalConnection(getTarget, ensureBackend) {
+  let target = getTarget()
+
+  if (target !== 'pending') {
+    return target
+  }
+
+  await ensureBackend()
+  target = getTarget()
+
+  if (target === 'pending') {
+    throw new Error('Remote connection is not ready yet. Try again in a moment.')
+  }
+
+  return target
+}
+
+async function resolveTerminalConnectionForSender(webContentsId, getTarget, ensureBackend) {
+  return resolveTerminalConnection(
+    () => getTarget(webContentsId),
+    () => ensureBackend(webContentsId)
+  )
+}
+
+/** A second before-quit must still wait for an in-flight remote kill.
+ *
+ *  teardownSshConnection deletes the sshConnections entry first, then
+ *  SSH-execs kill. backendShutdown's finally() calls app.quit() and
+ *  re-enters before-quit with an empty map. Without `inFlight`, Electron
+ *  exits while disconnect is running and the detached serve --isolated
+ *  stays at pid 1 (post-#95085 leftover on #91668: window X on Windows). */
+function sshQuitShouldBlock({ teardownDone, connectionCount, bootstrapPending, inFlight }) {
+  if (teardownDone) {
+    return false
+  }
+
+  return connectionCount > 0 || bootstrapPending > 0 || Boolean(inFlight)
+}
+
+async function teardownSshState(state, { cleanupRemote }) {
+  // Remote process first, while the SSH channel can still exec kill.
+  // Then drop the local forward and close the transport. Each step is
+  // best-effort so a failed remote cleanup cannot trap Cmd+Q (#91668).
+  try {
+    await cleanupRemote(state.ssh, state.ownershipId)
+  } catch {
+    // Remote teardown is best-effort; always release the local tunnel and SSH transport.
+  }
+
+  try {
+    if (state.localPort && state.remotePort) {
+      await state.ssh.cancelForward(state.localPort, state.remotePort)
+    }
+  } catch {
+    // Best effort; closing the transport below drops any remaining forwards.
+  }
+
+  try {
+    await state.ssh.close()
+  } catch {
+    // The app must still be able to quit when SSH teardown fails.
+  }
+}
+
+export {
+  applyConnectionChange,
+  commitConnectionFailure,
+  resolveTerminalConnection,
+  resolveTerminalConnectionForSender,
+  sshQuitShouldBlock,
+  teardownSshState
+}