orca/src/main/runtime/runtime-edge-command-controller.ts

import type { BrowserScreencastResult } from '../../shared/runtime-types'
import type { RuntimeBrowserCommands, RuntimeBrowserCommandHost } from './orca-runtime-browser'
import { RuntimeEmulatorCommands } from './orca-runtime-emulator'
import { RuntimeBrowserScreencastController } from './runtime-browser-screencast-controller'
import { createRuntimeBrowserCommands } from './runtime-browser-commands-factory'
import { RuntimeJiraCommands } from './runtime-jira-commands'
import type { RpcBinarySender } from './rpc/rpc-binary-sender'

type PublicMethods<T> = Pick<T, keyof T>
type BrowserSurface = Omit<PublicMethods<RuntimeBrowserCommands>, 'browserScreencast'> & {
  browserScreencast(
    params: Parameters<RuntimeBrowserCommands['browserScreencast']>[0],
    options: {
      connectionId?: string
      pairedDeviceId?: string
      clientKind?: 'mobile' | 'runtime'
      sendBinary?: RpcBinarySender
      signal?: AbortSignal
      emit: (result: BrowserScreencastResult) => void
    }
  ): Promise<void>
}

export type RuntimeEdgeCommandSurface = BrowserSurface &
  PublicMethods<RuntimeJiraCommands> &
  PublicMethods<RuntimeEmulatorCommands>

type ScreencastDependencies = ConstructorParameters<typeof RuntimeBrowserScreencastController>[0]
type EmulatorHost = ConstructorParameters<typeof RuntimeEmulatorCommands>[0]

const BROWSER_COMMAND_NAMES = [
  'browserSnapshot',
  'browserClick',
  'browserGoto',
  'browserFill',
  'browserType',
  'browserSelect',
  'browserScroll',
  'browserBack',
  'browserReload',
  'browserScreenshot',
  'browserScreencast',
  'browserEval',
  'browserTabList',
  'browserProceedCertificate',
  'browserTabShow',
  'browserTabCurrent',
  'browserTabSwitch',
  'browserHover',
  'browserDrag',
  'browserUpload',
  'browserWait',
  'browserCheck',
  'browserFocus',
  'browserClear',
  'browserSelectAll',
  'browserKeypress',
  'browserPdf',
  'browserFullScreenshot',
  'browserCookieGet',
  'browserCookieSet',
  'browserCookieDelete',
  'browserSetViewport',
  'browserSetGeolocation',
  'browserInterceptEnable',
  'browserInterceptDisable',
  'browserInterceptList',
  'browserCaptureStart',
  'browserCaptureStop',
  'browserConsoleLog',
  'browserNetworkLog',
  'browserDblclick',
  'browserForward',
  'browserScrollIntoView',
  'browserGet',
  'browserIs',
  'browserKeyboardInsertText',
  'browserMouseMove',
  'browserMouseDown',
  'browserMouseClick',
  'browserMouseUp',
  'browserMouseWheel',
  'browserFind',
  'browserSetDevice',
  'browserSetOffline',
  'browserSetHeaders',
  'browserSetCredentials',
  'browserSetMedia',
  'browserClipboardRead',
  'browserClipboardWrite',
  'browserDialogAccept',
  'browserDialogDismiss',
  'browserStorageLocalGet',
  'browserStorageLocalSet',
  'browserStorageLocalClear',
  'browserStorageSessionGet',
  'browserStorageSessionSet',
  'browserStorageSessionClear',
  'browserDownload',
  'browserHighlight',
  'browserExec',
  'browserTabCreate',
  'browserOpenUrlOnClient',
  'browserTabSetProfile',
  'browserTabProfileShow',
  'browserTabProfileClone',
  'browserProfileList',
  'browserProfileCreate',
  'browserProfileDelete',
  'browserProfileDetectBrowsers',
  'browserProfileImportFromBrowser',
  'browserProfileClearDefaultCookies',
  'browserTabClose'
] as const satisfies readonly (keyof RuntimeBrowserCommands)[]

function bindPrefixedMethods<T extends object>(
  instance: T,
  prefix: string
): Partial<PublicMethods<T>> {
  const bound: Record<string, unknown> = {}
  for (const name of Object.getOwnPropertyNames(Object.getPrototypeOf(instance))) {
    if (name.startsWith(prefix)) {
      bound[name] = (instance[name as keyof T] as (...args: unknown[]) => unknown).bind(instance)
    }
  }
  return bound as Partial<PublicMethods<T>>
}

function bindNamedMethods<T extends object>(
  instance: T,
  names: readonly (keyof T)[]
): Partial<PublicMethods<T>> {
  return Object.fromEntries(
    names.map((name) => [name, (instance[name] as (...args: unknown[]) => unknown).bind(instance)])
  ) as Partial<PublicMethods<T>>
}

export class RuntimeEdgeCommandController {
  private readonly jira = new RuntimeJiraCommands()
  private readonly browser: RuntimeBrowserCommands
  private readonly screencasts: RuntimeBrowserScreencastController
  private readonly emulator: RuntimeEmulatorCommands
  readonly surface: RuntimeEdgeCommandSurface

  constructor(args: {
    browserHost: RuntimeBrowserCommandHost
    screencast: Omit<ScreencastDependencies, 'getCommands'>
    emulatorHost: EmulatorHost
    getBrowserCommands?: () => RuntimeBrowserCommands
  }) {
    this.browser = createRuntimeBrowserCommands(args.browserHost)
    this.screencasts = new RuntimeBrowserScreencastController({
      ...args.screencast,
      getCommands: () => args.getBrowserCommands?.() ?? this.browser
    })
    this.emulator = new RuntimeEmulatorCommands(args.emulatorHost)
    this.surface = {
      ...bindPrefixedMethods(this.jira, 'jira'),
      ...bindNamedMethods(this.browser, BROWSER_COMMAND_NAMES),
      ...bindPrefixedMethods(this.emulator, 'emulator'),
      browserScreencast: (params, options) => this.screencasts.start(params, options)
    } as RuntimeEdgeCommandSurface
  }

  cancelScreencast(browserPageId: string): void {
    this.screencasts.cancelMobilePage(browserPageId, true)
  }

  getBrowserRemoteViewerPages(): string[] {
    return this.screencasts.getRemoteViewerPages()
  }

  getBrowserCommands(): RuntimeBrowserCommands {
    return this.browser
  }
}