zoushiyang/page-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(ext): sw calling robust

Browse Source
This commit is contained in:
Simon
2026-01-28 14:13:28 +08:00
parent 0cad8a1159
commit c80de93d3b
3 changed files with 5 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/extension/src/agent/RemotePageController.content.ts
View File

@@ -53,6 +53,10 @@ export function initPageController() {
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
if (message.type !== 'PAGE_CONTROL') {
sendResponse({
success: false,
error: `[RemotePageController.ContentScript]: Invalid message type: ${message.type}`,
})
return
}

5
packages/extension/src/agent/TabsController.background.ts
View File

@@ -8,11 +8,6 @@ export function handleTabControlMessage(
sender: chrome.runtime.MessageSender,
sendResponse: (response: unknown) => void
): boolean {
if (message.type !== 'TAB_CONTROL') {
sendResponse({ error: 'Invalid message type' })
return false
}
const { action, payload } = message
switch (action as TabAction) {

31
packages/extension/src/entrypoints/background.ts
View File

@@ -1,43 +1,14 @@
import { handlePageControlMessage } from '@/agent/RemotePageController.background'
import { handleTabControlMessage } from '@/agent/TabsController.background'
function handleUtilsMessage(
message: { type: 'UTILS'; action: string; payload: any },
sender: chrome.runtime.MessageSender,
sendResponse: (response: unknown) => void
): boolean {
const { action, payload } = message
switch (action) {
case 'get_tab_info': {
chrome.tabs
.get(payload.tabId)
.then((tab) => {
const result = { title: tab.title || '', url: tab.url || '' }
sendResponse(result)
})
.catch((error) => {
sendResponse({ error: error instanceof Error ? error.message : String(error) })
})
return true // async response
}
default:
sendResponse({ error: `Unknown TAB_CONTROL action: ${action}` })
return false
}
}
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
if (message.type === 'TAB_CONTROL') {
return handleTabControlMessage(message, sender, sendResponse)
} else if (message.type === 'PAGE_CONTROL') {
return handlePageControlMessage(message, sender, sendResponse)
} else if (message.type !== 'UTILS') {
return handleUtilsMessage(message, sender, sendResponse)
} else {
sendResponse({ error: 'Unknown message type' })
return false
return
}
})