初始化提交

WebDriverAgentLib/Utilities/FBXCAXClientProxy.m

@@ -0,0 +1,119 @@
+/**
+ * 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 "FBXCAXClientProxy.h"
+
+#import "FBXCAccessibilityElement.h"
+#import "FBLogger.h"
+#import "FBMacros.h"
+#import "XCAXClient_iOS+FBSnapshotReqParams.h"
+#import "XCUIDevice.h"
+#import "XCUIApplication.h"
+
+static id FBAXClient = nil;
+
+@interface FBXCAXClientProxy ()
+
+@property (nonatomic) NSMutableDictionary<NSNumber *, XCUIApplication *> *appsCache;
+
+@end
+
+@implementation FBXCAXClientProxy
+
++ (instancetype)sharedClient
+{
+  static FBXCAXClientProxy *instance = nil;
+  static dispatch_once_t onceToken;
+  dispatch_once(&onceToken, ^{
+    instance = [[self alloc] init];
+    instance.appsCache = [NSMutableDictionary dictionary];
+    FBAXClient = [XCUIDevice.sharedDevice accessibilityInterface];
+  });
+  return instance;
+}
+
+- (BOOL)setAXTimeout:(NSTimeInterval)timeout error:(NSError **)error
+{
+  return [FBAXClient _setAXTimeout:timeout error:error];
+}
+
+- (id<FBXCElementSnapshot>)snapshotForElement:(id<FBXCAccessibilityElement>)element
+                                   attributes:(NSArray<NSString *> *)attributes
+                                      inDepth:(BOOL)inDepth
+                                        error:(NSError **)error
+{
+  NSMutableDictionary *parameters = [NSMutableDictionary dictionaryWithDictionary:self.defaultParameters];
+  if (!inDepth) {
+    parameters[FBSnapshotMaxDepthKey] = @1;
+  }
+
+  id result = [FBAXClient requestSnapshotForElement:element
+                                         attributes:attributes
+                                         parameters:[parameters copy]
+                                              error:error];
+  id<FBXCElementSnapshot> snapshot = [result valueForKey:@"_rootElementSnapshot"];
+  return nil == snapshot ? result : snapshot;
+}
+
+- (NSArray<id<FBXCAccessibilityElement>> *)activeApplications
+{
+  return [FBAXClient activeApplications];
+}
+
+- (id<FBXCAccessibilityElement>)systemApplication
+{
+  return [FBAXClient systemApplication];
+}
+
+- (NSDictionary *)defaultParameters
+{
+  return [FBAXClient defaultParameters];
+}
+
+- (void)notifyWhenNoAnimationsAreActiveForApplication:(XCUIApplication *)application
+                                                reply:(void (^)(void))reply
+{
+  [FBAXClient notifyWhenNoAnimationsAreActiveForApplication:application reply:reply];
+}
+
+- (NSDictionary *)attributesForElement:(id<FBXCAccessibilityElement>)element
+                            attributes:(NSArray *)attributes
+                                 error:(NSError**)error;
+{
+  return [FBAXClient attributesForElement:element
+                               attributes:attributes
+                                    error:error];
+}
+
+- (XCUIApplication *)monitoredApplicationWithProcessIdentifier:(int)pid
+{
+  NSMutableSet *terminatedAppIds = [NSMutableSet set];
+  for (NSNumber *appPid in self.appsCache) {
+    if (![self.appsCache[appPid] running]) {
+      [terminatedAppIds addObject:appPid];
+    }
+  }
+  for (NSNumber *appPid in terminatedAppIds) {
+    [self.appsCache removeObjectForKey:appPid];
+  }
+
+  XCUIApplication *result = [self.appsCache objectForKey:@(pid)];
+  if (nil != result) {
+    return result;
+  }
+
+  XCUIApplication *app = [[FBAXClient applicationProcessTracker]
+                          monitoredApplicationWithProcessIdentifier:pid];
+  if (nil == app) {
+    return nil;
+  }
+  [self.appsCache setObject:app forKey:@(pid)];
+  return app;
+}
+
+@end