From b95c5c6619bdd9d2d42089d02f553a43bfb68dba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C4=90=E1=BB=97=20V=C4=83n=20Ho=C3=A0i=20Tu=C3=A2n?=
 <htuan03@gmail.com>
Date: Sat, 29 Apr 2023 15:08:04 -0700
Subject: [PATCH] Fix USRBG (#1000)

Co-authored-by: V <vendicated@riseup.net>
---
 src/plugins/usrbg/index.css | 21 ---------------------
 src/plugins/usrbg/index.tsx | 20 +++++++++++++++-----
 2 files changed, 15 insertions(+), 26 deletions(-)

diff --git a/src/plugins/usrbg/index.css b/src/plugins/usrbg/index.css
index 6d1bd9bd7..819d9751d 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 c898d929a..09ecd573e 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);