feat: improve effectiveness of padlock feature

Author: SoCuul

src/Controllers/SecurityViewController.h

@@ -1,5 +1,9 @@
 #import <UIKit/UIKit.h>
 #import <LocalAuthentication/LocalAuthentication.h>
 
+#import "../InstagramHeaders.h"
+#import "../Tweak.h"
+#import "../../modules/JGProgressHUD/JGProgressHUD.h"
+
 @interface SCISecurityViewController : UIViewController
 @end

src/Controllers/SecurityViewController.m

@@ -1,12 +1,24 @@
-#import "../../modules/JGProgressHUD/JGProgressHUD.h"
-#import "../InstagramHeaders.h"
 #import "SecurityViewController.h"
 
 @implementation SCISecurityViewController
 
-- (void)viewDidLoad {
-    [super viewDidLoad];
-    
+- (id)init {
+    self = [super init];
+    if (!self) return nil;
+
+    [[NSNotificationCenter defaultCenter] addObserver:self
+        selector:@selector(authenticate)
+        name:UIApplicationWillEnterForegroundNotification
+        object:[UIApplication sharedApplication]
+    ];
+
+    return self;
+}
+- (void)dealloc {
+    [[NSNotificationCenter defaultCenter] removeObserver:self];
+}
+
+- (void)viewDidLoad {    
     UIBlurEffect *blurEffect = [UIBlurEffect effectWithStyle:UIBlurEffectStyleDark];
     UIVisualEffectView *blurView = [[UIVisualEffectView alloc] initWithEffect:blurEffect];
     blurView.frame = self.view.bounds;
@@ -15,16 +27,12 @@ - (void)viewDidLoad {
     UIButton *authenticateButton = [[UIButton alloc] initWithFrame:CGRectMake(20, 20, 200, 60)];
     [authenticateButton setTitle:@"Click to unlock app" forState:UIControlStateNormal];
     authenticateButton.center = self.view.center;
-    [authenticateButton addTarget:self action:@selector(authenticateButtonTapped:) forControlEvents:UIControlEventTouchUpInside];
+    [authenticateButton addTarget:self action:@selector(authenticate) forControlEvents:UIControlEventTouchUpInside];
     [self.view addSubview:authenticateButton];
 
     [self authenticate];
 }
 
-- (void)authenticateButtonTapped:(id)sender {
-    [self authenticate];
-}
-
 - (void)authenticate {
     LAContext *context = [[LAContext alloc] init];
     NSError *error = nil;
@@ -39,6 +47,8 @@ - (void)authenticate {
                 if (success) {
                     [self dismissViewControllerAnimated:YES completion:nil];
 
+                    isAuthenticationBeingShown = NO;
+
                     NSLog(@"[SCInsta] Padlock authentication: Unlock success");
                 }
                 else {

src/InstagramHeaders.h

@@ -264,6 +264,7 @@
 @end
 
 @interface IGInstagramAppDelegate : NSObject <UIApplicationDelegate>
+- (void)authPrompt; // new
 @end
 
 @interface IGDirectInboxSearchAIAgentsPillsContainerCell : UIView

src/Tweak.h

@@ -5,4 +5,7 @@ extern NSString *SCIVersionString;
 
 // Variables that work across features
 extern BOOL seenButtonEnabled; // Whether dm message seen button is enabled
-extern BOOL dmVisualMsgsViewedButtonEnabled; // Whether story dm unlimited views button is enabled
+extern BOOL dmVisualMsgsViewedButtonEnabled; // Whether story dm unlimited views button is enabled
+
+// Biometric/passcode authentication
+extern BOOL isAuthenticationBeingShown;

src/Tweak.x

@@ -75,34 +75,38 @@ BOOL dmVisualMsgsViewedButtonEnabled = false;
     NSLog(@"[SCInsta] Cleaning cache...");
     [SCIManager cleanCache];
 
+    [self authPrompt];
+
     return true;
 }
 
 // Biometric/passcode authentication
-static BOOL isAuthenticationShowed = FALSE;
+BOOL isAuthenticationBeingShown = NO;
+
+- (void)applicationDidEnterBackground:(id)arg1 {
+    %orig;
 
-- (void)applicationDidBecomeActive:(id)arg1 {
+    [self authPrompt];
+}
+- (void)applicationWillEnterForeground:(id)arg1 {
     %orig;
 
+    [self authPrompt];
+}
+
+%new - (void)authPrompt {
     // Padlock (biometric auth)
-    if ([SCIManager getPref:@"padlock"] && !isAuthenticationShowed) {
+    if ([SCIManager getPref:@"padlock"] && !isAuthenticationBeingShown) {
         UIViewController *rootController = [[self window] rootViewController];
         SCISecurityViewController *securityViewController = [SCISecurityViewController new];
         securityViewController.modalPresentationStyle = UIModalPresentationOverFullScreen;
-        [rootController presentViewController:securityViewController animated:YES completion:nil];
+        [rootController presentViewController:securityViewController animated:NO completion:nil];
 
-        isAuthenticationShowed = TRUE;
+        isAuthenticationBeingShown = YES;
 
         NSLog(@"[SCInsta] Padlock authentication: App enabled");
     }
 }
-
-- (void)applicationWillEnterForeground:(id)arg1 {
-    %orig;
-
-    // Reset padlock status
-    isAuthenticationShowed = FALSE;
-}
 %end