chore(ext): refine MultiPageAgent api and clean up

packages/extension/src/agent/MultiPageAgent.ts

@@ -8,8 +8,7 @@ import { createTabTools } from './tabTools'
 export class MultiPageAgent extends PageAgentCore {
 	constructor(config: Omit<PageAgentConfig, 'pageController'>) {
 		const tabsController = new TabsController()
-		const pageController = new RemotePageController()
+		const pageController = new RemotePageController(tabsController)
-		pageController.tabsController = tabsController
 		const customTools = createTabTools(tabsController)
 
 		super({

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

@@ -3,11 +3,6 @@
  * - redirect messages from RemotePageController(Agent, extension pages) to ContentScript
  */
 
-// chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
-// 	if (message.type !== 'PAGE_CONTROL') {
-// 		return
-// 	}
-
 export function handlePageControlMessage(
 	message: { type: 'PAGE_CONTROL'; action: string; payload: any; targetTabId: number },
 	sender: chrome.runtime.MessageSender,

packages/extension/src/agent/RemotePageController.ts

@@ -7,9 +7,12 @@ import type { TabsController } from './TabsController'
  * - live in the agent env (extension page or content script)
  * - communicates with remote PageController via sw
  */
-
 export class RemotePageController {
-	tabsController!: TabsController
+	private tabsController: TabsController
+
+	constructor(tabsController: TabsController) {
+		this.tabsController = tabsController
+	}
 
 	get currentTabId(): number | null {
 		return this.tabsController.currentTabId
@@ -21,10 +24,6 @@ export class RemotePageController {
 		return url || ''
 	}
 
-	get currentTabUrl(): Promise<string> {
-		return this.getCurrentUrl()
-	}
-
 	async getCurrentTitle(): Promise<string> {
 		if (!this.currentTabId) return ''
 		const { title } = await this.tabsController.getTabInfo(this.currentTabId)
@@ -127,12 +126,11 @@ export class RemotePageController {
 		return this.remoteCallDomAction('execute_javascript', args)
 	}
 
-	/** @note Mask visibility is managed by content script via storage polling. */
+	/** @note Managed by content script via storage polling. */
 	async showMask(): Promise<void> {}
-	/** @note Mask visibility is managed by content script via storage polling. */
+	/** @note Managed by content script via storage polling. */
 	async hideMask(): Promise<void> {}
-
+	/** @note Managed by content script via storage polling. */
-	// dispose
 	dispose(): void {}
 
 	private async preCheck() {
@@ -160,6 +158,10 @@ export class RemotePageController {
 			payload,
 		})
 	}
+
+	private get currentTabUrl(): Promise<string> {
+		return this.getCurrentUrl()
+	}
 }
 
 interface DomActionReturn {
@@ -170,7 +172,7 @@ interface DomActionReturn {
 /**
  * Check if a URL can run content scripts.
  */
-export function isContentScriptAllowed(url: string | undefined): boolean {
+function isContentScriptAllowed(url: string | undefined): boolean {
 	if (!url) return false
 
 	const restrictedPatterns = [

packages/extension/src/agent/constants.ts

@@ -1,9 +1,4 @@
-// Demo build (auto-init with demo LLM, for quick testing)
+// Demo LLM for testing
-export const CDN_DEMO_URL = 'https://cdn.jsdelivr.net/npm/page-agent/dist/iife/page-agent.demo.js'
-export const CDN_DEMO_CN_URL =
-	'https://registry.npmmirror.com/page-agent/latest/files/dist/iife/page-agent.demo.js'
-
-// Demo LLM for website testing
 export const DEMO_MODEL = 'PAGE-AGENT-FREE-TESTING-RANDOM'
 export const DEMO_BASE_URL =
 	'https://hwcxiuzfylggtcktqgij.supabase.co/functions/v1/llm-testing-proxy'

packages/extension/src/agent/useAgent.ts

@@ -9,8 +9,6 @@ import { DEMO_API_KEY, DEMO_BASE_URL, DEMO_MODEL } from '@/agent/constants'
 
 import { MultiPageAgent } from './MultiPageAgent'
 
-// import { type AgentController, type LLMConfig, getAgentController } from './old/AgentController'
-
 export interface UseAgentResult {
 	status: AgentStatus
 	history: HistoricalEvent[]