moeka-project/packages/electron-screen-capture/src/renderer.ts

import type { createContext } from '@moeru/eventa/adapters/electron/renderer'
import type { SourcesOptions } from 'electron'

import type { ScreenCaptureSetSourceRequest, SerializableDesktopCapturerSource } from '.'

import { defineInvoke } from '@moeru/eventa'

import { screenCaptureCheckMacOSPermission, screenCaptureGetSources, screenCaptureRequestMacOSPermission, screenCaptureResetSource, screenCaptureSetSourceEx } from '.'

export interface SourceOptionsWithRequest {
  sourcesOptions?: SourcesOptions
  request?: Omit<ScreenCaptureSetSourceRequest, 'options' | 'sourceId'>
}

export function setupElectronScreenCapture(context: ReturnType<typeof createContext>['context']) {
  const invokeGetSources = defineInvoke(context, screenCaptureGetSources)
  const setSource = defineInvoke(context, screenCaptureSetSourceEx)
  const resetSource = defineInvoke(context, screenCaptureResetSource)

  const checkMacOSPermission = defineInvoke(context, screenCaptureCheckMacOSPermission)
  const requestMacOSPermission = defineInvoke(context, screenCaptureRequestMacOSPermission)

  async function getSources(sourcesOptions: SourcesOptions) {
    return invokeGetSources(sourcesOptions)
  }

  async function selectWithSource<R>(
    selectFn: (sources: SerializableDesktopCapturerSource[]) => string | Promise<string>,
    useFn: () => R | Promise<R>,
    options?: SourceOptionsWithRequest,
  ): Promise<R> {
    const sources = await getSources(options?.sourcesOptions)
    const sourceId = await selectFn(sources)

    let handle: string | undefined
    try {
      handle = await setSource({
        options: options?.sourcesOptions,
        sourceId,
        timeout: options?.request?.timeout,
      })
      return await useFn()
    }
    finally {
      if (handle) {
        await resetSource(handle)
      }
    }
  }

  return {
    getSources,
    setSource,
    selectWithSource,
    resetSource,
    checkMacOSPermission,
    requestMacOSPermission,
  }
}