Change category pill container

hanydd · Jan 21, 2024 · fee17ff · fee17ff
1 parent ca2bb63
commit fee17ff
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/maze-utils b/maze-utils
diff --git a/src/dearrowPromotion.ts b/src/dearrowPromotion.ts
@@ -1,5 +1,5 @@
 import { waitFor } from "../maze-utils/src";
-import { getYouTubeTitleNode } from "../maze-utils/src/elements";
+import { getBilibiliTitleNode } from "../maze-utils/src/elements";
 import { getHash } from "../maze-utils/src/hash";
 import { getVideoID, isOnInvidious, isOnMobileYouTube } from "../maze-utils/src/video";
 import Config from "./config";
@@ -22,7 +22,7 @@ export async function tryShowingDeArrowPromotion() {
 
             if (!await isDeArrowInstalled()) {
                 try {
-                    const element = await waitFor(() => getYouTubeTitleNode(), 5000, 500, (e) => isVisible(e)) as HTMLElement;
+                    const element = await waitFor(() => getBilibiliTitleNode(), 5000, 500, (e) => isVisible(e)) as HTMLElement;
                     if (element && element.innerText && badTitle(element.innerText)) {
                         const hashPrefix = (await getHash(getVideoID(), 1)).slice(0, 4);
                         const deArrowData = await asyncRequestToServer("GET", "/api/branding/" + hashPrefix);

diff --git a/src/render/CategoryPill.tsx b/src/render/CategoryPill.tsx
@@ -6,7 +6,7 @@ import { VoteResponse } from "../messageTypes";
 import { Category, SegmentUUID, SponsorTime } from "../types";
 import { Tooltip } from "./Tooltip";
 import { waitFor } from "../../maze-utils/src";
-import { getYouTubeTitleNode } from "../../maze-utils/src/elements";
+import { getBilibiliTitleNode } from "../../maze-utils/src/elements";
 import { addCleanupListener } from "../../maze-utils/src/cleanup";
 
 const id = "categoryPill";
@@ -44,7 +44,7 @@ export class CategoryPill {
 
     private async attachToPageInternal(): Promise<void> {
         const referenceNode = 
-            await waitFor(() => getYouTubeTitleNode());
+            await waitFor(() => getBilibiliTitleNode());
 
         if (referenceNode && !referenceNode.contains(this.container)) {
             if (!this.container) {