Tweak.x

#import <UIKit/UIKit.h>
#import "NSTask.h"

@interface SBHWidgetContainerViewController : UIViewController
@property (nonatomic,copy) NSString * applicationName;
-(void)createPowerWidget;
@end

UIView *overlay;
UIView *respringContainer;
UITapGestureRecognizer *respringTap;
UIImageView *respringButton;
UIView *safeModeContainer;
UITapGestureRecognizer *safeModeTap;
UIImageView *safeModeButton;
UIView *rebootContainer;
UITapGestureRecognizer *rebootTap;
UIImageView *rebootButton;
UIView *uicacheContainer;
UITapGestureRecognizer *uicacheTap;
UIImageView *uicacheButton;

static NSTimer *visibilityTimer = nil;

%hook SBHWidgetContainerViewController
-(void)viewWillAppear:(BOOL)arg1{
  %orig;
  if ([visibilityTimer isValid]){
    [visibilityTimer invalidate];
    visibilityTimer = nil;
  }
  [self createPowerWidget];

  overlay.alpha = 0;
  [UIView animateWithDuration:0.5 animations:^{
    overlay.alpha = 1;
  }];
}

-(void)viewWillDisappear:(BOOL)arg1{
  %orig;
  if (![visibilityTimer isValid]){
    [visibilityTimer isValid];
    visibilityTimer = [NSTimer scheduledTimerWithTimeInterval:2.0 target:self selector:@selector(freePW) userInfo:nil repeats:NO];
  }
}

%new -(void)freePW{
  [UIView animateWithDuration:0.5 animations:^{
    overlay.alpha = 0;
    } completion: ^(BOOL finished) {
    [overlay removeFromSuperview];
  }];

  if ([visibilityTimer isValid]){
    [visibilityTimer invalidate];
    visibilityTimer = nil;
  }
}

%new -(void)createPowerWidget{
  if ([self.applicationName isEqualToString:@"Power"]){
      overlay = [[UIView alloc] initWithFrame:CGRectMake(0,0,self.viewIfLoaded.frame.size.width, self.viewIfLoaded.frame.size.height)];
      overlay.backgroundColor = [UIColor blackColor];
      [self.view addSubview:overlay];
      [self.view bringSubviewToFront:overlay];

      //Respring
      respringContainer = [[UIView alloc] init];
      [respringContainer.layer setCornerRadius:13];
      respringContainer.clipsToBounds = YES;
      respringContainer.backgroundColor = [[UIColor grayColor] colorWithAlphaComponent:0.4];
      respringTap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(handleRespring:)];
      respringButton = [[UIImageView alloc]init];
      respringButton.image = [[UIImage imageNamed:@"/Library/Application Support/PowerWidget/respringIcon.png"] imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate];
      [respringButton setTintColor:[UIColor whiteColor]];


      //Safe mode
      safeModeContainer = [[UIView alloc] init];
      [safeModeContainer.layer setCornerRadius:13];
      safeModeContainer.clipsToBounds = YES;
      safeModeContainer.backgroundColor = [[UIColor grayColor] colorWithAlphaComponent:0.4];
      safeModeTap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(handleSafeMode:)];
      safeModeButton = [[UIImageView alloc]init];
      safeModeButton.image = [[UIImage imageNamed:@"/Library/Application Support/PowerWidget/safemodeIcon.png"] imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate];
      [safeModeButton setTintColor:[UIColor whiteColor]];


      //Reboot
      rebootContainer = [[UIView alloc] init];
      [rebootContainer.layer setCornerRadius:13];
      rebootContainer.clipsToBounds = YES;
      rebootContainer.backgroundColor = [[UIColor grayColor] colorWithAlphaComponent:0.4];
      rebootTap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(handleReboot:)];
      rebootButton = [[UIImageView alloc]init];
      rebootButton.image = [[UIImage imageNamed:@"/Library/Application Support/PowerWidget/rebootIcon.png"] imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate];
      [rebootButton setTintColor:[UIColor whiteColor]];


      //UICache
      uicacheContainer = [[UIView alloc] init];
      [uicacheContainer.layer setCornerRadius:13];
      uicacheContainer.clipsToBounds = YES;
      uicacheContainer.backgroundColor = [[UIColor grayColor] colorWithAlphaComponent:0.4];
      uicacheTap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(handleUIcache:)];
      uicacheButton = [[UIImageView alloc]init];
      uicacheButton.image = [[UIImage imageNamed:@"/Library/Application Support/PowerWidget/uicacheIcon.png"] imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate];
      [uicacheButton setTintColor:[UIColor whiteColor]];

      if (!(self.viewIfLoaded.frame.size.width > 300)){
        respringContainer.frame = CGRectMake(10,10, self.viewIfLoaded.frame.size.width/2 - 15, self.viewIfLoaded.frame.size.height/2 - 15);
        safeModeContainer.frame = CGRectMake(10,respringContainer.frame.origin.y + respringContainer.frame.size.height + 10, respringContainer.frame.size.width, respringContainer.frame.size.height);
        rebootContainer.frame = CGRectMake(respringContainer.frame.origin.x + respringContainer.frame.size.width + 10,10, respringContainer.frame.size.width, respringContainer.frame.size.height);
        uicacheContainer.frame = CGRectMake(rebootContainer.frame.origin.x,rebootContainer.frame.origin.y + rebootContainer.frame.size.height + 10, respringContainer.frame.size.width, respringContainer.frame.size.height);
      } else {
        respringContainer.frame = CGRectMake(10,10, self.viewIfLoaded.frame.size.width/4 - 12.5, self.viewIfLoaded.frame.size.height - 20);
        safeModeContainer.frame = CGRectMake(respringContainer.frame.origin.x + respringContainer.frame.size.width + 10,10, respringContainer.frame.size.width, respringContainer.frame.size.height);
        rebootContainer.frame = CGRectMake(safeModeContainer.frame.origin.x + safeModeContainer.frame.size.width + 10,10, respringContainer.frame.size.width, respringContainer.frame.size.height);
        uicacheContainer.frame = CGRectMake(rebootContainer.frame.origin.x + rebootContainer.frame.size.width + 10,10, respringContainer.frame.size.width, respringContainer.frame.size.height);
      }

      respringButton.frame = CGRectMake(respringContainer.frame.size.width/2 - 25,respringContainer.frame.size.height/2 - 25,50,50);
      safeModeButton.frame = CGRectMake(safeModeContainer.frame.size.width/2 - 25,safeModeContainer.frame.size.height/2 - 25,50,50);
      rebootButton.frame = CGRectMake(rebootContainer.frame.size.width/2 - 25,rebootContainer.frame.size.height/2 - 25,50,50);
      uicacheButton.frame = CGRectMake(rebootContainer.frame.size.width/2 - 25,rebootContainer.frame.size.height/2 - 25,50,50);

      [overlay addSubview:respringContainer];
      [respringContainer addGestureRecognizer:respringTap];
      [respringContainer addSubview:respringButton];
      [overlay addSubview:safeModeContainer];
      [safeModeContainer addGestureRecognizer:safeModeTap];
      [safeModeContainer addSubview:safeModeButton];
      [overlay addSubview:rebootContainer];
      [rebootContainer addGestureRecognizer:rebootTap];
      [rebootContainer addSubview:rebootButton];
      [overlay addSubview:uicacheContainer];
      [uicacheContainer addGestureRecognizer:uicacheTap];
      [uicacheContainer addSubview:uicacheButton];
  }
}

%new -(void)handleRespring:(UITapGestureRecognizer *)recognizer
{
  NSTask *task = [[NSTask alloc] init];
  [task setLaunchPath:@"/bin/bash"];
  [task setArguments:@[@"-c", @"/usr/bin/killall SpringBoard"]];
  [task launch];
}

%new -(void)handleSafeMode:(UITapGestureRecognizer *)recognizer
{
  NSTask *task = [[NSTask alloc] init];
  [task setLaunchPath:@"/bin/bash"];
  [task setArguments:@[@"-c", @"/usr/bin/killall -SEGV SpringBoard"]];
  [task launch];
}

%new -(void)handleReboot:(UITapGestureRecognizer *)recognizer
{
  NSTask *task = [[NSTask alloc] init];
  [task setLaunchPath:@"/bin/bash"];
  [task setArguments:@[@"-c", @"launchctl reboot userspace"]];
  [task launch];
}

%new -(void)handleUIcache:(UITapGestureRecognizer *)recognizer
{

  NSFileManager *fileManager = [NSFileManager defaultManager];
  NSString *pathForFile = @"/Applications/Sileo.app/Sileo";

  if ([fileManager fileExistsAtPath:pathForFile]){
    NSTask *task = [[NSTask alloc] init];
    [task setLaunchPath:@"/bin/bash"];
    [task setArguments:@[@"-c", @"uicache --all"]];
    [task launch];
  } else {
    NSTask *task = [[NSTask alloc] init];
    [task setLaunchPath:@"/bin/bash"];
    [task setArguments:@[@"-c", @"uicache"]];
    [task launch];
  }
}

%end