77 lines
2.6 KiB
JavaScript
77 lines
2.6 KiB
JavaScript
'use strict';
|
|
|
|
const ipcMain = require('electron').ipcMain;
|
|
const desktopCapturer = process.atomBinding('desktop_capturer').desktopCapturer;
|
|
|
|
var deepEqual = function(opt1, opt2) {
|
|
return JSON.stringify(opt1) === JSON.stringify(opt2);
|
|
};
|
|
|
|
// A queue for holding all requests from renderer process.
|
|
var requestsQueue = [];
|
|
|
|
ipcMain.on('ATOM_BROWSER_DESKTOP_CAPTURER_GET_SOURCES', function(event, captureWindow, captureScreen, thumbnailSize, id) {
|
|
var request;
|
|
request = {
|
|
id: id,
|
|
options: {
|
|
captureWindow: captureWindow,
|
|
captureScreen: captureScreen,
|
|
thumbnailSize: 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.
|
|
return event.sender.once('destroyed', function() {
|
|
return request.webContents = null;
|
|
});
|
|
});
|
|
|
|
desktopCapturer.emit = function(event, name, sources) {
|
|
// Receiving sources result from main process, now send them back to renderer.
|
|
var handledRequest, i, len, ref, ref1, request, result, source, unhandledRequestsQueue;
|
|
handledRequest = requestsQueue.shift(0);
|
|
result = (function() {
|
|
var i, len, results;
|
|
results = [];
|
|
for (i = 0, len = sources.length; i < len; i++) {
|
|
source = sources[i];
|
|
results.push({
|
|
id: source.id,
|
|
name: source.name,
|
|
thumbnail: source.thumbnail.toDataUrl()
|
|
});
|
|
}
|
|
return results;
|
|
})();
|
|
if ((ref = handledRequest.webContents) != null) {
|
|
ref.send("ATOM_RENDERER_DESKTOP_CAPTURER_RESULT_" + handledRequest.id, result);
|
|
}
|
|
|
|
// Check the queue to see whether there is other same request. If has, handle
|
|
// it for reducing redunplicated `desktopCaptuer.startHandling` calls.
|
|
unhandledRequestsQueue = [];
|
|
for (i = 0, len = requestsQueue.length; i < len; i++) {
|
|
request = requestsQueue[i];
|
|
if (deepEqual(handledRequest.options, request.options)) {
|
|
if ((ref1 = request.webContents) != null) {
|
|
ref1.send("ATOM_RENDERER_DESKTOP_CAPTURER_RESULT_" + 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);
|
|
}
|
|
};
|