fix(ext): sw calling robust

packages/extension/src/agent/RemotePageController.content.ts

@@ -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
 		}
 

packages/extension/src/agent/TabsController.background.ts

@@ -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) {

packages/extension/src/entrypoints/background.ts

@@ -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
 	}
 })