solargraph 0.58.1 → 0.59.0.dev.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (162) hide show
  1. checksums.yaml +4 -4
  2. data/.envrc +3 -0
  3. data/.github/workflows/linting.yml +4 -5
  4. data/.github/workflows/plugins.yml +40 -36
  5. data/.github/workflows/rspec.yml +45 -13
  6. data/.github/workflows/typecheck.yml +2 -2
  7. data/.rubocop_todo.yml +27 -49
  8. data/README.md +3 -3
  9. data/Rakefile +1 -0
  10. data/lib/solargraph/api_map/cache.rb +110 -110
  11. data/lib/solargraph/api_map/constants.rb +289 -279
  12. data/lib/solargraph/api_map/index.rb +204 -193
  13. data/lib/solargraph/api_map/source_to_yard.rb +109 -97
  14. data/lib/solargraph/api_map/store.rb +387 -384
  15. data/lib/solargraph/api_map.rb +1000 -945
  16. data/lib/solargraph/complex_type/conformance.rb +176 -0
  17. data/lib/solargraph/complex_type/type_methods.rb +242 -228
  18. data/lib/solargraph/complex_type/unique_type.rb +632 -482
  19. data/lib/solargraph/complex_type.rb +549 -444
  20. data/lib/solargraph/convention/data_definition/data_definition_node.rb +93 -91
  21. data/lib/solargraph/convention/data_definition.rb +108 -105
  22. data/lib/solargraph/convention/struct_definition/struct_assignment_node.rb +62 -61
  23. data/lib/solargraph/convention/struct_definition/struct_definition_node.rb +103 -102
  24. data/lib/solargraph/convention/struct_definition.rb +168 -164
  25. data/lib/solargraph/diagnostics/require_not_found.rb +54 -53
  26. data/lib/solargraph/diagnostics/rubocop.rb +119 -118
  27. data/lib/solargraph/diagnostics/rubocop_helpers.rb +70 -68
  28. data/lib/solargraph/diagnostics/type_check.rb +56 -55
  29. data/lib/solargraph/doc_map.rb +200 -439
  30. data/lib/solargraph/equality.rb +34 -34
  31. data/lib/solargraph/gem_pins.rb +97 -98
  32. data/lib/solargraph/language_server/host/dispatch.rb +131 -130
  33. data/lib/solargraph/language_server/host/message_worker.rb +113 -112
  34. data/lib/solargraph/language_server/host/sources.rb +100 -99
  35. data/lib/solargraph/language_server/host.rb +883 -878
  36. data/lib/solargraph/language_server/message/extended/check_gem_version.rb +109 -114
  37. data/lib/solargraph/language_server/message/extended/document.rb +24 -23
  38. data/lib/solargraph/language_server/message/text_document/completion.rb +58 -56
  39. data/lib/solargraph/language_server/message/text_document/definition.rb +42 -40
  40. data/lib/solargraph/language_server/message/text_document/document_symbol.rb +28 -26
  41. data/lib/solargraph/language_server/message/text_document/formatting.rb +150 -148
  42. data/lib/solargraph/language_server/message/text_document/hover.rb +60 -58
  43. data/lib/solargraph/language_server/message/text_document/signature_help.rb +25 -24
  44. data/lib/solargraph/language_server/message/text_document/type_definition.rb +27 -25
  45. data/lib/solargraph/language_server/message/workspace/workspace_symbol.rb +25 -23
  46. data/lib/solargraph/library.rb +729 -683
  47. data/lib/solargraph/location.rb +87 -82
  48. data/lib/solargraph/logging.rb +57 -37
  49. data/lib/solargraph/parser/comment_ripper.rb +76 -69
  50. data/lib/solargraph/parser/flow_sensitive_typing.rb +483 -255
  51. data/lib/solargraph/parser/node_processor/base.rb +122 -92
  52. data/lib/solargraph/parser/node_processor.rb +63 -62
  53. data/lib/solargraph/parser/parser_gem/class_methods.rb +167 -149
  54. data/lib/solargraph/parser/parser_gem/node_chainer.rb +191 -166
  55. data/lib/solargraph/parser/parser_gem/node_methods.rb +506 -486
  56. data/lib/solargraph/parser/parser_gem/node_processors/and_node.rb +22 -22
  57. data/lib/solargraph/parser/parser_gem/node_processors/args_node.rb +61 -59
  58. data/lib/solargraph/parser/parser_gem/node_processors/begin_node.rb +24 -15
  59. data/lib/solargraph/parser/parser_gem/node_processors/block_node.rb +46 -46
  60. data/lib/solargraph/parser/parser_gem/node_processors/def_node.rb +60 -53
  61. data/lib/solargraph/parser/parser_gem/node_processors/if_node.rb +53 -23
  62. data/lib/solargraph/parser/parser_gem/node_processors/ivasgn_node.rb +41 -40
  63. data/lib/solargraph/parser/parser_gem/node_processors/lvasgn_node.rb +30 -29
  64. data/lib/solargraph/parser/parser_gem/node_processors/masgn_node.rb +61 -59
  65. data/lib/solargraph/parser/parser_gem/node_processors/opasgn_node.rb +98 -98
  66. data/lib/solargraph/parser/parser_gem/node_processors/or_node.rb +22 -0
  67. data/lib/solargraph/parser/parser_gem/node_processors/orasgn_node.rb +17 -17
  68. data/lib/solargraph/parser/parser_gem/node_processors/resbody_node.rb +39 -38
  69. data/lib/solargraph/parser/parser_gem/node_processors/sclass_node.rb +53 -52
  70. data/lib/solargraph/parser/parser_gem/node_processors/send_node.rb +296 -291
  71. data/lib/solargraph/parser/parser_gem/node_processors/when_node.rb +23 -0
  72. data/lib/solargraph/parser/parser_gem/node_processors/while_node.rb +33 -29
  73. data/lib/solargraph/parser/parser_gem/node_processors.rb +74 -70
  74. data/lib/solargraph/parser/region.rb +75 -69
  75. data/lib/solargraph/parser/snippet.rb +17 -17
  76. data/lib/solargraph/pin/base.rb +761 -729
  77. data/lib/solargraph/pin/base_variable.rb +418 -126
  78. data/lib/solargraph/pin/block.rb +126 -104
  79. data/lib/solargraph/pin/breakable.rb +13 -9
  80. data/lib/solargraph/pin/callable.rb +278 -231
  81. data/lib/solargraph/pin/closure.rb +68 -72
  82. data/lib/solargraph/pin/common.rb +94 -79
  83. data/lib/solargraph/pin/compound_statement.rb +55 -0
  84. data/lib/solargraph/pin/conversions.rb +124 -123
  85. data/lib/solargraph/pin/delegated_method.rb +131 -120
  86. data/lib/solargraph/pin/documenting.rb +115 -114
  87. data/lib/solargraph/pin/instance_variable.rb +38 -34
  88. data/lib/solargraph/pin/keyword.rb +16 -20
  89. data/lib/solargraph/pin/local_variable.rb +31 -75
  90. data/lib/solargraph/pin/method.rb +720 -672
  91. data/lib/solargraph/pin/method_alias.rb +42 -34
  92. data/lib/solargraph/pin/namespace.rb +121 -115
  93. data/lib/solargraph/pin/parameter.rb +338 -275
  94. data/lib/solargraph/pin/proxy_type.rb +40 -39
  95. data/lib/solargraph/pin/reference/override.rb +47 -47
  96. data/lib/solargraph/pin/reference/superclass.rb +17 -15
  97. data/lib/solargraph/pin/reference.rb +41 -39
  98. data/lib/solargraph/pin/search.rb +62 -61
  99. data/lib/solargraph/pin/signature.rb +69 -61
  100. data/lib/solargraph/pin/symbol.rb +53 -53
  101. data/lib/solargraph/pin/until.rb +18 -18
  102. data/lib/solargraph/pin/while.rb +18 -18
  103. data/lib/solargraph/pin.rb +46 -44
  104. data/lib/solargraph/pin_cache.rb +665 -245
  105. data/lib/solargraph/position.rb +118 -119
  106. data/lib/solargraph/range.rb +112 -112
  107. data/lib/solargraph/rbs_map/conversions.rb +846 -823
  108. data/lib/solargraph/rbs_map/core_map.rb +65 -58
  109. data/lib/solargraph/rbs_map/stdlib_map.rb +72 -43
  110. data/lib/solargraph/rbs_map.rb +217 -163
  111. data/lib/solargraph/shell.rb +397 -352
  112. data/lib/solargraph/source/chain/call.rb +372 -337
  113. data/lib/solargraph/source/chain/constant.rb +28 -26
  114. data/lib/solargraph/source/chain/hash.rb +35 -34
  115. data/lib/solargraph/source/chain/if.rb +29 -28
  116. data/lib/solargraph/source/chain/instance_variable.rb +34 -13
  117. data/lib/solargraph/source/chain/literal.rb +53 -48
  118. data/lib/solargraph/source/chain/or.rb +31 -23
  119. data/lib/solargraph/source/chain.rb +294 -291
  120. data/lib/solargraph/source/change.rb +89 -82
  121. data/lib/solargraph/source/cursor.rb +172 -166
  122. data/lib/solargraph/source/source_chainer.rb +204 -194
  123. data/lib/solargraph/source/updater.rb +59 -55
  124. data/lib/solargraph/source.rb +524 -498
  125. data/lib/solargraph/source_map/clip.rb +237 -226
  126. data/lib/solargraph/source_map/data.rb +37 -34
  127. data/lib/solargraph/source_map/mapper.rb +282 -259
  128. data/lib/solargraph/source_map.rb +220 -212
  129. data/lib/solargraph/type_checker/problem.rb +34 -32
  130. data/lib/solargraph/type_checker/rules.rb +157 -84
  131. data/lib/solargraph/type_checker.rb +895 -814
  132. data/lib/solargraph/version.rb +1 -1
  133. data/lib/solargraph/workspace/config.rb +257 -255
  134. data/lib/solargraph/workspace/gemspecs.rb +367 -0
  135. data/lib/solargraph/workspace/require_paths.rb +98 -97
  136. data/lib/solargraph/workspace.rb +362 -220
  137. data/lib/solargraph/yard_map/helpers.rb +45 -44
  138. data/lib/solargraph/yard_map/mapper/to_method.rb +134 -130
  139. data/lib/solargraph/yard_map/mapper/to_namespace.rb +32 -31
  140. data/lib/solargraph/yard_map/mapper.rb +84 -79
  141. data/lib/solargraph/yardoc.rb +97 -87
  142. data/lib/solargraph.rb +126 -105
  143. data/rbs/fills/rubygems/0/dependency.rbs +193 -0
  144. data/rbs/fills/tuple/tuple.rbs +28 -0
  145. data/rbs/shims/ast/0/node.rbs +5 -0
  146. data/rbs/shims/diff-lcs/1.5/diff-lcs.rbs +11 -0
  147. data/rbs_collection.yaml +1 -1
  148. data/solargraph.gemspec +2 -1
  149. metadata +22 -17
  150. data/lib/solargraph/type_checker/checks.rb +0 -124
  151. data/lib/solargraph/type_checker/param_def.rb +0 -37
  152. data/lib/solargraph/yard_map/to_method.rb +0 -89
  153. data/sig/shims/ast/0/node.rbs +0 -5
  154. /data/{sig → rbs}/shims/ast/2.4/.rbs_meta.yaml +0 -0
  155. /data/{sig → rbs}/shims/ast/2.4/ast.rbs +0 -0
  156. /data/{sig → rbs}/shims/parser/3.2.0.1/builders/default.rbs +0 -0
  157. /data/{sig → rbs}/shims/parser/3.2.0.1/manifest.yaml +0 -0
  158. /data/{sig → rbs}/shims/parser/3.2.0.1/parser.rbs +0 -0
  159. /data/{sig → rbs}/shims/parser/3.2.0.1/polyfill.rbs +0 -0
  160. /data/{sig → rbs}/shims/thor/1.2.0.1/.rbs_meta.yaml +0 -0
  161. /data/{sig → rbs}/shims/thor/1.2.0.1/manifest.yaml +0 -0
  162. /data/{sig → rbs}/shims/thor/1.2.0.1/thor.rbs +0 -0
@@ -1,444 +1,549 @@
1
- # frozen_string_literal: true
2
-
3
- module Solargraph
4
- # A container for type data based on YARD type tags.
5
- #
6
- class ComplexType
7
- GENERIC_TAG_NAME = 'generic'.freeze
8
- # @!parse
9
- # include TypeMethods
10
- include Equality
11
-
12
- autoload :TypeMethods, 'solargraph/complex_type/type_methods'
13
- autoload :UniqueType, 'solargraph/complex_type/unique_type'
14
-
15
- # @param types [Array<UniqueType, ComplexType>]
16
- def initialize types = [UniqueType::UNDEFINED]
17
- # @todo @items here should not need an annotation
18
- # @type [Array<UniqueType>]
19
- items = types.flat_map(&:items).uniq(&:to_s)
20
- if items.any? { |i| i.name == 'false' } && items.any? { |i| i.name == 'true' }
21
- items.delete_if { |i| i.name == 'false' || i.name == 'true' }
22
- items.unshift(ComplexType::BOOLEAN)
23
- end
24
- items = [UniqueType::UNDEFINED] if items.any?(&:undefined?)
25
- @items = items
26
- end
27
-
28
- # @sg-ignore Fix "Not enough arguments to Module#protected"
29
- protected def equality_fields
30
- [self.class, items]
31
- end
32
-
33
- # @param api_map [ApiMap]
34
- # @param context [String]
35
- # @return [ComplexType]
36
- def qualify api_map, *gates
37
- red = reduce_object
38
- types = red.items.map do |t|
39
- next t if ['nil', 'void', 'undefined'].include?(t.name)
40
- next t if ['::Boolean'].include?(t.rooted_name)
41
- t.qualify api_map, *gates
42
- end
43
- ComplexType.new(types).reduce_object
44
- end
45
-
46
- # @param generics_to_resolve [Enumerable<String>]]
47
- # @param context_type [UniqueType, nil]
48
- # @param resolved_generic_values [Hash{String => ComplexType}] Added to as types are encountered or resolved
49
- # @return [self]
50
- def resolve_generics_from_context generics_to_resolve, context_type, resolved_generic_values: {}
51
- return self unless generic?
52
-
53
- ComplexType.new(@items.map { |i| i.resolve_generics_from_context(generics_to_resolve, context_type, resolved_generic_values: resolved_generic_values) })
54
- end
55
-
56
- # @return [UniqueType]
57
- def first
58
- @items.first
59
- end
60
-
61
- # @return [String]
62
- def to_rbs
63
- ((@items.length > 1 ? '(' : '') +
64
- @items.map(&:to_rbs).join(' | ') +
65
- (@items.length > 1 ? ')' : ''))
66
- end
67
-
68
- # @param dst [ComplexType]
69
- # @return [ComplexType]
70
- def self_to_type dst
71
- object_type_dst = dst.reduce_class_type
72
- transform do |t|
73
- next t if t.name != 'self'
74
- object_type_dst
75
- end
76
- end
77
-
78
- # @yieldparam [UniqueType]
79
- # @return [Array<UniqueType>]
80
- def map &block
81
- @items.map &block
82
- end
83
-
84
- # @yieldparam [UniqueType]
85
- # @return [Enumerable<UniqueType>]
86
- def each &block
87
- @items.each &block
88
- end
89
-
90
- # @yieldparam [UniqueType]
91
- # @return [void]
92
- # @overload each_unique_type()
93
- # @return [Enumerator<UniqueType>]
94
- def each_unique_type &block
95
- return enum_for(__method__) unless block_given?
96
-
97
- @items.each do |item|
98
- item.each_unique_type &block
99
- end
100
- end
101
-
102
- # @param atype [ComplexType] type which may be assigned to this type
103
- # @param api_map [ApiMap] The ApiMap that performs qualification
104
- def can_assign?(api_map, atype)
105
- any? { |ut| ut.can_assign?(api_map, atype) }
106
- end
107
-
108
- # @param new_name [String, nil]
109
- # @param make_rooted [Boolean, nil]
110
- # @param new_key_types [Array<ComplexType>, nil]
111
- # @param rooted [Boolean, nil]
112
- # @param new_subtypes [Array<ComplexType>, nil]
113
- # @return [self]
114
- def recreate(new_name: nil, make_rooted: nil, new_key_types: nil, new_subtypes: nil)
115
- ComplexType.new(map do |ut|
116
- ut.recreate(new_name: new_name,
117
- make_rooted: make_rooted,
118
- new_key_types: new_key_types,
119
- new_subtypes: new_subtypes)
120
- end)
121
- end
122
-
123
- # @return [Integer]
124
- def length
125
- @items.length
126
- end
127
-
128
- # @return [Array<UniqueType>]
129
- def to_a
130
- @items
131
- end
132
-
133
- # @param index [Integer]
134
- # @return [UniqueType]
135
- def [](index)
136
- @items[index]
137
- end
138
-
139
- # @return [Array<UniqueType>]
140
- def select &block
141
- @items.select &block
142
- end
143
-
144
- # @return [String]
145
- def namespace
146
- # cache this attr for high frequency call
147
- @namespace ||= method_missing(:namespace).to_s
148
- end
149
-
150
- # @return [Array<String>]
151
- def namespaces
152
- @items.map(&:namespace)
153
- end
154
-
155
- # @param name [Symbol]
156
- # @return [Object, nil]
157
- def method_missing name, *args, &block
158
- return if @items.first.nil?
159
- return @items.first.send(name, *args, &block) if respond_to_missing?(name)
160
- super
161
- end
162
-
163
- # @param name [Symbol]
164
- # @param include_private [Boolean]
165
- def respond_to_missing?(name, include_private = false)
166
- TypeMethods.public_instance_methods.include?(name) || super
167
- end
168
-
169
- def to_s
170
- map(&:tag).join(', ')
171
- end
172
-
173
- # @return [String]
174
- def tags
175
- map(&:tag).join(', ')
176
- end
177
-
178
- # @return [String]
179
- def simple_tags
180
- simplify_literals.tags
181
- end
182
-
183
- def literal?
184
- @items.any?(&:literal?)
185
- end
186
-
187
- # @return [ComplexType]
188
- def downcast_to_literal_if_possible
189
- ComplexType.new(items.map(&:downcast_to_literal_if_possible))
190
- end
191
-
192
- # @return [String]
193
- def desc
194
- rooted_tags
195
- end
196
-
197
- # @return [String]
198
- def rooted_tags
199
- map(&:rooted_tag).join(', ')
200
- end
201
-
202
- # @yieldparam [UniqueType]
203
- def all? &block
204
- @items.all? &block
205
- end
206
-
207
- # @yieldparam [UniqueType]
208
- # @yieldreturn [Boolean]
209
- # @return [Boolean]
210
- def any? &block
211
- @items.compact.any? &block
212
- end
213
-
214
- def selfy?
215
- @items.any?(&:selfy?)
216
- end
217
-
218
- def generic?
219
- any?(&:generic?)
220
- end
221
-
222
- # @return [self]
223
- def simplify_literals
224
- ComplexType.new(map(&:simplify_literals))
225
- end
226
-
227
- # @param new_name [String, nil]
228
- # @yieldparam t [UniqueType]
229
- # @yieldreturn [UniqueType]
230
- # @return [ComplexType]
231
- def transform(new_name = nil, &transform_type)
232
- raise "Please remove leading :: and set rooted with recreate() instead - #{new_name}" if new_name&.start_with?('::')
233
- ComplexType.new(map { |ut| ut.transform(new_name, &transform_type) })
234
- end
235
-
236
- # @return [self]
237
- def force_rooted
238
- transform do |t|
239
- t.recreate(make_rooted: true)
240
- end
241
- end
242
-
243
- # @param definitions [Pin::Namespace, Pin::Method]
244
- # @param context_type [ComplexType]
245
- # @return [ComplexType]
246
- def resolve_generics definitions, context_type
247
- result = @items.map { |i| i.resolve_generics(definitions, context_type) }
248
- ComplexType.new(result)
249
- end
250
-
251
- def nullable?
252
- @items.any?(&:nil_type?)
253
- end
254
-
255
- # @return [Array<ComplexType>]
256
- def all_params
257
- @items.first.all_params || []
258
- end
259
-
260
- # @return [ComplexType]
261
- def reduce_class_type
262
- new_items = items.flat_map do |type|
263
- next type unless ['Module', 'Class'].include?(type.name)
264
- next type if type.all_params.empty?
265
-
266
- type.all_params
267
- end
268
- ComplexType.new(new_items)
269
- end
270
-
271
- # every type and subtype in this union have been resolved to be
272
- # fully qualified
273
- def all_rooted?
274
- all?(&:all_rooted?)
275
- end
276
-
277
- # every top-level type has resolved to be fully qualified; see
278
- # #all_rooted? to check their subtypes as well
279
- def rooted?
280
- all?(&:rooted?)
281
- end
282
-
283
- attr_reader :items
284
-
285
- def rooted?
286
- @items.all?(&:rooted?)
287
- end
288
-
289
- protected
290
-
291
- # @return [ComplexType]
292
- def reduce_object
293
- new_items = items.flat_map do |ut|
294
- next [ut] if ut.name != 'Object' || ut.subtypes.empty?
295
- ut.subtypes
296
- end
297
- ComplexType.new(new_items)
298
- end
299
-
300
- def bottom?
301
- @items.all?(&:bot?)
302
- end
303
-
304
- class << self
305
- # Parse type strings into a ComplexType.
306
- #
307
- # @example
308
- # ComplexType.parse 'String', 'Foo', 'nil' #=> [String, Foo, nil]
309
- #
310
- # @note
311
- # The `partial` parameter is used to indicate that the method is
312
- # receiving a string that will be used inside another ComplexType.
313
- # It returns arrays of ComplexTypes instead of a single cohesive one.
314
- # Consumers should not need to use this parameter; it should only be
315
- # used internally.
316
- #
317
- # @param strings [Array<String>] The type definitions to parse
318
- # @return [ComplexType]
319
- # # @overload parse(*strings, partial: false)
320
- # # @todo Need ability to use a literal true as a type below
321
- # # @param partial [Boolean] True if the string is part of a another type
322
- # # @return [Array<UniqueType>]
323
- # @todo To be able to select the right signature above,
324
- # Chain::Call needs to know the decl type (:arg, :optarg,
325
- # :kwarg, etc) of the arguments given, instead of just having
326
- # an array of Chains as the arguments.
327
- def parse *strings, partial: false
328
- # @type [Hash{Array<String> => ComplexType}]
329
- @cache ||= {}
330
- unless partial
331
- cached = @cache[strings]
332
- return cached unless cached.nil?
333
- end
334
- types = []
335
- key_types = nil
336
- strings.each do |type_string|
337
- point_stack = 0
338
- curly_stack = 0
339
- paren_stack = 0
340
- base = String.new
341
- subtype_string = String.new
342
- # @param char [String]
343
- type_string&.each_char do |char|
344
- if char == '='
345
- #raise ComplexTypeError, "Invalid = in type #{type_string}" unless curly_stack > 0
346
- elsif char == '<'
347
- point_stack += 1
348
- elsif char == '>'
349
- if subtype_string.end_with?('=') && curly_stack > 0
350
- subtype_string += char
351
- elsif base.end_with?('=')
352
- raise ComplexTypeError, "Invalid hash thing" unless key_types.nil?
353
- # types.push ComplexType.new([UniqueType.new(base[0..-2].strip)])
354
- types.push UniqueType.parse(base[0..-2].strip, subtype_string)
355
- # @todo this should either expand key_type's type
356
- # automatically or complain about not being
357
- # compatible with key_type's type in type checking
358
- key_types = types
359
- types = []
360
- base.clear
361
- subtype_string.clear
362
- next
363
- else
364
- raise ComplexTypeError, "Invalid close in type #{type_string}" if point_stack == 0
365
- point_stack -= 1
366
- subtype_string += char
367
- end
368
- next
369
- elsif char == '{'
370
- curly_stack += 1
371
- elsif char == '}'
372
- curly_stack -= 1
373
- subtype_string += char
374
- raise ComplexTypeError, "Invalid close in type #{type_string}" if curly_stack < 0
375
- next
376
- elsif char == '('
377
- paren_stack += 1
378
- elsif char == ')'
379
- paren_stack -= 1
380
- subtype_string += char
381
- raise ComplexTypeError, "Invalid close in type #{type_string}" if paren_stack < 0
382
- next
383
- elsif char == ',' && point_stack == 0 && curly_stack == 0 && paren_stack == 0
384
- # types.push ComplexType.new([UniqueType.new(base.strip, subtype_string.strip)])
385
- types.push UniqueType.parse(base.strip, subtype_string.strip)
386
- base.clear
387
- subtype_string.clear
388
- next
389
- end
390
- if point_stack == 0 && curly_stack == 0 && paren_stack == 0
391
- base.concat char
392
- else
393
- subtype_string.concat char
394
- end
395
- end
396
- raise ComplexTypeError, "Unclosed subtype in #{type_string}" if point_stack != 0 || curly_stack != 0 || paren_stack != 0
397
- # types.push ComplexType.new([UniqueType.new(base, subtype_string)])
398
- types.push UniqueType.parse(base.strip, subtype_string.strip)
399
- end
400
- unless key_types.nil?
401
- raise ComplexTypeError, "Invalid use of key/value parameters" unless partial
402
- return key_types if types.empty?
403
- return [key_types, types]
404
- end
405
- result = partial ? types : ComplexType.new(types)
406
- @cache[strings] = result unless partial
407
- result
408
- end
409
-
410
- # @param strings [Array<String>]
411
- # @return [ComplexType]
412
- def try_parse *strings
413
- parse *strings
414
- rescue ComplexTypeError => e
415
- Solargraph.logger.info "Error parsing complex type `#{strings.join(', ')}`: #{e.message}"
416
- ComplexType::UNDEFINED
417
- end
418
- end
419
-
420
- VOID = ComplexType.parse('void')
421
- UNDEFINED = ComplexType.parse('undefined')
422
- SYMBOL = ComplexType.parse('::Symbol')
423
- ROOT = ComplexType.parse('::Class<>')
424
- NIL = ComplexType.parse('nil')
425
- SELF = ComplexType.parse('self')
426
- BOOLEAN = ComplexType.parse('::Boolean')
427
- BOT = ComplexType.parse('bot')
428
-
429
- private
430
-
431
- # @todo This is a quick and dirty hack that forces `self` keywords
432
- # to reference an instance of their class and never the class itself.
433
- # This behavior may change depending on which result is expected
434
- # from YARD conventions. See https://github.com/lsegal/yard/issues/1257
435
- # @param dst [String]
436
- # @return [String]
437
- def reduce_class dst
438
- while dst =~ /^(Class|Module)\<(.*?)\>$/
439
- dst = dst.sub(/^(Class|Module)\</, '').sub(/\>$/, '')
440
- end
441
- dst
442
- end
443
- end
444
- end
1
+ # frozen_string_literal: true
2
+
3
+ module Solargraph
4
+ # A container for type data based on YARD type tags.
5
+ #
6
+ class ComplexType
7
+ GENERIC_TAG_NAME = 'generic'.freeze
8
+ # @!parse
9
+ # include TypeMethods
10
+ include Equality
11
+
12
+ autoload :Conformance, 'solargraph/complex_type/conformance'
13
+ autoload :TypeMethods, 'solargraph/complex_type/type_methods'
14
+ autoload :UniqueType, 'solargraph/complex_type/unique_type'
15
+
16
+ # @param types [Array<UniqueType, ComplexType>]
17
+ def initialize types = [UniqueType::UNDEFINED]
18
+ # @todo @items here should not need an annotation
19
+ # @type [Array<UniqueType>]
20
+ items = types.flat_map(&:items).uniq(&:to_s)
21
+ if items.any? { |i| i.name == 'false' } && items.any? { |i| i.name == 'true' }
22
+ items.delete_if { |i| i.name == 'false' || i.name == 'true' }
23
+ items.unshift(UniqueType::BOOLEAN)
24
+ end
25
+ # @type [Array<UniqueType>]
26
+ items = [UniqueType::UNDEFINED] if items.any?(&:undefined?)
27
+ # @todo shouldn't need this cast - if statement above adds an 'Array' type
28
+ # @type [Array<UniqueType>]
29
+ @items = items
30
+ end
31
+
32
+ protected def equality_fields
33
+ [self.class, items]
34
+ end
35
+
36
+ # @param api_map [ApiMap]
37
+ # @param context [String]
38
+ # @return [ComplexType]
39
+ def qualify api_map, *gates
40
+ red = reduce_object
41
+ types = red.items.map do |t|
42
+ next t if ['nil', 'void', 'undefined'].include?(t.name)
43
+ next t if ['::Boolean'].include?(t.rooted_name)
44
+ t.qualify api_map, *gates
45
+ end
46
+ ComplexType.new(types).reduce_object
47
+ end
48
+
49
+ # @param generics_to_resolve [Enumerable<String>]]
50
+ # @param context_type [ComplexType, ComplexType::UniqueType, nil]
51
+ # @param resolved_generic_values [Hash{String => ComplexType}] Added to as types are encountered or resolved
52
+ # @return [self]
53
+ def resolve_generics_from_context generics_to_resolve, context_type, resolved_generic_values: {}
54
+ return self unless generic?
55
+
56
+ ComplexType.new(@items.map { |i| i.resolve_generics_from_context(generics_to_resolve, context_type, resolved_generic_values: resolved_generic_values) })
57
+ end
58
+
59
+ # @return [UniqueType]
60
+ def first
61
+ @items.first
62
+ end
63
+
64
+ # @return [String]
65
+ def to_rbs
66
+ ((@items.length > 1 ? '(' : '') +
67
+ @items.map(&:to_rbs).join(' | ') +
68
+ (@items.length > 1 ? ')' : ''))
69
+ end
70
+
71
+ # @param dst [ComplexType, ComplexType::UniqueType]
72
+ # @return [ComplexType]
73
+ def self_to_type dst
74
+ object_type_dst = dst.reduce_class_type
75
+ transform do |t|
76
+ next t if t.name != 'self'
77
+ object_type_dst
78
+ end
79
+ end
80
+
81
+ # @yieldparam [UniqueType]
82
+ # @yieldreturn [UniqueType]
83
+ # @return [Array<UniqueType>]
84
+ # @sg-ignore Declared return type
85
+ # ::Array<::Solargraph::ComplexType::UniqueType> does not match
86
+ # inferred type ::Array<::Proc> for Solargraph::ComplexType#map
87
+ def map(&block)
88
+ @items.map(&block)
89
+ end
90
+
91
+ # @yieldparam [UniqueType]
92
+ # @return [Enumerable<UniqueType>]
93
+ def each &block
94
+ @items.each &block
95
+ end
96
+
97
+ # @yieldparam [UniqueType]
98
+ # @return [void]
99
+ # @overload each_unique_type()
100
+ # @return [Enumerator<UniqueType>]
101
+ def each_unique_type &block
102
+ return enum_for(__method__) unless block_given?
103
+
104
+ @items.each do |item|
105
+ item.each_unique_type &block
106
+ end
107
+ end
108
+
109
+ # @param new_name [String, nil]
110
+ # @param make_rooted [Boolean, nil]
111
+ # @param new_key_types [Array<ComplexType>, nil]
112
+ # @param rooted [Boolean, nil]
113
+ # @param new_subtypes [Array<ComplexType>, nil]
114
+ # @return [self]
115
+ def recreate(new_name: nil, make_rooted: nil, new_key_types: nil, new_subtypes: nil)
116
+ ComplexType.new(map do |ut|
117
+ ut.recreate(new_name: new_name,
118
+ make_rooted: make_rooted,
119
+ new_key_types: new_key_types,
120
+ new_subtypes: new_subtypes)
121
+ end)
122
+ end
123
+
124
+ # @return [Integer]
125
+ def length
126
+ @items.length
127
+ end
128
+
129
+ # @return [Array<UniqueType>]
130
+ def to_a
131
+ @items
132
+ end
133
+
134
+ # @param index [Integer]
135
+ # @return [UniqueType]
136
+ def [](index)
137
+ @items[index]
138
+ end
139
+
140
+ # @return [Array<UniqueType>]
141
+ def select &block
142
+ @items.select &block
143
+ end
144
+
145
+ # @return [String]
146
+ def namespace
147
+ # cache this attr for high frequency call
148
+ @namespace ||= method_missing(:namespace).to_s
149
+ end
150
+
151
+ # @return [Array<String>]
152
+ def namespaces
153
+ @items.map(&:namespace)
154
+ end
155
+
156
+ # @param name [Symbol]
157
+ # @return [Object, nil]
158
+ def method_missing name, *args, &block
159
+ return if @items.first.nil?
160
+ return @items.first.send(name, *args, &block) if respond_to_missing?(name)
161
+ super
162
+ end
163
+
164
+ # @param name [Symbol]
165
+ # @param include_private [Boolean]
166
+ def respond_to_missing?(name, include_private = false)
167
+ TypeMethods.public_instance_methods.include?(name) || super
168
+ end
169
+
170
+ def to_s
171
+ map(&:tag).join(', ')
172
+ end
173
+
174
+ # @return [String]
175
+ def tags
176
+ map(&:tag).join(', ')
177
+ end
178
+
179
+ # @return [String]
180
+ def simple_tags
181
+ simplify_literals.tags
182
+ end
183
+
184
+ def literal?
185
+ @items.any?(&:literal?)
186
+ end
187
+
188
+ # @return [ComplexType]
189
+ def downcast_to_literal_if_possible
190
+ ComplexType.new(items.map(&:downcast_to_literal_if_possible))
191
+ end
192
+
193
+ # @return [String]
194
+ def desc
195
+ rooted_tags
196
+ end
197
+
198
+ # @param api_map [ApiMap]
199
+ # @param expected [ComplexType, ComplexType::UniqueType]
200
+ # @param situation [:method_call, :return_type, :assignment]
201
+ # @param allow_subtype_skew [Boolean] if false, check if any
202
+ # subtypes of the expected type match the inferred type
203
+ # @param allow_reverse_match [Boolean] if true, check if any subtypes
204
+ # of the expected type match the inferred type
205
+ # @param allow_empty_params [Boolean] if true, allow a general
206
+ # inferred type without parameters to conform to a more specific
207
+ # expected type
208
+ # @param allow_any_match [Boolean] if true, any unique type
209
+ # matched in the inferred qualifies as a match
210
+ # @param allow_undefined [Boolean] if true, treat undefined as a
211
+ # wildcard that matches anything
212
+ # @param rules [Array<:allow_subtype_skew, :allow_empty_params, :allow_reverse_match, :allow_any_match, :allow_undefined, :allow_unresolved_generic, :allow_unmatched_interface>]
213
+ # @param variance [:invariant, :covariant, :contravariant]
214
+ # @return [Boolean]
215
+ def conforms_to?(api_map, expected,
216
+ situation,
217
+ rules = [],
218
+ variance: erased_variance(situation))
219
+ expected = expected.downcast_to_literal_if_possible
220
+ inferred = downcast_to_literal_if_possible
221
+
222
+ return duck_types_match?(api_map, expected, inferred) if expected.duck_type?
223
+
224
+ if rules.include? :allow_any_match
225
+ inferred.any? do |inf|
226
+ inf.conforms_to?(api_map, expected, situation, rules,
227
+ variance: variance)
228
+ end
229
+ else
230
+ inferred.all? do |inf|
231
+ inf.conforms_to?(api_map, expected, situation, rules,
232
+ variance: variance)
233
+ end
234
+ end
235
+ end
236
+
237
+ # @param api_map [ApiMap]
238
+ # @param expected [ComplexType, UniqueType]
239
+ # @param inferred [ComplexType, UniqueType]
240
+ # @return [Boolean]
241
+ def duck_types_match? api_map, expected, inferred
242
+ raise ArgumentError, 'Expected type must be duck type' unless expected.duck_type?
243
+ expected.each do |exp|
244
+ next unless exp.duck_type?
245
+ quack = exp.to_s[1..]
246
+ # @sg-ignore Need to add nil check here
247
+ return false if api_map.get_method_stack(inferred.namespace, quack, scope: inferred.scope).empty?
248
+ end
249
+ true
250
+ end
251
+
252
+ # @return [String]
253
+ def rooted_tags
254
+ map(&:rooted_tag).join(', ')
255
+ end
256
+
257
+ # @yieldparam [UniqueType]
258
+ def all? &block
259
+ @items.all? &block
260
+ end
261
+
262
+ # @yieldparam [UniqueType]
263
+ # @yieldreturn [Boolean]
264
+ # @return [Boolean]
265
+ def any? &block
266
+ @items.compact.any? &block
267
+ end
268
+
269
+ def selfy?
270
+ @items.any?(&:selfy?)
271
+ end
272
+
273
+ def generic?
274
+ any?(&:generic?)
275
+ end
276
+
277
+ # @return [self]
278
+ def simplify_literals
279
+ ComplexType.new(map(&:simplify_literals))
280
+ end
281
+
282
+ # @param new_name [String, nil]
283
+ # @yieldparam t [UniqueType]
284
+ # @yieldreturn [UniqueType]
285
+ # @return [ComplexType]
286
+ def transform(new_name = nil, &transform_type)
287
+ raise "Please remove leading :: and set rooted with recreate() instead - #{new_name}" if new_name&.start_with?('::')
288
+ ComplexType.new(map { |ut| ut.transform(new_name, &transform_type) })
289
+ end
290
+
291
+ # @return [self]
292
+ def force_rooted
293
+ transform do |t|
294
+ t.recreate(make_rooted: true)
295
+ end
296
+ end
297
+
298
+ # @param definitions [Pin::Namespace, Pin::Method]
299
+ # @param context_type [ComplexType]
300
+ # @return [ComplexType]
301
+ def resolve_generics definitions, context_type
302
+ result = @items.map { |i| i.resolve_generics(definitions, context_type) }
303
+ ComplexType.new(result)
304
+ end
305
+
306
+ def nullable?
307
+ @items.any?(&:nil_type?)
308
+ end
309
+
310
+ # @return [ComplexType]
311
+ def without_nil
312
+ new_items = @items.reject(&:nil_type?)
313
+ return ComplexType::UNDEFINED if new_items.empty?
314
+ ComplexType.new(new_items)
315
+ end
316
+
317
+ # @return [Array<ComplexType>]
318
+ def all_params
319
+ @items.first.all_params || []
320
+ end
321
+
322
+ # @return [ComplexType]
323
+ def reduce_class_type
324
+ new_items = items.flat_map do |type|
325
+ next type unless ['Module', 'Class'].include?(type.name)
326
+ next type if type.all_params.empty?
327
+
328
+ type.all_params
329
+ end
330
+ ComplexType.new(new_items)
331
+ end
332
+
333
+ # every type and subtype in this union have been resolved to be
334
+ # fully qualified
335
+ def all_rooted?
336
+ all?(&:all_rooted?)
337
+ end
338
+
339
+ # @param other [ComplexType, UniqueType]
340
+ def erased_version_of?(other)
341
+ return false if items.length != 1 || other.items.length != 1
342
+
343
+ @items.first.erased_version_of?(other.items.first)
344
+ end
345
+
346
+ # every top-level type has resolved to be fully qualified; see
347
+ # #all_rooted? to check their subtypes as well
348
+ def rooted?
349
+ all?(&:rooted?)
350
+ end
351
+
352
+ attr_reader :items
353
+
354
+ def rooted?
355
+ @items.all?(&:rooted?)
356
+ end
357
+
358
+ # @param exclude_types [ComplexType, nil]
359
+ # @param api_map [ApiMap]
360
+ # @return [ComplexType, self]
361
+ def exclude exclude_types, api_map
362
+ return self if exclude_types.nil?
363
+
364
+ types = items - exclude_types.items
365
+ types = [ComplexType::UniqueType::UNDEFINED] if types.empty?
366
+ ComplexType.new(types)
367
+ end
368
+
369
+ # @see https://en.wikipedia.org/wiki/Intersection_type
370
+ #
371
+ # @param intersection_type [ComplexType, ComplexType::UniqueType, nil]
372
+ # @param api_map [ApiMap]
373
+ # @return [self, ComplexType::UniqueType]
374
+ def intersect_with intersection_type, api_map
375
+ return self if intersection_type.nil?
376
+ return intersection_type if undefined?
377
+ types = []
378
+ # try to find common types via conformance
379
+ items.each do |ut|
380
+ intersection_type.each do |int_type|
381
+ if int_type.conforms_to?(api_map, ut, :assignment)
382
+ types << int_type
383
+ elsif ut.conforms_to?(api_map, int_type, :assignment)
384
+ types << ut
385
+ end
386
+ end
387
+ end
388
+ types = [ComplexType::UniqueType::UNDEFINED] if types.empty?
389
+ ComplexType.new(types)
390
+ end
391
+
392
+ protected
393
+
394
+ # @return [ComplexType]
395
+ def reduce_object
396
+ new_items = items.flat_map do |ut|
397
+ next [ut] if ut.name != 'Object' || ut.subtypes.empty?
398
+ ut.subtypes
399
+ end
400
+ ComplexType.new(new_items)
401
+ end
402
+
403
+ def bottom?
404
+ @items.all?(&:bot?)
405
+ end
406
+
407
+ class << self
408
+ # Parse type strings into a ComplexType.
409
+ #
410
+ # @example
411
+ # ComplexType.parse 'String', 'Foo', 'nil' #=> [String, Foo, nil]
412
+ #
413
+ # @note
414
+ # The `partial` parameter is used to indicate that the method is
415
+ # receiving a string that will be used inside another ComplexType.
416
+ # It returns arrays of ComplexTypes instead of a single cohesive one.
417
+ # Consumers should not need to use this parameter; it should only be
418
+ # used internally.
419
+ #
420
+ # @param strings [Array<String>] The type definitions to parse
421
+ # @return [ComplexType]
422
+ # # @overload parse(*strings, partial: false)
423
+ # # @todo Need ability to use a literal true as a type below
424
+ # # @param partial [Boolean] True if the string is part of a another type
425
+ # # @return [Array<UniqueType>]
426
+ # @sg-ignore To be able to select the right signature above,
427
+ # Chain::Call needs to know the decl type (:arg, :optarg,
428
+ # :kwarg, etc) of the arguments given, instead of just having
429
+ # an array of Chains as the arguments.
430
+ def parse *strings, partial: false
431
+ # @type [Hash{Array<String> => ComplexType, Array<ComplexType::UniqueType>}]
432
+ @cache ||= {}
433
+ unless partial
434
+ cached = @cache[strings]
435
+ return cached unless cached.nil?
436
+ end
437
+ # @types [Array<ComplexType::UniqueType>]
438
+ types = []
439
+ key_types = nil
440
+ strings.each do |type_string|
441
+ point_stack = 0
442
+ curly_stack = 0
443
+ paren_stack = 0
444
+ base = String.new
445
+ subtype_string = String.new
446
+ # @param char [String]
447
+ type_string&.each_char do |char|
448
+ if char == '='
449
+ #raise ComplexTypeError, "Invalid = in type #{type_string}" unless curly_stack > 0
450
+ elsif char == '<'
451
+ point_stack += 1
452
+ elsif char == '>'
453
+ if subtype_string.end_with?('=') && curly_stack > 0
454
+ subtype_string += char
455
+ elsif base.end_with?('=')
456
+ raise ComplexTypeError, "Invalid hash thing" unless key_types.nil?
457
+ # types.push ComplexType.new([UniqueType.new(base[0..-2].strip)])
458
+ # @sg-ignore Need to add nil check here
459
+ types.push UniqueType.parse(base[0..-2].strip, subtype_string)
460
+ # @todo this should either expand key_type's type
461
+ # automatically or complain about not being
462
+ # compatible with key_type's type in type checking
463
+ key_types = types
464
+ types = []
465
+ base.clear
466
+ subtype_string.clear
467
+ next
468
+ else
469
+ raise ComplexTypeError, "Invalid close in type #{type_string}" if point_stack == 0
470
+ point_stack -= 1
471
+ subtype_string += char
472
+ end
473
+ next
474
+ elsif char == '{'
475
+ curly_stack += 1
476
+ elsif char == '}'
477
+ curly_stack -= 1
478
+ subtype_string += char
479
+ raise ComplexTypeError, "Invalid close in type #{type_string}" if curly_stack < 0
480
+ next
481
+ elsif char == '('
482
+ paren_stack += 1
483
+ elsif char == ')'
484
+ paren_stack -= 1
485
+ subtype_string += char
486
+ raise ComplexTypeError, "Invalid close in type #{type_string}" if paren_stack < 0
487
+ next
488
+ elsif char == ',' && point_stack == 0 && curly_stack == 0 && paren_stack == 0
489
+ # types.push ComplexType.new([UniqueType.new(base.strip, subtype_string.strip)])
490
+ types.push UniqueType.parse(base.strip, subtype_string.strip)
491
+ base.clear
492
+ subtype_string.clear
493
+ next
494
+ end
495
+ if point_stack == 0 && curly_stack == 0 && paren_stack == 0
496
+ base.concat char
497
+ else
498
+ subtype_string.concat char
499
+ end
500
+ end
501
+ raise ComplexTypeError, "Unclosed subtype in #{type_string}" if point_stack != 0 || curly_stack != 0 || paren_stack != 0
502
+ # types.push ComplexType.new([UniqueType.new(base, subtype_string)])
503
+ types.push UniqueType.parse(base.strip, subtype_string.strip)
504
+ end
505
+ unless key_types.nil?
506
+ raise ComplexTypeError, "Invalid use of key/value parameters" unless partial
507
+ return key_types if types.empty?
508
+ return [key_types, types]
509
+ end
510
+ result = partial ? types : ComplexType.new(types)
511
+ @cache[strings] = result unless partial
512
+ result
513
+ end
514
+
515
+ # @param strings [Array<String>]
516
+ # @return [ComplexType]
517
+ def try_parse *strings
518
+ parse *strings
519
+ rescue ComplexTypeError => e
520
+ Solargraph.logger.info "Error parsing complex type `#{strings.join(', ')}`: #{e.message}"
521
+ ComplexType::UNDEFINED
522
+ end
523
+ end
524
+
525
+ VOID = ComplexType.parse('void')
526
+ UNDEFINED = ComplexType.parse('undefined')
527
+ SYMBOL = ComplexType.parse('::Symbol')
528
+ ROOT = ComplexType.parse('::Class<>')
529
+ NIL = ComplexType.parse('nil')
530
+ SELF = ComplexType.parse('self')
531
+ BOOLEAN = ComplexType.parse('::Boolean')
532
+ BOT = ComplexType.parse('bot')
533
+
534
+ private
535
+
536
+ # @todo This is a quick and dirty hack that forces `self` keywords
537
+ # to reference an instance of their class and never the class itself.
538
+ # This behavior may change depending on which result is expected
539
+ # from YARD conventions. See https://github.com/lsegal/yard/issues/1257
540
+ # @param dst [String]
541
+ # @return [String]
542
+ def reduce_class dst
543
+ while dst =~ /^(Class|Module)\<(.*?)\>$/
544
+ dst = dst.sub(/^(Class|Module)\</, '').sub(/\>$/, '')
545
+ end
546
+ dst
547
+ end
548
+ end
549
+ end