orca/src/main/ipc/runtime-environment-abortable-requests.ts

import type { PairedRuntimeConnection } from '../../shared/paired-runtime-connection'
import type { RuntimeCapability } from '../../shared/protocol-version'
import { sendRemoteRuntimeRequest } from '../../shared/remote-runtime-client'
import type {
  RuntimeOrchestrationEnvelope,
  RuntimeRpcResponse
} from '../../shared/runtime-rpc-envelope'
import {
  sendRemoteRuntimeConnectionRequest,
  sendRemoteRuntimeSharedControlRequest
} from './runtime-environment-request-connections'

export function sendRemoteRuntimeRequestAbortable(
  connection: PairedRuntimeConnection,
  method: string,
  params: unknown,
  timeoutMs: number,
  envelope?: RuntimeOrchestrationEnvelope,
  signal?: AbortSignal,
  clientCapabilities: readonly RuntimeCapability[] = []
): Promise<RuntimeRpcResponse<unknown>> {
  if (signal || clientCapabilities.length > 0) {
    return sendRemoteRuntimeRequest(
      connection,
      method,
      params,
      timeoutMs,
      envelope,
      signal,
      clientCapabilities
    )
  }
  return envelope
    ? sendRemoteRuntimeRequest(connection, method, params, timeoutMs, envelope)
    : sendRemoteRuntimeRequest(connection, method, params, timeoutMs)
}

export function sendRemoteRuntimeConnectionRequestAbortable(
  environmentId: string,
  connection: PairedRuntimeConnection,
  method: string,
  params: unknown,
  timeoutMs: number,
  signal?: AbortSignal
): Promise<RuntimeRpcResponse<unknown>> {
  return signal
    ? sendRemoteRuntimeConnectionRequest(
        environmentId,
        connection,
        method,
        params,
        timeoutMs,
        signal
      )
    : sendRemoteRuntimeConnectionRequest(environmentId, connection, method, params, timeoutMs)
}

export function sendRemoteRuntimeSharedControlRequestAbortable(
  environmentId: string,
  connection: PairedRuntimeConnection,
  method: string,
  params: unknown,
  timeoutMs: number,
  envelope?: RuntimeOrchestrationEnvelope,
  signal?: AbortSignal
): Promise<RuntimeRpcResponse<unknown>> {
  if (signal) {
    return sendRemoteRuntimeSharedControlRequest(
      environmentId,
      connection,
      method,
      params,
      timeoutMs,
      envelope,
      signal
    )
  }
  return envelope
    ? sendRemoteRuntimeSharedControlRequest(
        environmentId,
        connection,
        method,
        params,
        timeoutMs,
        envelope
      )
    : sendRemoteRuntimeSharedControlRequest(environmentId, connection, method, params, timeoutMs)
}