bugfix: made Trieve script safer to load

devflowinc · Mar 8, 2025 · 6a1a3e6 · 6a1a3e6
1 parent 7375fe1
commit 6a1a3e6
Showing 1 changed file with 82 additions and 63 deletions.
diff --git a/fern/assets/trieve-user-script.js b/fern/assets/trieve-user-script.js
@@ -10,82 +10,101 @@
 // ==/UserScript==
 
 const removeAllClickListeners = (element) => {
-  const newElement = element.cloneNode(true);
-  element.parentNode.replaceChild(newElement, element);
-  return newElement;
+  try {
+    const newElement = element?.cloneNode(true);
+    element.parentNode.replaceChild(newElement, element);
+    return newElement;
+  } catch (e) {
+    return element;
+  }
 };
 
 const makeDefaultSearchTrieve = async () => {
-  let defaultSearchBar = null;
-  let retries = 0;
-  while (!defaultSearchBar && retries < 10) {
-    for (const el of document.querySelectorAll("*")) {
-      if (el.querySelector('[aria-label="Search"]')) {
-        defaultSearchBar = el.querySelector('[aria-label="Search"]');
-        break;
+  try {
+    let defaultSearchBar = null;
+    let retries = 0;
+    while (!defaultSearchBar && retries < 10) {
+      for (const el of document.querySelectorAll("*")) {
+        if (el.querySelector('[aria-label="Search"]')) {
+          defaultSearchBar = el.querySelector('[aria-label="Search"]');
+          break;
+        }
       }
+      retries++;
+      await new Promise((resolve) => setTimeout(resolve, 10));
     }
-    retries++;
-    await new Promise((resolve) => setTimeout(resolve, 10));
-  }
-  defaultSearchBar = removeAllClickListeners(defaultSearchBar);
+    defaultSearchBar = removeAllClickListeners(defaultSearchBar);
 
-  defaultSearchBar.onclick = () => {
-    const event = new CustomEvent("trieve-open-with-text", {
-      detail: { text: "" },
-    });
-    window.dispatchEvent(event);
-  };
+    defaultSearchBar.onclick = () => {
+      const event = new CustomEvent("trieve-open-with-text", {
+        detail: { text: "" },
+      });
+      window.dispatchEvent(event);
+    };
+  } catch (e) {
+    console.error(e);
+  }
 };
-await makeDefaultSearchTrieve();
-setTimeout(makeDefaultSearchTrieve, 50);
-setTimeout(makeDefaultSearchTrieve, 100);
-setTimeout(makeDefaultSearchTrieve, 250);
-setTimeout(makeDefaultSearchTrieve, 500);
-setTimeout(makeDefaultSearchTrieve, 1000);
-setTimeout(makeDefaultSearchTrieve, 2000);
-setTimeout(makeDefaultSearchTrieve, 3000);
+try {
+  await makeDefaultSearchTrieve();
+  setTimeout(makeDefaultSearchTrieve, 50);
+  setTimeout(makeDefaultSearchTrieve, 100);
+  setTimeout(makeDefaultSearchTrieve, 250);
+  setTimeout(makeDefaultSearchTrieve, 500);
+  setTimeout(makeDefaultSearchTrieve, 1000);
+  setTimeout(makeDefaultSearchTrieve, 2000);
+  setTimeout(makeDefaultSearchTrieve, 3000);
+} catch (e) {
+  console.error(e);
+}
 
 (async function () {
   "use strict";
-  const link = document.createElement("link");
-  link.rel = "stylesheet";
-  link.href = "https://cdn.trieve.ai/beta/search-component/index.css";
-  document.head.appendChild(link);
-
+  try {
+    const link = document.createElement("link");
+    link.rel = "stylesheet";
+    link.href = "https://cdn.trieve.ai/beta/search-component/index.css";
+    document.head.appendChild(link);
+  } catch (e) {
+    console.error(e);
+  }
   import("https://cdn.trieve.ai/beta/search-component/vanilla/index.js").then(
     async (module) => {
-      const { renderToDiv } = module;
-      const root = document.createElement("div");
-      root.classList.add("trigger");
-      document.body.appendChild(root);
-      const colorScheme = document.documentElement?.style?.colorScheme;
-      const brandColor = colorScheme === "dark" ? "#47ffb6d5" : "#00551dcd";
+      try {
+        const { renderToDiv } = module;
+        const root = document.createElement("div");
+        root.classList.add("trigger");
+        document.body.appendChild(root);
+        const colorScheme = document.documentElement?.style?.colorScheme;
+        const brandColor = colorScheme === "dark" ? "#47ffb6d5" : "#00551dcd";
 
-      renderToDiv(root, {
-        apiKey: "tr-hZMKSsTf3ML9hJbAAqPO8K91p9IVe9Oc",
-        datasetId: "d3493dc0-2b5c-4c6e-a8ee-b18feeed5b06",
-        baseUrl: "https://api.trieve.ai",
-        type: "docs",
-        analytics: true,
-        theme: colorScheme === "dark" ? "dark" : null,
-        brandLogoImgSrcUrl:
-          "https://storage.googleapis.com/organization-image-assets/vapi-botAvatarDarkSrcUrl-1709929110474.png",
-        brandName: "Vapi",
-        brandColor: brandColor,
-        placeholder: "How can I help?",
-        defaultSearchQueries: ["quickstart", "assistant", "tools"],
-        defaultAiQuestions: [
-          "What voices are supported?",
-          "What languages are supported?",
-          "How do I connect a custom LLM?",
-          "How do I fetch the prompt dynamically?",
-        ],
-        defaultSearchMode: "search",
-        showFloatingButton: "true",
-        cssRelease: "none",
-        hideOpenButton: true,
-      });
+        renderToDiv(root, {
+          apiKey: "tr-hZMKSsTf3ML9hJbAAqPO8K91p9IVe9Oc",
+          datasetId: "d3493dc0-2b5c-4c6e-a8ee-b18feeed5b06",
+          baseUrl: "https://api.trieve.ai",
+          type: "docs",
+          analytics: true,
+          theme: colorScheme === "dark" ? "dark" : null,
+          brandLogoImgSrcUrl:
+            "https://storage.googleapis.com/organization-image-assets/vapi-botAvatarDarkSrcUrl-1709929110474.png",
+          brandName: "Vapi",
+          brandColor: brandColor,
+          placeholder: "How can I help?",
+          defaultSearchQueries: ["quickstart", "assistant", "tools"],
+          defaultAiQuestions: [
+            "What voices are supported?",
+            "What languages are supported?",
+            "How do I connect a custom LLM?",
+            "How do I fetch the prompt dynamically?",
+          ],
+          defaultSearchMode: "search",
+          showFloatingButton: "true",
+          cssRelease: "none",
+          hideOpenButton: true,
+        });
+      } catch (e) {
+        console.error(e);
+      }
     },
     (error) => {
       console.error("Failed to load module:", error);