@devicelab/maestro-runner-darwin-x64 1.1.25

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBClassChain.m

@@ -0,0 +1,98 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import "XCUIElement+FBClassChain.h"
+
+#import "FBClassChainQueryParser.h"
+#import "FBXCodeCompatibility.h"
+#import "FBExceptions.h"
+
+@implementation XCUIElement (FBClassChain)
+
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingClassChain:(NSString *)classChainQuery shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch
+{
+  NSError *error;
+  FBClassChain *parsedChain = [FBClassChainQueryParser parseQuery:classChainQuery error:&error];
+  if (nil == parsedChain) {
+    @throw [NSException exceptionWithName:FBClassChainQueryParseException reason:error.localizedDescription userInfo:error.userInfo];
+    return nil;
+  }
+  NSMutableArray<FBClassChainItem *> *lookupChain = parsedChain.elements.mutableCopy;
+  FBClassChainItem *chainItem = lookupChain.firstObject;
+  XCUIElement *currentRoot = self;
+  XCUIElementQuery *query = [currentRoot fb_queryWithChainItem:chainItem query:nil];
+  [lookupChain removeObjectAtIndex:0];
+  while (lookupChain.count > 0) {
+    BOOL isRootChanged = NO;
+    if (nil != chainItem.position) {
+      // It is necessary to resolve the query if intermediate element index is not zero or one,
+      // because predicates don't support search by indexes
+      NSArray<XCUIElement *> *currentRootMatch = [self.class fb_matchingElementsWithItem:chainItem
+                                                                                   query:query
+                                                             shouldReturnAfterFirstMatch:nil];
+      if (0 == currentRootMatch.count) {
+        return @[];
+      }
+      currentRoot = currentRootMatch.firstObject;
+      isRootChanged = YES;
+    }
+    chainItem = [lookupChain firstObject];
+    query = [currentRoot fb_queryWithChainItem:chainItem query:isRootChanged ? nil : query];
+    [lookupChain removeObjectAtIndex:0];
+  }
+  return [self.class fb_matchingElementsWithItem:chainItem
+                                           query:query
+                     shouldReturnAfterFirstMatch:@(shouldReturnAfterFirstMatch)];
+}
+
+- (XCUIElementQuery *)fb_queryWithChainItem:(FBClassChainItem *)item query:(nullable XCUIElementQuery *)query
+{
+  if (item.isDescendant) {
+    if (query) {
+      query = [query descendantsMatchingType:item.type];
+    } else {
+      query = [self.fb_query descendantsMatchingType:item.type];
+    }
+  } else {
+    if (query) {
+      query = [query childrenMatchingType:item.type];
+    } else {
+      query = [self.fb_query childrenMatchingType:item.type];
+    }
+  }
+  if (item.predicates) {
+    for (FBAbstractPredicateItem *predicate in item.predicates) {
+      if ([predicate isKindOfClass:FBSelfPredicateItem.class]) {
+        query = [query matchingPredicate:predicate.value];
+      } else if ([predicate isKindOfClass:FBDescendantPredicateItem.class]) {
+        query = [query containingPredicate:predicate.value];
+      }
+    }
+  }
+  return query;
+}
+
++ (NSArray<XCUIElement *> *)fb_matchingElementsWithItem:(FBClassChainItem *)item query:(XCUIElementQuery *)query shouldReturnAfterFirstMatch:(nullable NSNumber *)shouldReturnAfterFirstMatch
+{
+  if (1 == item.position.integerValue || (0 == item.position.integerValue && shouldReturnAfterFirstMatch.boolValue)) {
+    XCUIElement *result = query.fb_firstMatch;
+    return result ? @[result] : @[];
+  }
+  NSArray<XCUIElement *> *allMatches = query.fb_allMatches;
+  if (0 == item.position.integerValue) {
+    return allMatches;
+  }
+  if (allMatches.count >= (NSUInteger)ABS(item.position.integerValue)) {
+    return item.position.integerValue > 0
+      ? @[[allMatches objectAtIndex:item.position.integerValue - 1]]
+      : @[[allMatches objectAtIndex:allMatches.count + item.position.integerValue]];
+  }
+  return @[];
+}
+
+@end

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBCustomActions.h

@@ -0,0 +1,27 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import <WebDriverAgentLib/FBXCElementSnapshotWrapper.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface XCUIElement (FBCustomActions)
+
+/*! Custom accessibility actions as a string – may be nil if the element does not have this attribute */
+@property (nonatomic, readonly, nullable) NSString *fb_customActions;
+
+@end
+
+@interface FBXCElementSnapshotWrapper (FBCustomActions)
+
+/*! Custom accessibility actions as a string – may be nil if the element does not have this attribute */
+@property (nonatomic, readonly, nullable) NSString *fb_customActions;
+
+@end
+
+NS_ASSUME_NONNULL_END

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBCustomActions.m

@@ -0,0 +1,127 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import "XCUIElement+FBCustomActions.h"
+#import "XCUIElement+FBUtilities.h"
+#import "FBXCElementSnapshotWrapper+Helpers.h"
+#import "FBLogger.h"
+#import "FBConfiguration.h"
+#import "XCTestPrivateSymbols.h"
+
+@interface FBXCElementSnapshotWrapper (FBCustomActionsInternal)
+
+- (NSString *)fb_stringAttribute:(NSString *)attributeName                            
+                          symbol:(NSNumber *)symbol;
+
+@end
+
+@implementation XCUIElement (FBCustomActions)
+
+- (NSString *)fb_customActions
+{
+  @autoreleasepool {
+    id<FBXCElementSnapshot> snapshot = [self fb_standardSnapshot];
+    return [[FBXCElementSnapshotWrapper ensureWrapped:snapshot]
+            fb_customActions];
+  }
+}
+
+@end
+
+
+@implementation FBXCElementSnapshotWrapper(FBCustomActions)
+
+- (NSString *)fb_customActions
+{
+  return [self fb_stringAttribute:FB_XCAXACustomActionsAttributeName
+                           symbol:FB_XCAXACustomActionsAttribute];
+}
+
+- (NSString *)fb_stringAttribute:(NSString *)attributeName
+                            symbol:(NSNumber *)symbol
+{
+  id cached = (self.snapshot.additionalAttributes ?: @{})[symbol];
+  if ([cached isKindOfClass:[NSString class]]) {
+    return cached;
+  }
+  
+  NSError *error = nil;
+  id raw = [self fb_attributeValue:attributeName error:&error];
+  if (raw == nil) {
+    [FBLogger logFmt: @"[FBCustomActions] Cannot determine string value for %@: %@",
+     attributeName, error.localizedDescription];
+    return nil;
+  }
+  
+  // Case 1: Already a string
+  if ([raw isKindOfClass:[NSString class]]) {
+    return [self retrieveCustomActionsFromString:raw forSymbol:symbol];
+  }
+  
+  // Case 2: Array of custom actions
+  if ([raw isKindOfClass:[NSArray class]]) {
+    return [self retrieveCustomActionsFromArray:raw forSymbol:symbol];
+  }
+  
+  // Fallback: Try to cast to string
+  return [self retrieveCustomActionsByCastingToString:raw forSymbol:symbol];
+}
+
+- (NSString *)retrieveCustomActionsFromString:(NSString *)stringValue
+                                    forSymbol:(NSNumber *)symbol
+{
+  NSMutableDictionary *updated =
+  (self.additionalAttributes ?: @{}).mutableCopy;
+  updated[symbol] = stringValue;
+  self.snapshot.additionalAttributes = updated.copy;
+  return stringValue;
+}
+
+- (NSString *)retrieveCustomActionsFromArray:(NSArray *)arrayValue
+                                   forSymbol:(NSNumber *)symbol
+{
+  NSMutableArray *stringified = [NSMutableArray array];
+  for (id action in arrayValue) {
+    NSString *title = nil;
+    
+    if ([action isKindOfClass:[NSDictionary class]]) {
+      title = ((NSDictionary *)action)[@"CustomActionName"];
+    }
+    
+    if (!title || ![title isKindOfClass:[NSString class]]) {
+      @try {
+        title = [action valueForKey:@"title"];
+      } @catch (__unused NSException * e) {
+        title = @"<unknown>";
+      }
+    }
+    
+    [stringified addObject:title ?: @"<null>"];
+    [FBLogger logFmt: @"[FBCustomActions] Custom action title: %@", title];
+  }
+  
+  NSString *joined = [stringified componentsJoinedByString:@","];
+  NSMutableDictionary *updated =
+  (self.additionalAttributes ?: @{}).mutableCopy;
+  updated[symbol] = joined;
+  self.snapshot.additionalAttributes = updated.copy;
+  return joined;
+}
+
+- (NSString *)retrieveCustomActionsByCastingToString:(id)raw
+                                           forSymbol:(NSNumber *)symbol
+{
+  NSString *stringValue = [NSString stringWithFormat:@"%@", raw];
+  NSMutableDictionary *updated =
+  (self.additionalAttributes ?: @{}).mutableCopy;
+  updated[symbol] = stringValue;
+  self.snapshot.additionalAttributes = updated.copy;
+  return stringValue;
+}
+
+@end

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBFind.h

@@ -0,0 +1,75 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+
+#import <XCTest/XCTest.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface XCUIElement (FBFind)
+
+/**
+ Returns an array of descendants matching given class name
+
+ @param className requested class name
+ @param shouldReturnAfterFirstMatch set it to YES if you want only the first found element to be
+ resolved and returned. This will speed up the search significantly if given class name matches multiple
+ nodes in the UI tree
+ @return an array of descendants matching given class name
+ */
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingClassName:(NSString *)className shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch;
+
+/**
+ Returns an array of descendants matching given accessibility id
+
+ @param accessibilityId requested accessibility id
+ @param shouldReturnAfterFirstMatch set it to YES if you want only the first found element to be
+ resolved and returned. This will speed up the search significantly if given id matches multiple
+ nodes in the UI tree
+ @return an array of descendants matching given accessibility id
+ */
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingIdentifier:(NSString *)accessibilityId shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch;
+
+/**
+ Returns an array of descendants matching given xpath query
+
+ @param xpathQuery requested xpath query
+ @param shouldReturnAfterFirstMatch set it to YES if you want only the first found element to be
+ resolved and returned. This will speed up the search significantly if given xpath matches multiple
+ nodes in the UI tree
+ @return an array of descendants matching given xpath query
+ */
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingXPathQuery:(NSString *)xpathQuery shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch;
+
+/**
+ Returns an array of descendants matching given predicate.
+ Allowed property names are only these declared in FBElement protocol (property names are received in runtime)
+ and their shortcuts (without 'wd' prefix). All other property names are considered as unknown.
+ 
+ @param predicate requested predicate
+ @param shouldReturnAfterFirstMatch set it to YES if you want only the first found element to be
+ resolved and returned. This will speed up the search significantly if given predicate matches multiple
+ nodes in the UI tree
+ @return an array of descendants matching given predicate
+ @throw FBUnknownPredicateKeyException in case the given property name is not declared in FBElement protocol
+ */
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingPredicate:(NSPredicate *)predicate shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch;
+
+/**
+ Returns an array of descendants with property matching given value
+
+ @param property requested property name
+ @param value requested value of the property
+ @param partialSearch determines whether it should be exact or partial match
+ @return an array of descendants with property matching given value
+ */
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingProperty:(NSString *)property value:(NSString *)value partialSearch:(BOOL)partialSearch;
+
+@end
+
+NS_ASSUME_NONNULL_END

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBFind.m

@@ -0,0 +1,133 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+
+#import "XCUIElement+FBFind.h"
+
+#import "FBMacros.h"
+#import "FBElementTypeTransformer.h"
+#import "FBConfiguration.h"
+#import "NSPredicate+FBFormat.h"
+#import "FBXCElementSnapshotWrapper+Helpers.h"
+#import "FBXCodeCompatibility.h"
+#import "XCUIElement+FBCaching.h"
+#import "XCUIElement+FBUID.h"
+#import "XCUIElement+FBUtilities.h"
+#import "XCUIElement+FBWebDriverAttributes.h"
+#import "XCUIElementQuery.h"
+#import "FBElementUtils.h"
+#import "FBXCodeCompatibility.h"
+#import "FBXPath.h"
+
+@implementation XCUIElement (FBFind)
+
++ (NSArray<XCUIElement *> *)fb_extractMatchingElementsFromQuery:(XCUIElementQuery *)query
+                                    shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch
+{
+  if (!shouldReturnAfterFirstMatch) {
+    return query.fb_allMatches;
+  }
+  XCUIElement *matchedElement = query.fb_firstMatch;
+  return matchedElement ? @[matchedElement] : @[];
+}
+
+- (id<FBXCElementSnapshot>)fb_cachedSnapshotWithQuery:(XCUIElementQuery *)query
+{
+  return [self isKindOfClass:XCUIApplication.class] ? query.rootElementSnapshot : self.fb_cachedSnapshot;
+}
+
+#pragma mark - Search by ClassName
+
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingClassName:(NSString *)className
+                                shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch
+{
+  XCUIElementType type = [FBElementTypeTransformer elementTypeWithTypeName:className];
+  XCUIElementQuery *query = [self.fb_query descendantsMatchingType:type];
+  NSMutableArray *result = [NSMutableArray array];
+  [result addObjectsFromArray:[self.class fb_extractMatchingElementsFromQuery:query
+                                                  shouldReturnAfterFirstMatch:shouldReturnAfterFirstMatch]];
+  id<FBXCElementSnapshot> cachedSnapshot = [self fb_cachedSnapshotWithQuery:query];
+  if (type == XCUIElementTypeAny || cachedSnapshot.elementType == type) {
+    if (shouldReturnAfterFirstMatch || result.count == 0) {
+      return @[self];
+    }
+    [result insertObject:self atIndex:0];
+  }
+  return result.copy;
+}
+
+
+#pragma mark - Search by property value
+
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingProperty:(NSString *)property
+                                                     value:(NSString *)value
+                                             partialSearch:(BOOL)partialSearch
+{
+  NSPredicate *searchPredicate = [NSPredicate predicateWithFormat:(partialSearch ? @"%K CONTAINS %@" : @"%K == %@"), property, value];
+  return [self fb_descendantsMatchingPredicate:searchPredicate shouldReturnAfterFirstMatch:NO];
+}
+
+#pragma mark - Search by Predicate String
+
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingPredicate:(NSPredicate *)predicate
+                                shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch
+{
+  NSPredicate *formattedPredicate = [NSPredicate fb_snapshotBlockPredicateWithPredicate:predicate];
+  XCUIElementQuery *query = [[self.fb_query descendantsMatchingType:XCUIElementTypeAny] matchingPredicate:formattedPredicate];
+  NSMutableArray<XCUIElement *> *result = [NSMutableArray array];
+  [result addObjectsFromArray:[self.class fb_extractMatchingElementsFromQuery:query
+                                                  shouldReturnAfterFirstMatch:shouldReturnAfterFirstMatch]];
+  id<FBXCElementSnapshot> cachedSnapshot = [self fb_cachedSnapshotWithQuery:query];
+  // Include self element into predicate search
+  if ([formattedPredicate evaluateWithObject:cachedSnapshot]) {
+    if (shouldReturnAfterFirstMatch || result.count == 0) {
+      return @[self];
+    }
+    [result insertObject:self atIndex:0];
+  }
+  return result.copy;
+}
+
+
+#pragma mark - Search by xpath
+
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingXPathQuery:(NSString *)xpathQuery
+                                 shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch
+{
+  // XPath will try to match elements only class name, so requesting elements by XCUIElementTypeAny will not work. We should use '*' instead.
+  xpathQuery = [xpathQuery stringByReplacingOccurrencesOfString:@"XCUIElementTypeAny" withString:@"*"];
+  NSArray<id<FBXCElementSnapshot>> *matchingSnapshots = [FBXPath matchesWithRootElement:self forQuery:xpathQuery];
+  if (0 == [matchingSnapshots count]) {
+    return @[];
+  }
+  if (shouldReturnAfterFirstMatch) {
+    id<FBXCElementSnapshot> snapshot = matchingSnapshots.firstObject;
+    matchingSnapshots = @[snapshot];
+  }
+  XCUIElement *scopeRoot = FBConfiguration.limitXpathContextScope ? self : self.application;
+  return [scopeRoot fb_filterDescendantsWithSnapshots:matchingSnapshots
+                                         onlyChildren:NO];
+}
+
+
+#pragma mark - Search by Accessibility Id
+
+- (NSArray<XCUIElement *> *)fb_descendantsMatchingIdentifier:(NSString *)accessibilityId
+                                 shouldReturnAfterFirstMatch:(BOOL)shouldReturnAfterFirstMatch
+{
+  NSPredicate *predicate = [NSPredicate predicateWithBlock:^BOOL(id<FBXCElementSnapshot> snapshot,
+                                                                 NSDictionary<NSString *,id> * _Nullable bindings) {
+    @autoreleasepool {
+      return [[FBXCElementSnapshotWrapper wdNameWithSnapshot:snapshot] isEqualToString:accessibilityId];
+    }
+  }];
+  return [self fb_descendantsMatchingPredicate:predicate
+                   shouldReturnAfterFirstMatch:shouldReturnAfterFirstMatch];
+}
+
+@end

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBForceTouch.h

@@ -0,0 +1,38 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import <WebDriverAgentLib/XCUIElement.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+#if !TARGET_OS_TV
+
+@interface XCUIElement (FBForceTouch)
+
+/**
+ Performs force touch on element
+ 
+ @param relativeCoordinate hit point coordinate relative to the current element position.
+ nil value means to use the default element hit point
+ @param pressure The pressure of the force touch – valid values are [0, touch.maximumPossibleForce]
+ nil value would use the default pressure value
+ @param duration The duration of the gesture in float seconds
+ nil value would use the default duration value
+ @param error If there is an error, upon return contains an NSError object that describes the problem.
+ @return YES if the operation succeeds, otherwise NO.
+ */
+- (BOOL)fb_forceTouchCoordinate:(nullable NSValue *)relativeCoordinate
+                       pressure:(nullable NSNumber *)pressure
+                       duration:(nullable NSNumber *)duration
+                          error:(NSError **)error;
+
+@end
+
+#endif
+
+NS_ASSUME_NONNULL_END

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBForceTouch.m

@@ -0,0 +1,52 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import "XCUIElement+FBForceTouch.h"
+
+#if !TARGET_OS_TV
+
+#import "FBErrorBuilder.h"
+#import "XCUICoordinate.h"
+#import "XCUIDevice.h"
+
+@implementation XCUIElement (FBForceTouch)
+
+- (BOOL)fb_forceTouchCoordinate:(NSValue *)relativeCoordinate
+                       pressure:(NSNumber *)pressure
+                       duration:(NSNumber *)duration
+                          error:(NSError **)error
+{
+  if (![XCUIDevice sharedDevice].supportsPressureInteraction) {
+    return [[[FBErrorBuilder builder]
+             withDescriptionFormat:@"Force press is not supported on this device"]
+            buildError:error];
+  }
+
+  if (nil == relativeCoordinate) {
+    if (nil == pressure || nil == duration) {
+      [self forcePress];
+    } else {
+      [self pressWithPressure:[pressure doubleValue] duration:[duration doubleValue]];
+    }
+  } else {
+    CGVector offset = CGVectorMake(relativeCoordinate.CGPointValue.x,
+                                   relativeCoordinate.CGPointValue.y);
+    XCUICoordinate *hitPoint = [[self coordinateWithNormalizedOffset:CGVectorMake(0, 0)]
+                                coordinateWithOffset:offset];
+    if (nil == pressure || nil == duration) {
+      [hitPoint forcePress];
+    } else {
+      [hitPoint pressWithPressure:[pressure doubleValue] duration:[duration doubleValue]];
+    }
+  }
+  return YES;
+}
+
+@end
+
+#endif

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBIsVisible.h

@@ -0,0 +1,28 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import <WebDriverAgentLib/FBXCElementSnapshotWrapper.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface XCUIElement (FBIsVisible)
+
+/*! Whether or not the element is visible */
+@property (atomic, readonly) BOOL fb_isVisible;
+
+@end
+
+
+@interface FBXCElementSnapshotWrapper (FBIsVisible)
+
+/*! Whether or not the element is visible */
+@property (atomic, readonly) BOOL fb_isVisible;
+
+@end
+
+NS_ASSUME_NONNULL_END

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBIsVisible.m

@@ -0,0 +1,78 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import "XCUIElement+FBIsVisible.h"
+
+#import "FBElementUtils.h"
+#import "FBXCodeCompatibility.h"
+#import "FBXCElementSnapshotWrapper+Helpers.h"
+#import "XCUIElement+FBUtilities.h"
+#import "XCUIElement+FBVisibleFrame.h"
+#import "XCTestPrivateSymbols.h"
+
+NSNumber* _Nullable fetchSnapshotVisibility(id<FBXCElementSnapshot> snapshot)
+{
+  return nil == snapshot.additionalAttributes ? nil : snapshot.additionalAttributes[FB_XCAXAIsVisibleAttribute];
+}
+
+@implementation XCUIElement (FBIsVisible)
+
+- (BOOL)fb_isVisible
+{
+  @autoreleasepool {
+    id<FBXCElementSnapshot> snapshot = [self fb_standardSnapshot];
+    return [FBXCElementSnapshotWrapper ensureWrapped:snapshot].fb_isVisible;
+  }
+}
+
+@end
+
+@implementation FBXCElementSnapshotWrapper (FBIsVisible)
+
+- (BOOL)fb_hasVisibleDescendants
+{
+  for (id<FBXCElementSnapshot> descendant in (self._allDescendants ?: @[])) {
+    if ([fetchSnapshotVisibility(descendant) boolValue]) {
+      return YES;
+    }
+  }
+  return NO;
+}
+
+- (BOOL)fb_isVisible
+{
+  NSNumber *isVisible = fetchSnapshotVisibility(self);
+  if (nil != isVisible) {
+    return isVisible.boolValue;
+  }
+
+  // Fetching the attribute value is expensive.
+  // Shortcircuit here to save time and assume if any of descendants
+  // is already determined as visible then the container should be visible as well
+  if ([self fb_hasVisibleDescendants]) {
+    return YES;
+  }
+
+  NSError *error;
+  NSNumber *attributeValue = [self fb_attributeValue:FB_XCAXAIsVisibleAttributeName
+                                               error:&error];
+  if (nil != attributeValue) {
+    NSMutableDictionary *updatedValue = [NSMutableDictionary dictionaryWithDictionary:self.additionalAttributes ?: @{}];
+    [updatedValue setObject:attributeValue forKey:FB_XCAXAIsVisibleAttribute];
+    self.snapshot.additionalAttributes = updatedValue.copy;
+    @autoreleasepool {
+      return [attributeValue boolValue];
+    }
+  }
+
+  NSLog(@"Cannot determine visiblity of %@ natively: %@. Defaulting to: %@",
+        self.fb_description, error.description, @(NO));
+  return NO;
+}
+
+@end