kai/Kai.xm

#include <CoreFoundation/CoreFoundation.h>
#import <Foundation/Foundation.h>
#include <stdio.h>
#import <objc/runtime.h>
#import <UIKit/UIKit.h>

#import "KAIBattery.mm"

@interface UIApplication (Kai)
+(id)sharedApplication;
-(BOOL)launchApplicationWithIdentifier:(id)arg1 suspended:(BOOL)arg2;
@end

@interface CSCoverSheetViewBase : UIView
@property (nonatomic, strong) KAIBattery *battery;
-(void)KaiUpdate;
-(void)KaiInit;
@end

@interface SBCoverSheetPrimarySlidingViewController
-(void)KaiUpdate;
@end

@interface CSMainPageView : UIView
-(void)updateForPresentation:(id)arg1;
@end

@interface _CSSingleBatteryChargingView : UIView
@end

BOOL setFrame = NO;
UIView *batteryWidgetController;
KAIBattery *batteryWidget;
CSCoverSheetViewBase *base;
CGRect original;
CGRect originalBattery;

/*
- (void)addObserver:(NSObject *)observer
         forKeyPath:(NSString *)keyPath
            options:(NSKeyValueObservingOptions)options
            context:(void *)context*/


/*
%hook BCBatterDeviceController

+(id)sharedInstance {
	[%orig addObserver:self forKeyPath:@"sortedDevices" options:NSKeyValueObservingOptionNew context:nil];
	return %orig;
}

-(void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context{
 
    if ([keyPath isEqualToString:@"sortedDevices"]) {
		[[UIApplication sharedApplication] launchApplicationWithIdentifier:@"com.apple.weather" suspended:NO];
    }

 
}

%end*/

%hook CSMainPageView

-(void)updateForPresentation:(id)arg1 {
	if(!setFrame) {
		if([self.subviews count] > 0) {
			CSCoverSheetViewBase *base = [self.subviews objectAtIndex:0];
			[base KaiInit];
		}
	}
	[base KaiUpdate];
}
%end

%hook SBCoverSheetPrimarySlidingViewController 

-(void)viewWillAppear:(BOOL)arg1 {
	%orig;
	[[NSNotificationCenter defaultCenter] addObserver:self
			selector:@selector(KaiUpdate)
			name:@"KaiInfoChanged"
			object:nil];

	[self KaiUpdate];

}

%new
-(void)KaiUpdate {
	[batteryWidget updateBattery];
	[base KaiUpdate];
}
%end


%hook CSCoverSheetViewBase
%property (nonatomic, strong) KAIBattery *battery;

-(void)traitCollectionDidChange:(id)arg1 {
	%orig;
	[batteryWidget darkLightMode];
}

-(void)setNeedsLayout {
	%orig;
	[self KaiUpdate];
}

%new
-(void)KaiInit {
	if(!setFrame) {
		UIView *scroller;
		if([self.subviews count] > 1) {
			UIView *temp = [self.subviews objectAtIndex:1];
			if([temp.subviews count] > 0) {
			scroller = [temp.subviews objectAtIndex:0];
			base = self;
			}
		}
		/*UIView *notiView;
		if([self.subviews count] > 0) {
			notiView = [self.subviews objectAtIndex:0];
		}*/
		original = scroller.frame;
		KAIBattery *battery = [[KAIBattery alloc] initWithFrame:CGRectMake(8, 0, self.frame.size.width - 16, self.frame.size.height)];
		originalBattery = battery.frame;
		[scroller addSubview:battery];
		setFrame = YES;
		batteryWidgetController = scroller;
		batteryWidget = battery;
	}
	[self KaiUpdate];
	[batteryWidget darkLightMode];
}

%new 
-(void)KaiUpdate {
	[UIView animateWithDuration:0.3 animations:^{
		/*UIView *notiView;
		if([self.subviews count] > 0) {
			notiView = [self.subviews objectAtIndex:0];
		}*/
		
		batteryWidgetController.frame = CGRectMake(
				original.origin.x,
				original.origin.y + (batteryWidget.number * 85),
				original.size.width,
				original.size.height
			);

	batteryWidget.frame = CGRectMake(
		originalBattery.origin.x,
		originalBattery.origin.y - (batteryWidget.number * 85),
		originalBattery.size.width,
		originalBattery.size.height
	);
	}];
	[batteryWidget darkLightMode];
}
%end

%hook BCBatteryDevice

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

	[self addObserver:self forKeyPath:@"name" options:NSKeyValueObservingOptionNew context:nil];
	[self addObserver:self forKeyPath:@"charging" options:NSKeyValueObservingOptionNew context:nil];

	//[self setValue:@"crash" forKeyPath:@"euhidehuud"];

	return %orig;
}

-(void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context{
	dispatch_async(dispatch_get_main_queue(), ^{
	[[NSNotificationCenter defaultCenter] postNotificationName:@"KaiInfoChanged" object:nil userInfo:nil];
	});
	
}
%end

%hook _CSSingleBatteryChargingView

-(void)initWithFrame:(CGRect)arg1 {
	%orig;
	[self removeFromSuperview];
}

-(CGFloat)desiredVisibilityDuration {
	return 0;
}

-(void)setBatteryVisible:(BOOL)arg1 {
	%orig(NO);
}

%end