electron/lib/browser/chrome-extension.js

436 lines
14 KiB
JavaScript
Raw Normal View History

'use strict'
const { app, webContents, BrowserWindow } = require('electron')
2018-09-13 16:10:51 +00:00
const { getAllWebContents } = process.atomBinding('web_contents')
const renderProcessPreferences = process.atomBinding('render_process_preferences').forAllWebContents()
const ipcMain = require('@electron/internal/browser/ipc-main-internal')
2016-05-27 00:47:37 +00:00
2018-09-13 16:10:51 +00:00
const { Buffer } = require('buffer')
const fs = require('fs')
const path = require('path')
const url = require('url')
2016-01-12 02:40:23 +00:00
// Mapping between extensionId(hostname) and manifest.
2018-09-13 16:10:51 +00:00
const manifestMap = {} // extensionId => manifest
const manifestNameMap = {} // name => manifest
const devToolsExtensionNames = new Set()
2016-01-12 02:40:23 +00:00
2016-05-29 02:57:20 +00:00
const generateExtensionIdFromName = function (name) {
return name.replace(/[\W_]+/g, '-').toLowerCase()
}
2016-01-12 02:40:23 +00:00
const isWindowOrWebView = function (webContents) {
const type = webContents.getType()
return type === 'window' || type === 'webview'
}
// Create or get manifest object from |srcDirectory|.
2016-05-26 07:57:23 +00:00
const getManifestFromPath = function (srcDirectory) {
let manifest
let manifestContent
try {
manifestContent = fs.readFileSync(path.join(srcDirectory, 'manifest.json'))
} catch (readError) {
console.warn(`Reading ${path.join(srcDirectory, 'manifest.json')} failed.`)
console.warn(readError.stack || readError)
throw readError
}
try {
manifest = JSON.parse(manifestContent)
} catch (parseError) {
console.warn(`Parsing ${path.join(srcDirectory, 'manifest.json')} failed.`)
console.warn(parseError.stack || parseError)
throw parseError
}
if (!manifestNameMap[manifest.name]) {
2016-05-29 02:57:20 +00:00
const extensionId = generateExtensionIdFromName(manifest.name)
manifestMap[extensionId] = manifestNameMap[manifest.name] = manifest
2016-05-26 09:58:18 +00:00
Object.assign(manifest, {
srcDirectory: srcDirectory,
extensionId: extensionId,
2016-05-26 09:58:18 +00:00
// We can not use 'file://' directly because all resources in the extension
// will be treated as relative to the root in Chrome.
startPage: url.format({
protocol: 'chrome-extension',
slashes: true,
hostname: extensionId,
2016-05-26 09:58:18 +00:00
pathname: manifest.devtools_page
})
})
2016-05-26 07:57:23 +00:00
return manifest
} else if (manifest && manifest.name) {
2016-06-09 17:08:21 +00:00
console.warn(`Attempted to load extension "${manifest.name}" that has already been loaded.`)
return manifest
2016-01-12 02:40:23 +00:00
}
}
2016-01-12 02:40:23 +00:00
2016-05-26 09:58:18 +00:00
// Manage the background pages.
2016-05-26 22:43:23 +00:00
const backgroundPages = {}
2016-05-26 09:58:18 +00:00
const startBackgroundPages = function (manifest) {
if (backgroundPages[manifest.extensionId] || !manifest.background) return
2016-05-26 09:58:18 +00:00
let html
let name
if (manifest.background.page) {
name = manifest.background.page
html = fs.readFileSync(path.join(manifest.srcDirectory, manifest.background.page))
} else {
name = '_generated_background_page.html'
const scripts = manifest.background.scripts.map((name) => {
return `<script src="${name}"></script>`
}).join('')
html = Buffer.from(`<html><body>${scripts}</body></html>`)
}
2016-05-26 09:58:18 +00:00
const contents = webContents.create({
partition: 'persist:__chrome_extension',
isBackgroundPage: true,
commandLineSwitches: ['--background-page']
})
backgroundPages[manifest.extensionId] = { html: html, webContents: contents, name: name }
2016-05-26 09:58:18 +00:00
contents.loadURL(url.format({
protocol: 'chrome-extension',
slashes: true,
hostname: manifest.extensionId,
pathname: name
2016-05-26 09:58:18 +00:00
}))
}
2016-05-27 00:55:59 +00:00
const removeBackgroundPages = function (manifest) {
if (!backgroundPages[manifest.extensionId]) return
2016-05-27 00:55:59 +00:00
backgroundPages[manifest.extensionId].webContents.destroy()
delete backgroundPages[manifest.extensionId]
2016-05-27 00:55:59 +00:00
}
const sendToBackgroundPages = function (...args) {
for (const page of Object.values(backgroundPages)) {
page.webContents._sendInternalToAll(...args)
}
}
// Dispatch web contents events to Chrome APIs
const hookWebContentsEvents = function (webContents) {
const tabId = webContents.id
sendToBackgroundPages('CHROME_TABS_ONCREATED')
webContents.on('will-navigate', (event, url) => {
sendToBackgroundPages('CHROME_WEBNAVIGATION_ONBEFORENAVIGATE', {
frameId: 0,
parentFrameId: -1,
processId: webContents.getProcessId(),
tabId: tabId,
timeStamp: Date.now(),
url: url
})
})
webContents.on('did-navigate', (event, url) => {
sendToBackgroundPages('CHROME_WEBNAVIGATION_ONCOMPLETED', {
frameId: 0,
parentFrameId: -1,
processId: webContents.getProcessId(),
tabId: tabId,
timeStamp: Date.now(),
url: url
})
})
webContents.once('destroyed', () => {
sendToBackgroundPages('CHROME_TABS_ONREMOVED', tabId)
})
}
2016-05-28 01:26:41 +00:00
// Handle the chrome.* API messages.
2016-05-28 03:07:08 +00:00
let nextId = 0
ipcMain.on('CHROME_RUNTIME_CONNECT', function (event, extensionId, connectInfo) {
const page = backgroundPages[extensionId]
2016-05-28 01:26:41 +00:00
if (!page) {
2016-06-14 22:39:07 +00:00
console.error(`Connect to unknown extension ${extensionId}`)
2016-05-28 01:26:41 +00:00
return
}
2016-05-28 03:07:08 +00:00
const portId = ++nextId
2018-09-13 16:10:51 +00:00
event.returnValue = { tabId: page.webContents.id, portId: portId }
2016-05-28 03:07:08 +00:00
event.sender.once('render-view-deleted', () => {
if (page.webContents.isDestroyed()) return
page.webContents._sendInternalToAll(`CHROME_PORT_DISCONNECT_${portId}`)
2016-05-28 03:07:08 +00:00
})
page.webContents._sendInternalToAll(`CHROME_RUNTIME_ONCONNECT_${extensionId}`, event.sender.id, portId, connectInfo)
2016-05-28 03:07:08 +00:00
})
2016-06-07 22:34:17 +00:00
ipcMain.on('CHROME_I18N_MANIFEST', function (event, extensionId) {
event.returnValue = manifestMap[extensionId]
})
let resultID = 1
ipcMain.on('CHROME_RUNTIME_SENDMESSAGE', function (event, extensionId, message, originResultID) {
2016-05-28 12:23:43 +00:00
const page = backgroundPages[extensionId]
if (!page) {
2016-06-14 22:39:07 +00:00
console.error(`Connect to unknown extension ${extensionId}`)
2016-05-28 12:23:43 +00:00
return
}
page.webContents._sendInternalToAll(`CHROME_RUNTIME_ONMESSAGE_${extensionId}`, event.sender.id, message, resultID)
ipcMain.once(`CHROME_RUNTIME_ONMESSAGE_RESULT_${resultID}`, (event, result) => {
event.sender._sendInternal(`CHROME_RUNTIME_SENDMESSAGE_RESULT_${originResultID}`, result)
})
resultID++
2016-05-28 12:23:43 +00:00
})
ipcMain.on('CHROME_TABS_SEND_MESSAGE', function (event, tabId, extensionId, isBackgroundPage, message, originResultID) {
const contents = webContents.fromId(tabId)
if (!contents) {
2016-06-14 22:39:07 +00:00
console.error(`Sending message to unknown tab ${tabId}`)
return
}
const senderTabId = isBackgroundPage ? null : event.sender.id
contents._sendInternalToAll(`CHROME_RUNTIME_ONMESSAGE_${extensionId}`, senderTabId, message, resultID)
ipcMain.once(`CHROME_RUNTIME_ONMESSAGE_RESULT_${resultID}`, (event, result) => {
event.sender._sendInternal(`CHROME_TABS_SEND_MESSAGE_RESULT_${originResultID}`, result)
})
resultID++
})
ipcMain.on('CHROME_TABS_EXECUTESCRIPT', function (event, requestId, tabId, extensionId, details) {
const contents = webContents.fromId(tabId)
2016-05-28 07:41:12 +00:00
if (!contents) {
2016-06-14 22:39:07 +00:00
console.error(`Sending message to unknown tab ${tabId}`)
2016-05-28 07:41:12 +00:00
return
}
let code, url
if (details.file) {
const manifest = manifestMap[extensionId]
2016-05-28 07:41:12 +00:00
code = String(fs.readFileSync(path.join(manifest.srcDirectory, details.file)))
url = `chrome-extension://${extensionId}${details.file}`
2016-05-28 07:41:12 +00:00
} else {
code = details.code
url = `chrome-extension://${extensionId}/${String(Math.random()).substr(2, 8)}.js`
2016-05-28 07:41:12 +00:00
}
contents._sendInternal('CHROME_TABS_EXECUTESCRIPT', event.sender.id, requestId, extensionId, url, code)
2016-05-28 07:41:12 +00:00
})
2016-05-27 00:47:37 +00:00
// Transfer the content scripts to renderer.
const contentScripts = {}
const injectContentScripts = function (manifest) {
if (contentScripts[manifest.name] || !manifest.content_scripts) return
const readArrayOfFiles = function (relativePath) {
2016-05-28 06:37:44 +00:00
return {
url: `chrome-extension://${manifest.extensionId}/${relativePath}`,
2016-05-28 06:37:44 +00:00
code: String(fs.readFileSync(path.join(manifest.srcDirectory, relativePath)))
}
2016-05-27 00:47:37 +00:00
}
const contentScriptToEntry = function (script) {
return {
matches: script.matches,
2017-07-20 18:24:39 +00:00
js: script.js ? script.js.map(readArrayOfFiles) : [],
css: script.css ? script.css.map(readArrayOfFiles) : [],
runAt: script.run_at || 'document_idle'
2016-05-27 00:47:37 +00:00
}
}
try {
const entry = {
extensionId: manifest.extensionId,
2016-05-27 00:47:37 +00:00
contentScripts: manifest.content_scripts.map(contentScriptToEntry)
}
contentScripts[manifest.name] = renderProcessPreferences.addEntry(entry)
} catch (e) {
console.error('Failed to read content scripts', e)
}
}
2016-05-27 00:55:59 +00:00
const removeContentScripts = function (manifest) {
if (!contentScripts[manifest.name]) return
renderProcessPreferences.removeEntry(contentScripts[manifest.name])
delete contentScripts[manifest.name]
}
2016-05-26 07:57:23 +00:00
// Transfer the |manifest| to a format that can be recognized by the
// |DevToolsAPI.addExtensions|.
const manifestToExtensionInfo = function (manifest) {
return {
startPage: manifest.startPage,
srcDirectory: manifest.srcDirectory,
name: manifest.name,
exposeExperimentalAPIs: true
}
}
2016-05-26 09:58:18 +00:00
// Load the extensions for the window.
const loadExtension = function (manifest) {
startBackgroundPages(manifest)
injectContentScripts(manifest)
}
2016-05-26 09:58:18 +00:00
const loadDevToolsExtensions = function (win, manifests) {
if (!win.devToolsWebContents) return
manifests.forEach(loadExtension)
2016-05-26 09:58:18 +00:00
const extensionInfoArray = manifests.map(manifestToExtensionInfo)
extensionInfoArray.forEach((extension) => {
win.devToolsWebContents._grantOriginAccess(extension.startPage)
})
2016-05-26 09:58:18 +00:00
win.devToolsWebContents.executeJavaScript(`DevToolsAPI.addExtensions(${JSON.stringify(extensionInfoArray)})`)
}
2016-06-13 15:59:03 +00:00
app.on('web-contents-created', function (event, webContents) {
if (!isWindowOrWebView(webContents)) return
hookWebContentsEvents(webContents)
webContents.on('devtools-opened', function () {
loadDevToolsExtensions(webContents, Object.values(manifestMap))
})
})
// The chrome-extension: can map a extension URL request to real file path.
const chromeExtensionHandler = function (request, callback) {
const parsed = url.parse(request.url)
if (!parsed.hostname || !parsed.path) return callback()
const manifest = manifestMap[parsed.hostname]
if (!manifest) return callback()
const page = backgroundPages[parsed.hostname]
if (page && parsed.path === `/${page.name}`) {
// Disabled due to false positive in StandardJS
// eslint-disable-next-line standard/no-callback-literal
return callback({
mimeType: 'text/html',
data: page.html
})
}
fs.readFile(path.join(manifest.srcDirectory, parsed.path), function (err, content) {
if (err) {
// Disabled due to false positive in StandardJS
// eslint-disable-next-line standard/no-callback-literal
2018-09-13 16:10:51 +00:00
return callback(-6) // FILE_NOT_FOUND
} else {
return callback(content)
}
})
}
app.on('session-created', function (ses) {
ses.protocol.registerBufferProtocol('chrome-extension', chromeExtensionHandler, function (error) {
if (error) {
console.error(`Unable to register chrome-extension protocol: ${error}`)
}
})
})
// The persistent path of "DevTools Extensions" preference file.
let loadedDevToolsExtensionsPath = null
2016-01-12 02:40:23 +00:00
app.on('will-quit', function () {
2016-01-12 02:40:23 +00:00
try {
const loadedDevToolsExtensions = Array.from(devToolsExtensionNames)
.map(name => manifestNameMap[name].srcDirectory)
if (loadedDevToolsExtensions.length > 0) {
try {
fs.mkdirSync(path.dirname(loadedDevToolsExtensionsPath))
} catch (error) {
// Ignore error
}
fs.writeFileSync(loadedDevToolsExtensionsPath, JSON.stringify(loadedDevToolsExtensions))
} else {
fs.unlinkSync(loadedDevToolsExtensionsPath)
2016-01-12 02:40:23 +00:00
}
2016-01-19 22:49:40 +00:00
} catch (error) {
// Ignore error
2016-01-12 02:40:23 +00:00
}
})
2016-01-12 02:40:23 +00:00
2016-01-14 18:35:29 +00:00
// We can not use protocol or BrowserWindow until app is ready.
app.once('ready', function () {
// The public API to add/remove extensions.
BrowserWindow.addExtension = function (srcDirectory) {
2016-05-26 07:57:23 +00:00
const manifest = getManifestFromPath(srcDirectory)
if (manifest) {
loadExtension(manifest)
2016-06-07 18:02:57 +00:00
for (const webContents of getAllWebContents()) {
if (isWindowOrWebView(webContents)) {
loadDevToolsExtensions(webContents, [manifest])
}
2016-01-12 02:40:23 +00:00
}
2016-05-26 07:57:23 +00:00
return manifest.name
2016-01-12 02:40:23 +00:00
}
}
2016-06-07 16:50:36 +00:00
BrowserWindow.removeExtension = function (name) {
const manifest = manifestNameMap[name]
2016-05-27 00:55:59 +00:00
if (!manifest) return
removeBackgroundPages(manifest)
removeContentScripts(manifest)
delete manifestMap[manifest.extensionId]
delete manifestNameMap[name]
}
BrowserWindow.getExtensions = function () {
2016-06-10 16:24:00 +00:00
const extensions = {}
Object.keys(manifestNameMap).forEach(function (name) {
const manifest = manifestNameMap[name]
2018-09-13 16:10:51 +00:00
extensions[name] = { name: manifest.name, version: manifest.version }
2016-06-10 16:24:00 +00:00
})
return extensions
}
BrowserWindow.addDevToolsExtension = function (srcDirectory) {
const manifestName = BrowserWindow.addExtension(srcDirectory)
if (manifestName) {
devToolsExtensionNames.add(manifestName)
}
return manifestName
}
BrowserWindow.removeDevToolsExtension = function (name) {
BrowserWindow.removeExtension(name)
devToolsExtensionNames.delete(name)
}
BrowserWindow.getDevToolsExtensions = function () {
const extensions = BrowserWindow.getExtensions()
const devExtensions = {}
Array.from(devToolsExtensionNames).forEach(function (name) {
2017-07-05 15:34:04 +00:00
if (!extensions[name]) return
devExtensions[name] = extensions[name]
})
return devExtensions
}
// Load persisted extensions.
loadedDevToolsExtensionsPath = path.join(app.getPath('userData'), 'DevTools Extensions')
try {
const loadedDevToolsExtensions = JSON.parse(fs.readFileSync(loadedDevToolsExtensionsPath))
if (Array.isArray(loadedDevToolsExtensions)) {
for (const srcDirectory of loadedDevToolsExtensions) {
// Start background pages and set content scripts.
BrowserWindow.addDevToolsExtension(srcDirectory)
}
}
} catch (error) {
if (process.env.ELECTRON_ENABLE_LOGGING && error.code !== 'ENOENT') {
console.error('Failed to load browser extensions from directory:', loadedDevToolsExtensionsPath)
console.error(error)
}
}
})