chore: fix aswebauthenticationsession issue for automated tests

Author: ImmutableJeffrey

Source/Immutable/Private/Immutable/Mac/ImmutableMac.cpp

@@ -7,8 +7,14 @@
 #include "Editor.h"
 #endif
 
+#import <AppKit/NSWorkspace.h>
+
 ASWebAuthenticationSession *_authSession;
 
+// Global variable to track if we're in automation mode
+static BOOL g_isInAutomationMode = NO;
+static NSString* _pendingRedirectScheme = nil;
+
 @implementation ImmutableMac
 
 - (instancetype)init {
@@ -67,6 +73,13 @@ ASWebAuthenticationSession *_authSession;
 }
 
 - (void)launchUrl:(const char *)url forRedirectUri:(const char *)redirectUri {
+  // For automation, use the browser-based method
+  if (g_isInAutomationMode) {
+    NSLog(@"Using automation mode for authentication");
+    [self launchUrlInBrowser:url forRedirectUri:redirectUri forAutomation:YES];
+    return;
+  }
+
   if (@available(macOS 10.15, *)) {
     NSURL *URL =
         [NSURL URLWithString:[[NSString alloc] initWithUTF8String:url]];
@@ -104,9 +117,56 @@ ASWebAuthenticationSession *_authSession;
   }
 }
 
+- (void)launchUrlInBrowser:(const char *)url forRedirectUri:(const char *)redirectUri forAutomation:(BOOL)forAutomation {
+    // Set automation mode
+    g_isInAutomationMode = forAutomation;
+    NSLog(@"Setting automation mode: %@", g_isInAutomationMode ? @"YES" : @"NO");
+    
+    // Create URL object
+    NSURL *URL = [NSURL URLWithString:[[NSString alloc] initWithUTF8String:url]];
+    
+    // Extract the callback scheme from redirect URI
+    NSString *redirectUriString = [[NSString alloc] initWithUTF8String:redirectUri];
+    NSString *callbackURLScheme = [[redirectUriString componentsSeparatedByString:@":"] objectAtIndex:0];
+    
+    // Store the redirect scheme for use when handling the callback
+    _pendingRedirectScheme = callbackURLScheme;
+    
+    // Register for custom URL scheme notifications (will need to be implemented in AppDelegate)
+    [[NSNotificationCenter defaultCenter] addObserverForName:@"ImmutableCustomURLSchemeCallback"
+                                                      object:nil
+                                                       queue:[NSOperationQueue mainQueue]
+                                                  usingBlock:^(NSNotification *notification) {
+        if (notification.userInfo) {
+            NSString *callbackURLString = notification.userInfo[@"url"];
+            if (callbackURLString) {
+                UImmutablePassport* passport = [ImmutableMac getPassport];
+                if (passport) {
+                    passport->HandleDeepLink(callbackURLString);
+                }
+            }
+        }
+    }];
+    
+    // Open URL in default browser
+    [[NSWorkspace sharedWorkspace] openURL:URL];
+    NSLog(@"Opened URL in browser for automation: %@", URL);
+}
+
 - (ASPresentationAnchor)presentationAnchorForWebAuthenticationSession:
     (ASWebAuthenticationSession *)session API_AVAILABLE(macos(10.15)) {
   return [[[NSApplication sharedApplication] windows] firstObject];
 }
 
+// Helper method to enable automation mode
++ (void)setAutomationMode:(BOOL)enabled {
+    g_isInAutomationMode = enabled;
+    NSLog(@"setAutomationMode called with: %@", enabled ? @"YES" : @"NO");
+}
+
+// Helper method to check if we're in automation mode
++ (BOOL)isInAutomationMode {
+    return g_isInAutomationMode;
+}
+
 @end

Source/Immutable/Private/Immutable/Mac/ImmutableMac.h

@@ -7,4 +7,9 @@
     : NSObject <ASWebAuthenticationPresentationContextProviding>
 + (ImmutableMac *)instance;
 - (void)launchUrl:(const char *)url forRedirectUri:(const char *)redirectUri;
+- (void)launchUrlInBrowser:(const char *)url forRedirectUri:(const char *)redirectUri forAutomation:(BOOL)forAutomation;
+
+// Helper methods for automation
++ (void)setAutomationMode:(BOOL)enabled;
++ (BOOL)isInAutomationMode;
 @end

Source/Immutable/Private/Immutable/Mac/ImmutableMacHelper.cpp

@@ -0,0 +1,21 @@
+#include "Immutable/Mac/ImmutableMacHelper.h"
+
+#if PLATFORM_MAC
+#import "ImmutableMac.h"
+#endif
+
+void FImmutableMacHelper::SetAutomationMode(bool bEnabled)
+{
+#if PLATFORM_MAC
+	[ImmutableMac setAutomationMode:bEnabled ? YES : NO];
+#endif
+}
+
+bool FImmutableMacHelper::IsInAutomationMode()
+{
+#if PLATFORM_MAC
+	return [ImmutableMac isInAutomationMode] == YES;
+#else
+	return false;
+#endif
+}

Source/Immutable/Public/Immutable/Mac/ImmutableMacHelper.h

@@ -0,0 +1,26 @@
+#pragma once
+
+#include "CoreMinimal.h"
+
+/**
+ * Helper class for interacting with the Immutable Mac implementation for automation purposes
+ */
+class IMMUTABLE_API FImmutableMacHelper
+{
+public:
+	/**
+	 * Enable or disable automation mode for authentication
+	 * In automation mode, authentication will use the system browser with a custom URI scheme
+	 * instead of ASWebAuthenticationSession
+	 *
+	 * @param bEnabled Whether automation mode should be enabled
+	 */
+	static void SetAutomationMode(bool bEnabled);
+
+	/**
+	 * Check if automation mode is currently enabled
+	 * 
+	 * @return True if automation mode is enabled, false otherwise
+	 */
+	static bool IsInAutomationMode();
+};