electron/lib/browser/desktop-capturer.js

88 lines
2.7 KiB
JavaScript
Raw Normal View History

'use strict'
2016-03-21 18:09:29 +00:00
const ipcMainUtils = require('@electron/internal/browser/ipc-main-internal-utils')
const { desktopCapturer } = process.electronBinding('desktop_capturer')
const eventBinding = process.electronBinding('event')
2016-01-12 02:40:23 +00:00
2017-10-24 16:49:37 +00:00
const deepEqual = (a, b) => JSON.stringify(a) === JSON.stringify(b)
2016-01-12 02:40:23 +00:00
2016-01-14 18:35:29 +00:00
// A queue for holding all requests from renderer process.
2017-10-24 16:49:37 +00:00
let requestsQueue = []
ipcMainUtils.handle('ELECTRON_BROWSER_DESKTOP_CAPTURER_GET_SOURCES', (event, captureWindow, captureScreen, thumbnailSize, fetchWindowIcons) => {
const customEvent = eventBinding.createWithSender(event.sender)
event.sender.emit('desktop-capturer-get-sources', customEvent)
if (customEvent.defaultPrevented) {
return []
}
return new Promise((resolve, reject) => {
const request = {
options: {
captureWindow,
captureScreen,
thumbnailSize,
fetchWindowIcons
},
resolve,
reject
}
requestsQueue.push(request)
if (requestsQueue.length === 1) {
desktopCapturer.startHandling(captureWindow, captureScreen, thumbnailSize, fetchWindowIcons)
}
2016-01-12 02:40:23 +00:00
// If the WebContents is destroyed before receiving result, just remove the
// reference from requestsQueue to make the module not send the result to it.
event.sender.once('destroyed', () => {
request.resolve = null
})
})
})
2016-01-12 02:40:23 +00:00
desktopCapturer.emit = (event, name, sources, fetchWindowIcons) => {
2016-01-14 18:35:29 +00:00
// Receiving sources result from main process, now send them back to renderer.
2017-10-24 19:47:09 +00:00
const handledRequest = requestsQueue.shift()
2017-10-24 16:49:37 +00:00
const unhandledRequestsQueue = []
if (name === 'error') {
const error = sources
handledRequest.reject(error)
return
}
2017-10-24 19:47:09 +00:00
const result = sources.map(source => {
return {
id: source.id,
name: source.name,
thumbnail: source.thumbnail.toDataURL(),
display_id: source.display_id,
appIcon: (fetchWindowIcons && source.appIcon) ? source.appIcon.toDataURL() : null
2017-10-24 19:47:09 +00:00
}
})
2017-10-24 16:49:37 +00:00
if (handledRequest.resolve) {
handledRequest.resolve(result)
2016-01-12 02:40:23 +00:00
}
2017-10-24 23:36:06 +00:00
// Check the queue to see whether there is another identical request & handle
2017-10-24 16:49:37 +00:00
requestsQueue.forEach(request => {
2016-01-12 02:40:23 +00:00
if (deepEqual(handledRequest.options, request.options)) {
if (request.resolve) {
request.resolve(result)
}
2016-01-12 02:40:23 +00:00
} else {
unhandledRequestsQueue.push(request)
2016-01-12 02:40:23 +00:00
}
2017-10-24 16:49:37 +00:00
})
requestsQueue = unhandledRequestsQueue
2016-01-12 02:40:23 +00:00
2016-01-14 18:35:29 +00:00
// If the requestsQueue is not empty, start a new request handling.
2016-01-12 02:40:23 +00:00
if (requestsQueue.length > 0) {
const { captureWindow, captureScreen, thumbnailSize, fetchWindowIcons } = requestsQueue[0].options
return desktopCapturer.startHandling(captureWindow, captureScreen, thumbnailSize, fetchWindowIcons)
2016-01-12 02:40:23 +00:00
}
}