From a29a6299efc03b0ea049a34f72dd1db16022bfef Mon Sep 17 00:00:00 2001
From: Jordan Wages <jordan@jordanwages.com>
Date: Fri, 27 Jun 2025 02:52:49 -0500
Subject: [PATCH] Use new scripting API for message display scripts

---
 background.js | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/background.js b/background.js
index 5b536a5..ca235c8 100644
--- a/background.js
+++ b/background.js
@@ -228,11 +228,27 @@ async function clearCacheForMessages(idsInput) {
             browser.messageDisplayAction.setLabel({ label: "Classify" });
         }
     }
-    if (browser.messageDisplayScripts) {
+    if (browser.scripting && browser.scripting.messageDisplay) {
+        try {
+            const scripts = [
+                {
+                    id: "clear-cache-button",
+                    js: [browser.runtime.getURL("resources/clearCacheButton.js")],
+                },
+                {
+                    id: "reason-button",
+                    js: [browser.runtime.getURL("resources/reasonButton.js")],
+                },
+            ];
+            await browser.scripting.messageDisplay.registerScripts(scripts);
+        } catch (e) {
+            logger.aiLog("failed to register message display script", { level: 'warn' }, e);
+        }
+    } else if (browser.messageDisplayScripts) {
         try {
             const scripts = [
                 { js: [browser.runtime.getURL("resources/clearCacheButton.js")] },
-                { js: [browser.runtime.getURL("resources/reasonButton.js")] }
+                { js: [browser.runtime.getURL("resources/reasonButton.js")] },
             ];
             if (browser.messageDisplayScripts.registerScripts) {
                 await browser.messageDisplayScripts.registerScripts(scripts);