52 lines
1.6 KiB
JavaScript
52 lines
1.6 KiB
JavaScript
const {runInThisContext} = require('vm')
|
|
|
|
// Check whether pattern matches.
|
|
// https://developer.chrome.com/extensions/match_patterns
|
|
const matchesPattern = function (pattern) {
|
|
if (pattern === '<all_urls>') return true
|
|
|
|
const regexp = new RegExp('^' + pattern.replace(/\*/g, '.*') + '$')
|
|
return location.href.match(regexp)
|
|
}
|
|
|
|
// Run the code with chrome API integrated.
|
|
const runContentScript = function (extensionId, url, code) {
|
|
const chrome = {}
|
|
require('./chrome-api').injectTo(extensionId, chrome)
|
|
const wrapper = `(function (chrome) {\n ${code}\n })`
|
|
const compiledWrapper = runInThisContext(wrapper, {
|
|
filename: url,
|
|
lineOffset: 1,
|
|
displayErrors: true
|
|
})
|
|
compiledWrapper.call(this, chrome)
|
|
}
|
|
|
|
// Run injected scripts.
|
|
// https://developer.chrome.com/extensions/content_scripts
|
|
const injectContentScript = function (script) {
|
|
for (const match of script.matches) {
|
|
if (!matchesPattern(match)) return
|
|
}
|
|
|
|
for (const {url, code} of script.js) {
|
|
const fire = runContentScript.bind(window, script.extensionId, url, code)
|
|
if (script.runAt === 'document_start') {
|
|
process.once('document-start', fire)
|
|
} else if (script.runAt === 'document_end') {
|
|
process.once('document-end', fire)
|
|
} else if (script.runAt === 'document_idle') {
|
|
document.addEventListener('DOMContentLoaded', fire)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Read the renderer process preferences.
|
|
const preferences = process.getRenderProcessPreferences()
|
|
if (preferences) {
|
|
for (const pref of preferences) {
|
|
if (pref.contentScripts) {
|
|
pref.contentScripts.forEach(injectContentScript)
|
|
}
|
|
}
|
|
}
|