Button now stores created node

YellowFish085 · Apr 26, 2020 · fe802a6 · fe802a6
1 parent c48a613
commit fe802a6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/content-scripts/web-integration/Button.ts b/src/content-scripts/web-integration/Button.ts
@@ -26,6 +26,9 @@ export default class Button {
   /** AniList data record */
   entry: ALSearch.AniList.Data | null;
 
+  /** Button node. */
+  button: HTMLElement | null;
+
   constructor(
     fixed: boolean,
     position: ButtonPosition,
@@ -39,6 +42,7 @@ export default class Button {
     this.value = value;
     this.title = title;
     this.entry = null;
+    this.button = null;
   }
 
   /**
@@ -428,7 +432,9 @@ export default class Button {
     // Try to find AniList related entry.
     this.entry = await this.findEntry();
 
-    // Create button.
-    return this.getButtonNode();
+    // Create button and return it.
+    this.button = this.getButtonNode();
+
+    return this.button;
   }
 }