electron/lib/browser/chrome-devtools.js

'use strict'

const { dialog, Menu } = require('electron')
const fs = require('fs')
const url = require('url')
const util = require('util')

const ipcMainUtils = require('@electron/internal/browser/ipc-main-internal-utils')

const readFile = util.promisify(fs.readFile)

const convertToMenuTemplate = function (event, items) {
  return items.map(function (item) {
    const transformed = item.type === 'subMenu' ? {
      type: 'submenu',
      label: item.label,
      enabled: item.enabled,
      submenu: convertToMenuTemplate(event, item.subItems)
    } : item.type === 'separator' ? {
      type: 'separator'
    } : item.type === 'checkbox' ? {
      type: 'checkbox',
      label: item.label,
      enabled: item.enabled,
      checked: item.checked
    } : {
      type: 'normal',
      label: item.label,
      enabled: item.enabled
    }

    if (item.id != null) {
      transformed.click = function () {
        event._replyInternal('ELECTRON_INSPECTOR_CONTEXT_MENU_CLICK', item.id)
      }
    }

    return transformed
  })
}

const getEditMenuItems = function () {
  return [
    { role: 'undo' },
    { role: 'redo' },
    { type: 'separator' },
    { role: 'cut' },
    { role: 'copy' },
    { role: 'paste' },
    { role: 'pasteAndMatchStyle' },
    { role: 'delete' },
    { role: 'selectAll' }
  ]
}

const isChromeDevTools = function (pageURL) {
  const { protocol } = url.parse(pageURL)
  return protocol === 'chrome-devtools:'
}

const assertChromeDevTools = function (contents, api) {
  const pageURL = contents._getURL()
  if (!isChromeDevTools(pageURL)) {
    console.error(`Blocked ${pageURL} from calling ${api}`)
    throw new Error(`Blocked ${api}`)
  }
}

ipcMainUtils.handle('ELECTRON_INSPECTOR_CONTEXT_MENU', function (event, items, isEditMenu) {
  assertChromeDevTools(event.sender, 'window.InspectorFrontendHost.showContextMenuAtPoint()')

  const template = isEditMenu ? getEditMenuItems() : convertToMenuTemplate(event, items)
  const menu = Menu.buildFromTemplate(template)
  const window = event.sender.getOwnerBrowserWindow()

  menu.once('menu-will-close', () => {
    // menu-will-close is emitted before the click handler, but needs to be sent after.
    // otherwise, DevToolsAPI.contextMenuCleared() would be called before
    // DevToolsAPI.contextMenuItemSelected(id) and the menu will not work properly.
    setTimeout(() => {
      event._replyInternal('ELECTRON_INSPECTOR_CONTEXT_MENU_CLOSE')
    })
  })

  menu.popup({ window })
})

ipcMainUtils.handle('ELECTRON_INSPECTOR_SELECT_FILE', async function (event) {
  assertChromeDevTools(event.sender, 'window.UI.createFileSelectorElement()')

  const result = await dialog.showOpenDialog({})
  if (result.canceled) return []

  const path = result.filePaths[0]
  const data = await readFile(path)

  return [path, data]
})

ipcMainUtils.handle('ELECTRON_INSPECTOR_CONFIRM', function (event, message, title) {
  return new Promise((resolve, reject) => {
    assertChromeDevTools(event.sender, 'window.confirm()')

    if (message == null) message = ''
    if (title == null) title = ''

    const options = {
      message: `${message}`,
      title: `${title}`,
      buttons: ['OK', 'Cancel'],
      cancelId: 1
    }
    const window = event.sender.getOwnerBrowserWindow()
    dialog.showMessageBox(window, options, (response) => {
      resolve(response === 0)
    })
  })
})