From ae2c2a5501e8bfb99cfd65ab8862567a239bf6bf Mon Sep 17 00:00:00 2001
From: Bryce Hackel <34104885+bhackel@users.noreply.github.com>
Date: Sun, 5 May 2024 01:09:00 -0700
Subject: [PATCH] Fix fake premium setting

---
 Sources/uYouPlus.xm         | 10 +++++-----
 Sources/uYouPlusSettings.xm | 14 +++++++++++++-
 2 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/Sources/uYouPlus.xm b/Sources/uYouPlus.xm
index 9fa1d1c..012ec21 100644
--- a/Sources/uYouPlus.xm
+++ b/Sources/uYouPlus.xm
@@ -1587,13 +1587,13 @@ static BOOL findCell(ASNodeController *nodeController, NSArray <NSString *> *ide
     // YTNoModernUI - @arichorn
     BOOL ytNoModernUIEnabled = IS_ENABLED(@"ytNoModernUI_enabled");
     if (ytNoModernUIEnabled) {
-    NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
-    [userDefaults setBool:NO forKey:@"enableVersionSpoofer_enabled"];
+        NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
+        [userDefaults setBool:NO forKey:@"enableVersionSpoofer_enabled"];
     } else {
-    BOOL enableVersionSpooferEnabled = IS_ENABLED(@"enableVersionSpoofer_enabled");
+        BOOL enableVersionSpooferEnabled = IS_ENABLED(@"enableVersionSpoofer_enabled");
 
-    NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
-    [userDefaults setBool:enableVersionSpooferEnabled forKey:@"enableVersionSpoofer_enabled"];
+        NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
+        [userDefaults setBool:enableVersionSpooferEnabled forKey:@"enableVersionSpoofer_enabled"];
     }
     NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
     [userDefaults setBool:ytNoModernUIEnabled ? ytNoModernUIEnabled : [userDefaults boolForKey:@"fixLowContrastMode_enabled"] forKey:@"fixLowContrastMode_enabled"];
diff --git a/Sources/uYouPlusSettings.xm b/Sources/uYouPlusSettings.xm
index 4f1d692..bcc6bb9 100644
--- a/Sources/uYouPlusSettings.xm
+++ b/Sources/uYouPlusSettings.xm
@@ -1020,8 +1020,20 @@ YTSettingsSectionItem *lowContrastModeButton = [%c(YTSettingsSectionItem)
     accessibilityIdentifier:nil
     switchOn:IS_ENABLED(@"youTabFakePremium_enabled")
     switchBlock:^BOOL (YTSettingsCell *cell, BOOL enabled) {
+        // Store the received value
+        [[NSUserDefaults standardUserDefaults] setBool:enabled forKey:@"youTabFakePremium_enabled"];
         NSLog(@"backel: Switch toggled: %@", enabled ? @"ON" : @"OFF");
-        NSString *appVersion = [[[NSBundle mainBundle] infoDictionary] objectForKey:@"CFBundleShortVersionString"];
+        // Get the current version (including spoofed versions)
+        NSString *appVersion;
+        Class YTVersionUtilsClass = %c(YTVersionUtils);
+        if ([YTVersionUtilsClass respondsToSelector:@selector(appVersion)]) {
+            appVersion = [YTVersionUtilsClass performSelector:@selector(appVersion)];
+            NSLog(@"backel: App Version: %@", appVersion);
+        } else {
+            NSLog(@"backel: The YTVersionUtils class does not respond to -appVersion");
+        }
+        // appVersion = [[[NSBundle mainBundle] infoDictionary] objectForKey:@"CFBundleShortVersionString"];
+        // Check if current version is less than the required version
         NSComparisonResult result = [appVersion compare:@"18.35.4" options:NSNumericSearch];
         NSLog(@"backel: App Version: %@, Required: 18.35.4, Result: %ld", appVersion, (long)result);
         if (result == NSOrderedAscending) {