PocAgent + refactor

manifest.json

@@ -1,7 +1,7 @@
 {
   "manifest_version": 3,
   "name": "Agent",
-  "version": "49.0.0.26",
+  "version": "49.0.0.27",
   "description": "Agent",
   "key": "MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAs1zULZz5eE0U8SEjr/R++dlx6WKFj7GbpnBiE1n17gaylMWDlw6uuBJNjcRrSGwOt53Z3PKf2T3g5DtNES8q6rQc11P/y8J8GKhKuqGrtRJyk5iXzcKJk4CHz6leFSMt8CsZY0r0b7wCZ5QuhomTHGQpNWNS0c13xfVqWt4dncfIRj7fMzfTkicq7Mqqx+JcdprLkiVfETvdkMwwEWmSNwQ6nCDzLtTbyyMiGUEBSJs+WlP1fO7LIX0sHesFVxfPhCZ2K4F1biwenbRL+YYD60ogpVppop2ee/W3D211IN1zYxgnhycFv3m8TrzG+MD/IZgcu13u0bHRn3V7IGW1iwIDAQAB",
   "permissions": [
@@ -51,4 +51,4 @@
     "48": "assets/icon48.png",
     "128": "assets/icon128.png"
   }
-}
+}

src/background/handlers/ExecutionHandler.ts

@@ -0,0 +1,216 @@
+import { MessageType, ExecuteQueryMessage, CancelTaskMessage, ResetConversationMessage } from '@/lib/types/messaging'
+import { PortMessage } from '@/lib/runtime/PortMessaging'
+import { ExecutionManager } from '@/lib/execution/ExecutionManager'
+import { Logging } from '@/lib/utils/Logging'
+import { PubSub } from '@/lib/pubsub'
+
+/**
+ * Handles execution-related messages:
+ * - EXECUTE_QUERY: Start a new query execution
+ * - CANCEL_TASK: Cancel running execution
+ * - RESET_CONVERSATION: Reset execution state
+ */
+export class ExecutionHandler {
+  private executionManager: ExecutionManager
+
+  constructor() {
+    this.executionManager = ExecutionManager.getInstance()
+  }
+
+  /**
+   * Handle EXECUTE_QUERY message
+   */
+  async handleExecuteQuery(
+    message: PortMessage,
+    port: chrome.runtime.Port,
+    executionId?: string
+  ): Promise<void> {
+    const payload = message.payload as ExecuteQueryMessage['payload']
+    const { query, tabIds, source, chatMode, metadata } = payload
+
+    // Use executionId from port or generate default
+    const execId = executionId || 'default'
+
+    Logging.log('ExecutionHandler', 
+      `Starting execution ${execId}: "${query}" (mode: ${chatMode ? 'chat' : 'browse'})`)
+
+    // Log metrics
+    Logging.logMetric('query_initiated', {
+      query,
+      source: source || metadata?.source || 'unknown',
+      mode: chatMode ? 'chat' : 'browse',
+      executionMode: metadata?.executionMode || 'dynamic',
+    })
+
+    try {
+      // Get or create execution
+      let execution = this.executionManager.get(execId)
+
+      if (!execution) {
+        // Create new execution
+        execution = this.executionManager.create(execId, {
+          mode: chatMode ? 'chat' : 'browse',
+          tabIds,
+          metadata,
+          debug: false
+        })
+      } else {
+        // If execution exists and is running, cancel it
+        // No need to wait - the new run() will handle it
+        if (execution.isRunning()) {
+          Logging.log('ExecutionHandler', `Cancelling previous task for execution ${execId}`)
+          execution.cancel()
+        }
+      }
+
+      // Run the query
+      await execution.run(query, metadata)
+
+      // Send success response
+      port.postMessage({
+        type: MessageType.WORKFLOW_STATUS,
+        payload: { 
+          status: 'success',
+          executionId: execId
+        },
+        id: message.id
+      })
+
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error)
+      Logging.log('ExecutionHandler', `Error executing query: ${errorMessage}`, 'error')
+
+      // Send error response
+      port.postMessage({
+        type: MessageType.WORKFLOW_STATUS,
+        payload: { 
+          status: 'error',
+          error: errorMessage,
+          executionId: execId
+        },
+        id: message.id
+      })
+    }
+  }
+
+  /**
+   * Handle CANCEL_TASK message
+   */
+  handleCancelTask(
+    message: PortMessage,
+    port: chrome.runtime.Port,
+    executionId?: string
+  ): void {
+    const execId = executionId || 'default'
+
+    Logging.log('ExecutionHandler', `Cancelling execution ${execId}`)
+
+    try {
+      this.executionManager.cancel(execId)
+      Logging.logMetric('task_cancelled', { executionId: execId })
+
+      // Send success response
+      port.postMessage({
+        type: MessageType.WORKFLOW_STATUS,
+        payload: { 
+          status: 'success',
+          message: 'Task cancelled',
+          executionId: execId
+        },
+        id: message.id
+      })
+
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error)
+      Logging.log('ExecutionHandler', `Error cancelling task: ${errorMessage}`, 'error')
+
+      // Send error response
+      port.postMessage({
+        type: MessageType.WORKFLOW_STATUS,
+        payload: { 
+          status: 'error',
+          error: errorMessage,
+          executionId: execId
+        },
+        id: message.id
+      })
+    }
+  }
+
+  /**
+   * Handle RESET_CONVERSATION message
+   */
+  handleResetConversation(
+    message: PortMessage,
+    port: chrome.runtime.Port,
+    executionId?: string
+  ): void {
+    const execId = executionId || 'default'
+
+    Logging.log('ExecutionHandler', `Resetting execution ${execId}`)
+
+    try {
+      this.executionManager.reset(execId)
+      Logging.logMetric('conversation_reset', { executionId: execId })
+
+      // Send success response
+      port.postMessage({
+        type: MessageType.WORKFLOW_STATUS,
+        payload: { 
+          status: 'success',
+          message: 'Conversation reset',
+          executionId: execId
+        },
+        id: message.id
+      })
+
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error)
+      Logging.log('ExecutionHandler', `Error resetting conversation: ${errorMessage}`, 'error')
+
+      // Send error response
+      port.postMessage({
+        type: MessageType.WORKFLOW_STATUS,
+        payload: { 
+          status: 'error',
+          error: errorMessage,
+          executionId: execId
+        },
+        id: message.id
+      })
+    }
+  }
+
+  /**
+   * Handle HUMAN_INPUT_RESPONSE message
+   */
+  handleHumanInputResponse(
+    message: PortMessage,
+    port: chrome.runtime.Port,
+    executionId?: string
+  ): void {
+    const execId = executionId || 'default'
+    const payload = message.payload as any
+
+    // Get the execution and forward the response
+    const execution = this.executionManager.get(execId)
+    if (execution) {
+      // Get the execution's PubSub channel
+      const pubsub = PubSub.getChannel(execId)
+      pubsub.publishHumanInputResponse(payload)
+
+      Logging.log('ExecutionHandler', 
+        `Forwarded human input response for execution ${execId}`)
+    } else {
+      Logging.log('ExecutionHandler', 
+        `No execution found for human input response: ${execId}`, 'warning')
+    }
+  }
+
+  /**
+   * Get execution statistics
+   */
+  getStats(): any {
+    return this.executionManager.getStats()
+  }
+}