9ea64322d112206d2918ab447cb369ac2584ea2f
[WebKit-https.git] / Tools / WebKitTestRunner / InjectedBundle / mac / AccessibilityCommonMac.mm
1 /*
2  * Copyright (C) 2012 Google Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #import "config.h"
32 #import "AccessibilityCommonMac.h"
33
34 #import <JavaScriptCore/JSRetainPtr.h>
35 #import <JavaScriptCore/JSStringRef.h>
36 #import <JavaScriptCore/JSStringRefCF.h>
37
38 @implementation NSString (JSStringRefAdditions)
39
40 + (NSString *)stringWithJSStringRef:(JSStringRef)jsStringRef
41 {
42     if (!jsStringRef)
43         return nil;
44
45     return CFBridgingRelease(JSStringCopyCFString(kCFAllocatorDefault, jsStringRef));
46 }
47
48 - (JSStringRef)createJSStringRef
49 {
50     return JSStringCreateWithCFString((__bridge CFStringRef)self);
51 }
52
53 namespace WTR {
54
55 NSDictionary *searchPredicateParameterizedAttributeForSearchCriteria(JSContextRef context, AccessibilityUIElement *startElement, bool isDirectionNext, unsigned resultsLimit, JSValueRef searchKey, JSStringRef searchText, bool visibleOnly, bool immediateDescendantsOnly)
56 {
57     NSMutableDictionary *parameterizedAttribute = [NSMutableDictionary dictionary];
58     
59     if (startElement && startElement->platformUIElement())
60         [parameterizedAttribute setObject:(id)startElement->platformUIElement() forKey:@"AXStartElement"];
61     
62     [parameterizedAttribute setObject:(isDirectionNext) ? @"AXDirectionNext" : @"AXDirectionPrevious" forKey:@"AXDirection"];
63     
64     [parameterizedAttribute setObject:@(resultsLimit) forKey:@"AXResultsLimit"];
65     
66     if (searchKey) {
67         id searchKeyParameter = nil;
68         if (JSValueIsString(context, searchKey)) {
69             JSRetainPtr<JSStringRef> searchKeyString(Adopt, JSValueToStringCopy(context, searchKey, nullptr));
70             if (searchKeyString)
71                 searchKeyParameter = [NSString stringWithJSStringRef:searchKeyString.get()];
72         } else if (JSValueIsObject(context, searchKey)) {
73             JSObjectRef searchKeyArray = JSValueToObject(context, searchKey, nullptr);
74             unsigned searchKeyArrayLength = 0;
75             
76             JSRetainPtr<JSStringRef> lengthPropertyString(Adopt, JSStringCreateWithUTF8CString("length"));
77             JSValueRef searchKeyArrayLengthValue = JSObjectGetProperty(context, searchKeyArray, lengthPropertyString.get(), nullptr);
78             if (searchKeyArrayLengthValue && JSValueIsNumber(context, searchKeyArrayLengthValue))
79                 searchKeyArrayLength = static_cast<unsigned>(JSValueToNumber(context, searchKeyArrayLengthValue, nullptr));
80             
81             for (unsigned i = 0; i < searchKeyArrayLength; ++i) {
82                 JSValueRef searchKeyValue = JSObjectGetPropertyAtIndex(context, searchKeyArray, i, nullptr);
83                 JSStringRef searchKeyString = JSValueToStringCopy(context, searchKeyValue, nullptr);
84                 if (searchKeyString) {
85                     if (!searchKeyParameter)
86                         searchKeyParameter = [NSMutableArray array];
87                     [searchKeyParameter addObject:[NSString stringWithJSStringRef:searchKeyString]];
88                     JSStringRelease(searchKeyString);
89                 }
90             }
91         }
92         if (searchKeyParameter)
93             [parameterizedAttribute setObject:searchKeyParameter forKey:@"AXSearchKey"];
94     }
95     
96     if (searchText && JSStringGetLength(searchText))
97         [parameterizedAttribute setObject:[NSString stringWithJSStringRef:searchText] forKey:@"AXSearchText"];
98     
99     [parameterizedAttribute setObject:@(visibleOnly) forKey:@"AXVisibleOnly"];
100     
101     [parameterizedAttribute setObject:@(immediateDescendantsOnly) forKey:@"AXImmediateDescendantsOnly"];
102     
103     return parameterizedAttribute;
104 }
105     
106 } // namespace WTR
107
108 @end