雨霖铃
127 lines
3.9 KiB
TypeScript
127 lines
3.9 KiB
TypeScript
import '../.wxt/types/imports.d.ts';
|
|
import { browser } from 'wxt/browser';
|
|
import { downloadHtmlInBackground } from '@/utils/recordUtils.tsx';
|
|
|
|
const events: unknown[] = [];
|
|
|
|
export default defineBackground(() => {
|
|
// 监听扩展安装或更新事件
|
|
browser.runtime.onInstalled.addListener(async ({ reason }) => {
|
|
if (reason === 'install') {
|
|
// 第一次安装扩展时触发
|
|
console.log('Extension installed for the first time');
|
|
} else if (reason === 'update') {
|
|
// 扩展更新时触发
|
|
console.log('Extension updated to a new version');
|
|
}
|
|
|
|
// 获取所有标签页
|
|
const tabs = await browser.tabs.query({});
|
|
|
|
// 过滤不合法或受限制的 URL
|
|
const targetTabs = tabs.filter((tab) => {
|
|
if (!tab.id || !tab.url) return false;
|
|
// 过滤掉浏览器内部页面和不支持注入的协议
|
|
const restrictedProtocols = [
|
|
'chrome:',
|
|
'chrome-extension:',
|
|
'about:',
|
|
'edge:',
|
|
'view-source:',
|
|
];
|
|
return !restrictedProtocols.some((protocol) => tab.url!.startsWith(protocol));
|
|
});
|
|
|
|
const results = await Promise.allSettled(
|
|
targetTabs.map((tab) =>
|
|
browser.scripting
|
|
.executeScript({
|
|
target: { tabId: tab.id! },
|
|
files: ['/content-scripts/content.js'],
|
|
})
|
|
.catch((err) => {
|
|
console.warn(`Failed to inject script into tab ${tab.id}:`, err.message);
|
|
}),
|
|
),
|
|
);
|
|
|
|
const successCount = results.filter((r) => r.status === 'fulfilled').length;
|
|
console.log(
|
|
`Successfully injected content script into ${successCount}/${targetTabs.length} tabs.`,
|
|
);
|
|
});
|
|
|
|
// 监听来自 popup script 的消息
|
|
chrome.runtime.onMessage.addListener((msg, _sender, sendResponse) => {
|
|
if (msg.type === messages.popup.checkStatus) {
|
|
console.log('[bg] checkStatus received');
|
|
sendToActiveTab({ type: messages.content.checkStatus })
|
|
.then((res) => {
|
|
sendResponse(res);
|
|
})
|
|
.catch(() => {
|
|
console.error('Failed to check status in content script');
|
|
sendResponse({ ok: false });
|
|
});
|
|
return true;
|
|
}
|
|
|
|
if (msg.type === messages.popup.from.start) {
|
|
console.log('[bg] startRecording received');
|
|
sendToActiveTab({ type: messages.content.to.startRecording })
|
|
.then(async () => {
|
|
await chrome.runtime.sendMessage({ type: messages.popup.to.started });
|
|
sendResponse({ ok: true });
|
|
})
|
|
.catch(() => {
|
|
console.error('Failed to start recording in content script');
|
|
sendResponse({ ok: false });
|
|
});
|
|
}
|
|
|
|
if (msg.type === messages.popup.from.stop) {
|
|
console.log('[bg] stopRecording received');
|
|
sendToActiveTab({ type: messages.content.to.stopRecording })
|
|
.then(async () => {
|
|
await chrome.runtime.sendMessage({ type: messages.popup.to.stopped });
|
|
downloadHtmlInBackground(events);
|
|
sendResponse({ ok: true });
|
|
})
|
|
.catch(() => {
|
|
sendResponse({ ok: false });
|
|
});
|
|
}
|
|
|
|
return true;
|
|
});
|
|
|
|
// 监听来自 content script 的消息
|
|
chrome.runtime.onMessage.addListener((msg, _sender, sendResponse) => {
|
|
if (msg.type === messages.content.from.saveTrackeEvents) {
|
|
console.log('[bg] saveTrackeEvents received:', msg.payload);
|
|
events.push(msg.payload);
|
|
sendResponse({ ok: true });
|
|
}
|
|
});
|
|
|
|
async function sendToActiveTab(message: {
|
|
type: string;
|
|
data?: unknown;
|
|
activeTabId?: number;
|
|
[key: string]: unknown;
|
|
}) {
|
|
const activeTabs = await chrome.tabs.query({
|
|
active: true,
|
|
currentWindow: true,
|
|
});
|
|
const activeTab = activeTabs[0];
|
|
const sendTo = message.activeTabId || activeTab.id;
|
|
try {
|
|
return await chrome.tabs.sendMessage(sendTo!, message);
|
|
} catch (error) {
|
|
console.error('Error sending message to active tab:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
});
|