AVLock/Tweak.xm

#define kBundlePath @"/Library/Application Support/AVLockBundle.bundle"

@interface SBOrientationLockManager
+(SBOrientationLockManager *)sharedInstance;
-(bool)isUserLocked;
-(void)lock;
-(void)unlock;
-(void)myIsUserLocked;
@end


@interface AVButton : UIView
@end

@interface AVTransportControlsView : UIView
-(void)deviceOrientationDidChange;
@property (assign, nonatomic) AVButton *skipBackButton;
@property (assign, nonatomic) AVButton *skipForwardButton;

@property (assign, nonatomic) AVButton *standardPlayPauseButton;

@end

//#import <AudioToolbox/AudioToolbox.h>
//#import <AudioToolbox/AudioServices.h>

extern "C" CFNotificationCenterRef CFNotificationCenterGetDistributedCenter(void);

static UIButton *button = nil;
static NSString *myObserver=@"anObserver";
static void toggle(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo){

    if([[%c(SBOrientationLockManager) sharedInstance] isUserLocked]){
        [[%c(SBOrientationLockManager) sharedInstance] unlock];
    }else{
        [[%c(SBOrientationLockManager) sharedInstance] lock];
    }
}
static void lockButton(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo){
    NSBundle *bundle = [[[NSBundle alloc] initWithPath:kBundlePath] autorelease];
    NSString *imagePath = [bundle pathForResource:@"Locked@3x" ofType:@"png"];
    UIImage *img = [UIImage imageWithContentsOfFile:imagePath];
    [button setImage:img forState:UIControlStateNormal];
    [button setImage:img forState:UIControlStateHighlighted];
    [button setImage:img forState:UIControlStateSelected];

}
static void unlockButton(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo){
    NSBundle *bundle = [[[NSBundle alloc] initWithPath:kBundlePath] autorelease];
    NSString *imagePath = [bundle pathForResource:@"Unlocked@3x" ofType:@"png"];
    UIImage *img = [UIImage imageWithContentsOfFile:imagePath];
    [button setImage:img forState:UIControlStateNormal];
    [button setImage:img forState:UIControlStateHighlighted];
    [button setImage:img forState:UIControlStateSelected];

}
static void firstUpdate(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo){
    [[%c(SBOrientationLockManager) sharedInstance] myIsUserLocked];

}


%hook SBOrientationLockManager
-(SBOrientationLockManager*)init{
	self = %orig;
CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
                                    (void*)myObserver,
                                    toggle,
                                    CFSTR("avlock.toggle"),
                                    NULL,
                                    CFNotificationSuspensionBehaviorDeliverImmediately);
CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
                                    (void*)myObserver,
                                    firstUpdate,
                                    CFSTR("avlock.firstUpdate"),
                                    NULL,
                                    CFNotificationSuspensionBehaviorDeliverImmediately);

  return self;
}
%new
-(void)myIsUserLocked{
    bool locked = [self isUserLocked];
    if(locked){
        CFNotificationCenterPostNotification(CFNotificationCenterGetDarwinNotifyCenter(), CFSTR("avlock.lockButton"), (void*)myObserver, NULL, true);
    }else{
        CFNotificationCenterPostNotification(CFNotificationCenterGetDarwinNotifyCenter(), CFSTR("avlock.unlockButton"), (void*)myObserver, NULL, true);
    }
}
-(void)unlock{
    %orig;
    CFNotificationCenterPostNotification(CFNotificationCenterGetDarwinNotifyCenter(), CFSTR("avlock.unlockButton"), (void*)myObserver, NULL, true);
}
-(void)lock{
    %orig;
    CFNotificationCenterPostNotification(CFNotificationCenterGetDarwinNotifyCenter(), CFSTR("avlock.lockButton"), (void*)myObserver, NULL, true);
}

%end
double delayInSeconds = 0.13;

%hook AVPlaybackControlsVisibilityControllerItem
-(BOOL)wantsAnimatedTransitionToHidden:(BOOL)arg1 alpha:(double)arg2{
  if(!arg1){
    delayInSeconds = 0.0;
    [[NSNotificationCenter defaultCenter] postNotificationName:UIDeviceOrientationDidChangeNotification object:nil];
    //AudioServicesPlaySystemSound(1103);
    delayInSeconds = 0.13;
  }
  return %orig;
}
%end

%hook AVTransportControlsView
int test = 30;

// iOS 13
-(AVTransportControlsView *)initWithFrame:(CGRect)frame styleSheet:(id)arg2{
    AVTransportControlsView *origself = %orig(frame,arg2);
	CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
									(void*)myObserver,
									lockButton,
									CFSTR("avlock.lockButton"),
									NULL,
									CFNotificationSuspensionBehaviorDeliverImmediately);
	CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
									(void*)myObserver,
									unlockButton,
									CFSTR("avlock.unlockButton"),
									NULL,
									CFNotificationSuspensionBehaviorDeliverImmediately);
	button = [UIButton buttonWithType:UIButtonTypeCustom];
	[button addTarget:self action:@selector(orientationButtonPressed) forControlEvents:UIControlEventTouchUpInside];
	CFNotificationCenterPostNotification(CFNotificationCenterGetDarwinNotifyCenter(), CFSTR("avlock.firstUpdate"), (void*)myObserver, NULL, true);
	button.contentMode = UIViewContentModeScaleAspectFit;

	button.imageEdgeInsets = UIEdgeInsetsMake(10, 10, 10, 10);

	[self addSubview:button];
	[[UIDevice currentDevice] beginGeneratingDeviceOrientationNotifications];
	[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(deviceOrientationDidChange) name:UIDeviceOrientationDidChangeNotification object:nil];

	return origself;
}



// iOS 12
-(AVTransportControlsView *)initWithFrame:(CGRect)frame styleSheet:(id)arg2 captureView:(id)arg3{
    AVTransportControlsView *origself = %orig(frame,arg2,arg3);
	CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
									(void*)myObserver,
									lockButton,
									CFSTR("avlock.lockButton"),
									NULL,
									CFNotificationSuspensionBehaviorDeliverImmediately);
	CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
									(void*)myObserver,
									unlockButton,
									CFSTR("avlock.unlockButton"),
									NULL,
									CFNotificationSuspensionBehaviorDeliverImmediately);
	button = [UIButton buttonWithType:UIButtonTypeCustom];
	[button addTarget:self action:@selector(orientationButtonPressed) forControlEvents:UIControlEventTouchUpInside];
	CFNotificationCenterPostNotification(CFNotificationCenterGetDarwinNotifyCenter(), CFSTR("avlock.firstUpdate"), (void*)myObserver, NULL, true);
	button.contentMode = UIViewContentModeScaleAspectFit;

	button.imageEdgeInsets = UIEdgeInsetsMake(10, 10, 10, 10);

	[self addSubview:button];
	[[UIDevice currentDevice] beginGeneratingDeviceOrientationNotifications];
	[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(deviceOrientationDidChange) name:UIDeviceOrientationDidChangeNotification object:nil];

	return origself;
}

// iOS 11
-(AVTransportControlsView *)initWithFrame:(CGRect)frame{
    AVTransportControlsView *origself = %orig(frame);
	CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
									(void*)myObserver,
									lockButton,
									CFSTR("avlock.lockButton"),
									NULL,
									CFNotificationSuspensionBehaviorDeliverImmediately);
	CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
									(void*)myObserver,
									unlockButton,
									CFSTR("avlock.unlockButton"),
									NULL,
									CFNotificationSuspensionBehaviorDeliverImmediately);
	button = [UIButton buttonWithType:UIButtonTypeCustom];
	[button addTarget:self action:@selector(orientationButtonPressed) forControlEvents:UIControlEventTouchUpInside];
	CFNotificationCenterPostNotification(CFNotificationCenterGetDarwinNotifyCenter(), CFSTR("avlock.firstUpdate"), (void*)myObserver, NULL, true);
	button.contentMode = UIViewContentModeScaleAspectFit;

	button.imageEdgeInsets = UIEdgeInsetsMake(10, 10, 10, 10);

	[self addSubview:button];
	[[UIDevice currentDevice] beginGeneratingDeviceOrientationNotifications];
	[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(deviceOrientationDidChange) name:UIDeviceOrientationDidChangeNotification object:nil];

	return origself;
}


-(void)dealloc{
    %orig;
    CFNotificationCenterRemoveObserver ( CFNotificationCenterGetDarwinNotifyCenter(), (void*)myObserver, NULL, NULL);
	test = 30;
}

-(void)layoutSubviews{
	%orig;
	if(test > 1){
		[self deviceOrientationDidChange];
		test--;;
	}
}

%new
-(void)deviceOrientationDidChange{

dispatch_time_t popTime = dispatch_time(DISPATCH_TIME_NOW, delayInSeconds * NSEC_PER_SEC);
dispatch_after(popTime, dispatch_get_main_queue(), ^(void){


	UIInterfaceOrientation orientation =[UIApplication sharedApplication].statusBarOrientation;
	if (orientation == UIInterfaceOrientationLandscapeRight || orientation == UIInterfaceOrientationLandscapeLeft){

		CGPoint point = self.standardPlayPauseButton.frame.origin;
		point.x =  (self.skipForwardButton.frame.origin.x - 40);
		[self.standardPlayPauseButton setFrame:CGRectMake(point.x, point.y, self.standardPlayPauseButton.frame.size.width, self.standardPlayPauseButton.frame.size.height)];

		point = self.skipBackButton.frame.origin;
		point.x =  (self.standardPlayPauseButton.frame.origin.x - 40);
		[self.skipBackButton setFrame:CGRectMake(point.x, point.y, self.skipBackButton.frame.size.width, self.skipBackButton.frame.size.height)];

		[button setFrame:CGRectMake(self.standardPlayPauseButton.frame.origin.x-93.5, self.standardPlayPauseButton.frame.origin.y, 46, 46)];

	}
	else if(orientation == UIInterfaceOrientationPortrait){
		button.frame = CGRectMake((self.skipBackButton.frame.origin.x - 62), self.skipBackButton.frame.origin.y, 46, 46);
	}
});
}

%new
-(void)orientationButtonPressed{
    CFNotificationCenterPostNotification(CFNotificationCenterGetDarwinNotifyCenter(), CFSTR("avlock.toggle"), (void*)myObserver, NULL, true);

}

%end