From 89d6835a99d52fc1a4f368e9c39cc1f69b96aabc Mon Sep 17 00:00:00 2001
From: Simon <10131203+gaomeng1900@users.noreply.github.com>
Date: Mon, 26 Jan 2026 19:25:00 +0800
Subject: [PATCH] chore(ext): highlight cleanup with mask

---
 .../extension/src/agent/RemotePageController.ts    | 14 +-------------
 1 file changed, 1 insertion(+), 13 deletions(-)

diff --git a/packages/extension/src/agent/RemotePageController.ts b/packages/extension/src/agent/RemotePageController.ts
index a0c6990..77437e8 100644
--- a/packages/extension/src/agent/RemotePageController.ts
+++ b/packages/extension/src/agent/RemotePageController.ts
@@ -84,19 +84,6 @@ export class RemotePageController {
 
 		console.debug(`${DEBUG_PREFIX} setTargetTab: ${previousTabId} → ${tabId}`)
 
-		// Clean up old tab highlights only (mask is controlled by AgentController)
-		if (previousTabId && previousTabId !== tabId && previousRpc) {
-			console.debug(`${DEBUG_PREFIX} Cleaning up highlights on previous tab ${previousTabId}`)
-			try {
-				await previousRpc.cleanUpHighlights()
-			} catch (e) {
-				console.debug(
-					`${DEBUG_PREFIX} cleanUpHighlights on tab ${previousTabId} failed (ignored):`,
-					e
-				)
-			}
-		}
-
 		// Get tab info to check URL
 		const tab = await chrome.tabs.get(tabId)
 		const tabUrl = tab.url
@@ -284,6 +271,7 @@ export class RemotePageController {
 	 */
 	async hideMask(): Promise<void> {
 		if (!this.rpc) return
+		await this.cleanUpHighlights()
 		return this.rpc.hideMask()
 	}