61 lines
1.4 KiB
JavaScript
61 lines
1.4 KiB
JavaScript
/* 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
|
|
}
|
|
}
|
|
})
|