Fix USRBG (#1000)

Co-authored-by: V <vendicated@riseup.net>

2 files changed, 15 insertions, 26 deletions

diff --git a/src/plugins/usrbg/index.css b/src/plugins/usrbg/index.css
index 6d1bd9b..819d975 100644
--- a/src/plugins/usrbg/index.css
+++ b/src/plugins/usrbg/index.css
@@ -4,27 +4,6 @@
     z-index: -1;
 }
 
-[class*="userPopout"] [class*="NonPremium"] [class*="bannerPremium"] {
-    top: -30px;
-}
-
-[class*="NonPremium"]:has([class*="bannerPremium"]) [class*="bannerSVGWrapper"] {
-    min-height: 120px !important;
-}
-
-[class*="NonPremium"]:has([class*="bannerPremium"]) [class*="bannerSVGWrapper"] foreignObject {
-    height: 360px;
-}
-
-[class*="userPopout"] [class*="NonPremium"]:has([class*="bannerPremium"]) [class*="bannerSVGWrapper"] rect {
-    height: 120px;
-    y: -30;
-}
-
-[class*="userPopout"] [class*="NonPremium"]:has([class*="bannerPremium"]) [class*="bannerSVGWrapper"] circle {
-    cy: 86;
-}
-
 [class*="NonPremium"]:has([class*="bannerPremium"]) [class*="avatarPositionNormal"],
 [class*="PremiumWithoutBanner"]:has([class*="bannerPremium"]) [class*="avatarPositionPremiumNoBanner"] {
     top: 76px;
diff --git a/src/plugins/usrbg/index.tsx b/src/plugins/usrbg/index.tsx
index c898d92..09ecd57 100644
--- a/src/plugins/usrbg/index.tsx
+++ b/src/plugins/usrbg/index.tsx
@@ -46,11 +46,17 @@ export default definePlugin({
     settings,
     patches: [
         {
-            find: ".bannerSrc,",
-            replacement: {
-                match: /(\i)\.bannerSrc,/,
-                replace: "$self.useBannerHook($1),"
-            }
+            find: ".NITRO_BANNER,",
+            replacement: [
+                {
+                    match: /(\i)\.premiumType/,
+                    replace: "$self.premiumHook($1)||$&"
+                },
+                {
+                    match: /(\i)\.bannerSrc,/,
+                    replace: "$self.useBannerHook($1),"
+                }
+            ]
         }
     ],
 
@@ -65,6 +71,10 @@ export default definePlugin({
         if (data[user.id]) return data[user.id];
     },
 
+    premiumHook({ userId }: any) {
+        if (data[userId]) return 2;
+    },
+
     async start() {
         enableStyle(style);