fix: nft profile for twitter

getAlby · Jan 9, 2024 · 23cac3c · 23cac3c
1 parent 6aef498
commit 23cac3c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/extension/content-script/batteries/Twitter.ts b/src/extension/content-script/batteries/Twitter.ts
@@ -38,9 +38,11 @@ function getUserData(username: string) {
     const element = document.querySelector(
       '[data-testid="primaryColumn"] [data-testid="UserDescription"]'
     );
-    const imageUrl = document.querySelector<HTMLImageElement>(
-      `[data-testid="primaryColumn"] a[href="/${username}/photo" i] img`
-    )?.src;
+    const imageUrl = document.querySelectorAll<HTMLImageElement>(
+      `[data-testid="primaryColumn"] a[href="/${username}/photo" i] img,
+       [data-testid="primaryColumn"] a[href="/${username}/nft" i] img` // for nft profile
+    )?.[0].src;
+
     const location = document.querySelector<HTMLElement>(
       `[data-testid="primaryColumn"] [data-testid="UserLocation"]`
     );