Aaron Klinker
43 lines
1.5 KiB
TypeScript
43 lines
1.5 KiB
TypeScript
import definition from 'virtual:user-background';
|
|
import { setupWebSocket } from '../client/utils/setup-web-socket';
|
|
import { logger } from '../client/utils/logger';
|
|
import browser from 'webextension-polyfill';
|
|
import { keepServiceWorkerAlive } from '../client/utils/keep-service-worker-alive';
|
|
import { reloadContentScript } from '../client/utils/reload-content-scripts';
|
|
|
|
if (__COMMAND__ === 'serve') {
|
|
try {
|
|
const ws = setupWebSocket((message) => {
|
|
if (message.event === 'wxt:reload-extension') browser.runtime.reload();
|
|
if (message.event === 'wxt:reload-content-script' && message.data != null)
|
|
reloadContentScript(message.data);
|
|
});
|
|
|
|
if (__MANIFEST_VERSION__ === 3) {
|
|
// Tell the server the background script is loaded and ready to go
|
|
ws.addEventListener('open', () => {
|
|
const msg = { type: 'custom', event: 'wxt:background-initialized' };
|
|
ws.send(JSON.stringify(msg));
|
|
});
|
|
|
|
// Web Socket will disconnect if the service worker is killed
|
|
keepServiceWorkerAlive();
|
|
}
|
|
} catch (err) {
|
|
logger.error('Failed to setup web socket connection with dev server', err);
|
|
}
|
|
}
|
|
|
|
try {
|
|
const res = definition.main();
|
|
// @ts-expect-error: res shouldn't be a promise, but we're checking it anyways
|
|
if (res instanceof Promise) {
|
|
console.warn(
|
|
"The background's main() function return a promise, but it must be synchonous",
|
|
);
|
|
}
|
|
} catch (err) {
|
|
logger.error('The background crashed on startup!');
|
|
throw err;
|
|
}
|