Tweak.xm

#import "Forcy.h"

static void loadPreferences() {
    preferences = [[NSUserDefaults alloc] initWithSuiteName:@"com.strayadevteam.forcyprefs"];

    [preferences registerDefaults:@{
        @"enabled": @YES,
        @"hapticFeedbackIsEnabled": @YES,
        @"removeBackgroundBlur": @NO,
        @"preferForceTouch": @NO,
        @"shortHoldTime": [NSNumber numberWithFloat:0.325],
        @"vibrationTime": [NSNumber numberWithFloat:50],
        @"invokeMethods": [NSNumber numberWithInteger:0],
        @"menuEnabled": @YES,
        @"peekAndPopSens": [NSNumber numberWithInteger:45],
        @"peekAndPopEnabled": @YES
    }];
    
    enabled = [preferences boolForKey:@"enabled"];
    hapticFeedbackIsEnabled = [preferences boolForKey:@"hapticFeedbackIsEnabled"];
    removeBackgroundBlur = [preferences boolForKey:@"removeBackgroundBlur"];
    preferForceTouch = [preferences boolForKey:@"preferForceTouch"];
    shortHoldTime = [preferences floatForKey:@"shortHoldTime"];
    vibrationTime = [preferences floatForKey:@"vibrationTime"];
    invokeMethods = [preferences integerForKey:@"invokeMethods"];
    menuEnabled = [preferences boolForKey:@"menuEnabled"];
    peekAndPopSens = [preferences integerForKey:@"peekAndPopSens"];
    peekAndPopEnabled = [preferences boolForKey:@"peekAndPopEnabled"];

    [preferences relees];
}

static void hapticFeedback(){
    if(hapticFeedbackIsEnabled){
        NSMutableDictionary* dict = [NSMutableDictionary dictionary];
        NSMutableArray* arr = [NSMutableArray array];
        [arr addObject:[NSNumber numberWithBool:YES]];
        [arr addObject:[NSNumber numberWithInt:vibrationTime]]; //vibrate for 50ms
        [dict setObject:arr forKey:@"VibePattern"];
        [dict setObject:[NSNumber numberWithInt:1] forKey:@"Intensity"];
        AudioServicesPlaySystemSoundWithVibration(4095,nil,dict);
    }
}

/*UIImage *getLatestPhoto() {
    PHImageManager *imgManager = [PHImageManager defaultManager];
    PHImageRequestOptions *requestOptions = [[PHImageRequestOptions alloc] init];
    requestOptions.synchronous = TRUE;

    PHFetchOptions *fetchOptions = [[PHFetchOptions alloc] init];
    fetchOptions.sortDescriptors = [[NSArray alloc] initWithObjects:[NSSortDescriptor sortDescriptorWithKey:@"creationDate" ascending: TRUE], nil];
    __block UIImage *finalImage = nil;

    if ([PHAsset fetchAssetsWithMediaType:PHAssetMediaTypeImage options:fetchOptions]) {
        PHFetchResult *fetchResult = [PHAsset fetchAssetsWithMediaType:PHAssetMediaTypeImage options:fetchOptions];
        if (fetchResult.count > 0) {
            [imgManager requestImageForAsset:[fetchResult objectAtIndex:(fetchResult.count-1)] targetSize:PHImageManagerMaximumSize contentMode:PHImageContentModeAspectFill options:requestOptions resultHandler:^(UIImage *result, NSDictionary *info){
                    finalImage = result;
                }];
        }
    }

    return finalImage;
}
*/
SBIconView *currentlyHighlightedIcon;

%group SpringBoardHooks
%hook SBIconView 

UISwipeGestureRecognizer *swipeUp;
UITapGestureRecognizer *doubleTap;

- (void)setLocation:(id)arg1 {
    //im trying mum - i did it you proud?

    //HBLogInfo(@"setLoaction:arg1 = %@", arg1);
    [self removeGestureRecognizer:swipeUp];
    [self removeGestureRecognizer:doubleTap];
    if(enabled && menuEnabled){
        if(invokeMethods == 0){
            /*UILongPressGestureRecognizer *shortcutMenuPeekGesture = MSHookIvar<UILongPressGestureRecognizer *>(self, "_shortcutMenuPeekGesture");
            shortcutMenuPeekGesture = [[UILongPressGestureRecognizer alloc] initWithTarget:[%c(SBIconController) sharedInstance] action:@selector(_handleShortcutMenuPeek:)];
            shortcutMenuPeekGesture.minimumPressDuration = shortHoldTime;*/
            UILongPressGestureRecognizer *shortcutPeekGesture = MSHookIvar<UILongPressGestureRecognizer *>(self, "_shortcutMenuPeekGesture");
            shortcutPeekGesture = [[UILongPressGestureRecognizer alloc] initWithTarget: [%c(SBIconController) sharedInstance] action:@selector(_handleShortcutMenuPeek:)];
            shortcutPeekGesture.minimumPressDuration = shortHoldTime;
            [self addGestureRecognizer:shortcutPeekGesture];

            swipeUp = [[%c(UISwipeGestureRecognizer) alloc] initWithTarget:self action:@selector(fc_swiped:)];
            swipeUp.direction = UISwipeGestureRecognizerDirectionUp;
            swipeUp.delegate = (id <UIGestureRecognizerDelegate>)self;
            [self addGestureRecognizer:swipeUp];
            [swipeUp release];
        } else if(invokeMethods == 1){
        //NSLog(@"invokeMethods == 1");
            swipeUp = [[%c(UISwipeGestureRecognizer) alloc] initWithTarget:self action:@selector(fc_swiped:)];
            swipeUp.direction = UISwipeGestureRecognizerDirectionUp;
            swipeUp.delegate = (id <UIGestureRecognizerDelegate>)self;
            [self addGestureRecognizer:swipeUp];
            [swipeUp release];

        } else if(invokeMethods == 2){
            doubleTap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(fc_handleDoubleTapGesture:)];
            doubleTap.numberOfTapsRequired = 2;
            [self addGestureRecognizer:doubleTap];
            [doubleTap release];
        }
    }
    %orig;
}

%new -(void)fc_handleDoubleTapGesture:(UITapGestureRecognizer *)gesture{
    if(gesture.state == UIGestureRecognizerStateRecognized){
            [[%c(SBIconController) sharedInstance] _revealMenuForIconView:self presentImmediately:true];
            [self cancelLongPressTimer];
    }
}

%new - (void)fc_swiped:(UISwipeGestureRecognizer *)gesture {
    if(invokeMethods == 0){
        [[%c(SBIconController) sharedInstance] setIsEditing:YES];
        [self _handleSecondHalfLongPressTimer:nil];
    } else if (invokeMethods == 1){
        [[%c(SBIconController) sharedInstance] _revealMenuForIconView:self presentImmediately:true];
        [self cancelLongPressTimer];
    }
}

%new - (BOOL)gestureRecognizer:(UIGestureRecognizer *)gestureRecognizer shouldReceiveTouch:(UITouch *)touch {
    if ([gestureRecognizer isKindOfClass:[UISwipeGestureRecognizer class]] 
                        && ![[%c(SBIconController) sharedInstance] _canRevealShortcutMenu])
        return NO;
    
    return YES;
}

- (void)_handleFirstHalfLongPressTimer:(id)timer{
    if(enabled && menuEnabled && [[%c(SBIconController) sharedInstance] _canRevealShortcutMenu] && timer != nil){
        if(invokeMethods == 0){
            [[%c(SBIconController) sharedInstance] _revealMenuForIconView:self presentImmediately:true];
            [self cancelLongPressTimer];
        } else if (invokeMethods == 1){
            [[%c(SBIconController) sharedInstance] setIsEditing:YES];
            [self _handleSecondHalfLongPressTimer:nil];
        }
    }
    %orig;
}

- (void)setHighlighted:(BOOL)highlighted {
    %orig;
    currentlyHighlightedIcon = highlighted ? self : nil;
}
%end

%hook SBApplicationShortcutMenu
-(void)_setupViews{
    %orig;
    if(enabled && menuEnabled && removeBackgroundBlur){
        _UIBackdropView *_blurView = MSHookIvar<_UIBackdropView*>(self, "_blurView");
        [_blurView setHidden:true];
    }
}
%end


%hook SBIconController
- (void)_revealMenuForIconView:(SBIconView *)iconView presentImmediately:(BOOL)imm {
    if(hapticFeedbackIsEnabled && !self.isEditing){
        hapticFeedback();
    }
    %orig(iconView, YES);
}
%end

%end



%hook UIScreen
- (int)_forceTouchCapability {
    return 2;
}
%end

%hook UITraitCollection
- (int)forceTouchCapability {
    return 2;
}
%end

%hook UIDevice
- (BOOL)_supportsForceTouch {
    return TRUE;
}
%end

%hook _UITouchForceMessage
- (void)setUnclampedTouchForce:(CGFloat)touchForce {
    if(peekAndPopEnabled){
        if (HardPress) {
            %orig((int) 200);
            //hapticFeedback();
        } else {
            %orig((int) 20);
            //hapticFeedback();
        }
    }
}
%end
%hook UITouch/*
- (void)setMajorRadiusTolerance:(float)arg1 {
    if (!FirstPress) {
        lightPress = peekAndPopSens;
        if (lightPress >= 15) {
            FirstPress = YES;
            NSLog(@"FIRST PRESS FUCK");
            //hapticFeedback();
        }
    }
    if ([self _pathMajorRadius] > lightPress) {
        HardPress = YES;
        NSLog(@"HARD PRESS FUCK");
        //hapticFeedback();
    }
    else {
        HardPress = NO;
        NSLog(@"NO HARD PRESS FUCK");
    }
    %orig;
}*/
- (void)setMajorRadius:(float)arg1 {
    if(peekAndPopEnabled){
    // NSLog(@"View: %@", self.view.gestureRecognizers);
        //if (![self.view isKindOfClass:[NSClassFromString(@"SBIconView") class]]) {
            if (!FirstPress) {
                lightPress = peekAndPopSens;
                
                if (lightPress >= 15) {
                    FirstPress = YES;
                }
            }
            if ([self _pathMajorRadius] > lightPress) {
                HardPress = 2;
            }
            if ([self _pathMajorRadius] > lightPress + lightPress /2) {
                HardPress = 3;
            }
            if ([self _pathMajorRadius] < lightPress) {
                HardPress = 1;
            }
        //}
    }
    %orig;
}
%end

%ctor{
    CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
                                NULL,
                                (CFNotificationCallback)loadPreferences,
                                CFSTR("com.strayadevteam.forcyprefs/prefsChanged"),
                                NULL,
                                CFNotificationSuspensionBehaviorDeliverImmediately);
    loadPreferences();
    if ([[[NSBundle mainBundle] bundleIdentifier] isEqualToString:@"com.apple.springboard"]) {
        %init(SpringBoardHooks);
    }
    %init;
}