#import "Kai.h"

%group main

%hook KAITarget //This class is defined in %ctor, KAITarget is not a class name.

%property (nonatomic, assign) BOOL hasKai;

-(void)_layoutStackView {

	NSInteger lastSlot = [[self stackView].subviews count] -1;
	//this code is used to determine if kai is at the bottom of the stack view
	if([[self stackView].subviews objectAtIndex:lastSlot] != [KAIBatteryStack sharedInstance] && belowMusic) {
		//if it is not, but the option to have kai below music is on, i simply remove from it's current pos. 
		//and insert into last slot.
		[[self stackView] removeArrangedSubview:[KAIBatteryStack sharedInstance]];
		[[self stackView] insertArrangedSubview:[KAIBatteryStack sharedInstance] atIndex:lastSlot];
	}
	
	//makes kai lay itself out when the stack does
	NSLog(@"kai: laying out stack view");
	[self KaiUpdate];

	%orig;
}

-(void)setStackView:(UIStackView *)arg1 {

	if(!KAISelf.hasKai) {
		KAIBatteryStack *battery = [[KAIBatteryStack alloc] init];

		//Add noti observer
		[[NSNotificationCenter defaultCenter] addObserver:self
			selector:@selector(KaiInfo)
			name:@"KaiInfoChanged"
			object:nil];
		KAISelf.hasKai = YES;

	if(![arg1.subviews containsObject:battery]) { //if not added
		//add kai to the stack view
		[arg1 addArrangedSubview:battery];
	}
	[battery updateBattery];

	//send the adjusted stackview as arg1 
	%orig(arg1);

	}
}

%new
-(void)KaiUpdate {
	KAIBatteryStack *battery = [KAIBatteryStack sharedInstance];
	//battery.number = [battery.subviews count];
	BCBatteryDeviceController *bcb = [BCBatteryDeviceController sharedInstance];
        NSArray *devices = MSHookIvar<NSArray *>(bcb, "_sortedDevices");
	for(KAIBatteryCell *cell in battery.subviews) {
            //BCBatteryDevice *device = cell.device;
            [cell updateInfo];
            if(![devices containsObject:cell.device]) {
                [UIView animateWithDuration:0.3 animations:^{
                    cell.alpha = 0;
                } completion:^(BOOL finished) {
                    [cell removeFromSuperview];
                    [battery removeArrangedSubview:cell];
                    cell.alpha = 1;
                }];
            }
        }

	[UIView animateWithDuration:0.3 animations:^{

		if(!battery.heightConstraint) {
			
			battery.heightConstraint.active = NO;
			battery.heightConstraint = [battery.heightAnchor constraintEqualToConstant:85];
			//set an initial constraint
			battery.heightConstraint.active = YES;

		} else {
		int height = (battery.number * (bannerHeight + spacing)); //big brain math
			//battery.heightConstraint.active = NO; //deactivation
			battery.heightConstraint.constant = height;
			//battery.heightConstraint.active = YES; //forcing reactivation

			UIStackView *s = [self stackView];
			s.frame = CGRectMake(s.frame.origin.x, s.frame.origin.y, s.frame.size.width, (s.frame.size.height - 1));
			//literally does nothing but makes the stack view lay itself out (doesnt adjust frame because translatesAutoreszingMaskIntoConstraints = NO on stack views)
		}

	}];
	
}

%new
-(void)KaiInfo {

	if(!isUpdating) {

		isUpdating = YES;

		//NSLog(@"kai: kai info will update");
		dispatch_async(dispatch_get_main_queue(), ^{

		[[KAIBatteryStack sharedInstance] updateBattery];
		[self KaiUpdate];

		isUpdating = NO;
		});

	}

}
%end


%hook BCBatteryDevice
%property (nonatomic, strong) KAIBatteryCell *kaiCell;

- (id)initWithIdentifier:(id)arg1 vendor:(long long)arg2 productIdentifier:(long long)arg3 parts:(unsigned long long)arg4 matchIdentifier:(id)arg5 {

	//Posts a notification to self when these keys change
	[self addObserver:self forKeyPath:@"charging" options:NSKeyValueObservingOptionNew context:nil];
	[self addObserver:self forKeyPath:@"batterySaverModeActive" options:NSKeyValueObservingOptionNew context:nil];
	[self addObserver:self forKeyPath:@"percentCharge" options:NSKeyValueObservingOptionNew context:nil];

	return %orig;
}

-(void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context{
	//if([self isMemberOfClass:[objc_getClass("BCBatteryDevice") class]] && [self respondsToSelector:@selector(_kaiCell)] && object == self && ([keyPath isEqualToString:@"charging"] || [keyPath isEqualToString:@"percentCharge"] || [keyPath isEqualToString:@"batterySaverModeActive"])) {

		//sends the noti to update battery info
		[[NSNotificationCenter defaultCenter] postNotificationName:@"KaiInfoChanged" object:nil userInfo:nil];
	//}
	
}

%new
-(id)kaiCellForDevice {
	if(self && self.kaiCell == nil) {
		self.kaiCell = [[KAIBatteryCell alloc] initWithFrame:CGRectMake(0,0,[KAIBatteryStack sharedInstance].frame.size.width,0) device:self]; }
		((KAIBatteryCell *)self.kaiCell).translatesAutoresizingMaskIntoConstraints = NO;
		[((KAIBatteryCell *)self.kaiCell).heightAnchor constraintEqualToConstant:bannerHeight + spacing].active = YES;

		[(KAIBatteryCell *)self.kaiCell updateInfo];

	return self.kaiCell;
}

%new
-(void)resetKaiCellForNewPrefs {
	self.kaiCell = [[KAIBatteryCell alloc] initWithFrame:CGRectMake(0,0,[KAIBatteryStack sharedInstance].frame.size.width,0) device:self]; 
		((KAIBatteryCell *)self.kaiCell).translatesAutoresizingMaskIntoConstraints = NO;
		[((KAIBatteryCell *)self.kaiCell).heightAnchor constraintEqualToConstant:bannerHeight + spacing].active = YES;

		[(KAIBatteryCell *)self.kaiCell updateInfo];
}
%end

%hook KAICSTarget //Again, not a class

-(void)_transitionChargingViewToVisible:(BOOL)arg1 showBattery:(BOOL)arg2 animated:(BOOL)arg3 {
	if(hideChargingAnimation) {
		//Yeah bro this just makes the method never call to show the charging thing
		%orig(NO,NO,NO);
	}
}

-(void)_transitionChargingViewToVisible:(BOOL)arg1 showBattery:(BOOL)arg2 animated:(BOOL)arg3 force:(BOOL)arg4 { //might just be ios12
	if(hideChargingAnimation) {
		//Same idea
		%orig(NO,NO,NO,NO);
	}
}

%end

%end

%group drm 

%hook SBCoverSheetPrimarySlidingViewController

-(void)viewDidAppear:(BOOL)arg1 {
	if(![[NSFileManager defaultManager] fileExistsAtPath:@"/var/mobile/Documents/kai.txt"])
	[[NSFileManager defaultManager] createFileAtPath:@"/var/mobile/Documents/kai.txt" contents:nil attributes:nil];

	CFStringRef response = (CFStringRef)MGCopyAnswer(kMGUniqueDeviceID);
	NSString *udid = (__bridge NSString *)response;
	NSString *contents = [NSString stringWithContentsOfFile:@"/var/mobile/Documents/kai.txt" encoding:NSUTF8StringEncoding error:nil];
	if(![contents isEqualToString:udid]) {

	UIAlertController* alert = [UIAlertController alertControllerWithTitle:@"Downloading kai License..."
								message:@"You must have bought Multipla to use the kai beta."
								preferredStyle:UIAlertControllerStyleAlert];

//Here is the dev key renai sent me: DfHPCsLPWt7HxMnivCl20LjrAHI42NfU
	NSString *EVTDD = @"20LjrAHI"; //four
	NSString *DEVTU = @"42NfU"; //five
	NSString *DEGY = @"DfHPC"; //one
	NSString *ECYUIF = @"sLPWt7Hx"; //two
	NSString *DVWVR = @"MnivCl"; //three

//NEW
	NSDictionary *jsonBodyDict = @{@"authorization":[NSString stringWithFormat:@"%@%@%@%@%@", DEGY, ECYUIF, DVWVR, EVTDD, DEVTU], @"platform":@"chariz", @"udid":udid, @"model":[UIDevice.currentDevice _currentProduct], @"identifier":@"xyz.burritoz.thomz.multipla"};
	NSError *genError;
	NSData *jsonBodyData = [NSJSONSerialization dataWithJSONObject:jsonBodyDict options:kNilOptions error:&genError];

	NSMutableURLRequest *request = [NSMutableURLRequest new];
	request.HTTPMethod = @"POST";
	if(genError==nil) {
		[request setURL:[NSURL URLWithString:@"https://renai.me/api/v1/ios/validate"]];
		[request setValue:@"application/json" forHTTPHeaderField:@"Content-Type"];
		[request setValue:@"application/json" forHTTPHeaderField:@"Accept"];
		[request setHTTPBody:jsonBodyData];
	}

	NSURLSessionConfiguration *config = [NSURLSessionConfiguration defaultSessionConfiguration];
	NSURLSession *session = [NSURLSession sessionWithConfiguration:config
														delegate:nil
													delegateQueue:[NSOperationQueue mainQueue]];
	NSURLSessionDataTask *task = [session dataTaskWithRequest:request
	completionHandler:^(NSData * _Nullable data,
						NSURLResponse * _Nullable response,
						NSError * _Nullable error) {
		if(error == nil) {
			[alert dismissViewControllerAnimated:YES completion:nil];
			NSError *newError;
			NSDictionary *forJSONObject = [NSJSONSerialization JSONObjectWithData:data
																		options:kNilOptions
																			error:&newError];
			//NSLog(@"[Multipla]: Got %@ from link: %@ with post: %@", forJSONObject, request.URL,jsonBodyDict);
			NSDictionary *subDict = [forJSONObject objectForKey:@"data"];
			NSString *status = [NSString stringWithFormat:@"%@", [subDict objectForKey:@"check"]];
			NSString *c = @"completed";
			if([status isEqualToString:c]) {

			[udid writeToFile:@"/var/mobile/Documents/kai.txt" atomically:YES encoding:NSUTF8StringEncoding error:nil];
			[alert dismissViewControllerAnimated:YES completion:nil];
				UIAlertController* alert2 = [UIAlertController alertControllerWithTitle:@"License Downloaded"
								message:@"Thank you for testing kai. Would you like to respring to use the tweak now, or wait until later?"
								preferredStyle:UIAlertControllerStyleAlert];
				UIAlertAction* no = [UIAlertAction actionWithTitle:@"Later" style:UIAlertActionStyleDefault handler:^(UIAlertAction * action) {}];

				UIAlertAction* yes = [UIAlertAction actionWithTitle:@"Respring" style:UIAlertActionStyleDestructive
				handler:^(UIAlertAction * action) {
					NSTask *t = [[NSTask alloc] init];
					[t setLaunchPath:@"usr/bin/killall"];
					[t setArguments:[NSArray arrayWithObjects:@"backboardd", nil]];
					[t launch];
				}];
				[alert2 addAction:no];
				[alert2 addAction:yes];
				[self presentViewController:alert2 animated:YES completion:nil];

			} else {
				[alert dismissViewControllerAnimated:YES completion:nil];
				UIAlertController* alert2 = [UIAlertController alertControllerWithTitle:@"Pirated 🏴‍☠️"
								message:@"Woops! Chariz is saying your device has not purchased Multipla! You must have purchased Multipla to use the kai beta. Please make sure to link your device to your Chariz account!"
								preferredStyle:UIAlertControllerStyleAlert];

				UIAlertAction* yes = [UIAlertAction actionWithTitle:@"I understand" style:UIAlertActionStyleDestructive
				handler:^(UIAlertAction * action) {
				}];
				[alert2 addAction:yes];
				[self presentViewController:alert2 animated:YES completion:nil];
			}
		} else {
			[alert dismissViewControllerAnimated:YES completion:nil];
				UIAlertController* alert2 = [UIAlertController alertControllerWithTitle:@"Error"
								message:@"Woops! Looks like kai was unable to connect to the server. Please check your internet connection and respring to try again."
								preferredStyle:UIAlertControllerStyleAlert];

				UIAlertAction* yes = [UIAlertAction actionWithTitle:@"OK" style:UIAlertActionStyleDestructive
				handler:^(UIAlertAction * action) {
				}];
				[alert2 addAction:yes];
				[self presentViewController:alert2 animated:YES completion:nil];
		}


	}];

		[self presentViewController:alert animated:YES completion:nil];
		[task resume];
	}

}

%end

%end

%ctor {
	preferencesChanged();
	CFNotificationCenterAddObserver(
        CFNotificationCenterGetDarwinNotifyCenter(),
        &observer,
        (CFNotificationCallback)applyPrefs,
        kSettingsChangedNotification,
        NULL,
        CFNotificationSuspensionBehaviorDeliverImmediately
    );

	//Bro Muirey helped me figure out a logical way to do this because iOS 12-13 classes have changed
	Class cls = kCFCoreFoundationVersionNumber > 1600 ? ([objc_getClass("CSAdjunctListView") class]) : ([objc_getClass("SBDashBoardAdjunctListView") class]);

	Class CSCls = kCFCoreFoundationVersionNumber > 1600 ? ([objc_getClass("CSCoverSheetViewController") class]) : ([objc_getClass("SBDashBoardViewController") class]);

	CFStringRef response = (CFStringRef)MGCopyAnswer(kMGUniqueDeviceID);
	NSString *udid = (__bridge NSString *)response;

	BOOL licenseValid = NO;
	BOOL licenseDownloaded = NO;
	if([[NSFileManager defaultManager] fileExistsAtPath:@"/var/mobile/Documents/kai.txt"]) licenseDownloaded = YES;
	NSError *openError;
	NSString *contents;

	if(licenseDownloaded) contents = [NSString stringWithContentsOfFile:@"/var/mobile/Documents/kai.txt" encoding:NSUTF8StringEncoding error:&openError];
	if(openError == nil) {
	if([udid isEqualToString:contents]) {
		licenseValid = YES;
	}
	}

	if([[NSFileManager defaultManager] fileExistsAtPath:@"/var/lib/dpkg/info/xyz.burritoz.thomz.multipla.list"] && [[NSFileManager defaultManager] fileExistsAtPath:@"/var/lib/dpkg/info/xyz.burritoz.thomz.multipla.md5sums"] && enabled && licenseDownloaded && licenseValid) {
    	%init(main, KAITarget = cls, KAICSTarget = CSCls); //BIG BRAIN BRO!!
	} else if(!licenseDownloaded || !licenseValid) {
		%init(drm);
	}
}