Module: Selenium::WebDriver::IE::Util

46
47
48
49
50
51
52

# File 'lib/selenium/webdriver/ie/util.rb', line 46
def create_element(&blk)
 element_ptr_ref = FFI::MemoryPointer.new :pointer
 yield element_ptr_ref
 Element .new (self, element_ptr_ref.get_pointer(0))
ensure
 element_ptr_ref.free
end

54
55
56
57
58
59

# File 'lib/selenium/webdriver/ie/util.rb', line 54
def create_element_collection(&blk)
 elements_ptr_ref = FFI::MemoryPointer.new :pointer
 yield elements_ptr_ref
 extract_elements_from elements_ptr_ref
end

61
62
63
64
65
66

# File 'lib/selenium/webdriver/ie/util.rb', line 61
def create_string(&blk)
 wrapper = FFI::MemoryPointer.new :pointer
 yield wrapper
 extract_string_from wrapper
end

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

# File 'lib/selenium/webdriver/ie/util.rb', line 91
def extract_elements_from(elements_ptr_ref)
 elements_ptr = elements_ptr_ref.get_pointer(0)
 length_ptr = FFI::MemoryPointer.new :int
 check_error_code Lib .wdcGetElementCollectionLength(elements_ptr, length_ptr),
 "Cannot extract elements from collection"
 arr = []
 length_ptr.get_int(0).times do |idx|
 arr << create_element do |ptr|
 result = Lib .wdcGetElementAtIndex(elements_ptr, idx, ptr)
 if e = WebDriver ::Error .for_code (result)
 Lib .wdFreeElementCollection(elements_ptr, 1)
 raise e, "unable to create element from collection at index #{idx} (#{result})"
 end
 end
 end
 Lib .wdFreeElementCollection(elements_ptr, 0)
 arr
ensure
 elements_ptr_ref.free
 length_ptr.free
end

68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

# File 'lib/selenium/webdriver/ie/util.rb', line 68
def extract_string_from(string_ptr_ref)
 string_ptr = string_ptr_ref.get_pointer(0)
 return if string_ptr.null? # getElementAttribute()

 length_ptr = FFI::MemoryPointer.new :int
 if Lib .wdStringLength(string_ptr, length_ptr) != 0
 raise Error ::WebDriverError , "cannot determine length of string"
 end
 length = length_ptr.get_int(0)
 raw_string = wstring_ptr(""*length)
 if Lib .wdCopyString(string_ptr, length, raw_string) != 0
 raise Error ::WebDriverError , "cannot copy string from native data to Ruby string"
 end
 wstring_to_bytestring raw_string
ensure
 Lib .wdFreeString(string_ptr) unless string_ptr.null?
 string_ptr_ref.free
end

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

# File 'lib/selenium/webdriver/ie/util.rb', line 18
def string_array_from(raw_strings)
 strings_ptr = raw_strings.get_pointer(0)
 length_ptr = FFI::MemoryPointer.new :int
 result = Lib .wdcGetStringCollectionLength(strings_ptr, length_ptr)
 if result != 0
 raise Error ::WebDriverError , "cannot extract strings from collection, error code: #{result.inspect}"
 end
 arr = []
 length_ptr.get_int(0).times do |idx|
 str_ptr_ref = FFI::MemoryPointer.new :pointer
 result = Lib .wdcGetStringAtIndex(strings_ptr, idx, str_ptr_ref)
 if result != 0
 raise Error ::WebDriverError , "unable to get string at index: #{idx}, error: #{result}"
 end
 arr << extract_string_from(str_ptr_ref)
 end
 arr
ensure
 Lib .wdFreeStringCollection(strings_ptr)
 raw_strings.free
end

120
121
122
123
124
125
126
127
128
129
130
131
132

# File 'lib/selenium/webdriver/ie/util.rb', line 120
def wstring_ptr(str)
 str = str.to_s
 size = Kernel32 .MultiByteToWideChar(CP_UTF8 , 0, str, -1, nil, 0)
 unless size > 0
 raise Error ::WebDriverError , "unable to convert #{str.inspect} to wchar ptr"
 end
 buf = FFI::MemoryPointer.new :pointer, size
 Kernel32 .MultiByteToWideChar(CP_UTF8 , 0, str, -1, buf, size)
 buf
end

134
135
136
137
138
139
140
141
142
143
144
145
146
147

# File 'lib/selenium/webdriver/ie/util.rb', line 134
def wstring_to_bytestring(wstring)
 size = Kernel32 .WideCharToMultiByte(CP_UTF8 , 0, wstring, -1, nil, 0, nil, nil)
 unless size > 0
 raise Error ::WebDriverError , "could not convert wstring pointer to bytestring"
 end
 buf = FFI::MemoryPointer.new :pointer, size
 Kernel32 .WideCharToMultiByte(CP_UTF8 , 0, wstring, -1, buf, size, nil, nil )
 buf.get_bytes(0, size - 1)
ensure
 buf.free if buf
end