feat: 新增包文件
This commit is contained in:
60
packages/electron-chrome-extensions/spec/fixtures/rpc/content-script.js
vendored
Normal file
60
packages/electron-chrome-extensions/spec/fixtures/rpc/content-script.js
vendored
Normal file
@@ -0,0 +1,60 @@
|
||||
/* eslint-disable */
|
||||
|
||||
function evalInMainWorld(fn) {
|
||||
const script = document.createElement('script')
|
||||
script.textContent = `((${fn})())`
|
||||
document.documentElement.appendChild(script)
|
||||
}
|
||||
|
||||
function sendIpc(name, ...args) {
|
||||
const jsonArgs = [name, ...args].map((arg) => JSON.stringify(arg))
|
||||
const funcStr = `() => { electronTest.sendIpc(${jsonArgs.join(', ')}) }`
|
||||
evalInMainWorld(funcStr)
|
||||
}
|
||||
|
||||
async function exec(action) {
|
||||
const send = async () => {
|
||||
return new Promise((resolve, reject) => {
|
||||
chrome.runtime.sendMessage(action, (result) => {
|
||||
if (chrome.runtime.lastError) {
|
||||
reject(chrome.runtime.lastError.message)
|
||||
} else {
|
||||
resolve(result)
|
||||
}
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
// Retry logic - the connection doesn't seem to always be available when
|
||||
// attempting to send. This started when upgrading to Electron 22 from 15.
|
||||
let result
|
||||
for (let i = 0; i < 3; i++) {
|
||||
try {
|
||||
result = await send()
|
||||
break
|
||||
} catch (e) {
|
||||
console.error(e)
|
||||
await new Promise((resolve) => setTimeout(resolve, 100)) // sleep
|
||||
}
|
||||
}
|
||||
|
||||
sendIpc('success', result)
|
||||
}
|
||||
|
||||
window.addEventListener('message', (event) => {
|
||||
exec(event.data)
|
||||
})
|
||||
|
||||
evalInMainWorld(() => {
|
||||
window.exec = (json) => window.postMessage(JSON.parse(json))
|
||||
})
|
||||
|
||||
chrome.runtime.onMessage.addListener((message) => {
|
||||
switch (message.type) {
|
||||
case 'send-ipc': {
|
||||
const [name] = message.args
|
||||
sendIpc(name)
|
||||
break
|
||||
}
|
||||
}
|
||||
})
|
||||
Reference in New Issue
Block a user