Remove redundant comments and add URL tracking for chat responses

Author: robertpiosik

apps/browser/src/background/config.ts

@@ -1,4 +1,4 @@
 export const CONFIG = {
   // WebSocket configuration
-  RECONNECT_DELAY: 5000 // 5 seconds
+  RECONNECT_DELAY: 5000
 }

apps/browser/src/background/keep-alive.ts

@@ -6,7 +6,7 @@ export const setup_keep_alive = () => {
       try {
         chrome.alarms.create('keep-alive', {
           when: Date.now() + 1000 * 30
-        }) // 30 seconds interval
+        })
       } catch (error) {
         console.error('Error creating KeepAlive alarm:', error)
       }

apps/browser/src/background/message-handler.ts

@@ -12,7 +12,6 @@ interface ChatQueueItem {
   timeout_id?: number
 }
 
-// Global queue of chat initialization requests
 const chat_queue: ChatQueueItem[] = []
 let is_processing = false
 
@@ -112,7 +111,6 @@ const process_next_chat = async () => {
     browser.tabs.create(create_tab_options)
   }
 
-  // Set a timeout to automatically proceed if no confirmation is received
   current_queue_item.timeout_id = setTimeout(() => {
     console.warn(
       `Chat initialization timeout for ${current_chat_message.url}. Moving to next chat.`
@@ -157,7 +155,8 @@ export const setup_message_listeners = () => {
             action: 'apply-chat-response',
             client_id: message.client_id,
             raw_instructions: message.raw_instructions,
-            edit_format: message.edit_format
+            edit_format: message.edit_format,
+            url: message.url
           } as ApplyChatResponseMessage)
         }
       }

apps/browser/src/content-scripts/send-prompt-content-script/chatbots/ai-studio.ts

@@ -456,7 +456,7 @@ export const ai_studio: Chatbot = {
 
 const open_panel = async () => {
   if (window.innerWidth <= 960) {
-    if (document.querySelector('ms-right-side-panel > div')) return // Already opened
+    if (document.querySelector('ms-right-side-panel > div')) return
 
     const tune_button = document.querySelector(
       'button.runsettings-toggle-button'
@@ -475,7 +475,8 @@ const open_panel = async () => {
 
 const close_panel = async () => {
   if (window.innerWidth <= 960) {
-    if (!document.querySelector('ms-right-side-panel > div')) return // Already closed
+    if (!document.querySelector('ms-right-side-panel > div')) return
+
     const close_button = document.querySelector(
       'ms-run-settings button[iconname="close"]'
     ) as HTMLButtonElement

apps/browser/src/content-scripts/send-prompt-content-script/chatbots/deepseek.ts

@@ -36,7 +36,6 @@ export const deepseek: Chatbot = {
       return
     }
 
-    // Uncheck deep think
     const deep_think_button_style = window.getComputedStyle(deep_think_button)
     if (
       deep_think_button_style.getPropertyValue('color') != 'rgb(15, 17, 21)' &&
@@ -45,7 +44,6 @@ export const deepseek: Chatbot = {
       deep_think_button.click()
     }
 
-    // Uncheck search
     const search_button_style = window.getComputedStyle(search_button)
     if (
       search_button_style.getPropertyValue('color') != 'rgb(15, 17, 21)' &&

apps/browser/src/content-scripts/send-prompt-content-script/send-prompt-content-script.ts

@@ -159,7 +159,6 @@ const initialize_chat = async (params: { message: string; chat: Chat }) => {
     })
   }
 
-  // User may send by pressing enter
   await new Promise<void>((resolve) => {
     const handle_key_press = (e: KeyboardEvent) => {
       if (e.key == 'Enter') {
@@ -170,7 +169,6 @@ const initialize_chat = async (params: { message: string; chat: Chat }) => {
     document.addEventListener('keydown', handle_key_press)
     setTimeout(resolve, 2000)
   })
-  // Process next chat from the queue
   browser.runtime.sendMessage<Message>({
     action: 'chat-initialized'
   })
@@ -180,14 +178,12 @@ const main = async () => {
   const session_data_key = 'cwc-session-data'
 
   if (is_cwc_hash) {
-    // Remove the hash from the URL to avoid reloading the content script if the page is refreshed
     history.replaceState(
       null,
       '',
       window.location.pathname + window.location.search
     )
 
-    // Get the message using the batch ID from the hash
     const storage_key = `chat-init:${batch_id}`
     const storage = await browser.storage.local.get(storage_key)
     const stored_data = storage[storage_key] as {
@@ -212,7 +208,6 @@ const main = async () => {
       return
     }
 
-    // Now directly use the current_chat instead of searching for it
     const message_text = stored_data.text
     const current_chat = stored_data.current_chat
 
@@ -230,7 +225,6 @@ const main = async () => {
       chat: current_chat
     })
 
-    // Clean up the storage entry after using it
     await browser.storage.local.remove(storage_key)
 
     if (

apps/browser/src/content-scripts/send-prompt-content-script/utils/add-apply-response-button.ts

@@ -51,7 +51,8 @@ export function add_apply_response_button(params: {
       action: 'apply-chat-response',
       client_id: params.client_id,
       raw_instructions: params.raw_instructions,
-      edit_format: params.edit_format
+      edit_format: params.edit_format,
+      url: window.location.href
     })
   })
 

apps/browser/src/types/messages.ts

@@ -7,6 +7,7 @@ type ApplyChatResponseMessage = {
   client_id: number
   raw_instructions?: string
   edit_format?: string
+  url?: string
 }
 
 export type Message = ChatInitializedMessage | ApplyChatResponseMessage

apps/editor/src/commands/apply-chat-response-command/apply-chat-response-command.ts

@@ -79,7 +79,8 @@ export const apply_chat_response_command = (params: {
         const new_item: ResponseHistoryItem = {
           response: chat_response,
           raw_instructions: args?.raw_instructions,
-          created_at: Date.now()
+          created_at: Date.now(),
+          url: args?.url
         }
 
         history.push(new_item)
@@ -316,7 +317,8 @@ export const apply_chat_response_command = (params: {
                   created_at: created_at_for_preview,
                   lines_added: total_lines_added,
                   lines_removed: total_lines_removed,
-                  files: files_for_history
+                  files: files_for_history,
+                  url: args?.url
                 }
 
                 history.push(new_item)
@@ -339,7 +341,8 @@ export const apply_chat_response_command = (params: {
             context: params.context,
             original_editor_state: args?.original_editor_state,
             raw_instructions: args?.raw_instructions,
-            created_at: created_at_for_preview
+            created_at: created_at_for_preview,
+            url: args?.url
           })
 
           if (changes_accepted) {

apps/editor/src/commands/apply-chat-response-command/response-processor.ts

@@ -46,6 +46,7 @@ export type CommandArgs = {
   }
   files_with_content?: FileInPreview[]
   created_at?: number
+  url?: string
 }
 
 export const process_chat_response = async (