diff --git a/lib/browser/chrome-extension.js b/lib/browser/chrome-extension.js index 9a86e2fc34dc..b0045294f38e 100644 --- a/lib/browser/chrome-extension.js +++ b/lib/browser/chrome-extension.js @@ -1,4 +1,4 @@ -const {app, protocol, webContents, BrowserWindow} = require('electron') +const {app, ipcMain, protocol, webContents, BrowserWindow} = require('electron') const renderProcessPreferences = process.atomBinding('render_process_preferences').forAllBrowserWindow() const fs = require('fs') @@ -67,6 +67,7 @@ const startBackgroundPages = function (manifest) { hostname: manifest.hostname, pathname: '_generated_background_page.html' })) + contents.openDevTools() } const removeBackgroundPages = function (manifest) { @@ -76,6 +77,28 @@ const removeBackgroundPages = function (manifest) { delete backgroundPages[manifest.hostname] } +// Handle the chrome.* API messages. +ipcMain.on('CHROME_RUNTIME_CONNECT', function (event, hostname, connectInfo) { + const page = backgroundPages[hostname] + if (!page) { + console.error(`Connect to unkown extension ${hostname}`) + return + } + + event.returnValue = page.webContents.id + page.webContents.sendToAll('CHROME_RUNTIME_ONCONNECT', event.sender.id, hostname, connectInfo) +}) + +ipcMain.on('CHROME_PORT_POSTMESSAGE', function (event, webContentsId, hostname, message) { + const contents = webContents.fromId(webContentsId) + if (!contents) { + console.error(`Sending message to extension ${hostname} with unkown webContentsId ${webContentsId}`) + return + } + + contents.sendToAll(`CHROME_PORT_ONMESSAGE_${hostname}`, message) +}) + // Transfer the content scripts to renderer. const contentScripts = {} diff --git a/lib/renderer/chrome-api.js b/lib/renderer/chrome-api.js index 719066a6fae7..84577a7c7941 100644 --- a/lib/renderer/chrome-api.js +++ b/lib/renderer/chrome-api.js @@ -1,4 +1,68 @@ +const {ipcRenderer} = require('electron') const url = require('url') + +// TODO(zcbenz): Set it to correct value for content scripts. +const currentExtensionId = window.location.hostname + +class Event { + constructor () { + this.listeners = [] + } + + addListener (callback) { + this.listeners.push(callback) + } + + emit (...args) { + for (const listener of this.listeners) { + listener(...args) + } + } +} + +class OnConnect extends Event { + constructor () { + super() + + ipcRenderer.on('CHROME_RUNTIME_ONCONNECT', (event, webContentsId, extensionId, connectInfo) => { + this.emit(new Port(webContentsId, extensionId, connectInfo.name)) + }) + } +} + +class MessageSender { + constructor () { + this.tab = null + this.frameId = null + this.id = null + this.url = null + this.tlsChannelId = null + } +} + +class Port { + constructor (webContentsId, extensionId, name) { + this.webContentsId = webContentsId + this.extensionId = extensionId + + this.name = name + this.onDisconnect = new Event() + this.onMessage = new Event() + this.sender = new MessageSender() + + ipcRenderer.on(`CHROME_PORT_ONMESSAGE_${extensionId}`, (event, message) => { + this.onMessage.emit(message, new MessageSender(), function () {}) + }) + } + + disconnect () { + } + + postMessage (message) { + ipcRenderer.send('CHROME_PORT_POSTMESSAGE', this.webContentsId, this.extensionId, message) + } +} + const chrome = window.chrome = window.chrome || {} chrome.extension = { @@ -6,8 +70,28 @@ chrome.extension = { return url.format({ protocol: window.location.protocol, slashes: true, - hostname: window.location.hostname, + hostname: currentExtensionId, pathname: path }) } } + +chrome.runtime = { + getURL: chrome.extension.getURL, + + onConnect: new OnConnect(), + + connect (...args) { + // Parse the optional args. + let extensionId = currentExtensionId + let connectInfo = {name: ''} + if (args.length === 1) { + connectInfo = args[0] + } else if (args.length === 2) { + [extensionId, connectInfo] = args + } + + const webContentsId = ipcRenderer.sendSync('CHROME_RUNTIME_CONNECT', extensionId, connectInfo) + return new Port(webContentsId, extensionId, connectInfo.name) + } +}