WebDriverAgent/WebDriverAgentLib/Utilities/FBXCAXClientProxy.m at 24f95c5256e582b152f20439cdad840e4511d28e · appium/WebDriverAgent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/**
 * 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. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

#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