refactor: Port chrome-api to TypeScript (#17014)

This commit is contained in:
Felix Rieseberg 2019-02-19 13:09:44 +00:00 committed by Samuel Attard
parent 7e7abc28f5
commit 2223114f20
3 changed files with 92 additions and 30 deletions

View file

@ -63,7 +63,7 @@ filenames = {
"lib/common/reset-search-paths.ts", "lib/common/reset-search-paths.ts",
"lib/common/web-view-methods.js", "lib/common/web-view-methods.js",
"lib/renderer/callbacks-registry.js", "lib/renderer/callbacks-registry.js",
"lib/renderer/chrome-api.js", "lib/renderer/chrome-api.ts",
"lib/renderer/content-scripts-injector.ts", "lib/renderer/content-scripts-injector.ts",
"lib/renderer/init.js", "lib/renderer/init.js",
"lib/renderer/inspector.js", "lib/renderer/inspector.js",

View file

@ -1,18 +1,24 @@
'use strict' import { ipcRendererInternal } from '@electron/internal/renderer/ipc-renderer-internal'
import * as ipcRendererUtils from '@electron/internal/renderer/ipc-renderer-internal-utils'
import * as url from 'url'
const { ipcRendererInternal } = require('@electron/internal/renderer/ipc-renderer-internal') // Todo: Import once extensions have been turned into TypeScript
const ipcRendererUtils = require('@electron/internal/renderer/ipc-renderer-internal-utils')
const Event = require('@electron/internal/renderer/extensions/event') const Event = require('@electron/internal/renderer/extensions/event')
const url = require('url')
class Tab { class Tab {
constructor (tabId) { public id: number
constructor (tabId: number) {
this.id = tabId this.id = tabId
} }
} }
class MessageSender { class MessageSender {
constructor (tabId, extensionId) { public tab: Tab | null
public id: string
public url: string
constructor (tabId: number, extensionId: string) {
this.tab = tabId ? new Tab(tabId) : null this.tab = tabId ? new Tab(tabId) : null
this.id = extensionId this.id = extensionId
this.url = `chrome-extension://${extensionId}` this.url = `chrome-extension://${extensionId}`
@ -20,12 +26,12 @@ class MessageSender {
} }
class Port { class Port {
constructor (tabId, portId, extensionId, name) { public disconnected: boolean = false
this.tabId = tabId public onDisconnect = new Event()
this.portId = portId public onMessage = new Event()
this.disconnected = false public sender: MessageSender
this.name = name constructor (public tabId: number, public portId: number, extensionId: string, public name: string) {
this.onDisconnect = new Event() this.onDisconnect = new Event()
this.onMessage = new Event() this.onMessage = new Event()
this.sender = new MessageSender(tabId, extensionId) this.sender = new MessageSender(tabId, extensionId)
@ -33,7 +39,10 @@ class Port {
ipcRendererInternal.once(`CHROME_PORT_DISCONNECT_${portId}`, () => { ipcRendererInternal.once(`CHROME_PORT_DISCONNECT_${portId}`, () => {
this._onDisconnect() this._onDisconnect()
}) })
ipcRendererInternal.on(`CHROME_PORT_POSTMESSAGE_${portId}`, (event, message) => {
ipcRendererInternal.on(`CHROME_PORT_POSTMESSAGE_${portId}`, (
_event: Electron.Event, message: string
) => {
const sendResponse = function () { console.error('sendResponse is not implemented') } const sendResponse = function () { console.error('sendResponse is not implemented') }
this.onMessage.emit(message, this.sender, sendResponse) this.onMessage.emit(message, this.sender, sendResponse)
}) })
@ -46,7 +55,7 @@ class Port {
this._onDisconnect() this._onDisconnect()
} }
postMessage (message) { postMessage (message: string) {
ipcRendererInternal.sendToAll(this.tabId, `CHROME_PORT_POSTMESSAGE_${this.portId}`, message) ipcRendererInternal.sendToAll(this.tabId, `CHROME_PORT_POSTMESSAGE_${this.portId}`, message)
} }
@ -58,32 +67,38 @@ class Port {
} }
// Inject chrome API to the |context| // Inject chrome API to the |context|
exports.injectTo = function (extensionId, isBackgroundPage, context) { export function injectTo (extensionId: string, isBackgroundPage: boolean, context: any) {
const chrome = context.chrome = context.chrome || {} const chrome = context.chrome = context.chrome || {}
let originResultID = 1 let originResultID = 1
ipcRendererInternal.on(`CHROME_RUNTIME_ONCONNECT_${extensionId}`, (event, tabId, portId, connectInfo) => { ipcRendererInternal.on(`CHROME_RUNTIME_ONCONNECT_${extensionId}`, (
_event: Electron.Event, tabId: number, portId: number, connectInfo: { name: string }
) => {
chrome.runtime.onConnect.emit(new Port(tabId, portId, extensionId, connectInfo.name)) chrome.runtime.onConnect.emit(new Port(tabId, portId, extensionId, connectInfo.name))
}) }
)
ipcRendererInternal.on(`CHROME_RUNTIME_ONMESSAGE_${extensionId}`, (event, tabId, message, resultID) => { ipcRendererInternal.on(`CHROME_RUNTIME_ONMESSAGE_${extensionId}`, (
chrome.runtime.onMessage.emit(message, new MessageSender(tabId, extensionId), (messageResult) => { _event: Electron.Event, tabId: number, message: string, resultID: number
) => {
chrome.runtime.onMessage.emit(message, new MessageSender(tabId, extensionId), (messageResult: any) => {
ipcRendererInternal.send(`CHROME_RUNTIME_ONMESSAGE_RESULT_${resultID}`, messageResult) ipcRendererInternal.send(`CHROME_RUNTIME_ONMESSAGE_RESULT_${resultID}`, messageResult)
}) })
}) })
ipcRendererInternal.on('CHROME_TABS_ONCREATED', (event, tabId) => { ipcRendererInternal.on('CHROME_TABS_ONCREATED', (_event: Electron.Event, tabId: number) => {
chrome.tabs.onCreated.emit(new Tab(tabId)) chrome.tabs.onCreated.emit(new Tab(tabId))
}) })
ipcRendererInternal.on('CHROME_TABS_ONREMOVED', (event, tabId) => { ipcRendererInternal.on('CHROME_TABS_ONREMOVED', (_event: Electron.Event, tabId: number) => {
chrome.tabs.onRemoved.emit(tabId) chrome.tabs.onRemoved.emit(tabId)
}) })
chrome.runtime = { chrome.runtime = {
id: extensionId, id: extensionId,
getURL: function (path) { // https://developer.chrome.com/extensions/runtime#method-getURL
getURL: function (path: string) {
return url.format({ return url.format({
protocol: 'chrome-extension', protocol: 'chrome-extension',
slashes: true, slashes: true,
@ -92,12 +107,14 @@ exports.injectTo = function (extensionId, isBackgroundPage, context) {
}) })
}, },
// https://developer.chrome.com/extensions/runtime#method-getManifest
getManifest: function () { getManifest: function () {
const manifest = ipcRendererUtils.invokeSync('CHROME_EXTENSION_MANIFEST', extensionId) const manifest = ipcRendererUtils.invokeSync('CHROME_EXTENSION_MANIFEST', extensionId)
return manifest return manifest
}, },
connect (...args) { // https://developer.chrome.com/extensions/runtime#method-connect
connect (...args: Array<any>) {
if (isBackgroundPage) { if (isBackgroundPage) {
console.error('chrome.runtime.connect is not supported in background page') console.error('chrome.runtime.connect is not supported in background page')
return return
@ -116,7 +133,8 @@ exports.injectTo = function (extensionId, isBackgroundPage, context) {
return new Port(tabId, portId, extensionId, connectInfo.name) return new Port(tabId, portId, extensionId, connectInfo.name)
}, },
sendMessage (...args) { // https://developer.chrome.com/extensions/runtime#method-sendMessage
sendMessage (...args: Array<any>) {
if (isBackgroundPage) { if (isBackgroundPage) {
console.error('chrome.runtime.sendMessage is not supported in background page') console.error('chrome.runtime.sendMessage is not supported in background page')
return return
@ -130,14 +148,19 @@ exports.injectTo = function (extensionId, isBackgroundPage, context) {
} else if (args.length === 2) { } else if (args.length === 2) {
// A case of not provide extension-id: (message, responseCallback) // A case of not provide extension-id: (message, responseCallback)
if (typeof args[1] === 'function') { if (typeof args[1] === 'function') {
ipcRendererInternal.once(`CHROME_RUNTIME_SENDMESSAGE_RESULT_${originResultID}`, (event, result) => args[1](result)) ipcRendererInternal.once(`CHROME_RUNTIME_SENDMESSAGE_RESULT_${originResultID}`,
(_event: Electron.Event, result: any) => args[1](result)
)
message = args[0] message = args[0]
} else { } else {
[targetExtensionId, message] = args [targetExtensionId, message] = args
} }
} else { } else {
console.error('options is not supported') console.error('options is not supported')
ipcRendererInternal.once(`CHROME_RUNTIME_SENDMESSAGE_RESULT_${originResultID}`, (event, result) => args[2](result)) ipcRendererInternal.once(`CHROME_RUNTIME_SENDMESSAGE_RESULT_${originResultID}`,
(event: Electron.Event, result: any) => args[2](result)
)
} }
ipcRendererInternal.send('CHROME_RUNTIME_SENDMESSAGE', targetExtensionId, message, originResultID) ipcRendererInternal.send('CHROME_RUNTIME_SENDMESSAGE', targetExtensionId, message, originResultID)
@ -150,17 +173,32 @@ exports.injectTo = function (extensionId, isBackgroundPage, context) {
} }
chrome.tabs = { chrome.tabs = {
executeScript (tabId, details, resultCallback) { // https://developer.chrome.com/extensions/tabs#method-executeScript
executeScript (
tabId: number,
details: Chrome.Tabs.ExecuteScriptDetails,
resultCallback: Chrome.Tabs.ExecuteScriptCallback
) {
if (resultCallback) { if (resultCallback) {
ipcRendererInternal.once(`CHROME_TABS_EXECUTESCRIPT_RESULT_${originResultID}`, (event, result) => resultCallback([result])) ipcRendererInternal.once(`CHROME_TABS_EXECUTESCRIPT_RESULT_${originResultID}`,
(_event: Electron.Event, result: any) => resultCallback([result])
)
} }
ipcRendererInternal.send('CHROME_TABS_EXECUTESCRIPT', originResultID, tabId, extensionId, details) ipcRendererInternal.send('CHROME_TABS_EXECUTESCRIPT', originResultID, tabId, extensionId, details)
originResultID++ originResultID++
}, },
sendMessage (tabId, message, options, responseCallback) { // https://developer.chrome.com/extensions/tabs#method-sendMessage
sendMessage (
tabId: number,
message: any,
_options: Chrome.Tabs.SendMessageDetails,
responseCallback: Chrome.Tabs.SendMessageCallback
) {
if (responseCallback) { if (responseCallback) {
ipcRendererInternal.once(`CHROME_TABS_SEND_MESSAGE_RESULT_${originResultID}`, (event, result) => responseCallback(result)) ipcRendererInternal.once(`CHROME_TABS_SEND_MESSAGE_RESULT_${originResultID}`,
(_event: Electron.Event, result: any) => responseCallback(result)
)
} }
ipcRendererInternal.send('CHROME_TABS_SEND_MESSAGE', tabId, extensionId, isBackgroundPage, message, originResultID) ipcRendererInternal.send('CHROME_TABS_SEND_MESSAGE', tabId, extensionId, isBackgroundPage, message, originResultID)
originResultID++ originResultID++

View file

@ -79,3 +79,27 @@ declare namespace ElectronInternal {
once(channel: string, listener: (event: IpcMainInternalEvent, ...args: any[]) => void): this; once(channel: string, listener: (event: IpcMainInternalEvent, ...args: any[]) => void): this;
} }
} }
declare namespace Chrome {
namespace Tabs {
// https://developer.chrome.com/extensions/tabs#method-executeScript
interface ExecuteScriptDetails {
code?: string;
file?: string;
allFrames?: boolean;
frameId?: number;
matchAboutBlank?: boolean;
runAt?: 'document-start' | 'document-end' | 'document_idle';
cssOrigin: 'author' | 'user';
}
type ExecuteScriptCallback = (result: Array<any>) => void;
// https://developer.chrome.com/extensions/tabs#method-sendMessage
interface SendMessageDetails {
frameId?: number;
}
type SendMessageCallback = (result: any) => void;
}
}