45 lines
1.5 KiB
TypeScript
45 lines
1.5 KiB
TypeScript
import { nativeImage } from 'electron'
|
|
import { ipcRendererInternal } from '@electron/internal/renderer/ipc-renderer-internal'
|
|
|
|
const { hasSwitch } = process.electronBinding('command_line')
|
|
|
|
// |options.types| can't be empty and must be an array
|
|
function isValid (options: Electron.SourcesOptions) {
|
|
const types = options ? options.types : undefined
|
|
return Array.isArray(types)
|
|
}
|
|
|
|
const enableStacks = hasSwitch('enable-api-filtering-logging')
|
|
|
|
function getCurrentStack () {
|
|
const target = {}
|
|
if (enableStacks) {
|
|
Error.captureStackTrace(target, getCurrentStack)
|
|
}
|
|
return (target as any).stack
|
|
}
|
|
|
|
export async function getSources (options: Electron.SourcesOptions) {
|
|
if (!isValid(options)) throw new Error('Invalid options')
|
|
|
|
const captureWindow = options.types.includes('window')
|
|
const captureScreen = options.types.includes('screen')
|
|
|
|
const { thumbnailSize = { width: 150, height: 150 } } = options
|
|
const { fetchWindowIcons = false } = options
|
|
|
|
const sources = await ipcRendererInternal.invoke<ElectronInternal.GetSourcesResult[]>('ELECTRON_BROWSER_DESKTOP_CAPTURER_GET_SOURCES', {
|
|
captureWindow,
|
|
captureScreen,
|
|
thumbnailSize,
|
|
fetchWindowIcons
|
|
} as ElectronInternal.GetSourcesOptions, getCurrentStack())
|
|
|
|
return sources.map(source => ({
|
|
id: source.id,
|
|
name: source.name,
|
|
thumbnail: nativeImage.createFromDataURL(source.thumbnail),
|
|
display_id: source.display_id,
|
|
appIcon: source.appIcon ? nativeImage.createFromDataURL(source.appIcon) : null
|
|
}))
|
|
}
|