Tweak.xm

#import "Tweak.h"

%group VibinTweak

%hook AXNView
-(id)initWithFrame:(CGRect)frame {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(hideView) name:@"hideNotificationsOnCS" object:nil];
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(showView) name:@"showNotificationsOnCS" object:nil];
    });
	return %orig;
}
%new
-(void)hideView {
	[UIView animateWithDuration:0.8 animations:^{
		self.alpha = 0;
	}];
}
%new
-(void)showView {
	[UIView animateWithDuration:0.8 animations:^{
		self.alpha = 1;
	}];
}
%end

%hook NCNotificationStructuredListViewController
-(void)viewDidAppear:(BOOL)animated {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(hideNotifs) name:@"hideNotificationsOnCS" object:nil];
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(showNotifs) name:@"showNotificationsOnCS" object:nil];
    });
	%orig;
}
%new
-(void)hideNotifs {
	for (UIView *view in self.masterListView.subviews) {
		if ([view isKindOfClass:%c(NCNotificationListView)]) {
			[UIView animateWithDuration:0.8 animations:^{
				view.alpha = 0;
			}];
			dispatch_time_t delay = dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 0.8);
			dispatch_after(delay, dispatch_get_main_queue(), ^(void){
				view.hidden = YES;
			});
		}
	}
}
%new
-(void)showNotifs {
	for (UIView *view in self.masterListView.subviews) {
		if ([view isKindOfClass:%c(NCNotificationListView)]) {
			view.hidden = NO; // Reenables notifs
			view.alpha = 0; // Keeps them hidden
			[UIView animateWithDuration:0.8 animations:^{				
				view.alpha = 1; // Slowly fades them in
			}];
		}
	}
}
-(void)dealloc {
	%orig;
	[[NSNotificationCenter defaultCenter] removeObserver:self];
}
%end

void turnOnDND() {
	if (!assertionService) {
		assertionService = (DNDModeAssertionService *)[objc_getClass("DNDModeAssertionService") serviceForClientIdentifier:@"com.apple.donotdisturb.control-center.module"];
	}
	DNDModeAssertionDetails *newAssertion = [objc_getClass("DNDModeAssertionDetails") userRequestedAssertionDetailsWithIdentifier:@"com.apple.control-center.manual-toggle" modeIdentifier:@"com.apple.donotdisturb.mode.default" lifetime:nil];
	[assertionService takeModeAssertionWithDetails:newAssertion error:NULL];
}

void turnOffDND() {
	if (!assertionService) {
    	assertionService = (DNDModeAssertionService *)[objc_getClass("DNDModeAssertionService") serviceForClientIdentifier:@"com.apple.donotdisturb.control-center.module"];
    }
    [assertionService invalidateAllActiveModeAssertionsWithError:NULL];
}

%hook DNDState
-(BOOL)isActive {
	// Fetch current dnd state
	DNDEnabled = %orig;
	return DNDEnabled;
}
%end

// %hook CXCall
// -(BOOL)hasConnected {
// 	NSLog(@"hasConnected");
// 	return %orig;
// }
// -(void)setHasConnected:(BOOL)arg1 {
// 	NSLog(@"setHasConnected");
// 	%orig;
// }
// -(BOOL)hasEnded {
// 	NSLog(@"hasEnded");
// 	return %orig;
// }
// -(void)setHasEnded:(BOOL)arg1 {
// 	NSLog(@"setHasEnded");
// 	%orig;
// }
// %end

SBMediaController *mediaController;

%hook SBUIController
-(id)init {
	[[NSNotificationCenter defaultCenter] addObserver:self
	selector:@selector(currentSongChanged:)
	name:@"SBMediaNowPlayingChangedNotification"
	object:nil];

	mediaController = [%c(SBMediaController) sharedInstance];

	return %orig;
}
%new
- (void)currentSongChanged:(NSNotification *)notification {
    if (mediaController.isPlaying) {
		if (enableDND && enabled) {
			turnOnDND();
		}
		if (enableHideNotifs && enabled) {
			[[NSNotificationCenter defaultCenter] postNotificationName:@"hideNotificationsOnCS" object:self];
		}
    } else {
		if (enableDND && enabled) {
			turnOffDND();
		}
		if (enableHideNotifs && enabled) {
			[[NSNotificationCenter defaultCenter] postNotificationName:@"showNotificationsOnCS" object:self];
		}
	}
}
%end

// Thanks MrGcGamer - NoDNDBanner
%hook DNDNotificationsService
-(void)_queue_postOrRemoveNotificationWithUpdatedBehavior:(BOOL)arg1 significantTimeChange:(BOOL)arg2 {
	// Absolutely Nothing :)
}
%end

%end // %group VibinTweak

%ctor {
	preferences = [[HBPreferences alloc] initWithIdentifier:@"com.chr1s.vibinprefs"];
	[preferences registerBool:&enabled default:YES forKey:@"enabled"];
	[preferences registerBool:&enableDND default:YES forKey:@"enableDND"];
	[preferences registerBool:&enableHideNotifs default:YES forKey:@"enableHideNotifs"];

	// This is for the CCToggle (If there is a better way of doing the whole toggle thing PLS let me know)
	[preferences registerPreferenceChangeBlock:^(){
		enabled = [preferences boolForKey:@"enabled"];
	}];

	NSLog(@"hi");

	if (enabled) {
		%init(VibinTweak);
	}
}