Refactor button injection to setup_observer and make it conditional based on prompt type

robertpiosik · robertpiosik · commit edd1ab63de0f · 2026-03-23T02:37:58.000+01:00
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/ai-studio.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/ai-studio.ts
@@ -375,7 +375,7 @@ export const ai_studio: Chatbot = {
 
     await close_panel()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -424,6 +424,7 @@ export const ai_studio: Chatbot = {
     // AI Studio is quite sluggish with showing already generated tokens,
     // therefore we handle waiting for finished response differently than
     // in other chatbots.
+    let has_sent_finished_responding = true
     let debounce_timer: NodeJS.Timeout
     const observer = new MutationObserver(() => {
       clearTimeout(debounce_timer)
@@ -438,16 +439,17 @@ export const ai_studio: Chatbot = {
               (span) => span.textContent?.trim() == 'thumb_up'
             )
           if (has_thumb_up) {
-            const has_apply_button = !!footer.querySelector(
-              '.cwc-apply-response-button'
-            )
-            if (!has_apply_button) {
+            const has_been_processed = footer.hasAttribute('data-cwc-processed')
+            if (!has_been_processed) {
               browser.runtime.sendMessage<Message>({
                 action: 'finished-responding'
               })
               show_response_ready_notification({ chatbot_name: 'AI Studio' })
+              footer.setAttribute('data-cwc-processed', 'true')
+            }
+            if (params.inject_button) {
+              add_buttons(footer)
             }
-            add_buttons(footer)
           }
         })
       }, 100)
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/arena.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/arena.ts
@@ -32,7 +32,7 @@ export const arena: Chatbot = {
     input_element.value = params.message
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -65,7 +65,7 @@ export const arena: Chatbot = {
       is_generating: () => !!document.querySelector('ol .animate-spin'),
       footer_selector:
         'ol .bg-surface-primary > div:last-child > div.text-text-primary',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/chatgpt.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/chatgpt.ts
@@ -130,7 +130,7 @@ export const chatgpt: Chatbot = {
     input_element.innerText = params.message
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -161,7 +161,7 @@ export const chatgpt: Chatbot = {
       is_generating: () =>
         !!document.querySelector('button[data-testid="stop-button"]'),
       footer_selector: '.agent-turn > div:nth-of-type(2) > div',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/claude.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/claude.ts
@@ -127,7 +127,7 @@ export const claude: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -162,7 +162,7 @@ export const claude: Chatbot = {
       chatbot_name: 'Claude',
       is_generating: () => !!document.querySelector(stop_button_selector),
       footer_selector,
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/copilot.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/copilot.ts
@@ -32,7 +32,7 @@ export const copilot: Chatbot = {
     input_element.value = params.message
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -62,7 +62,7 @@ export const copilot: Chatbot = {
       is_generating: () =>
         !!document.querySelector('button[data-testid="stop-button"]'),
       footer_selector: 'div[data-testid="message-item-reactions"]',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/deepseek.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/deepseek.ts
@@ -80,7 +80,7 @@ export const deepseek: Chatbot = {
     input_element.value = params.message
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -130,7 +130,7 @@ export const deepseek: Chatbot = {
         )
       },
       footer_selector: 'div.ds-message + div + div > div.ds-flex',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/doubao.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/doubao.ts
@@ -114,7 +114,7 @@ export const doubao: Chatbot = {
     input_element.value = params.message
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -148,7 +148,7 @@ export const doubao: Chatbot = {
           ?.classList.contains('!hidden'),
       footer_selector:
         'div[data-testid="message_action_bar"] > div > div > div',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/gemini.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/gemini.ts
@@ -151,7 +151,7 @@ export const gemini: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -182,7 +182,7 @@ export const gemini: Chatbot = {
       is_generating: () =>
         !!document.querySelector('mat-icon[fonticon="stop"]'),
       footer_selector: 'message-actions > div > div',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/github-copilot.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/github-copilot.ts
@@ -96,7 +96,7 @@ export const github_copilot: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -128,7 +128,7 @@ export const github_copilot: Chatbot = {
           'path[d="M5.75 4h4.5c.966 0 1.75.784 1.75 1.75v4.5A1.75 1.75 0 0 1 10.25 12h-4.5A1.75 1.75 0 0 1 4 10.25v-4.5C4 4.784 4.784 4 5.75 4Z"]'
         ),
       footer_selector: 'div[data-testid="nonshared-toolbar"]',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/grok.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/grok.ts
@@ -106,7 +106,7 @@ export const grok: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -139,7 +139,7 @@ export const grok: Chatbot = {
           'path[d="M4 9.2v5.6c0 1.116 0 1.673.11 2.134a4 4 0 0 0 2.956 2.956c.46.11 1.018.11 2.134.11h5.6c1.116 0 1.673 0 2.134-.11a4 4 0 0 0 2.956-2.956c.11-.46.11-1.018.11-2.134V9.2c0-1.116 0-1.673-.11-2.134a4 4 0 0 0-2.956-2.955C16.474 4 15.916 4 14.8 4H9.2c-1.116 0-1.673 0-2.134.11a4 4 0 0 0-2.955 2.956C4 7.526 4 8.084 4 9.2Z"]'
         ),
       footer_selector: 'div.items-start div.action-buttons > div',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/hugging-chat.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/hugging-chat.ts
@@ -116,7 +116,7 @@ export const hugging_chat: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -146,7 +146,7 @@ export const hugging_chat: Chatbot = {
       is_generating: () =>
         !!document.querySelector('button.stop-generating-btn'),
       footer_selector: 'div[data-message-role="assistant"] > div:nth-child(3)',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/kimi.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/kimi.ts
@@ -39,7 +39,7 @@ export const kimi: Chatbot = {
       })
     )
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -74,7 +74,7 @@ export const kimi: Chatbot = {
           'path[d="M331.946667 379.904c-11.946667 23.466667-11.946667 54.186667-11.946667 115.626667v32.938666c0 61.44 0 92.16 11.946667 115.626667 10.538667 20.650667 27.306667 37.418667 47.957333 47.957333 23.466667 11.946667 54.186667 11.946667 115.626667 11.946667h32.938666c61.44 0 92.16 0 115.626667-11.946667 20.650667-10.538667 37.418667-27.306667 47.957333-47.957333 11.946667-23.466667 11.946667-54.186667 11.946667-115.626667v-32.938666c0-61.44 0-92.16-11.946667-115.626667a109.696 109.696 0 0 0-47.957333-47.957333c-23.466667-11.946667-54.186667-11.946667-115.626667-11.946667h-32.938666c-61.44 0-92.16 0-115.626667 11.946667-20.650667 10.538667-37.418667 27.306667-47.957333 47.957333z"]'
         ),
       footer_selector: '.segment-assistant-actions-content',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/mistral.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/mistral.ts
@@ -89,7 +89,7 @@ export const mistral: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     await new Promise((r) => requestAnimationFrame(r))
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -134,7 +134,7 @@ export const mistral: Chatbot = {
           !document.querySelector(footer_selector) ||
           !!document.querySelector('form rect[rx="2"]'),
         footer_selector,
-        add_buttons
+        add_buttons: params.inject_button ? add_buttons : undefined
       })
     })
   }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/open-webui.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/open-webui.ts
@@ -288,7 +288,7 @@ export const open_webui: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -325,7 +325,7 @@ export const open_webui: Chatbot = {
           'path[d="M2.25 12c0-5.385 4.365-9.75 9.75-9.75s9.75 4.365 9.75 9.75-4.365 9.75-9.75 9.75S2.25 17.385 2.25 12zm6-2.438c0-.724.588-1.312 1.313-1.312h4.874c.725 0 1.313.588 1.313 1.313v4.874c0 .725-.588 1.313-1.313 1.313H9.564a1.312 1.312 0 01-1.313-1.313V9.564z"]'
         ),
       footer_selector: '.chat-assistant + div',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/openrouter.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/openrouter.ts
@@ -281,7 +281,7 @@ export const openrouter: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -316,7 +316,7 @@ export const openrouter: Chatbot = {
       is_generating: () => !!document.querySelector('.animate-spin'),
       footer_selector:
         'div[data-testid="assistant-message"] > div:last-child > div',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/qwen.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/qwen.ts
@@ -141,7 +141,7 @@ export const qwen: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -180,7 +180,7 @@ export const qwen: Chatbot = {
       chatbot_name: 'Qwen',
       is_generating: () => !document.querySelector('.send-button.disabled'),
       footer_selector: '.qwen-chat-package-comp-new-action-control-icons',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/together.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/together.ts
@@ -79,7 +79,7 @@ export const together: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -109,7 +109,7 @@ export const together: Chatbot = {
       is_generating: () =>
         !!document.querySelector('button[data-testid="stop-button"]'),
       footer_selector: 'div[data-testid="assistant-message-toolbar"]',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/yuanbao.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/yuanbao.ts
@@ -35,7 +35,7 @@ export const yuanbao: Chatbot = {
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
     input_element.focus()
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -64,7 +64,7 @@ export const yuanbao: Chatbot = {
       chatbot_name: 'Yuanbao',
       is_generating: () => !!document.querySelector('rect[x="7.71448"]'),
       footer_selector: '.agent-chat__toolbar__right',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/z-ai.ts b/apps/browser/src/content-scripts/send-prompt-content-script/chatbots/z-ai.ts
@@ -54,7 +54,7 @@ export const z_ai: Chatbot = {
     input_element.value = params.message
     input_element.dispatchEvent(new Event('input', { bubbles: true }))
   },
-  inject_apply_response_button: (params) => {
+  setup_observer: (params) => {
     const add_buttons = (footer: Element) => {
       add_apply_response_button({
         client_id: params.client_id,
@@ -83,7 +83,7 @@ export const z_ai: Chatbot = {
       is_generating: () =>
         !document.querySelector('button[id="send-message-button"]'),
       footer_selector: '.chat-assistant + div',
-      add_buttons
+      add_buttons: params.inject_button ? add_buttons : undefined
     })
   }
 }
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/send-prompt-content-script.ts b/apps/browser/src/content-scripts/send-prompt-content-script/send-prompt-content-script.ts
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/types/chatbot.ts b/apps/browser/src/content-scripts/send-prompt-content-script/types/chatbot.ts
diff --git a/apps/browser/src/content-scripts/send-prompt-content-script/utils/add-apply-response-button.ts b/apps/browser/src/content-scripts/send-prompt-content-script/utils/add-apply-response-button.ts

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,7 @@ export const deepseek: Chatbot = {`
`80`	`80`	`input_element.value = params.message`
`81`	`81`	`input_element.dispatchEvent(new Event('input', { bubbles: true }))`
`82`	`82`	`},`
`83`		`- inject_apply_response_button: (params) => {`
	`83`	`+ setup_observer: (params) => {`
`84`	`84`	`const add_buttons = (footer: Element) => {`
`85`	`85`	`add_apply_response_button({`
`86`	`86`	`client_id: params.client_id,`
`@@ -130,7 +130,7 @@ export const deepseek: Chatbot = {`
`130`	`130`	`)`
`131`	`131`	`},`
`132`	`132`	`footer_selector: 'div.ds-message + div + div > div.ds-flex',`
`133`		`- add_buttons`
	`133`	`+ add_buttons: params.inject_button ? add_buttons : undefined`
`134`	`134`	`})`
`135`	`135`	`}`
`136`	`136`	`}`