solargraph 0.54.4 → 0.58.3

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 (238) hide show
  1. checksums.yaml +4 -4
  2. data/.gitattributes +2 -0
  3. data/.github/workflows/linting.yml +127 -0
  4. data/.github/workflows/plugins.yml +185 -6
  5. data/.github/workflows/rspec.yml +55 -5
  6. data/.github/workflows/typecheck.yml +8 -3
  7. data/.gitignore +8 -0
  8. data/.overcommit.yml +72 -0
  9. data/.rspec +1 -0
  10. data/.rubocop.yml +66 -0
  11. data/.rubocop_todo.yml +1279 -0
  12. data/.yardopts +1 -0
  13. data/CHANGELOG.md +143 -0
  14. data/README.md +20 -6
  15. data/Rakefile +125 -13
  16. data/bin/solargraph +3 -0
  17. data/lib/solargraph/api_map/cache.rb +110 -109
  18. data/lib/solargraph/api_map/constants.rb +279 -0
  19. data/lib/solargraph/api_map/index.rb +193 -167
  20. data/lib/solargraph/api_map/source_to_yard.rb +97 -88
  21. data/lib/solargraph/api_map/store.rb +384 -241
  22. data/lib/solargraph/api_map.rb +945 -875
  23. data/lib/solargraph/bench.rb +45 -28
  24. data/lib/solargraph/complex_type/type_methods.rb +228 -217
  25. data/lib/solargraph/complex_type/unique_type.rb +482 -386
  26. data/lib/solargraph/complex_type.rb +444 -394
  27. data/lib/solargraph/convention/active_support_concern.rb +111 -0
  28. data/lib/solargraph/convention/base.rb +20 -3
  29. data/lib/solargraph/convention/data_definition/data_assignment_node.rb +61 -0
  30. data/lib/solargraph/convention/data_definition/data_definition_node.rb +91 -0
  31. data/lib/solargraph/convention/data_definition.rb +105 -0
  32. data/lib/solargraph/convention/gemfile.rb +15 -15
  33. data/lib/solargraph/convention/gemspec.rb +23 -22
  34. data/lib/solargraph/convention/rakefile.rb +17 -17
  35. data/lib/solargraph/convention/struct_definition/struct_assignment_node.rb +61 -0
  36. data/lib/solargraph/convention/struct_definition/struct_definition_node.rb +102 -0
  37. data/lib/solargraph/convention/struct_definition.rb +164 -0
  38. data/lib/solargraph/convention.rb +78 -47
  39. data/lib/solargraph/converters/dd.rb +17 -17
  40. data/lib/solargraph/converters/dl.rb +15 -15
  41. data/lib/solargraph/converters/dt.rb +15 -15
  42. data/lib/solargraph/converters/misc.rb +1 -1
  43. data/lib/solargraph/diagnostics/require_not_found.rb +53 -53
  44. data/lib/solargraph/diagnostics/rubocop.rb +118 -113
  45. data/lib/solargraph/diagnostics/rubocop_helpers.rb +68 -66
  46. data/lib/solargraph/diagnostics/type_check.rb +55 -55
  47. data/lib/solargraph/diagnostics/update_errors.rb +41 -41
  48. data/lib/solargraph/doc_map.rb +439 -188
  49. data/lib/solargraph/environ.rb +9 -2
  50. data/lib/solargraph/equality.rb +34 -33
  51. data/lib/solargraph/gem_pins.rb +98 -72
  52. data/lib/solargraph/language_server/error_codes.rb +20 -20
  53. data/lib/solargraph/language_server/host/diagnoser.rb +89 -89
  54. data/lib/solargraph/language_server/host/dispatch.rb +130 -128
  55. data/lib/solargraph/language_server/host/message_worker.rb +112 -106
  56. data/lib/solargraph/language_server/host/sources.rb +99 -99
  57. data/lib/solargraph/language_server/host.rb +878 -861
  58. data/lib/solargraph/language_server/message/base.rb +97 -96
  59. data/lib/solargraph/language_server/message/client/register_capability.rb +15 -15
  60. data/lib/solargraph/language_server/message/completion_item/resolve.rb +60 -60
  61. data/lib/solargraph/language_server/message/extended/check_gem_version.rb +114 -112
  62. data/lib/solargraph/language_server/message/extended/document.rb +23 -20
  63. data/lib/solargraph/language_server/message/extended/document_gems.rb +32 -32
  64. data/lib/solargraph/language_server/message/extended/download_core.rb +19 -19
  65. data/lib/solargraph/language_server/message/extended/search.rb +20 -20
  66. data/lib/solargraph/language_server/message/initialize.rb +191 -191
  67. data/lib/solargraph/language_server/message/text_document/completion.rb +56 -56
  68. data/lib/solargraph/language_server/message/text_document/definition.rb +40 -38
  69. data/lib/solargraph/language_server/message/text_document/document_highlight.rb +16 -16
  70. data/lib/solargraph/language_server/message/text_document/document_symbol.rb +26 -26
  71. data/lib/solargraph/language_server/message/text_document/formatting.rb +148 -131
  72. data/lib/solargraph/language_server/message/text_document/hover.rb +58 -58
  73. data/lib/solargraph/language_server/message/text_document/prepare_rename.rb +11 -11
  74. data/lib/solargraph/language_server/message/text_document/references.rb +16 -16
  75. data/lib/solargraph/language_server/message/text_document/rename.rb +19 -19
  76. data/lib/solargraph/language_server/message/text_document/signature_help.rb +24 -24
  77. data/lib/solargraph/language_server/message/text_document/type_definition.rb +25 -24
  78. data/lib/solargraph/language_server/message/workspace/did_change_configuration.rb +35 -35
  79. data/lib/solargraph/language_server/message/workspace/did_change_watched_files.rb +40 -40
  80. data/lib/solargraph/language_server/message/workspace/did_change_workspace_folders.rb +26 -24
  81. data/lib/solargraph/language_server/message/workspace/workspace_symbol.rb +23 -23
  82. data/lib/solargraph/language_server/message.rb +94 -94
  83. data/lib/solargraph/language_server/progress.rb +8 -0
  84. data/lib/solargraph/language_server/request.rb +27 -24
  85. data/lib/solargraph/language_server/transport/data_reader.rb +74 -74
  86. data/lib/solargraph/language_server/uri_helpers.rb +49 -49
  87. data/lib/solargraph/library.rb +683 -662
  88. data/lib/solargraph/location.rb +82 -58
  89. data/lib/solargraph/logging.rb +37 -27
  90. data/lib/solargraph/page.rb +92 -89
  91. data/lib/solargraph/parser/comment_ripper.rb +69 -56
  92. data/lib/solargraph/parser/flow_sensitive_typing.rb +255 -0
  93. data/lib/solargraph/parser/node_processor/base.rb +92 -87
  94. data/lib/solargraph/parser/node_processor.rb +62 -45
  95. data/lib/solargraph/parser/parser_gem/class_methods.rb +149 -157
  96. data/lib/solargraph/parser/parser_gem/flawed_builder.rb +19 -18
  97. data/lib/solargraph/parser/parser_gem/node_chainer.rb +166 -164
  98. data/lib/solargraph/parser/parser_gem/node_methods.rb +486 -495
  99. data/lib/solargraph/parser/parser_gem/node_processors/alias_node.rb +2 -1
  100. data/lib/solargraph/parser/parser_gem/node_processors/and_node.rb +22 -0
  101. data/lib/solargraph/parser/parser_gem/node_processors/args_node.rb +59 -57
  102. data/lib/solargraph/parser/parser_gem/node_processors/begin_node.rb +15 -15
  103. data/lib/solargraph/parser/parser_gem/node_processors/block_node.rb +46 -43
  104. data/lib/solargraph/parser/parser_gem/node_processors/casgn_node.rb +2 -1
  105. data/lib/solargraph/parser/parser_gem/node_processors/cvasgn_node.rb +2 -1
  106. data/lib/solargraph/parser/parser_gem/node_processors/def_node.rb +53 -50
  107. data/lib/solargraph/parser/parser_gem/node_processors/defs_node.rb +37 -36
  108. data/lib/solargraph/parser/parser_gem/node_processors/gvasgn_node.rb +2 -1
  109. data/lib/solargraph/parser/parser_gem/node_processors/if_node.rb +23 -0
  110. data/lib/solargraph/parser/parser_gem/node_processors/ivasgn_node.rb +40 -38
  111. data/lib/solargraph/parser/parser_gem/node_processors/lvasgn_node.rb +29 -28
  112. data/lib/solargraph/parser/parser_gem/node_processors/masgn_node.rb +59 -53
  113. data/lib/solargraph/parser/parser_gem/node_processors/namespace_node.rb +8 -7
  114. data/lib/solargraph/parser/parser_gem/node_processors/opasgn_node.rb +98 -0
  115. data/lib/solargraph/parser/parser_gem/node_processors/orasgn_node.rb +17 -16
  116. data/lib/solargraph/parser/parser_gem/node_processors/resbody_node.rb +38 -36
  117. data/lib/solargraph/parser/parser_gem/node_processors/sclass_node.rb +52 -42
  118. data/lib/solargraph/parser/parser_gem/node_processors/send_node.rb +291 -259
  119. data/lib/solargraph/parser/parser_gem/node_processors/sym_node.rb +3 -1
  120. data/lib/solargraph/parser/parser_gem/node_processors/until_node.rb +29 -0
  121. data/lib/solargraph/parser/parser_gem/node_processors/while_node.rb +29 -0
  122. data/lib/solargraph/parser/parser_gem/node_processors.rb +70 -56
  123. data/lib/solargraph/parser/parser_gem.rb +12 -12
  124. data/lib/solargraph/parser/region.rb +69 -66
  125. data/lib/solargraph/parser/snippet.rb +17 -15
  126. data/lib/solargraph/parser.rb +23 -22
  127. data/lib/solargraph/pin/base.rb +729 -378
  128. data/lib/solargraph/pin/base_variable.rb +126 -118
  129. data/lib/solargraph/pin/block.rb +104 -101
  130. data/lib/solargraph/pin/breakable.rb +9 -0
  131. data/lib/solargraph/pin/callable.rb +231 -147
  132. data/lib/solargraph/pin/closure.rb +72 -57
  133. data/lib/solargraph/pin/common.rb +79 -70
  134. data/lib/solargraph/pin/constant.rb +45 -43
  135. data/lib/solargraph/pin/conversions.rb +123 -123
  136. data/lib/solargraph/pin/delegated_method.rb +120 -101
  137. data/lib/solargraph/pin/documenting.rb +114 -98
  138. data/lib/solargraph/pin/instance_variable.rb +34 -34
  139. data/lib/solargraph/pin/keyword.rb +20 -15
  140. data/lib/solargraph/pin/local_variable.rb +75 -67
  141. data/lib/solargraph/pin/method.rb +672 -527
  142. data/lib/solargraph/pin/method_alias.rb +34 -31
  143. data/lib/solargraph/pin/namespace.rb +115 -107
  144. data/lib/solargraph/pin/parameter.rb +275 -212
  145. data/lib/solargraph/pin/proxy_type.rb +39 -29
  146. data/lib/solargraph/pin/reference/override.rb +47 -29
  147. data/lib/solargraph/pin/reference/require.rb +2 -2
  148. data/lib/solargraph/pin/reference/superclass.rb +15 -10
  149. data/lib/solargraph/pin/reference.rb +39 -22
  150. data/lib/solargraph/pin/search.rb +61 -56
  151. data/lib/solargraph/pin/signature.rb +61 -17
  152. data/lib/solargraph/pin/singleton.rb +1 -1
  153. data/lib/solargraph/pin/symbol.rb +53 -47
  154. data/lib/solargraph/pin/until.rb +18 -0
  155. data/lib/solargraph/pin/while.rb +18 -0
  156. data/lib/solargraph/pin.rb +44 -41
  157. data/lib/solargraph/pin_cache.rb +245 -0
  158. data/lib/solargraph/position.rb +132 -107
  159. data/lib/solargraph/range.rb +112 -98
  160. data/lib/solargraph/rbs_map/conversions.rb +823 -646
  161. data/lib/solargraph/rbs_map/core_fills.rb +84 -50
  162. data/lib/solargraph/rbs_map/core_map.rb +58 -28
  163. data/lib/solargraph/rbs_map/stdlib_map.rb +43 -33
  164. data/lib/solargraph/rbs_map.rb +163 -93
  165. data/lib/solargraph/server_methods.rb +16 -16
  166. data/lib/solargraph/shell.rb +363 -269
  167. data/lib/solargraph/source/chain/array.rb +37 -33
  168. data/lib/solargraph/source/chain/block_symbol.rb +1 -1
  169. data/lib/solargraph/source/chain/block_variable.rb +1 -1
  170. data/lib/solargraph/source/chain/call.rb +337 -303
  171. data/lib/solargraph/source/chain/class_variable.rb +13 -13
  172. data/lib/solargraph/source/chain/constant.rb +26 -89
  173. data/lib/solargraph/source/chain/global_variable.rb +13 -13
  174. data/lib/solargraph/source/chain/hash.rb +34 -33
  175. data/lib/solargraph/source/chain/head.rb +1 -1
  176. data/lib/solargraph/source/chain/if.rb +28 -28
  177. data/lib/solargraph/source/chain/instance_variable.rb +13 -13
  178. data/lib/solargraph/source/chain/link.rb +109 -98
  179. data/lib/solargraph/source/chain/literal.rb +48 -28
  180. data/lib/solargraph/source/chain/or.rb +23 -23
  181. data/lib/solargraph/source/chain/q_call.rb +11 -11
  182. data/lib/solargraph/source/chain/variable.rb +13 -13
  183. data/lib/solargraph/source/chain/z_super.rb +30 -30
  184. data/lib/solargraph/source/chain.rb +291 -252
  185. data/lib/solargraph/source/change.rb +82 -82
  186. data/lib/solargraph/source/cursor.rb +166 -167
  187. data/lib/solargraph/source/encoding_fixes.rb +23 -23
  188. data/lib/solargraph/source/source_chainer.rb +194 -194
  189. data/lib/solargraph/source/updater.rb +55 -55
  190. data/lib/solargraph/source.rb +498 -495
  191. data/lib/solargraph/source_map/clip.rb +226 -232
  192. data/lib/solargraph/source_map/data.rb +34 -30
  193. data/lib/solargraph/source_map/mapper.rb +259 -255
  194. data/lib/solargraph/source_map.rb +212 -217
  195. data/lib/solargraph/type_checker/checks.rb +124 -120
  196. data/lib/solargraph/type_checker/param_def.rb +37 -35
  197. data/lib/solargraph/type_checker/problem.rb +32 -32
  198. data/lib/solargraph/type_checker/rules.rb +84 -62
  199. data/lib/solargraph/type_checker.rb +814 -672
  200. data/lib/solargraph/version.rb +5 -5
  201. data/lib/solargraph/views/_method.erb +10 -10
  202. data/lib/solargraph/views/_namespace.erb +3 -3
  203. data/lib/solargraph/views/document.erb +10 -10
  204. data/lib/solargraph/workspace/config.rb +255 -239
  205. data/lib/solargraph/workspace/require_paths.rb +97 -0
  206. data/lib/solargraph/workspace.rb +220 -239
  207. data/lib/solargraph/yard_map/helpers.rb +44 -16
  208. data/lib/solargraph/yard_map/mapper/to_constant.rb +7 -5
  209. data/lib/solargraph/yard_map/mapper/to_method.rb +130 -94
  210. data/lib/solargraph/yard_map/mapper/to_namespace.rb +31 -28
  211. data/lib/solargraph/yard_map/mapper.rb +79 -78
  212. data/lib/solargraph/yard_map/to_method.rb +89 -86
  213. data/lib/solargraph/yard_tags.rb +20 -20
  214. data/lib/solargraph/yardoc.rb +87 -52
  215. data/lib/solargraph.rb +105 -72
  216. data/rbs/fills/bundler/0/bundler.rbs +4271 -0
  217. data/rbs/fills/open3/0/open3.rbs +172 -0
  218. data/rbs/fills/rubygems/0/basic_specification.rbs +326 -0
  219. data/rbs/fills/rubygems/0/errors.rbs +364 -0
  220. data/rbs/fills/rubygems/0/spec_fetcher.rbs +107 -0
  221. data/rbs/fills/rubygems/0/specification.rbs +1753 -0
  222. data/rbs/fills/tuple/tuple.rbs +149 -0
  223. data/rbs/shims/ast/0/node.rbs +5 -0
  224. data/rbs/shims/ast/2.4/.rbs_meta.yaml +9 -0
  225. data/rbs/shims/ast/2.4/ast.rbs +73 -0
  226. data/rbs/shims/parser/3.2.0.1/builders/default.rbs +195 -0
  227. data/rbs/shims/parser/3.2.0.1/manifest.yaml +7 -0
  228. data/rbs/shims/parser/3.2.0.1/parser.rbs +201 -0
  229. data/rbs/shims/parser/3.2.0.1/polyfill.rbs +4 -0
  230. data/rbs/shims/thor/1.2.0.1/.rbs_meta.yaml +9 -0
  231. data/rbs/shims/thor/1.2.0.1/manifest.yaml +7 -0
  232. data/rbs/shims/thor/1.2.0.1/thor.rbs +17 -0
  233. data/rbs_collection.yaml +19 -0
  234. data/solargraph.gemspec +27 -5
  235. metadata +215 -18
  236. data/lib/.rubocop.yml +0 -22
  237. data/lib/solargraph/cache.rb +0 -77
  238. data/lib/solargraph/parser/node_methods.rb +0 -83
@@ -1,495 +1,486 @@
1
- # frozen_string_literal: true
2
-
3
- require 'parser'
4
- require 'ast'
5
-
6
- # Teach AST::Node#children about its generic type
7
- #
8
- # @todo contribute back to https://github.com/ruby/gem_rbs_collection/blob/main/gems/ast/2.4/ast.rbs
9
- #
10
- # @!parse
11
- # module ::AST
12
- # class Node
13
- # # New children
14
- #
15
- # # @return [Array<self>]
16
- # attr_reader :children
17
- # end
18
- # end
19
-
20
- # https://github.com/whitequark/parser
21
- module Solargraph
22
- module Parser
23
- module ParserGem
24
- module NodeMethods
25
- module_function
26
-
27
- # @param node [Parser::AST::Node]
28
- # @return [String]
29
- def unpack_name(node)
30
- pack_name(node).join("::")
31
- end
32
-
33
- # @param node [Parser::AST::Node]
34
- # @return [Array<String>]
35
- def pack_name(node)
36
- # @type [Array<String>]
37
- parts = []
38
- if node.is_a?(AST::Node)
39
- node.children.each { |n|
40
- if n.is_a?(AST::Node)
41
- if n.type == :cbase
42
- parts = [''] + pack_name(n)
43
- else
44
- parts += pack_name(n)
45
- end
46
- else
47
- parts.push n unless n.nil?
48
- end
49
- }
50
- end
51
- parts
52
- end
53
-
54
- # @param node [Parser::AST::Node]
55
- # @return [String, nil]
56
- def infer_literal_node_type node
57
- return nil unless node.is_a?(AST::Node)
58
- if node.type == :str || node.type == :dstr
59
- return '::String'
60
- elsif node.type == :array
61
- return '::Array'
62
- elsif node.type == :hash
63
- return '::Hash'
64
- elsif node.type == :int
65
- return '::Integer'
66
- elsif node.type == :float
67
- return '::Float'
68
- elsif node.type == :sym || node.type == :dsym
69
- return '::Symbol'
70
- elsif node.type == :regexp
71
- return '::Regexp'
72
- elsif node.type == :irange
73
- return '::Range'
74
- elsif node.type == :true || node.type == :false
75
- return '::Boolean'
76
- # @todo Support `nil` keyword in types
77
- # elsif node.type == :nil
78
- # return 'NilClass'
79
- end
80
- nil
81
- end
82
-
83
- # @param node [Parser::AST::Node]
84
- # @return [Position]
85
- def get_node_start_position(node)
86
- Position.new(node.loc.line, node.loc.column)
87
- end
88
-
89
- # @param node [Parser::AST::Node]
90
- # @return [Position]
91
- def get_node_end_position(node)
92
- Position.new(node.loc.last_line, node.loc.last_column)
93
- end
94
-
95
- # @param node [Parser::AST::Node]
96
- # @param signature [String]
97
- #
98
- # @return [String]
99
- def drill_signature node, signature
100
- return signature unless node.is_a?(AST::Node)
101
- if node.type == :const or node.type == :cbase
102
- unless node.children[0].nil?
103
- signature += drill_signature(node.children[0], signature)
104
- end
105
- signature += '::' unless signature.empty?
106
- signature += node.children[1].to_s
107
- elsif node.type == :lvar or node.type == :ivar or node.type == :cvar
108
- signature += '.' unless signature.empty?
109
- signature += node.children[0].to_s
110
- elsif node.type == :send
111
- unless node.children[0].nil?
112
- signature += drill_signature(node.children[0], signature)
113
- end
114
- signature += '.' unless signature.empty?
115
- signature += node.children[1].to_s
116
- end
117
- signature
118
- end
119
-
120
- # @param node [Parser::AST::Node]
121
- # @return [Hash{Parser::AST::Node => Chain}]
122
- def convert_hash node
123
- return {} unless Parser.is_ast_node?(node)
124
- return convert_hash(node.children[0]) if node.type == :kwsplat
125
- return convert_hash(node.children[0]) if Parser.is_ast_node?(node.children[0]) && node.children[0].type == :kwsplat
126
- return {} unless node.type == :hash
127
- result = {}
128
- node.children.each do |pair|
129
- result[pair.children[0].children[0]] = Solargraph::Parser.chain(pair.children[1])
130
- end
131
- result
132
- end
133
-
134
- NIL_NODE = ::Parser::AST::Node.new(:nil)
135
-
136
- # @param node [Parser::AST::Node]
137
- #
138
- # @return [Array<Parser::AST::Node>]
139
- def const_nodes_from node
140
- return [] unless Parser.is_ast_node?(node)
141
- result = []
142
- if node.type == :const
143
- result.push node
144
- else
145
- node.children.each { |child| result.concat const_nodes_from(child) }
146
- end
147
- result
148
- end
149
-
150
- # @param node [Parser::AST::Node]
151
- def splatted_hash? node
152
- Parser.is_ast_node?(node.children[0]) && node.children[0].type == :kwsplat
153
- end
154
-
155
- # @param node [Parser::AST::Node]
156
- def splatted_call? node
157
- return false unless Parser.is_ast_node?(node)
158
- Parser.is_ast_node?(node.children[0]) && node.children[0].type == :kwsplat && node.children[0].children[0].type != :hash
159
- end
160
-
161
- # @param nodes [Enumerable<Parser::AST::Node>]
162
- def any_splatted_call?(nodes)
163
- nodes.any? { |n| splatted_call?(n) }
164
- end
165
-
166
- # @todo Temporarily here for testing. Move to Solargraph::Parser.
167
- # @param node [Parser::AST::Node]
168
- # @return [Array<Parser::AST::Node>]
169
- def call_nodes_from node
170
- return [] unless node.is_a?(::Parser::AST::Node)
171
- result = []
172
- if node.type == :block
173
- result.push node
174
- if Parser.is_ast_node?(node.children[0]) && node.children[0].children.length > 2
175
- node.children[0].children[2..-1].each { |child| result.concat call_nodes_from(child) }
176
- end
177
- node.children[1..-1].each { |child| result.concat call_nodes_from(child) }
178
- elsif node.type == :send
179
- result.push node
180
- node.children[2..-1].each { |child| result.concat call_nodes_from(child) }
181
- elsif [:super, :zsuper].include?(node.type)
182
- result.push node
183
- node.children.each { |child| result.concat call_nodes_from(child) }
184
- else
185
- node.children.each { |child| result.concat call_nodes_from(child) }
186
- end
187
- result
188
- end
189
-
190
- # Find all the nodes within the provided node that potentially return a
191
- # value.
192
- #
193
- # The node parameter typically represents a method's logic, e.g., the
194
- # second child (after the :args node) of a :def node. A simple one-line
195
- # method would typically return itself, while a node with conditions
196
- # would return the resulting node from each conditional branch. Nodes
197
- # that follow a :return node are assumed to be unreachable. Nil values
198
- # are converted to nil node types.
199
- #
200
- # @param node [Parser::AST::Node]
201
- # @return [Array<Parser::AST::Node>]
202
- def returns_from_method_body node
203
- # @todo is the || NIL_NODE necessary?
204
- # STDERR.puts("Evaluating expression: #{node.inspect}")
205
- DeepInference.from_method_body(node).map { |n| n || NIL_NODE }
206
- end
207
-
208
- # @param node [Parser::AST::Node]
209
- # @return [Array<AST::Node>] low-level value nodes in
210
- # value position. Does not include explicit return
211
- # statements
212
- def value_position_nodes_only(node)
213
- DeepInference.value_position_nodes_only(node).map { |n| n || NIL_NODE }
214
- end
215
-
216
- # @param cursor [Solargraph::Source::Cursor]
217
- # @return [Parser::AST::Node, nil]
218
- def find_recipient_node cursor
219
- return repaired_find_recipient_node(cursor) if cursor.source.repaired? && cursor.source.code[cursor.offset - 1] == '('
220
- source = cursor.source
221
- position = cursor.position
222
- offset = cursor.offset
223
- tree = if source.synchronized?
224
- match = source.code[0..offset-1].match(/,\s*\z/)
225
- if match
226
- source.tree_at(position.line, position.column - match[0].length)
227
- else
228
- source.tree_at(position.line, position.column)
229
- end
230
- else
231
- source.tree_at(position.line, position.column - 1)
232
- end
233
- prev = nil
234
- tree.each do |node|
235
- if node.type == :send
236
- args = node.children[2..-1]
237
- if !args.empty?
238
- return node if prev && args.include?(prev)
239
- else
240
- if source.synchronized?
241
- return node if source.code[0..offset-1] =~ /\(\s*\z/ && source.code[offset..-1] =~ /^\s*\)/
242
- else
243
- return node if source.code[0..offset-1] =~ /\([^\(]*\z/
244
- end
245
- end
246
- end
247
- prev = node
248
- end
249
- nil
250
- end
251
-
252
- # @param cursor [Solargraph::Source::Cursor]
253
- # @return [Parser::AST::Node, nil]
254
- def repaired_find_recipient_node cursor
255
- cursor = cursor.source.cursor_at([cursor.position.line, cursor.position.column - 1])
256
- node = cursor.source.tree_at(cursor.position.line, cursor.position.column).first
257
- return node if node && node.type == :send
258
- end
259
-
260
- #
261
- # Concepts:
262
- #
263
- # - statement - one single node in the AST. Generally used
264
- # synonymously with how the Parser gem uses the term
265
- # 'expression'. This may have side effects (e.g.,
266
- # registering a method in the namespace, modifying
267
- # variables or doing I/O). It may encapsulate multiple
268
- # other statements (see compound statement).
269
- #
270
- # - value - something that can be assigned to a variable by
271
- # evaluating a statement
272
- #
273
- # - value node - the 'lowest level' AST node whose return
274
- # type is a subset of the value type of the overall
275
- # statement. Might be a literal, a method call, etc - the
276
- # goal is to find the lowest level node, which we can use
277
- # Chains and Pins later on to determine the type of.
278
- #
279
- # e.g., if the node 'b ? 123 : 456' were a return value, we
280
- # know the actual return values possible are 123 and 456,
281
- # and can disregard the rest.
282
- #
283
- # - value type - the type representing the multiple possible
284
- # values that can result from evaluation of the statement.
285
- #
286
- # - return type - the type describing the values a statement
287
- # might evaluate to. When used with a method, the term
288
- # describes the values that may result from the method
289
- # being called, and includes explicit return statements
290
- # within the method body's closure.
291
- #
292
- # - method body - a compound statement with parameters whose
293
- # return value type must account both for the explicit
294
- # 'return' statemnts as well as the final statements
295
- # executed in any given control flow through the method.
296
- #
297
- # - explicit return statement - a statement which, when part of a
298
- # method body, is a possible value of a call to that method -
299
- # e.g., "return 123"
300
- #
301
- # - compound statement - a statement which can be expanded to
302
- # be multiple statements in a row, executed in the context
303
- # of a method which can be explicitly returned from.
304
- #
305
- # - value position - the positions in the AST where the
306
- # return type of the statement would be one of the return
307
- # types of any compound statements it is a part of. For a
308
- # compound statement, the last of the child statements
309
- # would be in return position. This concept can be applied
310
- # recursively through e.g. conditionals to find a list of
311
- # statements in value positions.
312
- module DeepInference
313
- class << self
314
- CONDITIONAL_ALL_BUT_FIRST = [:if, :unless]
315
- CONDITIONAL_ALL = [:or]
316
- ONLY_ONE_CHILD = [:return]
317
- FIRST_TWO_CHILDREN = [:rescue]
318
- COMPOUND_STATEMENTS = [:begin, :kwbegin]
319
- SKIPPABLE = [:def, :defs, :class, :sclass, :module]
320
- FUNCTION_VALUE = [:block]
321
- CASE_STATEMENT = [:case]
322
-
323
- # @param node [AST::Node] a method body compound statement
324
- # @param include_explicit_returns [Boolean] If true,
325
- # include the value nodes of the parameter of the
326
- # 'return' statements in the type returned.
327
- # @return [Array<AST::Node>] low-level value nodes from
328
- # both nodes in value position as well as explicit
329
- # return statements in the method's closure.
330
- def from_method_body node
331
- from_value_position_statement(node, include_explicit_returns: true)
332
- end
333
-
334
- # @param node [AST::Node] an individual statement, to be
335
- # evaluated outside the context of a containing method
336
- # @return [Array<AST::Node>] low-level value nodes in
337
- # value position. Does not include explicit return
338
- # statements
339
- def value_position_nodes_only(node)
340
- from_value_position_statement(node, include_explicit_returns: false)
341
- end
342
-
343
- # Look at known control statements and use them to find
344
- # more specific return nodes.
345
- #
346
- # @param node [Parser::AST::Node] Statement which is in
347
- # value position for a method body
348
- # @param include_explicit_returns [Boolean] If true,
349
- # include the value nodes of the parameter of the
350
- # 'return' statements in the type returned.
351
- # @return [Array<Parser::AST::Node>]
352
- def from_value_position_statement node, include_explicit_returns: true
353
- # STDERR.puts("from_expression called on #{node.inspect}")
354
- return [] unless node.is_a?(::Parser::AST::Node)
355
- # @type [Array<Parser::AST::Node>]
356
- result = []
357
- if COMPOUND_STATEMENTS.include?(node.type)
358
- result.concat from_value_position_compound_statement node
359
- elsif CONDITIONAL_ALL_BUT_FIRST.include?(node.type)
360
- result.concat reduce_to_value_nodes(node.children[1..-1])
361
- # result.push NIL_NODE unless node.children[2]
362
- elsif CONDITIONAL_ALL.include?(node.type)
363
- result.concat reduce_to_value_nodes(node.children)
364
- elsif ONLY_ONE_CHILD.include?(node.type)
365
- result.concat reduce_to_value_nodes([node.children[0]])
366
- elsif FIRST_TWO_CHILDREN.include?(node.type)
367
- result.concat reduce_to_value_nodes([node.children[0], node.children[1]])
368
- elsif FUNCTION_VALUE.include?(node.type)
369
- # the block itself is a first class value that could be returned
370
- result.push node
371
- # @todo any explicit returns actually return from
372
- # scope in which the proc is run. This asssumes
373
- # that the function is executed here.
374
- result.concat explicit_return_values_from_compound_statement(node.children[2]) if include_explicit_returns
375
- elsif CASE_STATEMENT.include?(node.type)
376
- node.children[1..-1].each do |cc|
377
- if cc.nil?
378
- result.push NIL_NODE
379
- elsif cc.type == :when
380
- result.concat reduce_to_value_nodes([cc.children.last])
381
- else
382
- # else clause in case
383
- result.concat reduce_to_value_nodes([cc])
384
- end
385
- end
386
- elsif node.type == :resbody
387
- result.concat reduce_to_value_nodes([node.children[2]])
388
- else
389
- result.push node
390
- end
391
- result
392
- end
393
-
394
- # Treat parent as as a begin block and use the last node's
395
- # return node plus any explicit return nodes' return nodes. e.g.,
396
- #
397
- # 123
398
- # 456
399
- # return 'a' if foo == bar
400
- # 789
401
- #
402
- # would return 'a' and 789.
403
- #
404
- # @param parent [Parser::AST::Node]
405
- #
406
- # @return [Array<Parser::AST::Node>]
407
- def from_value_position_compound_statement parent
408
- result = []
409
- nodes = parent.children.select{|n| n.is_a?(AST::Node)}
410
- nodes.each_with_index do |node, idx|
411
- if node.type == :block
412
- result.concat explicit_return_values_from_compound_statement(node.children[2])
413
- elsif node.type == :rescue
414
- # body statements
415
- result.concat from_value_position_statement(node.children[0])
416
- # rescue statements
417
- result.concat from_value_position_statement(node.children[1])
418
- elsif SKIPPABLE.include?(node.type)
419
- next
420
- elsif node.type == :resbody
421
- result.concat reduce_to_value_nodes([node.children[2]])
422
- elsif node.type == :return
423
- result.concat reduce_to_value_nodes([node.children[0]])
424
- # Return here because the rest of the code is
425
- # unreachable and shouldn't be looked at
426
- return result
427
- else
428
- result.concat explicit_return_values_from_compound_statement(node)
429
- end
430
- # handle last line of compound statements, which is in
431
- # value position. we already have the explicit values
432
- # from above; now we need to also gather the value
433
- # position nodes
434
- result.concat from_value_position_statement(nodes.last, include_explicit_returns: false) if idx == nodes.length - 1
435
- end
436
- result
437
- end
438
-
439
- private
440
-
441
- # Useful when this statement isn't in value position, but
442
- # we care explicit return statements nonetheless.
443
- #
444
- # @param parent [Parser::AST::Node]
445
- #
446
- # @return [Array<Parser::AST::Node>]
447
- def explicit_return_values_from_compound_statement parent
448
- return [] unless parent.is_a?(::Parser::AST::Node)
449
- result = []
450
- nodes = parent.children.select{|n| n.is_a?(::Parser::AST::Node)}
451
- nodes.each do |node|
452
- next if SKIPPABLE.include?(node.type)
453
- if node.type == :return
454
- result.concat reduce_to_value_nodes([node.children[0]])
455
- # Return the result here because the rest of the code is
456
- # unreachable
457
- return result
458
- else
459
- result.concat explicit_return_values_from_compound_statement(node)
460
- end
461
- end
462
- result
463
- end
464
-
465
- # @param nodes [Enumerable<Parser::AST::Node, BasicObject>]
466
- # @return [Array<Parser::AST::Node, nil>]
467
- def reduce_to_value_nodes nodes
468
- result = []
469
- nodes.each do |node|
470
- if !node.is_a?(::Parser::AST::Node)
471
- result.push nil
472
- elsif COMPOUND_STATEMENTS.include?(node.type)
473
- result.concat from_value_position_compound_statement(node)
474
- elsif CONDITIONAL_ALL_BUT_FIRST.include?(node.type)
475
- result.concat reduce_to_value_nodes(node.children[1..-1])
476
- elsif node.type == :return
477
- result.concat reduce_to_value_nodes([node.children[0]])
478
- elsif node.type == :or
479
- result.concat reduce_to_value_nodes(node.children)
480
- elsif node.type == :block
481
- result.concat explicit_return_values_from_compound_statement(node.children[2])
482
- elsif node.type == :resbody
483
- result.concat reduce_to_value_nodes([node.children[2]])
484
- else
485
- result.push node
486
- end
487
- end
488
- result
489
- end
490
- end
491
- end
492
- end
493
- end
494
- end
495
- end
1
+ # frozen_string_literal: true
2
+
3
+ require 'parser'
4
+ require 'ast'
5
+
6
+ # https://github.com/whitequark/parser
7
+ module Solargraph
8
+ module Parser
9
+ module ParserGem
10
+ module NodeMethods
11
+ module_function
12
+
13
+ # @param node [Parser::AST::Node]
14
+ # @return [String]
15
+ def unpack_name(node)
16
+ pack_name(node).join("::")
17
+ end
18
+
19
+ # @param node [Parser::AST::Node]
20
+ # @return [Array<String>]
21
+ def pack_name(node)
22
+ # @type [Array<String>]
23
+ parts = []
24
+ if node.is_a?(AST::Node)
25
+ node.children.each { |n|
26
+ if n.is_a?(AST::Node)
27
+ if n.type == :cbase
28
+ parts = [''] + pack_name(n)
29
+ elsif n.type == :const
30
+ parts += pack_name(n)
31
+ end
32
+ else
33
+ parts.push n unless n.nil?
34
+ end
35
+ }
36
+ end
37
+ parts
38
+ end
39
+
40
+ # @param node [Parser::AST::Node]
41
+ # @return [String, nil]
42
+ def infer_literal_node_type node
43
+ return nil unless node.is_a?(AST::Node)
44
+ if node.type == :str || node.type == :dstr
45
+ return '::String'
46
+ elsif node.type == :array
47
+ return '::Array'
48
+ elsif node.type == :nil
49
+ return '::NilClass'
50
+ elsif node.type == :hash
51
+ return '::Hash'
52
+ elsif node.type == :int
53
+ return '::Integer'
54
+ elsif node.type == :float
55
+ return '::Float'
56
+ elsif node.type == :sym || node.type == :dsym
57
+ return '::Symbol'
58
+ elsif node.type == :regexp
59
+ return '::Regexp'
60
+ elsif node.type == :irange
61
+ return '::Range'
62
+ elsif node.type == :true || node.type == :false
63
+ return '::Boolean'
64
+ # @todo Support `nil` keyword in types
65
+ # elsif node.type == :nil
66
+ # return 'NilClass'
67
+ end
68
+ nil
69
+ end
70
+
71
+ # @param node [Parser::AST::Node]
72
+ # @return [Position]
73
+ def get_node_start_position(node)
74
+ Position.new(node.loc.line, node.loc.column)
75
+ end
76
+
77
+ # @param node [Parser::AST::Node]
78
+ # @return [Position]
79
+ def get_node_end_position(node)
80
+ Position.new(node.loc.last_line, node.loc.last_column)
81
+ end
82
+
83
+ # @param node [Parser::AST::Node]
84
+ # @param signature [String]
85
+ #
86
+ # @return [String]
87
+ def drill_signature node, signature
88
+ return signature unless node.is_a?(AST::Node)
89
+ if node.type == :const or node.type == :cbase
90
+ unless node.children[0].nil?
91
+ signature += drill_signature(node.children[0], signature)
92
+ end
93
+ signature += '::' unless signature.empty?
94
+ signature += node.children[1].to_s
95
+ elsif node.type == :lvar or node.type == :ivar or node.type == :cvar
96
+ signature += '.' unless signature.empty?
97
+ signature += node.children[0].to_s
98
+ elsif node.type == :send
99
+ unless node.children[0].nil?
100
+ signature += drill_signature(node.children[0], signature)
101
+ end
102
+ signature += '.' unless signature.empty?
103
+ signature += node.children[1].to_s
104
+ end
105
+ signature
106
+ end
107
+
108
+ # @param node [Parser::AST::Node]
109
+ # @return [Hash{Symbol => Chain}]
110
+ def convert_hash node
111
+ return {} unless Parser.is_ast_node?(node)
112
+ return convert_hash(node.children[0]) if node.type == :kwsplat
113
+ return convert_hash(node.children[0]) if Parser.is_ast_node?(node.children[0]) && node.children[0].type == :kwsplat
114
+ return {} unless node.type == :hash
115
+ result = {}
116
+ node.children.each do |pair|
117
+ result[pair.children[0].children[0]] = Solargraph::Parser.chain(pair.children[1])
118
+ end
119
+ result
120
+ end
121
+
122
+ # @sg-ignore Wrong argument type for AST::Node.new: type expected AST::_ToSym, received :nil
123
+ NIL_NODE = ::Parser::AST::Node.new(:nil)
124
+
125
+ # @param node [Parser::AST::Node]
126
+ #
127
+ # @return [Array<Parser::AST::Node>]
128
+ def const_nodes_from node
129
+ return [] unless Parser.is_ast_node?(node)
130
+ result = []
131
+ if node.type == :const
132
+ result.push node
133
+ else
134
+ node.children.each { |child| result.concat const_nodes_from(child) }
135
+ end
136
+ result
137
+ end
138
+
139
+ # @param node [Parser::AST::Node]
140
+ def splatted_hash? node
141
+ Parser.is_ast_node?(node.children[0]) && node.children[0].type == :kwsplat
142
+ end
143
+
144
+ # @param node [Parser::AST::Node]
145
+ def splatted_call? node
146
+ return false unless Parser.is_ast_node?(node)
147
+ Parser.is_ast_node?(node.children[0]) && node.children[0].type == :kwsplat && node.children[0].children[0].type != :hash
148
+ end
149
+
150
+ # @param nodes [Enumerable<Parser::AST::Node>]
151
+ def any_splatted_call?(nodes)
152
+ nodes.any? { |n| splatted_call?(n) }
153
+ end
154
+
155
+ # @todo Temporarily here for testing. Move to Solargraph::Parser.
156
+ # @param node [Parser::AST::Node]
157
+ # @return [Array<Parser::AST::Node>]
158
+ def call_nodes_from node
159
+ return [] unless node.is_a?(::Parser::AST::Node)
160
+ result = []
161
+ if node.type == :block
162
+ result.push node
163
+ if Parser.is_ast_node?(node.children[0]) && node.children[0].children.length > 2
164
+ node.children[0].children[2..-1].each { |child| result.concat call_nodes_from(child) }
165
+ end
166
+ node.children[1..-1].each { |child| result.concat call_nodes_from(child) }
167
+ elsif node.type == :send
168
+ result.push node
169
+ result.concat call_nodes_from(node.children.first)
170
+ node.children[2..-1].each { |child| result.concat call_nodes_from(child) }
171
+ elsif [:super, :zsuper].include?(node.type)
172
+ result.push node
173
+ node.children.each { |child| result.concat call_nodes_from(child) }
174
+ else
175
+ node.children.each { |child| result.concat call_nodes_from(child) }
176
+ end
177
+ result
178
+ end
179
+
180
+ # Find all the nodes within the provided node that potentially return a
181
+ # value.
182
+ #
183
+ # The node parameter typically represents a method's logic, e.g., the
184
+ # second child (after the :args node) of a :def node. A simple one-line
185
+ # method would typically return itself, while a node with conditions
186
+ # would return the resulting node from each conditional branch. Nodes
187
+ # that follow a :return node are assumed to be unreachable. Nil values
188
+ # are converted to nil node types.
189
+ #
190
+ # @param node [Parser::AST::Node]
191
+ # @return [Array<Parser::AST::Node>]
192
+ def returns_from_method_body node
193
+ # @todo is the || NIL_NODE necessary?
194
+ # STDERR.puts("Evaluating expression: #{node.inspect}")
195
+ DeepInference.from_method_body(node).map { |n| n || NIL_NODE }
196
+ end
197
+
198
+ # @param node [Parser::AST::Node]
199
+ # @return [Array<AST::Node>] low-level value nodes in
200
+ # value position. Does not include explicit return
201
+ # statements
202
+ def value_position_nodes_only(node)
203
+ DeepInference.value_position_nodes_only(node).map { |n| n || NIL_NODE }
204
+ end
205
+
206
+ # @param cursor [Solargraph::Source::Cursor]
207
+ # @return [Parser::AST::Node, nil]
208
+ def find_recipient_node cursor
209
+ return repaired_find_recipient_node(cursor) if cursor.source.repaired? && cursor.source.code[cursor.offset - 1] == '('
210
+ source = cursor.source
211
+ position = cursor.position
212
+ offset = cursor.offset
213
+ tree = if source.synchronized?
214
+ match = source.code[0..offset-1].match(/,\s*\z/)
215
+ if match
216
+ source.tree_at(position.line, position.column - match[0].length)
217
+ else
218
+ source.tree_at(position.line, position.column)
219
+ end
220
+ else
221
+ source.tree_at(position.line, position.column - 1)
222
+ end
223
+ # @type [AST::Node, nil]
224
+ prev = nil
225
+ tree.each do |node|
226
+ if node.type == :send
227
+ args = node.children[2..-1]
228
+ if !args.empty?
229
+ return node if prev && args.include?(prev)
230
+ else
231
+ if source.synchronized?
232
+ return node if source.code[0..offset-1] =~ /\(\s*\z/ && source.code[offset..-1] =~ /^\s*\)/
233
+ else
234
+ return node if source.code[0..offset-1] =~ /\([^(]*\z/
235
+ end
236
+ end
237
+ end
238
+ prev = node
239
+ end
240
+ nil
241
+ end
242
+
243
+ # @param cursor [Solargraph::Source::Cursor]
244
+ # @return [Parser::AST::Node, nil]
245
+ def repaired_find_recipient_node cursor
246
+ cursor = cursor.source.cursor_at([cursor.position.line, cursor.position.column - 1])
247
+ node = cursor.source.tree_at(cursor.position.line, cursor.position.column).first
248
+ return node if node && node.type == :send
249
+ end
250
+
251
+ #
252
+ # Concepts:
253
+ #
254
+ # - statement - one single node in the AST. Generally used
255
+ # synonymously with how the Parser gem uses the term
256
+ # 'expression'. This may have side effects (e.g.,
257
+ # registering a method in the namespace, modifying
258
+ # variables or doing I/O). It may encapsulate multiple
259
+ # other statements (see compound statement).
260
+ #
261
+ # - value - something that can be assigned to a variable by
262
+ # evaluating a statement
263
+ #
264
+ # - value node - the 'lowest level' AST node whose return
265
+ # type is a subset of the value type of the overall
266
+ # statement. Might be a literal, a method call, etc - the
267
+ # goal is to find the lowest level node, which we can use
268
+ # Chains and Pins later on to determine the type of.
269
+ #
270
+ # e.g., if the node 'b ? 123 : 456' were a return value, we
271
+ # know the actual return values possible are 123 and 456,
272
+ # and can disregard the rest.
273
+ #
274
+ # - value type - the type representing the multiple possible
275
+ # values that can result from evaluation of the statement.
276
+ #
277
+ # - return type - the type describing the values a statement
278
+ # might evaluate to. When used with a method, the term
279
+ # describes the values that may result from the method
280
+ # being called, and includes explicit return statements
281
+ # within the method body's closure.
282
+ #
283
+ # - method body - a compound statement with parameters whose
284
+ # return value type must account both for the explicit
285
+ # 'return' statemnts as well as the final statements
286
+ # executed in any given control flow through the method.
287
+ #
288
+ # - explicit return statement - a statement which, when part of a
289
+ # method body, is a possible value of a call to that method -
290
+ # e.g., "return 123"
291
+ #
292
+ # - compound statement - a statement which can be expanded to
293
+ # be multiple statements in a row, executed in the context
294
+ # of a method which can be explicitly returned from.
295
+ #
296
+ # - value position - the positions in the AST where the
297
+ # return type of the statement would be one of the return
298
+ # types of any compound statements it is a part of. For a
299
+ # compound statement, the last of the child statements
300
+ # would be in return position. This concept can be applied
301
+ # recursively through e.g. conditionals to find a list of
302
+ # statements in value positions.
303
+ module DeepInference
304
+ class << self
305
+ CONDITIONAL_ALL_BUT_FIRST = [:if, :unless]
306
+ CONDITIONAL_ALL = [:or]
307
+ ONLY_ONE_CHILD = [:return]
308
+ FIRST_TWO_CHILDREN = [:rescue]
309
+ COMPOUND_STATEMENTS = [:begin, :kwbegin]
310
+ SKIPPABLE = [:def, :defs, :class, :sclass, :module]
311
+ FUNCTION_VALUE = [:block]
312
+ CASE_STATEMENT = [:case]
313
+
314
+ # @param node [AST::Node] a method body compound statement
315
+ # @param include_explicit_returns [Boolean] If true,
316
+ # include the value nodes of the parameter of the
317
+ # 'return' statements in the type returned.
318
+ # @return [Array<AST::Node>] low-level value nodes from
319
+ # both nodes in value position as well as explicit
320
+ # return statements in the method's closure.
321
+ def from_method_body node
322
+ from_value_position_statement(node, include_explicit_returns: true)
323
+ end
324
+
325
+ # @param node [AST::Node] an individual statement, to be
326
+ # evaluated outside the context of a containing method
327
+ # @return [Array<AST::Node>] low-level value nodes in
328
+ # value position. Does not include explicit return
329
+ # statements
330
+ def value_position_nodes_only(node)
331
+ from_value_position_statement(node, include_explicit_returns: false)
332
+ end
333
+
334
+ # Look at known control statements and use them to find
335
+ # more specific return nodes.
336
+ #
337
+ # @param node [Parser::AST::Node] Statement which is in
338
+ # value position for a method body
339
+ # @param include_explicit_returns [Boolean] If true,
340
+ # include the value nodes of the parameter of the
341
+ # 'return' statements in the type returned.
342
+ # @return [Array<Parser::AST::Node>]
343
+ def from_value_position_statement node, include_explicit_returns: true
344
+ # STDERR.puts("from_expression called on #{node.inspect}")
345
+ return [] unless node.is_a?(::Parser::AST::Node)
346
+ # @type [Array<Parser::AST::Node>]
347
+ result = []
348
+ if COMPOUND_STATEMENTS.include?(node.type)
349
+ result.concat from_value_position_compound_statement node
350
+ elsif CONDITIONAL_ALL_BUT_FIRST.include?(node.type)
351
+ result.concat reduce_to_value_nodes(node.children[1..-1])
352
+ # result.push NIL_NODE unless node.children[2]
353
+ elsif CONDITIONAL_ALL.include?(node.type)
354
+ result.concat reduce_to_value_nodes(node.children)
355
+ elsif ONLY_ONE_CHILD.include?(node.type)
356
+ result.concat reduce_to_value_nodes([node.children[0]])
357
+ elsif FIRST_TWO_CHILDREN.include?(node.type)
358
+ result.concat reduce_to_value_nodes([node.children[0], node.children[1]])
359
+ elsif FUNCTION_VALUE.include?(node.type)
360
+ # the block itself is a first class value that could be returned
361
+ result.push node
362
+ # @todo any explicit returns actually return from
363
+ # scope in which the proc is run. This asssumes
364
+ # that the function is executed here.
365
+ result.concat explicit_return_values_from_compound_statement(node.children[2]) if include_explicit_returns
366
+ elsif CASE_STATEMENT.include?(node.type)
367
+ node.children[1..-1].each do |cc|
368
+ if cc.nil?
369
+ result.push NIL_NODE
370
+ elsif cc.type == :when
371
+ result.concat reduce_to_value_nodes([cc.children.last])
372
+ else
373
+ # else clause in case
374
+ result.concat reduce_to_value_nodes([cc])
375
+ end
376
+ end
377
+ elsif node.type == :resbody
378
+ result.concat reduce_to_value_nodes([node.children[2]])
379
+ else
380
+ result.push node
381
+ end
382
+ result
383
+ end
384
+
385
+ # Treat parent as as a begin block and use the last node's
386
+ # return node plus any explicit return nodes' return nodes. e.g.,
387
+ #
388
+ # 123
389
+ # 456
390
+ # return 'a' if foo == bar
391
+ # 789
392
+ #
393
+ # would return 'a' and 789.
394
+ #
395
+ # @param parent [Parser::AST::Node]
396
+ #
397
+ # @return [Array<Parser::AST::Node>]
398
+ def from_value_position_compound_statement parent
399
+ result = []
400
+ nodes = parent.children.select{|n| n.is_a?(AST::Node)}
401
+ nodes.each_with_index do |node, idx|
402
+ if node.type == :block
403
+ result.concat explicit_return_values_from_compound_statement(node.children[2])
404
+ elsif node.type == :rescue
405
+ # body statements
406
+ result.concat from_value_position_statement(node.children[0])
407
+ # rescue statements
408
+ result.concat from_value_position_statement(node.children[1])
409
+ elsif SKIPPABLE.include?(node.type)
410
+ next
411
+ elsif node.type == :resbody
412
+ result.concat reduce_to_value_nodes([node.children[2]])
413
+ elsif node.type == :return
414
+ result.concat reduce_to_value_nodes([node.children[0]])
415
+ # Return here because the rest of the code is
416
+ # unreachable and shouldn't be looked at
417
+ return result
418
+ else
419
+ result.concat explicit_return_values_from_compound_statement(node)
420
+ end
421
+ # handle last line of compound statements, which is in
422
+ # value position. we already have the explicit values
423
+ # from above; now we need to also gather the value
424
+ # position nodes
425
+ result.concat from_value_position_statement(nodes.last, include_explicit_returns: false) if idx == nodes.length - 1
426
+ end
427
+ result
428
+ end
429
+
430
+ private
431
+
432
+ # Useful when this statement isn't in value position, but
433
+ # we care explicit return statements nonetheless.
434
+ #
435
+ # @param parent [Parser::AST::Node]
436
+ #
437
+ # @return [Array<Parser::AST::Node>]
438
+ def explicit_return_values_from_compound_statement parent
439
+ return [] unless parent.is_a?(::Parser::AST::Node)
440
+ result = []
441
+ nodes = parent.children.select{|n| n.is_a?(::Parser::AST::Node)}
442
+ nodes.each do |node|
443
+ next if SKIPPABLE.include?(node.type)
444
+ if node.type == :return
445
+ result.concat reduce_to_value_nodes([node.children[0]])
446
+ # Return the result here because the rest of the code is
447
+ # unreachable
448
+ return result
449
+ else
450
+ result.concat explicit_return_values_from_compound_statement(node)
451
+ end
452
+ end
453
+ result
454
+ end
455
+
456
+ # @param nodes [Enumerable<Parser::AST::Node, BasicObject>]
457
+ # @return [Array<Parser::AST::Node, nil>]
458
+ def reduce_to_value_nodes nodes
459
+ result = []
460
+ nodes.each do |node|
461
+ if !node.is_a?(::Parser::AST::Node)
462
+ result.push nil
463
+ elsif COMPOUND_STATEMENTS.include?(node.type)
464
+ result.concat from_value_position_compound_statement(node)
465
+ elsif CONDITIONAL_ALL_BUT_FIRST.include?(node.type)
466
+ result.concat reduce_to_value_nodes(node.children[1..-1])
467
+ elsif node.type == :return
468
+ result.concat reduce_to_value_nodes([node.children[0]])
469
+ elsif node.type == :or
470
+ result.concat reduce_to_value_nodes(node.children)
471
+ elsif node.type == :block
472
+ result.concat explicit_return_values_from_compound_statement(node.children[2])
473
+ elsif node.type == :resbody
474
+ result.concat reduce_to_value_nodes([node.children[2]])
475
+ else
476
+ result.push node
477
+ end
478
+ end
479
+ result
480
+ end
481
+ end
482
+ end
483
+ end
484
+ end
485
+ end
486
+ end