79 lines
2.4 KiB
JavaScript
79 lines
2.4 KiB
JavaScript
'use strict'
|
|
|
|
const {ipcMain} = require('electron')
|
|
const {desktopCapturer} = process.atomBinding('desktop_capturer')
|
|
|
|
const deepEqual = (a, b) => JSON.stringify(a) === JSON.stringify(b)
|
|
|
|
// A queue for holding all requests from renderer process.
|
|
let requestsQueue = []
|
|
|
|
const electronSources = 'ELECTRON_BROWSER_DESKTOP_CAPTURER_GET_SOURCES'
|
|
const capturerResult = (id) => `ELECTRON_RENDERER_DESKTOP_CAPTURER_RESULT_${id}`
|
|
|
|
ipcMain.on(electronSources, (event, captureWindow, captureScreen, thumbnailSize, id) => {
|
|
const request = {
|
|
id,
|
|
options: {
|
|
captureWindow,
|
|
captureScreen,
|
|
thumbnailSize
|
|
},
|
|
webContents: event.sender
|
|
}
|
|
requestsQueue.push(request)
|
|
if (requestsQueue.length === 1) {
|
|
desktopCapturer.startHandling(captureWindow, captureScreen, thumbnailSize)
|
|
}
|
|
|
|
// 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.webContents = null
|
|
return
|
|
})
|
|
})
|
|
|
|
desktopCapturer.emit = (event, name, sources) => {
|
|
// Receiving sources result from main process, now send them back to renderer.
|
|
const handledRequest = requestsQueue.shift(0)
|
|
const handledWebContents = handledRequest.webContents
|
|
const unhandledRequestsQueue = []
|
|
|
|
const result = () => {
|
|
const results = []
|
|
sources.forEach(source => {
|
|
results.push({
|
|
id: source.id,
|
|
name: source.name,
|
|
thumbnail: source.thumbnail.toDataURL()
|
|
})
|
|
})
|
|
return results
|
|
}
|
|
|
|
if (handledWebContents != null) {
|
|
handledWebContents.send(capturerResult(handledRequest.id), result())
|
|
}
|
|
|
|
// Check the queue to see whether there is other same request. If has, handle
|
|
// it for reducing redunplicated `desktopCaptuer.startHandling` calls.
|
|
|
|
requestsQueue.forEach(request => {
|
|
const webContents = request.webContents
|
|
if (deepEqual(handledRequest.options, request.options)) {
|
|
if (webContents != null) {
|
|
webContents.send(capturerResult(request.id), result())
|
|
}
|
|
} else {
|
|
unhandledRequestsQueue.push(request)
|
|
}
|
|
})
|
|
requestsQueue = unhandledRequestsQueue
|
|
|
|
// If the requestsQueue is not empty, start a new request handling.
|
|
if (requestsQueue.length > 0) {
|
|
const {captureWindow, captureScreen, thumbnailSize} = requestsQueue[0].options
|
|
return desktopCapturer.startHandling(captureWindow, captureScreen, thumbnailSize)
|
|
}
|
|
}
|