orca/src/main/ipc/agent-pane-authority-ownership.ts

export type AgentPaneAuthorityOwnershipSources = {
  getPtyIdForPaneKey?: (paneKey: string) => string | undefined
  getRuntimeTerminalHandleForPaneKey?: (paneKey: string) => string | undefined
}

function getRemoteRuntimeHandle(ptyId: string): string | null {
  if (!ptyId.startsWith('remote:')) {
    return null
  }
  const rest = ptyId.slice(7)
  const separatorIndex = rest.indexOf('@@')
  if (separatorIndex === -1) {
    return rest.length > 0 && rest.trim() === rest ? rest : null
  }
  const encodedOwner = rest.slice(0, separatorIndex)
  const encodedHandle = rest.slice(separatorIndex + 2)
  if (!encodedOwner || !encodedHandle) {
    return null
  }
  try {
    const owner = decodeURIComponent(encodedOwner)
    const handle = decodeURIComponent(encodedHandle)
    if (!owner || owner.trim() !== owner || !handle || handle.trim() !== handle) {
      return null
    }
    return `remote:${encodeURIComponent(owner)}@@${encodeURIComponent(handle)}` === ptyId
      ? handle
      : null
  } catch {
    return null
  }
}

export function createAgentPaneAuthorityOwnership(
  sources: AgentPaneAuthorityOwnershipSources
): (paneKey: string, ptyId: string) => boolean {
  return (paneKey, ptyId) => {
    if (sources.getPtyIdForPaneKey?.(paneKey) === ptyId) {
      return true
    }
    const runtimeHandle = sources.getRuntimeTerminalHandleForPaneKey?.(paneKey)
    return Boolean(runtimeHandle && getRemoteRuntimeHandle(ptyId) === runtimeHandle)
  }
}