fix: auto-handle iOS 18+ limited access permission prompt (#1150)

Author: abhinvv1

WebDriverAgentLib/Categories/XCUIApplication+FBAlert.h

@@ -8,6 +8,8 @@
 
 #import <XCTest/XCTest.h>
 
+NS_ASSUME_NONNULL_BEGIN
+
 @interface XCUIApplication (FBAlert)
 
 /* The accessiblity label used for Safari app */
@@ -18,6 +20,16 @@ extern NSString *const FB_SAFARI_APP_NAME;
 
  @return Alert element instance
  */
-- (XCUIElement *)fb_alertElement;
+- (nullable XCUIElement *)fb_alertElement;
+
+/**
+ Retrieve an alert element hosted by the iOS 18+ limited access permission prompt
+ process. See https://github.com/appium/appium/issues/20591
+
+ @return Alert element instance if the prompt is present, otherwise nil
+ */
++ (nullable XCUIElement *)fb_limitedAccessPromptAlertElement;
 
 @end
+
+NS_ASSUME_NONNULL_END

WebDriverAgentLib/Categories/XCUIApplication+FBAlert.m

@@ -17,9 +17,22 @@
 
 NSString *const FB_SAFARI_APP_NAME = @"Safari";
 
+// The iOS 18+ limited access permission prompt (e.g. the "Select Contacts" view)
+// runs in a dedicated process that is not reported by fb_activeApplications.
+static NSString *const FB_LIMITED_ACCESS_PROMPT_BUNDLE_ID = @"com.apple.ContactsUI.LimitedAccessPromptView";
+
 
 @implementation XCUIApplication (FBAlert)
 
++ (nullable XCUIElement *)fb_limitedAccessPromptAlertElement
+{
+  XCUIApplication *promptApp = [[XCUIApplication alloc] initWithBundleIdentifier:FB_LIMITED_ACCESS_PROMPT_BUNDLE_ID];
+  if (promptApp.state < XCUIApplicationStateRunningForeground) {
+    return nil;
+  }
+  return promptApp.fb_alertElement;
+}
+
 - (nullable XCUIElement *)fb_alertElementFromSafariWithScrollView:(XCUIElement *)scrollView
                                                      viewSnapshot:(id<FBXCElementSnapshot>)viewSnapshot
 {

WebDriverAgentLib/FBAlert.m

@@ -267,6 +267,9 @@ - (XCUIElement *)alertElement
     } else {
       self.element = systemApp.fb_alertElement ?: self.application.fb_alertElement;
     }
+    if (nil == self.element) {
+      self.element = [XCUIApplication fb_limitedAccessPromptAlertElement];
+    }
   }
   return self.element;
 }

WebDriverAgentLib/Utilities/FBAlertsMonitor.m

@@ -48,22 +48,36 @@ - (void)scheduleNextTick
   }
 
   dispatch_async(dispatch_get_main_queue(), ^{
+    id<FBAlertsMonitorDelegate> delegate = self.delegate;
     NSArray<XCUIApplication *> *activeApps = XCUIApplication.fb_activeApplications;
+    BOOL didDetectAlert = NO;
     for (XCUIApplication *activeApp in activeApps) {
       XCUIElement *alertElement = nil;
       @try {
         alertElement = activeApp.fb_alertElement;
         if (nil != alertElement) {
-          [self.delegate didDetectAlert:[FBAlert alertWithElement:alertElement]];
+          [delegate didDetectAlert:[FBAlert alertWithElement:alertElement]];
         }
       } @catch (NSException *e) {
         [FBLogger logFmt:@"Got an unexpected exception while monitoring alerts: %@\n%@", e.reason, e.callStackSymbols];
       }
       if (nil != alertElement) {
+        didDetectAlert = YES;
         break;
       }
     }
 
+    if (!didDetectAlert) {
+      @try {
+        XCUIElement *alertElement = [XCUIApplication fb_limitedAccessPromptAlertElement];
+        if (nil != alertElement) {
+          [delegate didDetectAlert:[FBAlert alertWithElement:alertElement]];
+        }
+      } @catch (NSException *e) {
+        [FBLogger logFmt:@"Got an unexpected exception while monitoring alerts: %@\n%@", e.reason, e.callStackSymbols];
+      }
+    }
+
     if (self.isMonitoring) {
       dispatch_after(dispatch_time(DISPATCH_TIME_NOW, delta), dispatch_get_main_queue(), ^{
         [self scheduleNextTick];