chore: refactor the code

open-sauced · Nov 9, 2023 · 22a7a6d · 22a7a6d
1 parent 50486f1
commit 22a7a6d
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/utils/dom-utils/addDescriptionGenerator.ts b/src/utils/dom-utils/addDescriptionGenerator.ts
@@ -21,13 +21,11 @@ const injectDescriptionGeneratorButton = async () => {
         }
     }
 
-    const prActionsClassName = "ActionBar-item-container";
-    const actionBars = document.getElementsByClassName(prActionsClassName);
-    const actionBarsArray = Array.from(actionBars);
-    const addGeneratorButton = DescriptionGeneratorButton();
-    const firstPrDescription = actionBarsArray[0];
+    const firstPrDescription = document.querySelector(".ActionBar-item-container");
+
+    if (firstPrDescription && !firstPrDescription.querySelector("#ai-description-button")) {
+        const addGeneratorButton = DescriptionGeneratorButton();
 
-    if (!firstPrDescription.querySelector("#ai-description-button")) {
         firstPrDescription.insertBefore(addGeneratorButton, firstPrDescription.firstChild);
     }
 };