solargraph 0.56.0 → 0.58.2

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 (191) 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 +183 -7
  5. data/.github/workflows/rspec.yml +55 -5
  6. data/.github/workflows/typecheck.yml +6 -3
  7. data/.gitignore +6 -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 +92 -1
  14. data/README.md +8 -4
  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 -175
  20. data/lib/solargraph/api_map/source_to_yard.rb +97 -88
  21. data/lib/solargraph/api_map/store.rb +384 -266
  22. data/lib/solargraph/api_map.rb +945 -973
  23. data/lib/solargraph/bench.rb +1 -0
  24. data/lib/solargraph/complex_type/type_methods.rb +228 -222
  25. data/lib/solargraph/complex_type/unique_type.rb +482 -475
  26. data/lib/solargraph/complex_type.rb +444 -423
  27. data/lib/solargraph/convention/active_support_concern.rb +111 -0
  28. data/lib/solargraph/convention/base.rb +17 -0
  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/gemspec.rb +3 -2
  33. data/lib/solargraph/convention/struct_definition/struct_assignment_node.rb +61 -60
  34. data/lib/solargraph/convention/struct_definition/struct_definition_node.rb +102 -100
  35. data/lib/solargraph/convention/struct_definition.rb +164 -101
  36. data/lib/solargraph/convention.rb +32 -2
  37. data/lib/solargraph/diagnostics/require_not_found.rb +53 -53
  38. data/lib/solargraph/diagnostics/rubocop.rb +118 -113
  39. data/lib/solargraph/diagnostics/rubocop_helpers.rb +68 -66
  40. data/lib/solargraph/diagnostics/type_check.rb +55 -55
  41. data/lib/solargraph/doc_map.rb +439 -405
  42. data/lib/solargraph/environ.rb +9 -2
  43. data/lib/solargraph/equality.rb +34 -33
  44. data/lib/solargraph/gem_pins.rb +98 -88
  45. data/lib/solargraph/language_server/host/diagnoser.rb +89 -89
  46. data/lib/solargraph/language_server/host/dispatch.rb +130 -128
  47. data/lib/solargraph/language_server/host/message_worker.rb +112 -109
  48. data/lib/solargraph/language_server/host/sources.rb +99 -99
  49. data/lib/solargraph/language_server/host.rb +878 -871
  50. data/lib/solargraph/language_server/message/base.rb +2 -1
  51. data/lib/solargraph/language_server/message/extended/check_gem_version.rb +114 -114
  52. data/lib/solargraph/language_server/message/extended/document.rb +23 -23
  53. data/lib/solargraph/language_server/message/text_document/completion.rb +56 -56
  54. data/lib/solargraph/language_server/message/text_document/definition.rb +40 -38
  55. data/lib/solargraph/language_server/message/text_document/document_symbol.rb +26 -26
  56. data/lib/solargraph/language_server/message/text_document/formatting.rb +148 -131
  57. data/lib/solargraph/language_server/message/text_document/hover.rb +58 -58
  58. data/lib/solargraph/language_server/message/text_document/signature_help.rb +24 -24
  59. data/lib/solargraph/language_server/message/text_document/type_definition.rb +25 -24
  60. data/lib/solargraph/language_server/message/workspace/did_change_workspace_folders.rb +2 -0
  61. data/lib/solargraph/language_server/message/workspace/workspace_symbol.rb +23 -23
  62. data/lib/solargraph/language_server/progress.rb +8 -0
  63. data/lib/solargraph/language_server/request.rb +4 -1
  64. data/lib/solargraph/library.rb +683 -666
  65. data/lib/solargraph/location.rb +82 -79
  66. data/lib/solargraph/logging.rb +37 -28
  67. data/lib/solargraph/page.rb +3 -0
  68. data/lib/solargraph/parser/comment_ripper.rb +69 -62
  69. data/lib/solargraph/parser/flow_sensitive_typing.rb +255 -227
  70. data/lib/solargraph/parser/node_processor/base.rb +92 -87
  71. data/lib/solargraph/parser/node_processor.rb +62 -46
  72. data/lib/solargraph/parser/parser_gem/class_methods.rb +149 -159
  73. data/lib/solargraph/parser/parser_gem/flawed_builder.rb +1 -0
  74. data/lib/solargraph/parser/parser_gem/node_chainer.rb +166 -164
  75. data/lib/solargraph/parser/parser_gem/node_methods.rb +486 -497
  76. data/lib/solargraph/parser/parser_gem/node_processors/and_node.rb +22 -21
  77. data/lib/solargraph/parser/parser_gem/node_processors/args_node.rb +59 -59
  78. data/lib/solargraph/parser/parser_gem/node_processors/begin_node.rb +15 -15
  79. data/lib/solargraph/parser/parser_gem/node_processors/block_node.rb +46 -45
  80. data/lib/solargraph/parser/parser_gem/node_processors/casgn_node.rb +1 -21
  81. data/lib/solargraph/parser/parser_gem/node_processors/def_node.rb +53 -53
  82. data/lib/solargraph/parser/parser_gem/node_processors/if_node.rb +23 -21
  83. data/lib/solargraph/parser/parser_gem/node_processors/ivasgn_node.rb +40 -40
  84. data/lib/solargraph/parser/parser_gem/node_processors/lvasgn_node.rb +29 -29
  85. data/lib/solargraph/parser/parser_gem/node_processors/masgn_node.rb +59 -53
  86. data/lib/solargraph/parser/parser_gem/node_processors/namespace_node.rb +0 -22
  87. data/lib/solargraph/parser/parser_gem/node_processors/opasgn_node.rb +98 -41
  88. data/lib/solargraph/parser/parser_gem/node_processors/orasgn_node.rb +17 -16
  89. data/lib/solargraph/parser/parser_gem/node_processors/resbody_node.rb +38 -37
  90. data/lib/solargraph/parser/parser_gem/node_processors/sclass_node.rb +52 -43
  91. data/lib/solargraph/parser/parser_gem/node_processors/send_node.rb +291 -271
  92. data/lib/solargraph/parser/parser_gem/node_processors/sym_node.rb +1 -0
  93. data/lib/solargraph/parser/parser_gem/node_processors/while_node.rb +29 -29
  94. data/lib/solargraph/parser/parser_gem/node_processors.rb +70 -66
  95. data/lib/solargraph/parser/region.rb +69 -66
  96. data/lib/solargraph/parser/snippet.rb +17 -15
  97. data/lib/solargraph/pin/base.rb +729 -651
  98. data/lib/solargraph/pin/base_variable.rb +126 -125
  99. data/lib/solargraph/pin/block.rb +104 -103
  100. data/lib/solargraph/pin/breakable.rb +9 -9
  101. data/lib/solargraph/pin/callable.rb +231 -218
  102. data/lib/solargraph/pin/closure.rb +72 -74
  103. data/lib/solargraph/pin/common.rb +79 -75
  104. data/lib/solargraph/pin/constant.rb +2 -0
  105. data/lib/solargraph/pin/conversions.rb +123 -123
  106. data/lib/solargraph/pin/delegated_method.rb +120 -120
  107. data/lib/solargraph/pin/documenting.rb +114 -114
  108. data/lib/solargraph/pin/instance_variable.rb +34 -34
  109. data/lib/solargraph/pin/keyword.rb +20 -20
  110. data/lib/solargraph/pin/local_variable.rb +75 -76
  111. data/lib/solargraph/pin/method.rb +672 -651
  112. data/lib/solargraph/pin/method_alias.rb +34 -31
  113. data/lib/solargraph/pin/namespace.rb +115 -115
  114. data/lib/solargraph/pin/parameter.rb +275 -261
  115. data/lib/solargraph/pin/proxy_type.rb +39 -35
  116. data/lib/solargraph/pin/reference/override.rb +47 -33
  117. data/lib/solargraph/pin/reference/superclass.rb +15 -10
  118. data/lib/solargraph/pin/reference.rb +39 -22
  119. data/lib/solargraph/pin/search.rb +61 -56
  120. data/lib/solargraph/pin/signature.rb +61 -59
  121. data/lib/solargraph/pin/symbol.rb +53 -48
  122. data/lib/solargraph/pin/until.rb +18 -18
  123. data/lib/solargraph/pin/while.rb +18 -18
  124. data/lib/solargraph/pin.rb +44 -44
  125. data/lib/solargraph/pin_cache.rb +245 -185
  126. data/lib/solargraph/position.rb +132 -116
  127. data/lib/solargraph/range.rb +112 -107
  128. data/lib/solargraph/rbs_map/conversions.rb +823 -773
  129. data/lib/solargraph/rbs_map/core_fills.rb +18 -0
  130. data/lib/solargraph/rbs_map/core_map.rb +58 -51
  131. data/lib/solargraph/rbs_map/stdlib_map.rb +43 -43
  132. data/lib/solargraph/rbs_map.rb +163 -150
  133. data/lib/solargraph/shell.rb +352 -268
  134. data/lib/solargraph/source/chain/call.rb +337 -333
  135. data/lib/solargraph/source/chain/constant.rb +26 -89
  136. data/lib/solargraph/source/chain/hash.rb +34 -34
  137. data/lib/solargraph/source/chain/if.rb +28 -28
  138. data/lib/solargraph/source/chain/instance_variable.rb +13 -13
  139. data/lib/solargraph/source/chain/link.rb +11 -2
  140. data/lib/solargraph/source/chain/literal.rb +48 -48
  141. data/lib/solargraph/source/chain/or.rb +23 -23
  142. data/lib/solargraph/source/chain.rb +291 -282
  143. data/lib/solargraph/source/change.rb +82 -82
  144. data/lib/solargraph/source/cursor.rb +166 -167
  145. data/lib/solargraph/source/encoding_fixes.rb +23 -23
  146. data/lib/solargraph/source/source_chainer.rb +194 -194
  147. data/lib/solargraph/source/updater.rb +55 -55
  148. data/lib/solargraph/source.rb +498 -495
  149. data/lib/solargraph/source_map/clip.rb +226 -234
  150. data/lib/solargraph/source_map/data.rb +34 -30
  151. data/lib/solargraph/source_map/mapper.rb +259 -259
  152. data/lib/solargraph/source_map.rb +212 -200
  153. data/lib/solargraph/type_checker/checks.rb +124 -124
  154. data/lib/solargraph/type_checker/param_def.rb +37 -35
  155. data/lib/solargraph/type_checker/problem.rb +32 -32
  156. data/lib/solargraph/type_checker/rules.rb +84 -62
  157. data/lib/solargraph/type_checker.rb +814 -699
  158. data/lib/solargraph/version.rb +5 -5
  159. data/lib/solargraph/workspace/config.rb +255 -239
  160. data/lib/solargraph/workspace/require_paths.rb +97 -0
  161. data/lib/solargraph/workspace.rb +220 -249
  162. data/lib/solargraph/yard_map/helpers.rb +44 -16
  163. data/lib/solargraph/yard_map/mapper/to_constant.rb +5 -5
  164. data/lib/solargraph/yard_map/mapper/to_method.rb +130 -134
  165. data/lib/solargraph/yard_map/mapper/to_namespace.rb +31 -30
  166. data/lib/solargraph/yard_map/mapper.rb +79 -79
  167. data/lib/solargraph/yard_map/to_method.rb +89 -88
  168. data/lib/solargraph/yardoc.rb +87 -49
  169. data/lib/solargraph.rb +105 -90
  170. data/rbs/fills/bundler/0/bundler.rbs +4271 -0
  171. data/rbs/fills/open3/0/open3.rbs +172 -0
  172. data/rbs/fills/rubygems/0/basic_specification.rbs +326 -0
  173. data/rbs/fills/rubygems/0/errors.rbs +364 -0
  174. data/rbs/fills/rubygems/0/spec_fetcher.rbs +107 -0
  175. data/rbs/fills/rubygems/0/specification.rbs +1753 -0
  176. data/rbs/fills/{tuple.rbs → tuple/tuple.rbs} +2 -3
  177. data/rbs/shims/ast/0/node.rbs +5 -0
  178. data/rbs/shims/ast/2.4/.rbs_meta.yaml +9 -0
  179. data/rbs/shims/ast/2.4/ast.rbs +73 -0
  180. data/rbs/shims/parser/3.2.0.1/builders/default.rbs +195 -0
  181. data/rbs/shims/parser/3.2.0.1/manifest.yaml +7 -0
  182. data/rbs/shims/parser/3.2.0.1/parser.rbs +201 -0
  183. data/rbs/shims/parser/3.2.0.1/polyfill.rbs +4 -0
  184. data/rbs/shims/thor/1.2.0.1/.rbs_meta.yaml +9 -0
  185. data/rbs/shims/thor/1.2.0.1/manifest.yaml +7 -0
  186. data/rbs/shims/thor/1.2.0.1/thor.rbs +17 -0
  187. data/rbs_collection.yaml +4 -4
  188. data/solargraph.gemspec +26 -5
  189. metadata +187 -15
  190. data/lib/.rubocop.yml +0 -22
  191. data/lib/solargraph/parser/node_methods.rb +0 -97
@@ -1,495 +1,498 @@
1
- # frozen_string_literal: true
2
-
3
- require 'yard'
4
-
5
- module Solargraph
6
- # A Ruby file that has been parsed into an AST.
7
- #
8
- class Source
9
- autoload :Updater, 'solargraph/source/updater'
10
- autoload :Change, 'solargraph/source/change'
11
- autoload :EncodingFixes, 'solargraph/source/encoding_fixes'
12
- autoload :Cursor, 'solargraph/source/cursor'
13
- autoload :Chain, 'solargraph/source/chain'
14
- autoload :SourceChainer, 'solargraph/source/source_chainer'
15
-
16
- include EncodingFixes
17
-
18
- # @return [String]
19
- attr_reader :filename
20
-
21
- # @return [String]
22
- def code
23
- finalize
24
- @code
25
- end
26
-
27
- # @return [Parser::AST::Node, nil]
28
- def node
29
- finalize
30
- @node
31
- end
32
-
33
- # @return [Hash{Integer => Array<String>}]
34
- def comments
35
- finalize
36
- @comments
37
- end
38
-
39
- # @todo Deprecate?
40
- # @return [Integer]
41
- attr_reader :version
42
-
43
- # @param code [String]
44
- # @param filename [String, nil]
45
- # @param version [Integer]
46
- def initialize code, filename = nil, version = 0
47
- @code = normalize(code)
48
- @repaired = code
49
- @filename = filename
50
- @version = version
51
- end
52
-
53
- # @param range [Solargraph::Range]
54
- # @return [String]
55
- def at range
56
- from_to range.start.line, range.start.character, range.ending.line, range.ending.character
57
- end
58
-
59
- # @param l1 [Integer]
60
- # @param c1 [Integer]
61
- # @param l2 [Integer]
62
- # @param c2 [Integer]
63
- # @return [String]
64
- def from_to l1, c1, l2, c2
65
- b = Solargraph::Position.line_char_to_offset(code, l1, c1)
66
- e = Solargraph::Position.line_char_to_offset(code, l2, c2)
67
- code[b..e-1]
68
- end
69
-
70
- # Get the nearest node that contains the specified index.
71
- #
72
- # @param line [Integer]
73
- # @param column [Integer]
74
- # @return [AST::Node]
75
- def node_at(line, column)
76
- tree_at(line, column).first
77
- end
78
-
79
- # Get an array of nodes containing the specified index, starting with the
80
- # nearest node and ending with the root.
81
- #
82
- # @param line [Integer]
83
- # @param column [Integer]
84
- # @return [Array<AST::Node>]
85
- def tree_at(line, column)
86
- position = Position.new(line, column)
87
- stack = []
88
- inner_tree_at node, position, stack
89
- stack
90
- end
91
-
92
- # Synchronize the Source with an update. This method applies changes to the
93
- # code, parses the new code's AST, and returns the resulting Source object.
94
- #
95
- # @param updater [Source::Updater]
96
- # @return [Source]
97
- def synchronize updater
98
- raise 'Invalid synchronization' unless updater.filename == filename
99
- real_code = updater.write(@code)
100
- if real_code == @code
101
- @version = updater.version
102
- return self
103
- end
104
- Source.new(@code, filename, updater.version).tap do |src|
105
- src.repaired = @repaired
106
- src.error_ranges.concat error_ranges
107
- src.changes.concat(changes + updater.changes)
108
- end
109
- end
110
-
111
- # @param position [Position, Array(Integer, Integer)]
112
- # @return [Source::Cursor]
113
- def cursor_at position
114
- finalize
115
- Cursor.new(self, position)
116
- end
117
-
118
- # @return [Boolean]
119
- def parsed?
120
- finalize
121
- @parsed
122
- end
123
-
124
- def repaired?
125
- code != @repaired
126
- end
127
-
128
- # @param position [Position]
129
- # @return [Boolean]
130
- def string_at? position
131
- return false if Position.to_offset(code, position) >= code.length
132
- string_nodes.each do |node|
133
- range = Range.from_node(node)
134
- next if range.ending.line < position.line
135
- break if range.ending.line > position.line
136
- return true if node.type == :str && range.include?(position) && range.start != position
137
- return true if [:STR, :str].include?(node.type) && range.include?(position) && range.start != position
138
- if node.type == :dstr
139
- inner = node_at(position.line, position.column)
140
- next if inner.nil?
141
- inner_range = Range.from_node(inner)
142
- next unless range.include?(inner_range.ending)
143
- return true if inner.type == :str
144
- inner_code = at(Solargraph::Range.new(inner_range.start, position))
145
- return true if (inner.type == :dstr && inner_range.ending.character <= position.character) && !inner_code.end_with?('}') ||
146
- (inner.type != :dstr && inner_range.ending.line == position.line && position.character <= inner_range.ending.character && inner_code.end_with?('}'))
147
- end
148
- break if range.ending.line > position.line
149
- end
150
- false
151
- end
152
-
153
- # @return [::Array<Range>]
154
- def string_ranges
155
- @string_ranges ||= Parser.string_ranges(node)
156
- end
157
-
158
- # @param position [Position]
159
- # @return [Boolean]
160
- def comment_at? position
161
- comment_ranges.each do |range|
162
- return true if range.include?(position) ||
163
- (range.ending.line == position.line && range.ending.column < position.column)
164
- break if range.ending.line > position.line
165
- end
166
- false
167
- end
168
-
169
- # @param name [String]
170
- # @return [Array<Location>]
171
- def references name
172
- Parser.references self, name
173
- end
174
-
175
- # @return [Array<Range>]
176
- def error_ranges
177
- @error_ranges ||= []
178
- end
179
-
180
- # @param node [Parser::AST::Node]
181
- # @return [String]
182
- def code_for(node)
183
- rng = Range.from_node(node)
184
- b = Position.line_char_to_offset(code, rng.start.line, rng.start.column)
185
- e = Position.line_char_to_offset(code, rng.ending.line, rng.ending.column)
186
- frag = code[b..e-1].to_s
187
- frag.strip.gsub(/,$/, '')
188
- end
189
-
190
- # @param node [Parser::AST::Node]
191
- # @return [String, nil]
192
- def comments_for node
193
- rng = Range.from_node(node)
194
- stringified_comments[rng.start.line] ||= begin
195
- buff = associated_comments[rng.start.line]
196
- buff ? stringify_comment_array(buff) : nil
197
- end
198
- end
199
-
200
- # A location representing the file in its entirety.
201
- #
202
- # @return [Location]
203
- def location
204
- st = Position.new(0, 0)
205
- en = Position.from_offset(code, code.length)
206
- range = Range.new(st, en)
207
- Location.new(filename, range)
208
- end
209
-
210
- FOLDING_NODE_TYPES = %i[
211
- class sclass module def defs if str dstr array while unless kwbegin hash block
212
- ].freeze
213
-
214
- # Get an array of ranges that can be folded, e.g., the range of a class
215
- # definition or an if condition.
216
- #
217
- # See FOLDING_NODE_TYPES for the list of node types that can be folded.
218
- #
219
- # @return [Array<Range>]
220
- def folding_ranges
221
- @folding_ranges ||= begin
222
- result = []
223
- inner_folding_ranges node, result
224
- result.concat foldable_comment_block_ranges
225
- result
226
- end
227
- end
228
-
229
- def synchronized?
230
- true
231
- end
232
-
233
- # Get a hash of comments grouped by the line numbers of the associated code.
234
- #
235
- # @return [Hash{Integer => String}]
236
- def associated_comments
237
- @associated_comments ||= begin
238
- result = {}
239
- buffer = String.new('')
240
- # @type [Integer, nil]
241
- last = nil
242
- comments.each_pair do |num, snip|
243
- if !last || num == last + 1
244
- buffer.concat "#{snip.text}\n"
245
- else
246
- result[first_not_empty_from(last + 1)] = buffer.clone
247
- buffer.replace "#{snip.text}\n"
248
- end
249
- last = num
250
- end
251
- result[first_not_empty_from(last + 1)] = buffer unless buffer.empty? || last.nil?
252
- result
253
- end
254
- end
255
-
256
- private
257
-
258
- # @param line [Integer]
259
- # @return [Integer]
260
- def first_not_empty_from line
261
- cursor = line
262
- cursor += 1 while cursor < code_lines.length && code_lines[cursor].strip.empty?
263
- cursor = line if cursor > code_lines.length - 1
264
- cursor
265
- end
266
-
267
- # @param top [Parser::AST::Node]
268
- # @param result [Array<Range>]
269
- # @param parent [Symbol, nil]
270
- # @return [void]
271
- def inner_folding_ranges top, result = [], parent = nil
272
- return unless Parser.is_ast_node?(top)
273
- if FOLDING_NODE_TYPES.include?(top.type)
274
- range = Range.from_node(top)
275
- if result.empty? || range.start.line > result.last.start.line
276
- result.push range unless range.ending.line - range.start.line < 2
277
- end
278
- end
279
- top.children.each do |child|
280
- inner_folding_ranges(child, result, top.type)
281
- end
282
- end
283
-
284
- # Get a string representation of an array of comments.
285
- #
286
- # @param comments [String]
287
- # @return [String]
288
- def stringify_comment_array comments
289
- ctxt = String.new('')
290
- started = false
291
- skip = nil
292
- comments.lines.each { |l|
293
- # Trim the comment and minimum leading whitespace
294
- p = l.force_encoding('UTF-8').encode('UTF-8', invalid: :replace, replace: '?').gsub(/^#+/, '')
295
- if p.strip.empty?
296
- next unless started
297
- ctxt.concat p
298
- else
299
- here = p.index(/[^ \t]/)
300
- skip = here if skip.nil? || here < skip
301
- ctxt.concat p[skip..-1]
302
- end
303
- started = true
304
- }
305
- ctxt
306
- end
307
-
308
- # A hash of line numbers and their associated comments.
309
- #
310
- # @return [Hash{Integer => Array<String>, nil}]
311
- def stringified_comments
312
- @stringified_comments ||= {}
313
- end
314
-
315
- # @return [Array<Parser::AST::Node>]
316
- def string_nodes
317
- @string_nodes ||= string_nodes_in(node)
318
- end
319
-
320
- # @return [Array<Range>]
321
- def comment_ranges
322
- @comment_ranges ||= comments.values.map(&:range)
323
- end
324
-
325
- # Get an array of foldable comment block ranges. Blocks are excluded if
326
- # they are less than 3 lines long.
327
- #
328
- # @return [Array<Range>]
329
- def foldable_comment_block_ranges
330
- return [] unless synchronized?
331
- result = []
332
- grouped = []
333
- comments.keys.each do |l|
334
- if grouped.empty? || l == grouped.last + 1
335
- grouped.push l
336
- else
337
- result.push Range.from_to(grouped.first, 0, grouped.last, 0) unless grouped.length < 3
338
- grouped = [l]
339
- end
340
- end
341
- result.push Range.from_to(grouped.first, 0, grouped.last, 0) unless grouped.length < 3
342
- result
343
- end
344
-
345
- # @param n [Parser::AST::Node, nil]
346
- # @return [Array<Parser::AST::Node>]
347
- def string_nodes_in n
348
- result = []
349
- if Parser.is_ast_node?(n)
350
- if n.type == :str || n.type == :dstr || n.type == :STR || n.type == :DSTR
351
- result.push n
352
- else
353
- n.children.each{ |c| result.concat string_nodes_in(c) }
354
- end
355
- end
356
- result
357
- end
358
-
359
- # @param node [Parser::AST::Node, nil]
360
- # @param position [Position]
361
- # @param stack [Array<Parser::AST::Node>]
362
- # @return [void]
363
- def inner_tree_at node, position, stack
364
- return if node.nil?
365
- here = Range.from_node(node)
366
- if here.contain?(position)
367
- stack.unshift node
368
- node.children.each do |c|
369
- next unless Parser.is_ast_node?(c)
370
- next if c.loc.expression.nil?
371
- inner_tree_at(c, position, stack)
372
- end
373
- end
374
- end
375
-
376
- protected
377
-
378
- # @return [Array<Change>]
379
- def changes
380
- @changes ||= []
381
- end
382
-
383
- # @return [String]
384
- attr_writer :filename
385
-
386
- # @return [Integer]
387
- attr_writer :version
388
-
389
- def finalize
390
- return if @finalized && changes.empty?
391
-
392
- changes.each do |change|
393
- @code = change.write(@code)
394
- end
395
- @finalized = true
396
- begin
397
- @node, @comments = Solargraph::Parser.parse_with_comments(@code, filename)
398
- @parsed = true
399
- @repaired = @code
400
- rescue Parser::SyntaxError, EncodingError => e
401
- @node = nil
402
- @comments = {}
403
- @parsed = false
404
- ensure
405
- @code.freeze
406
- end
407
- if !@parsed && !changes.empty?
408
- changes.each do |change|
409
- @repaired = change.repair(@repaired)
410
- end
411
- error_ranges.concat(changes.map(&:range))
412
- begin
413
- @node, @comments = Solargraph::Parser.parse_with_comments(@repaired, filename)
414
- @parsed = true
415
- rescue Parser::SyntaxError, EncodingError => e
416
- @node = nil
417
- @comments = {}
418
- @parsed = false
419
- end
420
- elsif @parsed
421
- error_ranges.clear
422
- end
423
- changes.clear
424
- end
425
-
426
- # @param val [String]
427
- # @return [String]
428
- def code=(val)
429
- @code_lines = nil
430
- @finalized = false
431
- @code = val
432
- end
433
-
434
- # @return [Parser::AST::Node, nil]
435
- attr_writer :node
436
-
437
- # @return [Array<Range>]
438
- attr_writer :error_ranges
439
-
440
- # @return [String]
441
- attr_writer :repaired
442
-
443
- def repaired
444
- finalize
445
- @repaired
446
- end
447
-
448
- # @return [Boolean]
449
- attr_writer :parsed
450
-
451
- # @return [Hash{Integer => String}
452
- attr_writer :comments
453
-
454
- # @return [Boolean]
455
- attr_writer :synchronized
456
-
457
- private
458
-
459
- # @return [Array<String>]
460
- def code_lines
461
- @code_lines ||= code.lines
462
- end
463
-
464
- class << self
465
- # @param filename [String]
466
- # @return [Solargraph::Source]
467
- def load filename
468
- file = File.open(filename)
469
- code = file.read
470
- file.close
471
- Source.load_string(code, filename)
472
- end
473
-
474
- # @param code [String]
475
- # @param filename [String, nil]
476
- # @param version [Integer]
477
- # @return [Solargraph::Source]
478
- def load_string code, filename = nil, version = 0
479
- Source.new code, filename, version
480
- end
481
-
482
- # @param comments [String]
483
- # @return [YARD::DocstringParser]
484
- def parse_docstring comments
485
- # HACK: Pass a dummy code object to the parser for plugins that
486
- # expect it not to be nil
487
- YARD::Docstring.parser.parse(comments, YARD::CodeObjects::Base.new(:root, 'stub'))
488
- rescue StandardError => e
489
- Solargraph.logger.info "YARD failed to parse docstring: [#{e.class}] #{e.message}"
490
- Solargraph.logger.debug "Unparsed comment: #{comments}"
491
- YARD::Docstring.parser
492
- end
493
- end
494
- end
495
- end
1
+ # frozen_string_literal: true
2
+
3
+ require 'yard'
4
+
5
+ module Solargraph
6
+ # A Ruby file that has been parsed into an AST.
7
+ #
8
+ class Source
9
+ autoload :Updater, 'solargraph/source/updater'
10
+ autoload :Change, 'solargraph/source/change'
11
+ autoload :EncodingFixes, 'solargraph/source/encoding_fixes'
12
+ autoload :Cursor, 'solargraph/source/cursor'
13
+ autoload :Chain, 'solargraph/source/chain'
14
+ autoload :SourceChainer, 'solargraph/source/source_chainer'
15
+
16
+ include EncodingFixes
17
+
18
+ # @return [String, nil]
19
+ attr_reader :filename
20
+
21
+ # @return [String]
22
+ def code
23
+ finalize
24
+ @code
25
+ end
26
+
27
+ # @return [Parser::AST::Node, nil]
28
+ def node
29
+ finalize
30
+ @node
31
+ end
32
+
33
+ # @return [Hash{Integer => Solargraph::Parser::Snippet}]
34
+ def comments
35
+ finalize
36
+ @comments
37
+ end
38
+
39
+ # @todo Deprecate?
40
+ # @return [Integer]
41
+ attr_reader :version
42
+
43
+ # @param code [String]
44
+ # @param filename [String, nil]
45
+ # @param version [Integer]
46
+ def initialize code, filename = nil, version = 0
47
+ @code = normalize(code)
48
+ @repaired = code
49
+ @filename = filename
50
+ @version = version
51
+ end
52
+
53
+ # @param range [Solargraph::Range]
54
+ # @return [String]
55
+ def at range
56
+ from_to range.start.line, range.start.character, range.ending.line, range.ending.character
57
+ end
58
+
59
+ # @param l1 [Integer]
60
+ # @param c1 [Integer]
61
+ # @param l2 [Integer]
62
+ # @param c2 [Integer]
63
+ # @return [String]
64
+ def from_to l1, c1, l2, c2
65
+ b = Solargraph::Position.line_char_to_offset(code, l1, c1)
66
+ e = Solargraph::Position.line_char_to_offset(code, l2, c2)
67
+ code[b..e-1]
68
+ end
69
+
70
+ # Get the nearest node that contains the specified index.
71
+ #
72
+ # @param line [Integer]
73
+ # @param column [Integer]
74
+ # @return [AST::Node]
75
+ def node_at(line, column)
76
+ tree_at(line, column).first
77
+ end
78
+
79
+ # Get an array of nodes containing the specified index, starting with the
80
+ # nearest node and ending with the root.
81
+ #
82
+ # @param line [Integer]
83
+ # @param column [Integer]
84
+ # @return [Array<AST::Node>]
85
+ def tree_at(line, column)
86
+ position = Position.new(line, column)
87
+ stack = []
88
+ inner_tree_at node, position, stack
89
+ stack
90
+ end
91
+
92
+ # Synchronize the Source with an update. This method applies changes to the
93
+ # code, parses the new code's AST, and returns the resulting Source object.
94
+ #
95
+ # @param updater [Source::Updater]
96
+ # @return [Source]
97
+ def synchronize updater
98
+ raise 'Invalid synchronization' unless updater.filename == filename
99
+ real_code = updater.write(@code)
100
+ if real_code == @code
101
+ @version = updater.version
102
+ return self
103
+ end
104
+ Source.new(@code, filename, updater.version).tap do |src|
105
+ src.repaired = @repaired
106
+ src.error_ranges.concat error_ranges
107
+ src.changes.concat(changes + updater.changes)
108
+ end
109
+ end
110
+
111
+ # @param position [Position, Array(Integer, Integer)]
112
+ # @return [Source::Cursor]
113
+ def cursor_at position
114
+ finalize
115
+ Cursor.new(self, position)
116
+ end
117
+
118
+ # @return [Boolean]
119
+ def parsed?
120
+ finalize
121
+ @parsed
122
+ end
123
+
124
+ def repaired?
125
+ code != @repaired
126
+ end
127
+
128
+ # @param position [Position]
129
+ # @return [Boolean]
130
+ def string_at? position
131
+ return false if Position.to_offset(code, position) >= code.length
132
+ string_nodes.each do |node|
133
+ range = Range.from_node(node)
134
+ next if range.ending.line < position.line
135
+ break if range.ending.line > position.line
136
+ return true if node.type == :str && range.include?(position) && range.start != position
137
+ return true if [:STR, :str].include?(node.type) && range.include?(position) && range.start != position
138
+ if node.type == :dstr
139
+ inner = node_at(position.line, position.column)
140
+ next if inner.nil?
141
+ inner_range = Range.from_node(inner)
142
+ next unless range.include?(inner_range.ending)
143
+ return true if inner.type == :str
144
+ inner_code = at(Solargraph::Range.new(inner_range.start, position))
145
+ return true if (inner.type == :dstr && inner_range.ending.character <= position.character) && !inner_code.end_with?('}') ||
146
+ (inner.type != :dstr && inner_range.ending.line == position.line && position.character <= inner_range.ending.character && inner_code.end_with?('}'))
147
+ end
148
+ break if range.ending.line > position.line
149
+ end
150
+ false
151
+ end
152
+
153
+ # @return [::Array<Range>]
154
+ def string_ranges
155
+ @string_ranges ||= Parser.string_ranges(node)
156
+ end
157
+
158
+ # @param position [Position]
159
+ # @return [Boolean]
160
+ def comment_at? position
161
+ comment_ranges.each do |range|
162
+ return true if range.include?(position) ||
163
+ (range.ending.line == position.line && range.ending.column < position.column)
164
+ break if range.ending.line > position.line
165
+ end
166
+ false
167
+ end
168
+
169
+ # @param name [String]
170
+ # @return [Array<Location>]
171
+ def references name
172
+ Parser.references self, name
173
+ end
174
+
175
+ # @return [Array<Range>]
176
+ def error_ranges
177
+ @error_ranges ||= []
178
+ end
179
+
180
+ # @param node [Parser::AST::Node]
181
+ # @return [String]
182
+ def code_for(node)
183
+ rng = Range.from_node(node)
184
+ b = Position.line_char_to_offset(code, rng.start.line, rng.start.column)
185
+ e = Position.line_char_to_offset(code, rng.ending.line, rng.ending.column)
186
+ frag = code[b..e-1].to_s
187
+ frag.strip.gsub(/,$/, '')
188
+ end
189
+
190
+ # @param node [Parser::AST::Node]
191
+ # @return [String, nil]
192
+ def comments_for node
193
+ rng = Range.from_node(node)
194
+ stringified_comments[rng.start.line] ||= begin
195
+ buff = associated_comments[rng.start.line]
196
+ buff ? stringify_comment_array(buff) : nil
197
+ end
198
+ end
199
+
200
+ # A location representing the file in its entirety.
201
+ #
202
+ # @return [Location]
203
+ def location
204
+ st = Position.new(0, 0)
205
+ en = Position.from_offset(code, code.length)
206
+ range = Range.new(st, en)
207
+ Location.new(filename, range)
208
+ end
209
+
210
+ FOLDING_NODE_TYPES = %i[
211
+ class sclass module def defs if str dstr array while unless kwbegin hash block
212
+ ].freeze
213
+
214
+ # Get an array of ranges that can be folded, e.g., the range of a class
215
+ # definition or an if condition.
216
+ #
217
+ # See FOLDING_NODE_TYPES for the list of node types that can be folded.
218
+ #
219
+ # @return [Array<Range>]
220
+ def folding_ranges
221
+ @folding_ranges ||= begin
222
+ result = []
223
+ inner_folding_ranges node, result
224
+ result.concat foldable_comment_block_ranges
225
+ result
226
+ end
227
+ end
228
+
229
+ def synchronized?
230
+ true
231
+ end
232
+
233
+ # Get a hash of comments grouped by the line numbers of the associated code.
234
+ #
235
+ # @return [Hash{Integer => String}]
236
+ def associated_comments
237
+ @associated_comments ||= begin
238
+ # @type [Hash{Integer => String}]
239
+ result = {}
240
+ buffer = String.new('')
241
+ # @type [Integer, nil]
242
+ last = nil
243
+ comments.each_pair do |num, snip|
244
+ if !last || num == last + 1
245
+ buffer.concat "#{snip.text}\n"
246
+ else
247
+ result[first_not_empty_from(last + 1)] = buffer.clone
248
+ buffer.replace "#{snip.text}\n"
249
+ end
250
+ last = num
251
+ end
252
+ result[first_not_empty_from(last + 1)] = buffer unless buffer.empty? || last.nil?
253
+ result
254
+ end
255
+ end
256
+
257
+ private
258
+
259
+ # @param line [Integer]
260
+ # @return [Integer]
261
+ def first_not_empty_from line
262
+ cursor = line
263
+ cursor += 1 while cursor < code_lines.length && code_lines[cursor].strip.empty?
264
+ cursor = line if cursor > code_lines.length - 1
265
+ cursor
266
+ end
267
+
268
+ # @param top [Parser::AST::Node]
269
+ # @param result [Array<Range>]
270
+ # @param parent [Symbol, nil]
271
+ # @return [void]
272
+ def inner_folding_ranges top, result = [], parent = nil
273
+ return unless Parser.is_ast_node?(top)
274
+ if FOLDING_NODE_TYPES.include?(top.type)
275
+ range = Range.from_node(top)
276
+ if result.empty? || range.start.line > result.last.start.line
277
+ result.push range unless range.ending.line - range.start.line < 2
278
+ end
279
+ end
280
+ top.children.each do |child|
281
+ inner_folding_ranges(child, result, top.type)
282
+ end
283
+ end
284
+
285
+ # Get a string representation of an array of comments.
286
+ #
287
+ # @param comments [String]
288
+ # @return [String]
289
+ def stringify_comment_array comments
290
+ ctxt = String.new('')
291
+ started = false
292
+ skip = nil
293
+ comments.lines.each { |l|
294
+ # Trim the comment and minimum leading whitespace
295
+ p = l.force_encoding('UTF-8').encode('UTF-8', invalid: :replace, replace: '?').gsub(/^#+/, '')
296
+ if p.strip.empty?
297
+ next unless started
298
+ ctxt.concat p
299
+ else
300
+ here = p.index(/[^ \t]/)
301
+ skip = here if skip.nil? || here < skip
302
+ ctxt.concat p[skip..-1]
303
+ end
304
+ started = true
305
+ }
306
+ ctxt
307
+ end
308
+
309
+ # A hash of line numbers and their associated comments.
310
+ #
311
+ # @return [Hash{Integer => Array<String>, nil}]
312
+ def stringified_comments
313
+ @stringified_comments ||= {}
314
+ end
315
+
316
+ # @return [Array<Parser::AST::Node>]
317
+ def string_nodes
318
+ @string_nodes ||= string_nodes_in(node)
319
+ end
320
+
321
+ # @return [Array<Solargraph::Range>]
322
+ def comment_ranges
323
+ @comment_ranges ||= comments.values.map(&:range)
324
+ end
325
+
326
+ # Get an array of foldable comment block ranges. Blocks are excluded if
327
+ # they are less than 3 lines long.
328
+ #
329
+ # @return [Array<Range>]
330
+ def foldable_comment_block_ranges
331
+ return [] unless synchronized?
332
+ result = []
333
+ grouped = []
334
+ comments.keys.each do |l|
335
+ if grouped.empty? || l == grouped.last + 1
336
+ grouped.push l
337
+ else
338
+ result.push Range.from_to(grouped.first, 0, grouped.last, 0) unless grouped.length < 3
339
+ grouped = [l]
340
+ end
341
+ end
342
+ result.push Range.from_to(grouped.first, 0, grouped.last, 0) unless grouped.length < 3
343
+ result
344
+ end
345
+
346
+ # @param n [Parser::AST::Node, nil]
347
+ # @return [Array<Parser::AST::Node>]
348
+ def string_nodes_in n
349
+ result = []
350
+ if Parser.is_ast_node?(n)
351
+ if n.type == :str || n.type == :dstr || n.type == :STR || n.type == :DSTR
352
+ result.push n
353
+ else
354
+ n.children.each{ |c| result.concat string_nodes_in(c) }
355
+ end
356
+ end
357
+ result
358
+ end
359
+
360
+ # @param node [Parser::AST::Node, nil]
361
+ # @param position [Position]
362
+ # @param stack [Array<Parser::AST::Node>]
363
+ # @return [void]
364
+ def inner_tree_at node, position, stack
365
+ return if node.nil?
366
+ here = Range.from_node(node)
367
+ if here.contain?(position)
368
+ stack.unshift node
369
+ node.children.each do |c|
370
+ next unless Parser.is_ast_node?(c)
371
+ next if c.loc.expression.nil?
372
+ inner_tree_at(c, position, stack)
373
+ end
374
+ end
375
+ end
376
+
377
+ protected
378
+
379
+ # @return [Array<Change>]
380
+ def changes
381
+ @changes ||= []
382
+ end
383
+
384
+ # @return [String]
385
+ attr_writer :filename
386
+
387
+ # @return [Integer]
388
+ attr_writer :version
389
+
390
+ # @return [void]
391
+ def finalize
392
+ return if @finalized && changes.empty?
393
+
394
+ changes.each do |change|
395
+ @code = change.write(@code)
396
+ end
397
+ @finalized = true
398
+ begin
399
+ @node, @comments = Solargraph::Parser.parse_with_comments(@code, filename)
400
+ @parsed = true
401
+ @repaired = @code
402
+ rescue Parser::SyntaxError, EncodingError => e
403
+ @node = nil
404
+ @comments = {}
405
+ @parsed = false
406
+ ensure
407
+ @code.freeze
408
+ end
409
+ if !@parsed && !changes.empty?
410
+ changes.each do |change|
411
+ @repaired = change.repair(@repaired)
412
+ end
413
+ error_ranges.concat(changes.map(&:range))
414
+ begin
415
+ @node, @comments = Solargraph::Parser.parse_with_comments(@repaired, filename)
416
+ @parsed = true
417
+ rescue Parser::SyntaxError, EncodingError => e
418
+ @node = nil
419
+ @comments = {}
420
+ @parsed = false
421
+ end
422
+ elsif @parsed
423
+ error_ranges.clear
424
+ end
425
+ changes.clear
426
+ end
427
+
428
+ # @param val [String]
429
+ # @return [String]
430
+ def code=(val)
431
+ @code_lines = nil
432
+ @finalized = false
433
+ @code = val
434
+ end
435
+
436
+ # @return [Parser::AST::Node, nil]
437
+ attr_writer :node
438
+
439
+ # @return [Array<Range>]
440
+ attr_writer :error_ranges
441
+
442
+ # @return [String]
443
+ attr_writer :repaired
444
+
445
+ # @return [String]
446
+ def repaired
447
+ finalize
448
+ @repaired
449
+ end
450
+
451
+ # @return [Boolean]
452
+ attr_writer :parsed
453
+
454
+ # @return [Hash{Integer => String}
455
+ attr_writer :comments
456
+
457
+ # @return [Boolean]
458
+ attr_writer :synchronized
459
+
460
+ private
461
+
462
+ # @return [Array<String>]
463
+ def code_lines
464
+ @code_lines ||= code.lines
465
+ end
466
+
467
+ class << self
468
+ # @param filename [String]
469
+ # @return [Solargraph::Source]
470
+ def load filename
471
+ file = File.open(filename)
472
+ code = file.read
473
+ file.close
474
+ Source.load_string(code, filename)
475
+ end
476
+
477
+ # @param code [String]
478
+ # @param filename [String, nil]
479
+ # @param version [Integer]
480
+ # @return [Solargraph::Source]
481
+ def load_string code, filename = nil, version = 0
482
+ Source.new code, filename, version
483
+ end
484
+
485
+ # @param comments [String]
486
+ # @return [YARD::DocstringParser]
487
+ def parse_docstring comments
488
+ # HACK: Pass a dummy code object to the parser for plugins that
489
+ # expect it not to be nil
490
+ YARD::Docstring.parser.parse(comments, YARD::CodeObjects::Base.new(:root, 'stub'))
491
+ rescue StandardError => e
492
+ Solargraph.logger.info "YARD failed to parse docstring: [#{e.class}] #{e.message}"
493
+ Solargraph.logger.debug "Unparsed comment: #{comments}"
494
+ YARD::Docstring.parser
495
+ end
496
+ end
497
+ end
498
+ end