#import <UIKit/UIKit.h>
#import <Foundation/Foundation.h>
#import <objc/runtime.h>
#import "Header.h"
#import "Tweaks/YouTubeHeader/YTVideoQualitySwitchOriginalController.h"
#import "Tweaks/YouTubeHeader/YTSettingsSectionItem.h"
#import "Tweaks/YouTubeHeader/YTPlayerViewController.h"
#import "Tweaks/YouTubeHeader/YTWatchController.h"
#import "Tweaks/YouTubeHeader/YTIGuideResponse.h"
#import "Tweaks/YouTubeHeader/YTIGuideResponseSupportedRenderers.h"
#import "Tweaks/YouTubeHeader/YTIPivotBarSupportedRenderers.h"
#import "Tweaks/YouTubeHeader/YTIPivotBarRenderer.h"
#import "Tweaks/YouTubeHeader/YTIBrowseRequest.h"

BOOL hideHUD() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"hideHUD_enabled"];
}
BOOL oled() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"oled_enabled"];
}
BOOL oledKB() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"oledKeyBoard_enabled"];
}
BOOL autoFullScreen() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"autofull_enabled"];
}
BOOL noHoverCard() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"hover_cards_enabled"];
}
BOOL ReExplore() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"reExplore_enabled"];
}
BOOL bigYTMiniPlayer() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"bigYTMiniPlayer_enabled"];
}
BOOL hideCC() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"hideCC_enabled"];
}
BOOL hideAutoplaySwitch() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"hideAutoplaySwitch_enabled"];
}
BOOL castConfirm() {
    return [[NSUserDefaults standardUserDefaults] boolForKey:@"castConfirm_enabled"];
}

UIColor* oledColor = [UIColor colorWithRed:0.0 green:0.0 blue:0.0 alpha:1.0];

// Hide CC / Autoplay switch
%hook YTMainAppControlsOverlayView
- (void)layoutSubviews {
    %orig;
    if (hideAutoplaySwitch())
        MSHookIvar<UIView *>(self, "_autonavSwitch").hidden = YES;
    if (hideCC())
        MSHookIvar<UIView *>(self, "_closedCaptionsOrSubtitlesButton").hidden = YES;
}
%end

// Hide HUD Messages
%hook YTHUDMessageView
- (id)initWithMessage:(id)arg1 dismissHandler:(id)arg2 {
    return hideHUD() ? nil : %orig;
}
%end

// YTAutoFullScreen: https://github.com/PoomSmart/YTAutoFullScreen/
%hook YTPlayerViewController
- (void)loadWithPlayerTransition:(id)arg1 playbackConfig:(id)arg2 {
    %orig;
    if (autoFullScreen())
        [NSTimer scheduledTimerWithTimeInterval:0.75 target:self selector:@selector(autoFullscreen) userInfo:nil repeats:NO];
}
%new
- (void)autoFullscreen {
    YTWatchController *watchController = [self valueForKey:@"_UIDelegate"];
    [watchController showFullScreen];
}
%end

// YTNoHoverCards: https://github.com/level3tjg/YTNoHoverCards
%hook YTCreatorEndscreenView
- (void)setHidden:(BOOL)hidden {
    if (noHoverCard())
        hidden = YES;
    %orig;
}
%end
 
//YTCastConfirm: https://github.com/JamieBerghmans/YTCastConfirm
%hook MDXPlaybackRouteButtonController
- (void)didPressButton:(id)arg1 {
    if (castConfirm()) {
        UIAlertController* alertController = [%c(UIAlertController) alertControllerWithTitle:@"Casting"
                                message:@"Are you sure you want to start casting?"
                                preferredStyle:UIAlertControllerStyleAlert];
        UIAlertAction* defaultAction = [%c(UIAlertAction) actionWithTitle:@"Yes" style:UIAlertActionStyleDefault handler:^(UIAlertAction * action) {
            %orig;
        }];

        UIAlertAction* noButton = [%c(UIAlertAction)
                                actionWithTitle:@"Cancel"
                                style:UIAlertActionStyleDefault
                                handler: ^(UIAlertAction * action) {
            return;
        }];

        [alertController addAction:defaultAction];
        [alertController addAction:noButton];
        
        id rootViewController = [%c(UIApplication) sharedApplication].delegate.window.rootViewController;
        if ([rootViewController isKindOfClass:[%c(UINavigationController) class]]) {
            rootViewController = ((UINavigationController *)rootViewController).viewControllers.firstObject;
        }
        if ([rootViewController isKindOfClass:[%c(UITabBarController) class]]) {
            rootViewController = ((UITabBarController *)rootViewController).selectedViewController;
        }
        if ([rootViewController presentedViewController] != nil) {
            rootViewController = [rootViewController presentedViewController];
        }
        [rootViewController presentViewController:alertController animated:YES completion:nil];
    }
    return %orig;
}
%end

// Workaround for https://github.com/MiRO92/uYou-for-YouTube/issues/12
%hook YTAdsInnerTubeContextDecorator
- (void)decorateContext:(id)arg1 { %orig(nil); }
%end

// YTClassicVideoQuality: https://github.com/PoomSmart/YTClassicVideoQuality
%hook YTVideoQualitySwitchControllerFactory
- (id)videoQualitySwitchControllerWithParentResponder:(id)responder {
    Class originalClass = %c(YTVideoQualitySwitchOriginalController);
    return originalClass ? [[originalClass alloc] initWithParentResponder:responder] : %orig;
}
%end

// YTNoCheckLocalNetwork: https://poomsmart.github.io/repo/depictions/ytnochecklocalnetwork.html
%hook YTHotConfig
- (BOOL)isPromptForLocalNetworkPermissionsEnabled { return NO; }
%end

// YTSystemAppearance: https://poomsmart.github.io/repo/depictions/ytsystemappearance.html
// YouRememberCaption: https://poomsmart.github.io/repo/depictions/youremembercaption.html
%hook YTColdConfig
- (BOOL)shouldUseAppThemeSetting { return YES; }
- (BOOL)respectDeviceCaptionSetting { return NO; }
- (BOOL)isEnhancedSearchBarEnabled { return YES; }
%end

// NOYTPremium - https://github.com/PoomSmart/NoYTPremium/
%hook YTCommerceEventGroupHandler
- (void)addEventHandlers {}
%end

%hook YTInterstitialPromoEventGroupHandler
- (void)addEventHandlers {}
%end

%hook YTPromosheetEventGroupHandler
- (void)addEventHandlers {}
%end

%hook YTPromoThrottleController
- (BOOL)canShowThrottledPromo { return NO; }
- (BOOL)canShowThrottledPromoWithFrequencyCap:(id)arg1 { return NO; }
- (BOOL)canShowThrottledPromoWithFrequencyCaps:(id)arg1 { return NO; }
%end

%hook YTIShowFullscreenInterstitialCommand
- (BOOL)shouldThrottleInterstitial { return YES; }
%end

%hook YTSurveyController
- (void)showSurveyWithRenderer:(id)arg1 surveyParentResponder:(id)arg2 {}
%end

// OLED 
// Thanks u/DGh0st for his very well explained comment - https://www.reddit.com/r/jailbreakdevelopers/comments/9uape7/comment/e94sq80/
// Thanks sinfool for his flex patch which brings OLED Dark mode for YouTube - "Color Customizer (YouTube) OLED"
%group gOLED
%hook UIView
- (void)setBackgroundColor:(id)arg1 {
    if ([self.nextResponder isKindOfClass:%c(YTLinkCell)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTCommentsHeaderView)]) 
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTSearchView)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTSearchBoxView)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTEngagementPanelHeaderView)])
    arg1 = oledColor;	
    if ([self.nextResponder isKindOfClass:%c(YTEngagementPanelView)])
    arg1 = oledColor;	
    if ([self.nextResponder isKindOfClass:%c(YTPivotBarView)])
    arg1 = oledColor;	
    if ([self.nextResponder isKindOfClass:%c(YTHUDMessageView)])
    arg1 = oledColor;	
    if ([self.nextResponder isKindOfClass:%c(YTChipCloudCell)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YCHLiveChatTextCell)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YCHLiveChatView)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YCHLiveChatViewerEngagementCell)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTSlideForActionsView)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTPlaylistHeaderView)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTAsyncCollectionView)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTFeedHeaderView)])
    arg1 = oledColor;
    if ([self.nextResponder isKindOfClass:%c(YTMessageCell)])
    arg1 = oledColor;	
    if ([self.nextResponder isKindOfClass:%c(YTPlaylistPanelProminentThumbnailVideoCell)])
    arg1 = oledColor;	
    if ([self.nextResponder isKindOfClass:%c(ASWAppSwitcherCollectionViewCell)])
    arg1 = oledColor;	
    if ([self.nextResponder isKindOfClass:%c(YTEditSheetControllerHeader)])
    arg1 = oledColor;	
    if ([self.nextResponder isKindOfClass:%c(YTDialogContainerScrollView)])
    arg1 = oledColor;	
    %orig;
}
%end

%hook YTLightweightQTMButton
- (void)setBackgroundColor:(id)arg1 {
    if ([self.nextResponder isKindOfClass:%c(YTShareMainView)])
    arg1 = oledColor;
    %orig;
}
- (void)setCustomTitleColor:(id)arg1 { %orig([UIColor whiteColor]); }
%end

%hook NIAttributedLabel
- (void)setBackgroundColor:(id)arg1 {
    if ([self.nextResponder isKindOfClass:%c(UIScrollView)])
    arg1 = oledColor;
    %orig;
}
%end

%hook UIControl // this sucks I know :/
- (void)setBackgroundColor:(id)arg1 {
    if ([self.nextResponder isKindOfClass:%c(YTShareMainView)]) 
    arg1 = oledColor;
    %orig;
}
%end

%hook YTAsyncCollectionView
- (void)setBackgroundColor:(id)arg1 {
    if([self.nextResponder isKindOfClass:%c(YTRelatedVideosCollectionViewController)]) {
        arg1 = [oledColor colorWithAlphaComponent:0.0];
    } else if([self.nextResponder isKindOfClass:%c(YTFullscreenMetadataHighlightsCollectionViewController)]) {
        arg1 = [oledColor colorWithAlphaComponent:0.0];
    } else {
        arg1 = oledColor;
    }
    %orig;
}
%end

%hook YTWatchMiniBarView 
- (void)setBackgroundColor:(id)arg1 {
    arg1 = [UIColor colorWithRed:0.0 green:0.0 blue:0.0 alpha:0.9];
    %orig;
}
%end

%hook YTPlaylistMiniBarView 
- (void)setBackgroundColor:(id)arg1 {
    arg1 = [UIColor colorWithRed:0.0 green:0.0 blue:0.0 alpha:0.9];
    %orig;
}
%end

%hook _LNPopupBarContentView // uYou player
- (void)setBackgroundColor:(id)arg1 {
    arg1 = [UIColor colorWithRed:0.0 green:0.0 blue:0.0 alpha:0.9];
    %orig;
}
%end

%hook YTTopAlignedView 
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
- (void)didMoveToWindow {  // Dune - https://github.com/Skittyblock/Dune/blob/9b1df9790230115b7553cc9dbadf36889018d7f9/Tweak.xm#L70
    MSHookIvar<UIView *>(self, "_contentView").backgroundColor = oledColor;
	%orig;
}
%end

%hook DownloadedVC
- (id)ytBackgroundColor { return oledColor; }
%end

%hook DownloadsPagerVC
- (id)ytBackgroundColor { return oledColor; }
%end

%hook DownloadingVC
- (id)ytBackgroundColor { return oledColor; }
%end

%hook PlayerVC
- (id)ytBackgroundColor { return oledColor; }
%end

%hook ASScrollView  // Explore
- (void)didMoveToWindow { 
    self.backgroundColor = oledColor;
    %orig;
}
%end

%hook ASCollectionView  // your videos
- (void)didMoveToWindow {
    self.backgroundColor = oledColor;
    %orig;
}
%end

%hook YTDialogContainerScrollView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook MDXQueueView // Cast queue
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTChannelProfileEditorView 
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTChannelSubMenuView // 
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTChannelListSubMenuView // sub - 
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTCommentView 
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTCreateCommentAccessoryView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTCreateCommentTextView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
- (void)setTextColor:(id)arg1       { %orig([UIColor whiteColor]); }
%end

%hook YCHLiveChatActionPanelView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTEmojiTextView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTShareTitleView // Share sheet
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTNavigationBar
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
- (void)setBarTintColor:(id)arg1    { %orig(oledColor); }
%end

%hook YTPrivacyTosFooterView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTEngagementPanelHeaderView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTChannelMobileHeaderView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTInlineSignInView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTHeaderView //Stt bar
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTTabTitlesView // Tab bar - mychannel
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTSettingsCell // Settings 
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTHorizontalCardListView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTShortsGalleryHeaderView  // upload videos heaer (gallery)
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook GOODialogView // 3 dots menu
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTCollectionView //sharesheet
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook ASWAppSwitchingSheetHeaderView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook ASWAppSwitchingSheetFooterView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTCollectionSeparatorView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
- (void)didMoveToWindow {}
%end

%hook YTShareBusyView // sharesheet load
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTPageView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTWatchView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTSearchBarView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTSearchBoxView
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTSearchSuggestionCollectionViewCell
- (void)updateColors {}
%end

%hook UISearchBarBackground
- (void)setBarTintColor:(id)arg1    { %orig(oledColor); }
%end

%hook YTMealBarPromoView // Offline
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

////
/*
%hook UICollectionView
- (void)setBackgroundColor:(id)arg1 {
    if ([self.nextResponder isKindOfClass:%c(UICollectionViewControllerWrapperView)])
    arg1 = oledColor;
    %orig;
}
%end

%hook _ASDisplayView // edit your videos
- (void)layoutSubviews {
    if ([self.nextResponder isKindOfClass:%c(ELMView)])
    self.backgroundColor = oledColor;
}
%end

%hook YTChannelProfileDescriptionEditorView // edit profile Description
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

%hook YTChannelProfileNameEditorView  // edit profile Name
- (void)setBackgroundColor:(id)arg1 { %orig(oledColor); }
%end

hook GOOTextField 
- (void)setBackgroundColor:(id)arg1 {  // edit profile Description
    arg1 = oledColor;
    %orig;
}
%end

%hook GOOMultilineTextField// 
- (void)setBackgroundColor:(id)arg1 { // edit profile Name
    arg1 = oledColor;
    %orig;
}
%end
*/
%end

%group gOLEDKB // OLED keyboard by @ichitaso <3 - http://gist.github.com/ichitaso/935100fd53a26f18a9060f7195a1be0e
%hook UIPredictionViewController
- (void)loadView {
    %orig;
    [self.view setBackgroundColor:oledColor];
}
%end

%hook UICandidateViewController
- (void)loadView {
    %orig;
    [self.view setBackgroundColor:oledColor];
}
%end

%hook UIKeyboardDockView
- (void)didMoveToWindow {
    self.backgroundColor = oledColor;
    %orig;
}
%end

%hook UIKeyboardLayoutStar 
- (void)didMoveToWindow {
    self.backgroundColor = oledColor;
    %orig;
}
%end

%hook UIKBRenderConfig // Prediction text color
- (void)setLightKeyboard:(BOOL)arg1 { %orig(NO); }
%end
%end

// YTReExplore: https://github.com/PoomSmart/YTReExplore/
%group gReExplore
static void replaceTab(YTIGuideResponse *response) {
    NSMutableArray <YTIGuideResponseSupportedRenderers *> *renderers = [response itemsArray];
    for (YTIGuideResponseSupportedRenderers *guideRenderers in renderers) {
        YTIPivotBarRenderer *pivotBarRenderer = [guideRenderers pivotBarRenderer];
        NSMutableArray <YTIPivotBarSupportedRenderers *> *items = [pivotBarRenderer itemsArray];
        NSUInteger shortIndex = [items indexOfObjectPassingTest:^BOOL(YTIPivotBarSupportedRenderers *renderers, NSUInteger idx, BOOL *stop) {
            return [[[renderers pivotBarItemRenderer] pivotIdentifier] isEqualToString:@"FEshorts"];
        }];
        if (shortIndex != NSNotFound) {
            [items removeObjectAtIndex:shortIndex];
            NSUInteger exploreIndex = [items indexOfObjectPassingTest:^BOOL(YTIPivotBarSupportedRenderers *renderers, NSUInteger idx, BOOL *stop) {
                return [[[renderers pivotBarItemRenderer] pivotIdentifier] isEqualToString:[%c(YTIBrowseRequest) browseIDForExploreTab]];
            }];
            if (exploreIndex == NSNotFound) {
                YTIPivotBarSupportedRenderers *exploreTab = [%c(YTIPivotBarRenderer) pivotSupportedRenderersWithBrowseId:[%c(YTIBrowseRequest) browseIDForExploreTab] title:@"Explore" iconType:292];
                [items insertObject:exploreTab atIndex:1];
            }
        }
    }
}
%hook YTGuideServiceCoordinator
- (void)handleResponse:(YTIGuideResponse *)response withCompletion:(id)completion {
    replaceTab(response);
    %orig(response, completion);
}
- (void)handleResponse:(YTIGuideResponse *)response error:(id)error completion:(id)completion {
    replaceTab(response);
    %orig(response, error, completion);
}
%end
%end

// BigYTMiniPlayer: https://github.com/Galactic-Dev/BigYTMiniPlayer
%group Main
%hook YTWatchMiniBarView
- (void)setWatchMiniPlayerLayout:(int)arg1 {
    %orig(1);
}
- (int)watchMiniPlayerLayout {
    return 1;
}
- (void)layoutSubviews {
    %orig;
    self.frame = CGRectMake(([UIScreen mainScreen].bounds.size.width - self.frame.size.width), self.frame.origin.y, self.frame.size.width, self.frame.size.height);
}
%end

%hook YTMainAppVideoPlayerOverlayView
- (BOOL)isUserInteractionEnabled {
    if([[self _viewControllerForAncestor].parentViewController.parentViewController isKindOfClass:%c(YTWatchMiniBarViewController)]) {
        return NO;
    }
    return %orig;
}
%end
%end

%ctor {
    %init;
    if (oled() && ([[NSUserDefaults standardUserDefaults] integerForKey:@"page_style"] == 1)) {
       %init(gOLED);
    }
    if (oledKB()) {
       %init(gOLEDKB);
    }
    if (ReExplore()) {
       %init(gReExplore);
    }
    if (bigYTMiniPlayer() && (UIDevice.currentDevice.userInterfaceIdiom != UIUserInterfaceIdiomPad)) {
       %init(Main);
    }
}