1
0
mirror da https://github.com/Burrit0z/kai synced 2025-07-02 01:56:47 +00:00
Files
kai/KAIBatteryPlatter.mm
2020-12-25 00:23:34 -05:00

360 righe
13 KiB
Plaintext
Executable File

KAIBatteryPlatter *instance;
NSTimer *queueTimer = nil;
NSMutableArray *deviceNames = [[NSMutableArray alloc] init];
NSMutableArray *cellsForDeviceNames = [[NSMutableArray alloc] init];
@implementation KAIBatteryPlatter
- (instancetype)initWithFrame:(CGRect)arg1 {
self = [super initWithFrame:arg1];
instance = self;
if (self) {
self.stackHolder = [[UIView alloc] initWithFrame:arg1];
self.stack = [[KAIStackView alloc] init];
self.stack.axis = kaiAlign == 0 ? 1 : 0;
self.stack.distribution = UIStackViewDistributionFillEqually;
self.stack.spacing = kaiAlign == 0 ? 0 : spacingHorizontal;
self.stack.alignment = 0;
self.oldCountOfDevices = -100;
self.queued = NO;
[self setShowsHorizontalScrollIndicator:NO];
[self setShowsVerticalScrollIndicator:NO];
[self setMinimumZoomScale:1];
[self setMaximumZoomScale:1];
[self addSubview:self.stackHolder];
[self.stackHolder addSubview:self.stack];
[self setContentSize:self.stack.frame.size];
[self resetOffset];
// Add noti observer
[[NSNotificationCenter defaultCenter] addObserver:self
selector:@selector(resetOffset)
name:@"KaiResetOffset"
object:nil];
self.stackHolder.translatesAutoresizingMaskIntoConstraints = NO;
[self.stackHolder.heightAnchor constraintEqualToAnchor:self.heightAnchor].active = YES;
[self.stackHolder.widthAnchor constraintEqualToAnchor:self.widthAnchor].active = YES;
[self.stackHolder.centerYAnchor constraintEqualToAnchor:self.centerYAnchor].active = YES;
if (kaiAlign == 0) {
if (bannerAlign == 2) { // center
self.subviewAligner = [self.stack.centerXAnchor constraintEqualToAnchor:self.stackHolder.centerXAnchor constant:horizontalOffset];
} else if (bannerAlign == 1) { // left
self.subviewAligner = [self.stack.leftAnchor constraintEqualToAnchor:self.stackHolder.leftAnchor constant:horizontalOffset];
} else if (bannerAlign == 3) { // right
self.subviewAligner = [self.stack.rightAnchor constraintEqualToAnchor:self.stackHolder.rightAnchor constant:horizontalOffset];
}
self.subviewAligner.active = YES;
}
[self updateBattery];
}
return self;
}
- (void)resetOffset { // holy fucking shit i just read this method, what is this garbage????
if (kaiAlign != 0 && reAlignSelf) {
[UIView animateWithDuration:0.2 animations:^{
if (bannerAlign == 1) { // left
[self setContentOffset:CGPointMake(0 + horizontalOffset, self.contentOffset.y)];
self.contentInset = UIEdgeInsetsMake(0, 0, 0, 0);
} else if (bannerAlign == 2) { // center
[self setContentOffset:CGPointMake(((-1 * self.stackHolder.frame.size.width) / 2) + (self.stack.frame.size.width / 2) + horizontalOffset, self.contentOffset.y)];
CGFloat top = 0, left = 0;
if (self.contentSize.width < self.bounds.size.width) left = (self.bounds.size.width - self.contentSize.width) * 0.5f;
if (self.contentSize.height < self.bounds.size.height) top = (self.bounds.size.height - self.contentSize.height) * 0.5f;
self.contentInset = UIEdgeInsetsMake(top, left, top, left);
} else if (bannerAlign == 3) { // right
[self setContentOffset:CGPointMake((-1 * self.stackHolder.frame.size.width) + self.stack.frame.size.width + horizontalOffset, self.contentOffset.y)];
CGFloat top = 0, left = 0;
if(self.contentSize.width < self.bounds.size.width) left = (self.bounds.size.width - self.contentSize.width);
if(self.contentSize.height < self.bounds.size.height) top = (self.bounds.size.height - self.contentSize.height);
self.contentInset = UIEdgeInsetsMake(top, left, top, left);
}
}];
}
}
- (void)setClipsToBounds:(BOOL)arg1 {
[super setClipsToBounds:NO];
}
- (void)updateBattery {
if(self.isUpdating == YES) {
self.queued = YES;
return;
}
dispatch_async(dispatch_get_main_queue(), ^{
BCBatteryDeviceController *bcb = [BCBatteryDeviceController sharedInstance];
NSArray *devices = [bcb connectedDevices];
if (self.oldCountOfDevices == -100) {
self.oldCountOfDevices = [devices count] + 1;
}
for (KAIBatteryCell *cell in self.stack.subviews) {
[cell updateInfo];
}
if (!self.isUpdating && self.oldCountOfDevices != 0 && ([devices count] + 1 == self.oldCountOfDevices || [devices count] - 1 == self.oldCountOfDevices || [devices count] == self.oldCountOfDevices)) {
self.isUpdating = YES;
NSMutableArray *cellsToAdd = [[NSMutableArray alloc] init];
NSMutableArray *cellsToRemove = [[NSMutableArray alloc] init];
for (BCBatteryDevice *device in devices) {
KAIBatteryCell *cell = [device kaiCellForDevice];
BOOL charging = [device isCharging];
BOOL internal = [device isInternal];
BOOL shouldAdd = NO;
BOOL fake = [device isFake];
NSString *deviceName = [device name];
if (!fake) {
if (showAll) {
shouldAdd = YES;
} else if (showAllMinusInternal && !internal) {
shouldAdd = YES;
} else if (!showAll && charging) {
shouldAdd = YES;
}
}
if (!showPhone && internal) {
shouldAdd = NO;
}
if (![self.stack.subviews containsObject:cell] && shouldAdd && [devices containsObject:device]) {
if (![deviceNames containsObject:deviceName]) {
[cellsToAdd addObject:cell];
[deviceNames addObject:deviceName];
[cellsForDeviceNames addObject:cell];
} else {
for (int i = 0; i < [deviceNames count]; i++) {
if ([[deviceNames objectAtIndex:i] isEqualToString:deviceName]) {
KAIBatteryCell *cell = [cellsForDeviceNames objectAtIndex:i];
cell.device = device;
if ([cellsToRemove containsObject:cell])
[cellsToRemove removeObject:cell];
[cell updateInfo];
}
}
}
} else if ([self.stack.subviews containsObject:cell] && !shouldAdd) {
[cellsToRemove addObject:cell];
// this is where i stupidly removed from deviceName names :kekw:
}
}
for (KAIBatteryCell *cell in cellsToAdd) {
if ([cellsToRemove containsObject:cell])
[cellsToRemove removeObject:cell];
if (![self.stack.subviews containsObject:cell] && [devices containsObject:cell.device]) {
// add cell
cell.alpha = 0;
[self.stack addSubview:cell];
[self.stack addArrangedSubview:cell];
[UIView animateWithDuration:0.3
animations:^{
cell.alpha = 1;
}];
}
}
for (KAIBatteryCell *cell in cellsToRemove) {
// remove cell
[UIView animateWithDuration:0.3
animations:^{
cell.alpha = 0;
}
completion:^(BOOL finished) {
[cell removeFromSuperview];
[self.stack removeArrangedSubview:cell];
cell.alpha = 1;
}];
NSString *deviceName = [cell.device name];
[deviceNames removeObject:deviceName];
[cellsForDeviceNames removeObject:cell];
}
for (KAIBatteryCell *cell in self.stack.subviews) {
if (![devices containsObject:cell.device] || cell.device == nil) { // not existing, remove
NSString *deviceName = cell.label.text;
[UIView animateWithDuration:0.3
animations:^{
cell.alpha = 0;
}
completion:^(BOOL finished) {
[cell removeFromSuperview];
[self.stack removeArrangedSubview:cell];
cell.alpha = 1;
}];
[deviceNames removeObject:deviceName];
[cellsForDeviceNames removeObject:cell];
}
}
// isUpdating is set to NO in this block
// adds a cooldown, essentially
queueTimer = [NSTimer scheduledTimerWithTimeInterval:1.0 target:self selector:@selector(dispatchQueue) userInfo:nil repeats:NO];
}
self.oldCountOfDevices = [devices count];
[self calculateHeight];
if ([self.superview.superview.superview respondsToSelector:@selector(fixComplicationsViewFrame)]) {
[(NCNotificationListView *)(self.superview.superview.superview) fixComplicationsViewFrame];
}
[self setContentSize:self.stack.frame.size];
[self performSelector:@selector(resetOffset) withObject:self afterDelay:0.2];
});
}
- (void)setContentOffset:(CGPoint)arg1 {
[self setContentSize:self.stack.frame.size]; // sometimes the view gets "stuck", this fixes it
[super setContentOffset:CGPointMake(arg1.x, 0)];
}
- (void)calculateHeight {
self.number = [self.stack.subviews count];
if (self.number == 0) {
UIStackView *s = (UIStackView *)(self.superview);
s.frame = CGRectMake(s.frame.origin.x, s.frame.origin.y, s.frame.size.width, (s.frame.size.height - 1));
[s removeArrangedSubview:self];
[self removeFromSuperview];
} else if (self.number != 0 && self.superview == nil && shouldBeAdded == YES) {
Class cls = kCFCoreFoundationVersionNumber > 1600 ? ([objc_getClass("CSAdjunctListView") class]) : ([objc_getClass("SBDashBoardAdjunctListView") class]);
[cls reorderKai];
}
[UIView animateWithDuration:0.3
animations:^{
if (!self.heightConstraint) {
int height = (self.number * (bannerHeight + spacing));
if (kaiAlign != 0) {
height = bannerHeight + spacing;
}
if ([self.superview.subviews count] > 1) {
height = (height - spacing) + 1;
}
self.heightConstraint = [self.heightAnchor constraintEqualToConstant:height];
self.stack.heightConstraint = [self.stack.heightAnchor constraintEqualToConstant:height];
self.heightConstraint.active = YES;
self.stack.heightConstraint.active = YES;
[self setContentSize:self.stack.frame.size];
[self resetOffset];
} else {
int height = (self.number * (bannerHeight + spacing));
int extra = extraPaddingAfter ? spacing : 0;
if (kaiAlign == 0) {
// self.stack.widthConstraint.constant = bannerWidthFactor;
} else {
height = bannerHeight + spacing;
}
height = height + extra;
if ([self.superview.subviews count] > 1) {
height = (height - spacing + 1);
}
self.heightConstraint.constant = height;
self.stack.heightConstraint.constant = height - extra; // minus extra because it will stretch cell spacing otherwise
UIStackView *s = (UIStackView *)(self.superview);
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)
}
[self setContentSize:self.stack.frame.size];
[self resetOffset];
}];
}
- (void)removeFromSuperview {
[self.superview setNeedsLayout];
[super removeFromSuperview];
}
- (void)refreshForPrefs {
self.stack.spacing = kaiAlign == 0 ? 0 : spacingHorizontal;
[self setContentSize:self.stack.frame.size];
for (UIView *view in self.stack.subviews) {
@try {
[view removeFromSuperview];
} @catch (NSException *exception) {
// Panik
}
}
BCBatteryDeviceController *bcb = [BCBatteryDeviceController sharedInstance];
NSArray *devices = [bcb connectedDevices];
for (BCBatteryDevice *device in devices) {
[device resetKaiCellForNewPrefs];
}
if (kaiAlign == 0) {
self.subviewAligner.active = NO;
if (bannerAlign == 2) { // center
self.subviewAligner = [self.stack.centerXAnchor constraintEqualToAnchor:self.stackHolder.centerXAnchor constant:horizontalOffset];
} else if (bannerAlign == 1) { // left
self.subviewAligner = [self.stack.leftAnchor constraintEqualToAnchor:self.stackHolder.leftAnchor constant:horizontalOffset];
} else if (bannerAlign == 3) { // right
self.subviewAligner = [self.stack.rightAnchor constraintEqualToAnchor:self.stackHolder.rightAnchor constant:horizontalOffset];
}
self.subviewAligner.active = YES;
}
[cellsForDeviceNames removeAllObjects];
[deviceNames removeAllObjects];
[self updateBattery];
}
- (void)dispatchQueue {
[queueTimer invalidate];
queueTimer = nil;
self.isUpdating = NO;
if (self.queued) {
self.queued = NO;
[self updateBattery];
if ([self.superview.superview.superview respondsToSelector:@selector(fixComplicationsViewFrame)]) {
[(NCNotificationListView *)(self.superview.superview.superview) fixComplicationsViewFrame];
}
}
}
+ (KAIBatteryPlatter *)sharedInstance {
return instance;
}
// This is for compatibility (did i spell that right?)
// basically this fixes it crashing in landscape:
- (void)setSizeToMimic:(CGSize)arg1 {
}
@end