electron/lib/browser/desktop-capturer.js

'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)
  }
}