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,646 +1,823 @@
1
- # frozen_string_literal: true
2
-
3
- require 'rbs'
4
-
5
- module Solargraph
6
- class RbsMap
7
- # Functions for converting RBS declarations to Solargraph pins
8
- #
9
- module Conversions
10
- # A container for tracking the current context of the RBS conversion
11
- # process, e.g., what visibility is declared for methods in the current
12
- # scope
13
- #
14
- class Context
15
- attr_reader :visibility
16
-
17
- # @param visibility [Symbol]
18
- def initialize visibility = :public
19
- @visibility = visibility
20
- end
21
- end
22
-
23
- # @return [Array<Pin::Base>]
24
- def pins
25
- @pins ||= []
26
- end
27
-
28
- private
29
-
30
- # @return [Hash{String => RBS::AST::Declarations::TypeAlias}]
31
- def type_aliases
32
- @type_aliases ||= {}
33
- end
34
-
35
- # @param loader [RBS::EnvironmentLoader]
36
- # @return [void]
37
- def load_environment_to_pins(loader)
38
- environment = RBS::Environment.from_loader(loader).resolve_type_names
39
- cursor = pins.length
40
- environment.declarations.each { |decl| convert_decl_to_pin(decl, Solargraph::Pin::ROOT_PIN) }
41
- added_pins = pins[cursor..-1]
42
- added_pins.each { |pin| pin.source = :rbs }
43
- end
44
-
45
- # @param decl [RBS::AST::Declarations::Base]
46
- # @param closure [Pin::Closure]
47
- # @return [void]
48
- def convert_decl_to_pin decl, closure
49
- case decl
50
- when RBS::AST::Declarations::Class
51
- class_decl_to_pin decl
52
- when RBS::AST::Declarations::Interface
53
- # STDERR.puts "Skipping interface #{decl.name.relative!}"
54
- interface_decl_to_pin decl, closure
55
- when RBS::AST::Declarations::TypeAlias
56
- type_aliases[decl.name.to_s] = decl
57
- when RBS::AST::Declarations::Module
58
- module_decl_to_pin decl
59
- when RBS::AST::Declarations::Constant
60
- constant_decl_to_pin decl
61
- when RBS::AST::Declarations::ClassAlias
62
- class_alias_decl_to_pin decl
63
- when RBS::AST::Declarations::ModuleAlias
64
- module_alias_decl_to_pin decl
65
- when RBS::AST::Declarations::Global
66
- global_decl_to_pin decl
67
- else
68
- Solargraph.logger.warn "Skipping declaration #{decl.class}"
69
- end
70
- end
71
-
72
- # @param decl [RBS::AST::Declarations::Module]
73
- # @param module_pin [Pin::Namespace]
74
- # @return [void]
75
- def convert_self_types_to_pins decl, module_pin
76
- decl.self_types.each { |self_type| context = convert_self_type_to_pins(self_type, module_pin) }
77
- end
78
-
79
- # @param decl [RBS::AST::Declarations::Module::Self]
80
- # @param closure [Pin::Namespace]
81
- # @return [void]
82
- def convert_self_type_to_pins decl, closure
83
- type = build_type(decl.name, decl.args)
84
- generic_values = type.all_params.map(&:to_s)
85
- include_pin = Solargraph::Pin::Reference::Include.new(
86
- name: decl.name.relative!.to_s,
87
- type_location: location_decl_to_pin_location(decl.location),
88
- generic_values: generic_values,
89
- closure: closure
90
- )
91
- pins.push include_pin
92
- end
93
-
94
- # @param decl [RBS::AST::Declarations::Module,RBS::AST::Declarations::Class,RBS::AST::Declarations::Interface]
95
- # @param closure [Pin::Namespace]
96
- # @return [void]
97
- def convert_members_to_pins decl, closure
98
- context = Context.new
99
- decl.members.each { |m| context = convert_member_to_pin(m, closure, context) }
100
- end
101
-
102
- # @param member [RBS::AST::Members::Base,RBS::AST::Declarations::Base]
103
- # @param closure [Pin::Namespace]
104
- # @param context [Context]
105
- # @return [void]
106
- def convert_member_to_pin member, closure, context
107
- case member
108
- when RBS::AST::Members::MethodDefinition
109
- method_def_to_pin(member, closure)
110
- when RBS::AST::Members::AttrReader
111
- attr_reader_to_pin(member, closure)
112
- when RBS::AST::Members::AttrWriter
113
- attr_writer_to_pin(member, closure)
114
- when RBS::AST::Members::AttrAccessor
115
- attr_accessor_to_pin(member, closure)
116
- when RBS::AST::Members::Include
117
- include_to_pin(member, closure)
118
- when RBS::AST::Members::Prepend
119
- prepend_to_pin(member, closure)
120
- when RBS::AST::Members::Extend
121
- extend_to_pin(member, closure)
122
- when RBS::AST::Members::Alias
123
- alias_to_pin(member, closure)
124
- when RBS::AST::Members::ClassInstanceVariable
125
- civar_to_pin(member, closure)
126
- when RBS::AST::Members::ClassVariable
127
- cvar_to_pin(member, closure)
128
- when RBS::AST::Members::InstanceVariable
129
- ivar_to_pin(member, closure)
130
- when RBS::AST::Members::Public
131
- return Context.new(visibility: :public)
132
- when RBS::AST::Members::Private
133
- return Context.new(visibility: :private)
134
- when RBS::AST::Declarations::Base
135
- convert_decl_to_pin(member, closure)
136
- else
137
- Solargraph.logger.warn "Skipping member type #{member.class}"
138
- end
139
- context
140
- end
141
-
142
- # @param decl [RBS::AST::Declarations::Class]
143
- # @return [void]
144
- def class_decl_to_pin decl
145
- class_pin = Solargraph::Pin::Namespace.new(
146
- type: :class,
147
- name: decl.name.relative!.to_s,
148
- closure: Solargraph::Pin::ROOT_PIN,
149
- comments: decl.comment&.string,
150
- type_location: location_decl_to_pin_location(decl.location),
151
- # @todo some type parameters in core/stdlib have default
152
- # values; Solargraph doesn't support that yet as so these
153
- # get treated as undefined if not specified
154
- generics: decl.type_params.map(&:name).map(&:to_s)
155
- )
156
- pins.push class_pin
157
- if decl.super_class
158
- pins.push Solargraph::Pin::Reference::Superclass.new(
159
- type_location: location_decl_to_pin_location(decl.super_class.location),
160
- closure: class_pin,
161
- name: decl.super_class.name.relative!.to_s
162
- )
163
- end
164
- add_mixins decl, class_pin
165
- convert_members_to_pins decl, class_pin
166
- end
167
-
168
- # @param decl [RBS::AST::Declarations::Interface]
169
- # @param closure [Pin::Closure]
170
- # @return [void]
171
- def interface_decl_to_pin decl, closure
172
- class_pin = Solargraph::Pin::Namespace.new(
173
- type: :module,
174
- type_location: location_decl_to_pin_location(decl.location),
175
- name: decl.name.relative!.to_s,
176
- closure: Solargraph::Pin::ROOT_PIN,
177
- comments: decl.comment&.string,
178
- generics: decl.type_params.map(&:name).map(&:to_s),
179
- # HACK: Using :hidden to keep interfaces from appearing in
180
- # autocompletion
181
- visibility: :hidden
182
- )
183
- class_pin.docstring.add_tag(YARD::Tags::Tag.new(:abstract, '(RBS interface)'))
184
- pins.push class_pin
185
- convert_members_to_pins decl, class_pin
186
- end
187
-
188
- # @param decl [RBS::AST::Declarations::Module]
189
- # @return [void]
190
- def module_decl_to_pin decl
191
- module_pin = Solargraph::Pin::Namespace.new(
192
- type: :module,
193
- name: decl.name.relative!.to_s,
194
- type_location: location_decl_to_pin_location(decl.location),
195
- closure: Solargraph::Pin::ROOT_PIN,
196
- comments: decl.comment&.string,
197
- generics: decl.type_params.map(&:name).map(&:to_s),
198
- )
199
- pins.push module_pin
200
- convert_self_types_to_pins decl, module_pin
201
- convert_members_to_pins decl, module_pin
202
-
203
- add_mixins decl, module_pin.closure
204
- end
205
-
206
- # @param name [String]
207
- # @param tag [String]
208
- # @param comments [String]
209
- # @param decl [RBS::AST::Declarations::ClassAlias, RBS::AST::Declarations::Constant, RBS::AST::Declarations::ModuleAlias]
210
- # @param base [String, nil] Optional conversion of tag to base<tag>
211
- #
212
- # @return [Solargraph::Pin::Constant]
213
- def create_constant(name, tag, comments, decl, base = nil)
214
- parts = name.split('::')
215
- if parts.length > 1
216
- name = parts.last
217
- closure = pins.select { |pin| pin && pin.path == parts[0..-2].join('::') }.first
218
- else
219
- name = parts.first
220
- closure = Solargraph::Pin::ROOT_PIN
221
- end
222
- constant_pin = Solargraph::Pin::Constant.new(
223
- name: name,
224
- closure: closure,
225
- type_location: location_decl_to_pin_location(decl.location),
226
- comments: comments
227
- )
228
- tag = "#{base}<#{tag}>" if base
229
- rooted_tag = ComplexType.parse(tag).force_rooted.rooted_tags
230
- constant_pin.docstring.add_tag(YARD::Tags::Tag.new(:return, '', rooted_tag))
231
- constant_pin
232
- end
233
-
234
- # @param decl [RBS::AST::Declarations::ClassAlias]
235
- # @return [void]
236
- def class_alias_decl_to_pin decl
237
- # See https://www.rubydoc.info/gems/rbs/3.4.3/RBS/AST/Declarations/ClassAlias
238
- new_name = decl.new_name.relative!.to_s
239
- old_name = decl.old_name.relative!.to_s
240
-
241
- pins.push create_constant(new_name, old_name, decl.comment&.string, decl, 'Class')
242
- end
243
-
244
- # @param decl [RBS::AST::Declarations::ModuleAlias]
245
- # @return [void]
246
- def module_alias_decl_to_pin decl
247
- # See https://www.rubydoc.info/gems/rbs/3.4.3/RBS/AST/Declarations/ModuleAlias
248
- new_name = decl.new_name.relative!.to_s
249
- old_name = decl.old_name.relative!.to_s
250
-
251
- pins.push create_constant(new_name, old_name, decl.comment&.string, decl, 'Module')
252
- end
253
-
254
- # @param decl [RBS::AST::Declarations::Constant]
255
- # @return [void]
256
- def constant_decl_to_pin decl
257
- tag = other_type_to_tag(decl.type)
258
- pins.push create_constant(decl.name.relative!.to_s, tag, decl.comment&.string, decl)
259
- end
260
-
261
- # @param decl [RBS::AST::Declarations::Global]
262
- # @return [void]
263
- def global_decl_to_pin decl
264
- closure = Solargraph::Pin::ROOT_PIN
265
- name = decl.name.to_s
266
- pin = Solargraph::Pin::GlobalVariable.new(
267
- name: name,
268
- closure: closure,
269
- comments: decl.comment&.string,
270
- )
271
- rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
272
- pin.docstring.add_tag(YARD::Tags::Tag.new(:type, '', rooted_tag))
273
- pins.push pin
274
- end
275
-
276
- # @param decl [RBS::AST::Members::MethodDefinition]
277
- # @param closure [Pin::Closure]
278
- # @return [void]
279
- def method_def_to_pin decl, closure
280
- # there may be edge cases here around different signatures
281
- # having different type params / orders - we may need to match
282
- # this data model and have generics live in signatures to
283
- # handle those correctly
284
- generics = decl.overloads.map(&:method_type).flat_map(&:type_params).map(&:name).map(&:to_s).uniq
285
- if decl.instance?
286
- pin = Solargraph::Pin::Method.new(
287
- name: decl.name.to_s,
288
- closure: closure,
289
- type_location: location_decl_to_pin_location(decl.location),
290
- comments: decl.comment&.string,
291
- scope: :instance,
292
- signatures: [],
293
- generics: generics,
294
- # @todo RBS core has unreliable visibility definitions
295
- visibility: closure.path == 'Kernel' && Kernel.private_instance_methods(false).include?(decl.name) ? :private : :public
296
- )
297
- pin.signatures.concat method_def_to_sigs(decl, pin)
298
- pins.push pin
299
- if pin.name == 'initialize'
300
- pin.instance_variable_set(:@visibility, :private)
301
- pin.instance_variable_set(:@return_type, ComplexType::VOID)
302
- end
303
- end
304
- if decl.singleton?
305
- pin = Solargraph::Pin::Method.new(
306
- name: decl.name.to_s,
307
- closure: closure,
308
- comments: decl.comment&.string,
309
- type_location: location_decl_to_pin_location(decl.location),
310
- scope: :class,
311
- signatures: [],
312
- generics: generics
313
- )
314
- pin.signatures.concat method_def_to_sigs(decl, pin)
315
- pins.push pin
316
- end
317
- end
318
-
319
- # @param decl [RBS::AST::Members::MethodDefinition]
320
- # @param pin [Pin::Method]
321
- # @return [void]
322
- def method_def_to_sigs decl, pin
323
- decl.overloads.map do |overload|
324
- generics = overload.method_type.type_params.map(&:name).map(&:to_s)
325
- signature_parameters, signature_return_type = parts_of_function(overload.method_type, pin)
326
- block = if overload.method_type.block
327
- block_parameters, block_return_type = parts_of_function(overload.method_type.block, pin)
328
- Pin::Signature.new(generics: generics, parameters: block_parameters, return_type: block_return_type)
329
- end
330
- Pin::Signature.new(generics: generics, parameters: signature_parameters, return_type: signature_return_type, block: block)
331
- end
332
- end
333
-
334
- # @param location [RBS::Location, nil]
335
- # @return [Solargraph::Location, nil]
336
- def location_decl_to_pin_location(location)
337
- return nil if location&.name.nil?
338
-
339
- start_pos = Position.new(location.start_line - 1, location.start_column)
340
- end_pos = Position.new(location.end_line - 1, location.end_column)
341
- range = Range.new(start_pos, end_pos)
342
- Location.new(location.name.to_s, range)
343
- end
344
-
345
- # @param type [RBS::MethodType,RBS::Types::Block]
346
- # @param pin [Pin::Method]
347
- # @return [Array(Array<Pin::Parameter>, ComplexType)]
348
- def parts_of_function type, pin
349
- return [[Solargraph::Pin::Parameter.new(decl: :restarg, name: 'arg', closure: pin)], ComplexType.try_parse(method_type_to_tag(type)).force_rooted] if defined?(RBS::Types::UntypedFunction) && type.type.is_a?(RBS::Types::UntypedFunction)
350
-
351
- parameters = []
352
- arg_num = -1
353
- type.type.required_positionals.each do |param|
354
- name = param.name ? param.name.to_s : "arg#{arg_num += 1}"
355
- parameters.push Solargraph::Pin::Parameter.new(decl: :arg, name: name, closure: pin, return_type: ComplexType.try_parse(other_type_to_tag(param.type)).force_rooted)
356
- end
357
- type.type.optional_positionals.each do |param|
358
- name = param.name ? param.name.to_s : "arg#{arg_num += 1}"
359
- parameters.push Solargraph::Pin::Parameter.new(decl: :optarg, name: name, closure: pin,
360
- return_type: ComplexType.try_parse(other_type_to_tag(param.type)).force_rooted)
361
- end
362
- if type.type.rest_positionals
363
- name = type.type.rest_positionals.name ? type.type.rest_positionals.name.to_s : "arg#{arg_num += 1}"
364
- parameters.push Solargraph::Pin::Parameter.new(decl: :restarg, name: name, closure: pin)
365
- end
366
- type.type.trailing_positionals.each do |param|
367
- name = param.name ? param.name.to_s : "arg#{arg_num += 1}"
368
- parameters.push Solargraph::Pin::Parameter.new(decl: :arg, name: name, closure: pin)
369
- end
370
- type.type.required_keywords.each do |orig, param|
371
- name = orig ? orig.to_s : "arg#{arg_num += 1}"
372
- parameters.push Solargraph::Pin::Parameter.new(decl: :kwarg, name: name, closure: pin,
373
- return_type: ComplexType.try_parse(other_type_to_tag(param.type)).force_rooted)
374
- end
375
- type.type.optional_keywords.each do |orig, param|
376
- name = orig ? orig.to_s : "arg#{arg_num += 1}"
377
- parameters.push Solargraph::Pin::Parameter.new(decl: :kwoptarg, name: name, closure: pin,
378
- return_type: ComplexType.try_parse(other_type_to_tag(param.type)).force_rooted)
379
- end
380
- if type.type.rest_keywords
381
- name = type.type.rest_keywords.name ? type.type.rest_keywords.name.to_s : "arg#{arg_num += 1}"
382
- parameters.push Solargraph::Pin::Parameter.new(decl: :kwrestarg, name: type.type.rest_keywords.name.to_s, closure: pin)
383
- end
384
-
385
- rooted_tag = method_type_to_tag(type)
386
- return_type = ComplexType.try_parse(rooted_tag).force_rooted
387
- [parameters, return_type]
388
- end
389
-
390
- # @param decl [RBS::AST::Members::AttrReader,RBS::AST::Members::AttrAccessor]
391
- # @param closure [Pin::Namespace]
392
- # @return [void]
393
- def attr_reader_to_pin(decl, closure)
394
- pin = Solargraph::Pin::Method.new(
395
- name: decl.name.to_s,
396
- type_location: location_decl_to_pin_location(decl.location),
397
- closure: closure,
398
- comments: decl.comment&.string,
399
- scope: :instance,
400
- attribute: true
401
- )
402
- rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
403
- pin.docstring.add_tag(YARD::Tags::Tag.new(:return, '', rooted_tag))
404
- pins.push pin
405
- end
406
-
407
- # @param decl [RBS::AST::Members::AttrWriter, RBS::AST::Members::AttrAccessor]
408
- # @param closure [Pin::Namespace]
409
- # @return [void]
410
- def attr_writer_to_pin(decl, closure)
411
- pin = Solargraph::Pin::Method.new(
412
- name: "#{decl.name.to_s}=",
413
- type_location: location_decl_to_pin_location(decl.location),
414
- closure: closure,
415
- comments: decl.comment&.string,
416
- scope: :instance,
417
- attribute: true
418
- )
419
- rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
420
- pin.docstring.add_tag(YARD::Tags::Tag.new(:return, '', rooted_tag))
421
- pins.push pin
422
- end
423
-
424
- # @param decl [RBS::AST::Members::AttrAccessor]
425
- # @param closure [Pin::Namespace]
426
- # @return [void]
427
- def attr_accessor_to_pin(decl, closure)
428
- attr_reader_to_pin(decl, closure)
429
- attr_writer_to_pin(decl, closure)
430
- end
431
-
432
- # @param decl [RBS::AST::Members::InstanceVariable]
433
- # @param closure [Pin::Namespace]
434
- # @return [void]
435
- def ivar_to_pin(decl, closure)
436
- pin = Solargraph::Pin::InstanceVariable.new(
437
- name: decl.name.to_s,
438
- closure: closure,
439
- type_location: location_decl_to_pin_location(decl.location),
440
- comments: decl.comment&.string
441
- )
442
- rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
443
- pin.docstring.add_tag(YARD::Tags::Tag.new(:type, '', rooted_tag))
444
- pins.push pin
445
- end
446
-
447
- # @param decl [RBS::AST::Members::ClassVariable]
448
- # @param closure [Pin::Namespace]
449
- # @return [void]
450
- def cvar_to_pin(decl, closure)
451
- name = decl.name.to_s
452
- pin = Solargraph::Pin::ClassVariable.new(
453
- name: name,
454
- closure: closure,
455
- comments: decl.comment&.string
456
- )
457
- rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
458
- pin.docstring.add_tag(YARD::Tags::Tag.new(:type, '', rooted_tag))
459
- pins.push pin
460
- end
461
-
462
- # @param decl [RBS::AST::Members::ClassInstanceVariable]
463
- # @param closure [Pin::Namespace]
464
- # @return [void]
465
- def civar_to_pin(decl, closure)
466
- name = decl.name.to_s
467
- pin = Solargraph::Pin::InstanceVariable.new(
468
- name: name,
469
- closure: closure,
470
- comments: decl.comment&.string
471
- )
472
- rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
473
- pin.docstring.add_tag(YARD::Tags::Tag.new(:type, '', rooted_tag))
474
- pins.push pin
475
- end
476
-
477
- # @param decl [RBS::AST::Members::Include]
478
- # @param closure [Pin::Namespace]
479
- # @return [void]
480
- def include_to_pin decl, closure
481
- type = build_type(decl.name, decl.args)
482
- generic_values = type.all_params.map(&:to_s)
483
- pins.push Solargraph::Pin::Reference::Include.new(
484
- name: decl.name.relative!.to_s,
485
- type_location: location_decl_to_pin_location(decl.location),
486
- generic_values: generic_values,
487
- closure: closure
488
- )
489
- end
490
-
491
- # @param decl [RBS::AST::Members::Prepend]
492
- # @param closure [Pin::Namespace]
493
- # @return [void]
494
- def prepend_to_pin decl, closure
495
- pins.push Solargraph::Pin::Reference::Prepend.new(
496
- name: decl.name.relative!.to_s,
497
- type_location: location_decl_to_pin_location(decl.location),
498
- closure: closure
499
- )
500
- end
501
-
502
- # @param decl [RBS::AST::Members::Extend]
503
- # @param closure [Pin::Namespace]
504
- # @return [void]
505
- def extend_to_pin decl, closure
506
- pins.push Solargraph::Pin::Reference::Extend.new(
507
- name: decl.name.relative!.to_s,
508
- type_location: location_decl_to_pin_location(decl.location),
509
- closure: closure
510
- )
511
- end
512
-
513
- # @param decl [RBS::AST::Members::Alias]
514
- # @param closure [Pin::Namespace]
515
- # @return [void]
516
- def alias_to_pin decl, closure
517
- pins.push Solargraph::Pin::MethodAlias.new(
518
- name: decl.new_name.to_s,
519
- type_location: location_decl_to_pin_location(decl.location),
520
- original: decl.old_name.to_s,
521
- closure: closure
522
- )
523
- end
524
-
525
- RBS_TO_YARD_TYPE = {
526
- 'bool' => 'Boolean',
527
- 'string' => 'String',
528
- 'int' => 'Integer',
529
- 'untyped' => '',
530
- 'NilClass' => 'nil'
531
- }
532
-
533
- # @param type [RBS::MethodType]
534
- # @return [String]
535
- def method_type_to_tag type
536
- if type_aliases.key?(type.type.return_type.to_s)
537
- other_type_to_tag(type_aliases[type.type.return_type.to_s].type)
538
- else
539
- other_type_to_tag type.type.return_type
540
- end
541
- end
542
-
543
- # @param type_name [RBS::TypeName]
544
- # @param type_args [Enumerable<RBS::Types::Bases::Base>]
545
- # @return [ComplexType::UniqueType]
546
- def build_type(type_name, type_args = [])
547
- base = RBS_TO_YARD_TYPE[type_name.relative!.to_s] || type_name.relative!.to_s
548
- params = type_args.map { |a| other_type_to_tag(a) }.reject { |t| t == 'undefined' }.map do |t|
549
- ComplexType.try_parse(t).force_rooted
550
- end
551
- if base == 'Hash' && params.length == 2
552
- ComplexType::UniqueType.new(base, [params.first], [params.last], rooted: true, parameters_type: :hash)
553
- else
554
- ComplexType::UniqueType.new(base, [], params, rooted: true, parameters_type: :list)
555
- end
556
- end
557
-
558
- # @param type_name [RBS::TypeName]
559
- # @param type_args [Enumerable<RBS::Types::Bases::Base>]
560
- # @return [String]
561
- def type_tag(type_name, type_args = [])
562
- build_type(type_name, type_args).tags
563
- end
564
-
565
- # @param type [RBS::Types::Bases::Base]
566
- # @return [String]
567
- def other_type_to_tag type
568
- if type.is_a?(RBS::Types::Optional)
569
- "#{other_type_to_tag(type.type)}, nil"
570
- elsif type.is_a?(RBS::Types::Bases::Any)
571
- # @todo Not sure what to do with Any yet
572
- 'BasicObject'
573
- elsif type.is_a?(RBS::Types::Bases::Bool)
574
- 'Boolean'
575
- elsif type.is_a?(RBS::Types::Tuple)
576
- "Array(#{type.types.map { |t| other_type_to_tag(t) }.join(', ')})"
577
- elsif type.is_a?(RBS::Types::Literal)
578
- type.literal.to_s
579
- elsif type.is_a?(RBS::Types::Union)
580
- type.types.map { |t| other_type_to_tag(t) }.join(', ')
581
- elsif type.is_a?(RBS::Types::Record)
582
- # @todo Better record support
583
- 'Hash'
584
- elsif type.is_a?(RBS::Types::Bases::Nil)
585
- 'nil'
586
- elsif type.is_a?(RBS::Types::Bases::Self)
587
- 'self'
588
- elsif type.is_a?(RBS::Types::Bases::Void)
589
- 'void'
590
- elsif type.is_a?(RBS::Types::Variable)
591
- "#{Solargraph::ComplexType::GENERIC_TAG_NAME}<#{type.name}>"
592
- elsif type.is_a?(RBS::Types::ClassInstance) #&& !type.args.empty?
593
- type_tag(type.name, type.args)
594
- elsif type.is_a?(RBS::Types::Bases::Instance)
595
- 'self'
596
- elsif type.is_a?(RBS::Types::Bases::Top)
597
- # top is the most super superclass
598
- 'BasicObject'
599
- elsif type.is_a?(RBS::Types::Bases::Bottom)
600
- # bottom is used in contexts where nothing will ever return
601
- # - e.g., it could be the return type of 'exit()' or 'raise'
602
- #
603
- # @todo define a specific bottom type and use it to
604
- # determine dead code
605
- 'undefined'
606
- elsif type.is_a?(RBS::Types::Intersection)
607
- type.types.map { |member| other_type_to_tag(member) }.join(', ')
608
- elsif type.is_a?(RBS::Types::Proc)
609
- 'Proc'
610
- elsif type.is_a?(RBS::Types::Alias)
611
- # type-level alias use - e.g., 'bool' in "type bool = true | false"
612
- # @todo ensure these get resolved after processing all aliases
613
- # @todo handle recursive aliases
614
- type_tag(type.name, type.args)
615
- elsif type.is_a?(RBS::Types::Interface)
616
- # represents a mix-in module which can be considered a
617
- # subtype of a consumer of it
618
- type_tag(type.name, type.args)
619
- elsif type.is_a?(RBS::Types::ClassSingleton)
620
- # e.g., singleton(String)
621
- type_tag(type.name)
622
- else
623
- Solargraph.logger.warn "Unrecognized RBS type: #{type.class} at #{type.location}"
624
- 'undefined'
625
- end
626
- end
627
-
628
- # @param decl [RBS::AST::Declarations::Class, RBS::AST::Declarations::Module]
629
- # @param namespace [Pin::Namespace]
630
- # @return [void]
631
- def add_mixins decl, namespace
632
- decl.each_mixin do |mixin|
633
- klass = mixin.is_a?(RBS::AST::Members::Include) ? Pin::Reference::Include : Pin::Reference::Extend
634
- type = build_type(mixin.name, mixin.args)
635
- generic_values = type.all_params.map(&:to_s)
636
- pins.push klass.new(
637
- name: mixin.name.relative!.to_s,
638
- type_location: location_decl_to_pin_location(mixin.location),
639
- generic_values: generic_values,
640
- closure: namespace
641
- )
642
- end
643
- end
644
- end
645
- end
646
- end
1
+ # frozen_string_literal: true
2
+
3
+ require 'rbs'
4
+
5
+ module Solargraph
6
+ class RbsMap
7
+ # Functions for converting RBS declarations to Solargraph pins
8
+ #
9
+ class Conversions
10
+ include Logging
11
+
12
+ # A container for tracking the current context of the RBS conversion
13
+ # process, e.g., what visibility is declared for methods in the current
14
+ # scope
15
+ #
16
+ class Context
17
+ attr_reader :visibility
18
+
19
+ # @param visibility [Symbol]
20
+ def initialize visibility = :public
21
+ @visibility = visibility
22
+ end
23
+ end
24
+
25
+ # @param loader [RBS::EnvironmentLoader]
26
+ def initialize(loader:)
27
+ @loader = loader
28
+ @pins = []
29
+ load_environment_to_pins(loader)
30
+ end
31
+
32
+ # @return [RBS::EnvironmentLoader]
33
+ attr_reader :loader
34
+
35
+ # @return [Array<Pin::Base>]
36
+ attr_reader :pins
37
+
38
+ private
39
+
40
+ # @return [Hash{String => RBS::AST::Declarations::TypeAlias}]
41
+ def type_aliases
42
+ @type_aliases ||= {}
43
+ end
44
+
45
+ # @param loader [RBS::EnvironmentLoader]
46
+ # @return [void]
47
+ def load_environment_to_pins(loader)
48
+ environment = RBS::Environment.from_loader(loader).resolve_type_names
49
+ cursor = pins.length
50
+ if environment.declarations.empty?
51
+ Solargraph.logger.info "No RBS declarations found in environment for core_root #{loader.core_root.inspect}, libraries #{loader.libs} and directories #{loader.dirs}"
52
+ return
53
+ end
54
+ environment.declarations.each { |decl| convert_decl_to_pin(decl, Solargraph::Pin::ROOT_PIN) }
55
+ end
56
+
57
+ # @param decl [RBS::AST::Declarations::Base]
58
+ # @param closure [Pin::Closure]
59
+ # @return [void]
60
+ def convert_decl_to_pin decl, closure
61
+ case decl
62
+ when RBS::AST::Declarations::Class
63
+ class_decl_to_pin decl
64
+ when RBS::AST::Declarations::Interface
65
+ # STDERR.puts "Skipping interface #{decl.name.relative!}"
66
+ interface_decl_to_pin decl, closure
67
+ when RBS::AST::Declarations::TypeAlias
68
+ # @sg-ignore https://github.com/castwide/solargraph/pull/1114
69
+ type_aliases[decl.name.to_s] = decl
70
+ when RBS::AST::Declarations::Module
71
+ module_decl_to_pin decl
72
+ when RBS::AST::Declarations::Constant
73
+ constant_decl_to_pin decl
74
+ when RBS::AST::Declarations::ClassAlias
75
+ class_alias_decl_to_pin decl
76
+ when RBS::AST::Declarations::ModuleAlias
77
+ module_alias_decl_to_pin decl
78
+ when RBS::AST::Declarations::Global
79
+ global_decl_to_pin decl
80
+ else
81
+ Solargraph.logger.warn "Skipping declaration #{decl.class}"
82
+ end
83
+ end
84
+
85
+ # @param decl [RBS::AST::Declarations::Module]
86
+ # @param module_pin [Pin::Namespace]
87
+ # @return [void]
88
+ def convert_self_types_to_pins decl, module_pin
89
+ decl.self_types.each { |self_type| context = convert_self_type_to_pins(self_type, module_pin) }
90
+ end
91
+
92
+ # @param decl [RBS::AST::Declarations::Module::Self]
93
+ # @param closure [Pin::Namespace]
94
+ # @return [void]
95
+ def convert_self_type_to_pins decl, closure
96
+ type = build_type(decl.name, decl.args)
97
+ generic_values = type.all_params.map(&:to_s)
98
+ include_pin = Solargraph::Pin::Reference::Include.new(
99
+ name: decl.name.relative!.to_s,
100
+ type_location: location_decl_to_pin_location(decl.location),
101
+ generic_values: generic_values,
102
+ closure: closure,
103
+ source: :rbs
104
+ )
105
+ pins.push include_pin
106
+ end
107
+
108
+ # @param decl [RBS::AST::Declarations::Module,RBS::AST::Declarations::Class,RBS::AST::Declarations::Interface]
109
+ # @param closure [Pin::Namespace]
110
+ # @return [void]
111
+ def convert_members_to_pins decl, closure
112
+ context = Conversions::Context.new
113
+ decl.members.each { |m| context = convert_member_to_pin(m, closure, context) }
114
+ end
115
+
116
+ # @param member [RBS::AST::Members::Base,RBS::AST::Declarations::Base]
117
+ # @param closure [Pin::Namespace]
118
+ # @param context [Context]
119
+ # @return [Context]
120
+ def convert_member_to_pin member, closure, context
121
+ case member
122
+ when RBS::AST::Members::MethodDefinition
123
+ method_def_to_pin(member, closure, context)
124
+ when RBS::AST::Members::AttrReader
125
+ attr_reader_to_pin(member, closure, context)
126
+ when RBS::AST::Members::AttrWriter
127
+ attr_writer_to_pin(member, closure, context)
128
+ when RBS::AST::Members::AttrAccessor
129
+ attr_accessor_to_pin(member, closure, context)
130
+ when RBS::AST::Members::Include
131
+ include_to_pin(member, closure)
132
+ when RBS::AST::Members::Prepend
133
+ prepend_to_pin(member, closure)
134
+ when RBS::AST::Members::Extend
135
+ extend_to_pin(member, closure)
136
+ when RBS::AST::Members::Alias
137
+ alias_to_pin(member, closure)
138
+ when RBS::AST::Members::ClassInstanceVariable
139
+ civar_to_pin(member, closure)
140
+ when RBS::AST::Members::ClassVariable
141
+ cvar_to_pin(member, closure)
142
+ when RBS::AST::Members::InstanceVariable
143
+ ivar_to_pin(member, closure)
144
+ when RBS::AST::Members::Public
145
+ return Context.new(:public)
146
+ when RBS::AST::Members::Private
147
+ return Context.new(:private)
148
+ when RBS::AST::Declarations::Base
149
+ convert_decl_to_pin(member, closure)
150
+ else
151
+ Solargraph.logger.warn "Skipping member type #{member.class}"
152
+ end
153
+ context
154
+ end
155
+
156
+ # @param decl [RBS::AST::Declarations::Class]
157
+ # @return [void]
158
+ def class_decl_to_pin decl
159
+ generics = decl.type_params.map(&:name).map(&:to_s)
160
+ generic_defaults = {}
161
+ decl.type_params.each do |param|
162
+ if param.default_type
163
+ tag = other_type_to_tag param.default_type
164
+ generic_defaults[param.name.to_s] = ComplexType.parse(tag).force_rooted
165
+ end
166
+ end
167
+ class_name = decl.name.relative!.to_s
168
+ class_pin = Solargraph::Pin::Namespace.new(
169
+ type: :class,
170
+ name: class_name,
171
+ closure: Solargraph::Pin::ROOT_PIN,
172
+ comments: decl.comment&.string,
173
+ type_location: location_decl_to_pin_location(decl.location),
174
+ # @todo some type parameters in core/stdlib have default
175
+ # values; Solargraph doesn't support that yet as so these
176
+ # get treated as undefined if not specified
177
+ generics: generics,
178
+ generic_defaults: generic_defaults,
179
+ source: :rbs
180
+ )
181
+ pins.push class_pin
182
+ if decl.super_class
183
+ type = build_type(decl.super_class.name, decl.super_class.args)
184
+ generic_values = type.all_params.map(&:to_s)
185
+ superclass_name = decl.super_class.name.to_s
186
+ pins.push Solargraph::Pin::Reference::Superclass.new(
187
+ type_location: location_decl_to_pin_location(decl.super_class.location),
188
+ closure: class_pin,
189
+ generic_values: generic_values,
190
+ name: superclass_name,
191
+ source: :rbs
192
+ )
193
+ end
194
+ add_mixins decl, class_pin
195
+ convert_members_to_pins decl, class_pin
196
+ end
197
+
198
+ # @param decl [RBS::AST::Declarations::Interface]
199
+ # @param closure [Pin::Closure]
200
+ # @return [void]
201
+ def interface_decl_to_pin decl, closure
202
+ class_pin = Solargraph::Pin::Namespace.new(
203
+ type: :module,
204
+ type_location: location_decl_to_pin_location(decl.location),
205
+ name: decl.name.relative!.to_s,
206
+ closure: Solargraph::Pin::ROOT_PIN,
207
+ comments: decl.comment&.string,
208
+ generics: decl.type_params.map(&:name).map(&:to_s),
209
+ # HACK: Using :hidden to keep interfaces from appearing in
210
+ # autocompletion
211
+ visibility: :hidden,
212
+ source: :rbs
213
+ )
214
+ class_pin.docstring.add_tag(YARD::Tags::Tag.new(:abstract, '(RBS interface)'))
215
+ pins.push class_pin
216
+ convert_members_to_pins decl, class_pin
217
+ end
218
+
219
+ # @param decl [RBS::AST::Declarations::Module]
220
+ # @return [void]
221
+ def module_decl_to_pin decl
222
+ module_pin = Solargraph::Pin::Namespace.new(
223
+ type: :module,
224
+ name: decl.name.relative!.to_s,
225
+ type_location: location_decl_to_pin_location(decl.location),
226
+ closure: Solargraph::Pin::ROOT_PIN,
227
+ comments: decl.comment&.string,
228
+ generics: decl.type_params.map(&:name).map(&:to_s),
229
+ source: :rbs
230
+ )
231
+ pins.push module_pin
232
+ convert_self_types_to_pins decl, module_pin
233
+ convert_members_to_pins decl, module_pin
234
+
235
+ add_mixins decl, module_pin.closure
236
+ end
237
+
238
+ # @param name [String]
239
+ # @param tag [String]
240
+ # @param comments [String]
241
+ # @param decl [RBS::AST::Declarations::ClassAlias, RBS::AST::Declarations::Constant, RBS::AST::Declarations::ModuleAlias]
242
+ # @param base [String, nil] Optional conversion of tag to base<tag>
243
+ #
244
+ # @return [Solargraph::Pin::Constant]
245
+ def create_constant(name, tag, comments, decl, base = nil)
246
+ parts = name.split('::')
247
+ if parts.length > 1
248
+ name = parts.last
249
+ closure = pins.select { |pin| pin && pin.path == parts[0..-2].join('::') }.first
250
+ else
251
+ name = parts.first
252
+ closure = Solargraph::Pin::ROOT_PIN
253
+ end
254
+ constant_pin = Solargraph::Pin::Constant.new(
255
+ name: name,
256
+ closure: closure,
257
+ type_location: location_decl_to_pin_location(decl.location),
258
+ comments: comments,
259
+ source: :rbs
260
+ )
261
+ tag = "#{base}<#{tag}>" if base
262
+ rooted_tag = ComplexType.parse(tag).force_rooted.rooted_tags
263
+ constant_pin.docstring.add_tag(YARD::Tags::Tag.new(:return, '', rooted_tag))
264
+ constant_pin
265
+ end
266
+
267
+ # @param decl [RBS::AST::Declarations::ClassAlias]
268
+ # @return [void]
269
+ def class_alias_decl_to_pin decl
270
+ # See https://www.rubydoc.info/gems/rbs/3.4.3/RBS/AST/Declarations/ClassAlias
271
+ new_name = decl.new_name.relative!.to_s
272
+ old_name = decl.old_name.relative!.to_s
273
+
274
+ pins.push create_constant(new_name, old_name, decl.comment&.string, decl, 'Class')
275
+ end
276
+
277
+ # @param decl [RBS::AST::Declarations::ModuleAlias]
278
+ # @return [void]
279
+ def module_alias_decl_to_pin decl
280
+ # See https://www.rubydoc.info/gems/rbs/3.4.3/RBS/AST/Declarations/ModuleAlias
281
+ new_name = decl.new_name.relative!.to_s
282
+ old_name = decl.old_name.relative!.to_s
283
+
284
+ pins.push create_constant(new_name, old_name, decl.comment&.string, decl, 'Module')
285
+ end
286
+
287
+ # @param decl [RBS::AST::Declarations::Constant]
288
+ # @return [void]
289
+ def constant_decl_to_pin decl
290
+ tag = other_type_to_tag(decl.type)
291
+ pins.push create_constant(decl.name.relative!.to_s, tag, decl.comment&.string, decl)
292
+ end
293
+
294
+ # @param decl [RBS::AST::Declarations::Global]
295
+ # @return [void]
296
+ def global_decl_to_pin decl
297
+ closure = Solargraph::Pin::ROOT_PIN
298
+ name = decl.name.to_s
299
+ pin = Solargraph::Pin::GlobalVariable.new(
300
+ name: name,
301
+ closure: closure,
302
+ comments: decl.comment&.string,
303
+ type_location: location_decl_to_pin_location(decl.location),
304
+ source: :rbs
305
+ )
306
+ rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
307
+ pin.docstring.add_tag(YARD::Tags::Tag.new(:type, '', rooted_tag))
308
+ pins.push pin
309
+ end
310
+
311
+
312
+ # Visibility overrides that will allow the Solargraph project
313
+ # and plugins to pass typechecking using SOLARGRAPH_ASSERTS=on,
314
+ # so that we can detect any regressions/issues elsewhere in the
315
+ # visibility logic.
316
+ #
317
+ # These should either reflect a bug upstream in the RBS
318
+ # definitions, or include a @todo indicating what needs to be
319
+ # fixed in Solargraph to properly understand it.
320
+ #
321
+ # @todo PR these fixes upstream and list open PRs here above
322
+ # related overrides
323
+ # @todo externalize remaining overrides into yaml file, then
324
+ # allow that to be extended via .solargraph.yml
325
+ # @type [Hash{Array(String, Symbol, String) => Symbol}
326
+ VISIBILITY_OVERRIDE = {
327
+ ["Rails::Engine", :instance, "run_tasks_blocks"] => :protected,
328
+ # Should have been marked as both instance and class method in module -e.g., 'module_function'
329
+ ["Kernel", :instance, "pretty_inspect"] => :private,
330
+ # marked incorrectly in RBS
331
+ ["WEBrick::HTTPUtils::FormData", :instance, "next_data"] => :protected,
332
+ ["Rails::Command", :class, "command_type"] => :private,
333
+ ["Rails::Command", :class, "lookup_paths"] => :private,
334
+ ["Rails::Command", :class, "file_lookup_paths"] => :private,
335
+ ["Rails::Railtie", :instance, "run_console_blocks"] => :protected,
336
+ ["Rails::Railtie", :instance, "run_generators_blocks"] => :protected,
337
+ ["Rails::Railtie", :instance, "run_runner_blocks"] => :protected,
338
+ ["Rails::Railtie", :instance, "run_tasks_blocks"] => :protected,
339
+ ["ActionController::Base", :instance, "_protected_ivars"] => :private,
340
+ ["ActionView::Template", :instance, "method_name"] => :public,
341
+ ["Module", :instance, "ruby2_keywords"] => :private,
342
+ ["Nokogiri::XML::Node", :instance, "coerce"] => :protected,
343
+ ["Nokogiri::XML::Document", :class, "empty_doc?"] => :private,
344
+ ["Nokogiri::Decorators::Slop", :instance, "respond_to_missing?"] => :public,
345
+ ["RuboCop::Cop::RangeHelp", :instance, "source_range"] => :private,
346
+ ["AST::Node", :instance, "original_dup"] => :private,
347
+ ["Rainbow::Presenter", :instance, "wrap_with_sgr"] => :private,
348
+ }
349
+
350
+ # @param decl [RBS::AST::Members::MethodDefinition, RBS::AST::Members::AttrReader, RBS::AST::Members::AttrAccessor]
351
+ # @param closure [Pin::Closure]
352
+ # @param context [Context]
353
+ # @param scope [Symbol] :instance or :class
354
+ # @param name [String] The name of the method
355
+ # @sg-ignore
356
+ # @return [Symbol]
357
+ def calculate_method_visibility(decl, context, closure, scope, name)
358
+ override_key = [closure.path, scope, name]
359
+ visibility = VISIBILITY_OVERRIDE[override_key]
360
+ simple_override_key = [closure.path, scope]
361
+ visibility ||= VISIBILITY_OVERRIDE[simple_override_key]
362
+ visibility ||= :private if closure.path == 'Kernel' && Kernel.private_instance_methods(false).include?(decl.name)
363
+ if decl.kind == :singleton_instance
364
+ # this is a 'module function'
365
+ visibility ||= :private
366
+ end
367
+ visibility ||= decl.visibility
368
+ visibility ||= context.visibility
369
+ visibility ||= :public
370
+ visibility
371
+ end
372
+
373
+ # @param decl [RBS::AST::Members::MethodDefinition]
374
+ # @param closure [Pin::Closure]
375
+ # @param context [Context]
376
+ # @return [void]
377
+ def method_def_to_pin decl, closure, context
378
+ # there may be edge cases here around different signatures
379
+ # having different type params / orders - we may need to match
380
+ # this data model and have generics live in signatures to
381
+ # handle those correctly
382
+ generics = decl.overloads.map(&:method_type).flat_map(&:type_params).map(&:name).map(&:to_s).uniq
383
+
384
+ if decl.instance?
385
+ name = decl.name.to_s
386
+ final_scope = :instance
387
+ visibility = calculate_method_visibility(decl, context, closure, final_scope, name)
388
+ pin = Solargraph::Pin::Method.new(
389
+ name: name,
390
+ closure: closure,
391
+ type_location: location_decl_to_pin_location(decl.location),
392
+ comments: decl.comment&.string,
393
+ scope: final_scope,
394
+ signatures: [],
395
+ generics: generics,
396
+ visibility: visibility,
397
+ source: :rbs
398
+ )
399
+ pin.signatures.concat method_def_to_sigs(decl, pin)
400
+ pins.push pin
401
+ if pin.name == 'initialize'
402
+ pin.instance_variable_set(:@visibility, :private)
403
+ pin.instance_variable_set(:@return_type, ComplexType::VOID)
404
+ end
405
+ end
406
+ if decl.singleton?
407
+ final_scope = :class
408
+ name = decl.name.to_s
409
+ visibility = calculate_method_visibility(decl, context, closure, final_scope, name)
410
+ pin = Solargraph::Pin::Method.new(
411
+ name: name,
412
+ closure: closure,
413
+ comments: decl.comment&.string,
414
+ type_location: location_decl_to_pin_location(decl.location),
415
+ visibility: visibility,
416
+ scope: final_scope,
417
+ signatures: [],
418
+ generics: generics,
419
+ source: :rbs
420
+ )
421
+ pin.signatures.concat method_def_to_sigs(decl, pin)
422
+ pins.push pin
423
+ end
424
+ end
425
+
426
+ # @param decl [RBS::AST::Members::MethodDefinition]
427
+ # @param pin [Pin::Method]
428
+ # @return [void]
429
+ def method_def_to_sigs decl, pin
430
+ # @param overload [RBS::AST::Members::MethodDefinition::Overload]
431
+ decl.overloads.map do |overload|
432
+ type_location = location_decl_to_pin_location(overload.method_type.location)
433
+ generics = overload.method_type.type_params.map(&:name).map(&:to_s)
434
+ signature_parameters, signature_return_type = parts_of_function(overload.method_type, pin)
435
+ block = if overload.method_type.block
436
+ block_parameters, block_return_type = parts_of_function(overload.method_type.block, pin)
437
+ Pin::Signature.new(generics: generics, parameters: block_parameters, return_type: block_return_type, source: :rbs,
438
+ type_location: type_location, closure: pin)
439
+ end
440
+ Pin::Signature.new(generics: generics, parameters: signature_parameters, return_type: signature_return_type, block: block, source: :rbs,
441
+ type_location: type_location, closure: pin)
442
+ end
443
+ end
444
+
445
+ # @param location [RBS::Location, nil]
446
+ # @return [Solargraph::Location, nil]
447
+ def location_decl_to_pin_location(location)
448
+ return nil if location&.name.nil?
449
+
450
+ start_pos = Position.new(location.start_line - 1, location.start_column)
451
+ end_pos = Position.new(location.end_line - 1, location.end_column)
452
+ range = Range.new(start_pos, end_pos)
453
+ Location.new(location.name.to_s, range)
454
+ end
455
+
456
+ # @param type [RBS::MethodType,RBS::Types::Block]
457
+ # @param pin [Pin::Method]
458
+ # @return [Array(Array<Pin::Parameter>, ComplexType)]
459
+ def parts_of_function type, pin
460
+ type_location = pin.type_location
461
+ if defined?(RBS::Types::UntypedFunction) && type.type.is_a?(RBS::Types::UntypedFunction)
462
+ return [
463
+ [Solargraph::Pin::Parameter.new(decl: :restarg, name: 'arg', closure: pin, source: :rbs, type_location: type_location)],
464
+ ComplexType.try_parse(method_type_to_tag(type)).force_rooted
465
+ ]
466
+ end
467
+
468
+ parameters = []
469
+ arg_num = -1
470
+ type.type.required_positionals.each do |param|
471
+ # @sg-ignore RBS generic type understanding issue
472
+ name = param.name ? param.name.to_s : "arg_#{arg_num += 1}"
473
+ # @sg-ignore RBS generic type understanding issue
474
+ parameters.push Solargraph::Pin::Parameter.new(decl: :arg, name: name, closure: pin, return_type: ComplexType.try_parse(other_type_to_tag(param.type)).force_rooted, source: :rbs, type_location: type_location)
475
+ end
476
+ type.type.optional_positionals.each do |param|
477
+ # @sg-ignore RBS generic type understanding issue
478
+ name = param.name ? param.name.to_s : "arg_#{arg_num += 1}"
479
+ parameters.push Solargraph::Pin::Parameter.new(decl: :optarg, name: name, closure: pin,
480
+ # @sg-ignore RBS generic type understanding issue
481
+ return_type: ComplexType.try_parse(other_type_to_tag(param.type)).force_rooted,
482
+ type_location: type_location,
483
+ source: :rbs)
484
+ end
485
+ if type.type.rest_positionals
486
+ name = type.type.rest_positionals.name ? type.type.rest_positionals.name.to_s : "arg_#{arg_num += 1}"
487
+ inner_rest_positional_type =
488
+ ComplexType.try_parse(other_type_to_tag(type.type.rest_positionals.type))
489
+ rest_positional_type = ComplexType::UniqueType.new('Array',
490
+ [],
491
+ [inner_rest_positional_type],
492
+ rooted: true, parameters_type: :list)
493
+ parameters.push Solargraph::Pin::Parameter.new(decl: :restarg, name: name, closure: pin,
494
+ source: :rbs, type_location: type_location,
495
+ return_type: rest_positional_type,)
496
+ end
497
+ type.type.trailing_positionals.each do |param|
498
+ # @sg-ignore RBS generic type understanding issue
499
+ name = param.name ? param.name.to_s : "arg_#{arg_num += 1}"
500
+ parameters.push Solargraph::Pin::Parameter.new(decl: :arg, name: name, closure: pin, source: :rbs, type_location: type_location)
501
+ end
502
+ type.type.required_keywords.each do |orig, param|
503
+ # @sg-ignore RBS generic type understanding issue
504
+ name = orig ? orig.to_s : "arg_#{arg_num += 1}"
505
+ parameters.push Solargraph::Pin::Parameter.new(decl: :kwarg, name: name, closure: pin,
506
+ # @sg-ignore RBS generic type understanding issue
507
+ return_type: ComplexType.try_parse(other_type_to_tag(param.type)).force_rooted,
508
+ source: :rbs, type_location: type_location)
509
+ end
510
+ type.type.optional_keywords.each do |orig, param|
511
+ # @sg-ignore RBS generic type understanding issue
512
+ name = orig ? orig.to_s : "arg_#{arg_num += 1}"
513
+ parameters.push Solargraph::Pin::Parameter.new(decl: :kwoptarg, name: name, closure: pin,
514
+ # @sg-ignore RBS generic type understanding issue
515
+ return_type: ComplexType.try_parse(other_type_to_tag(param.type)).force_rooted,
516
+ type_location: type_location,
517
+ source: :rbs)
518
+ end
519
+ if type.type.rest_keywords
520
+ name = type.type.rest_keywords.name ? type.type.rest_keywords.name.to_s : "arg_#{arg_num += 1}"
521
+ parameters.push Solargraph::Pin::Parameter.new(decl: :kwrestarg, name: type.type.rest_keywords.name.to_s, closure: pin,
522
+ source: :rbs, type_location: type_location)
523
+ end
524
+
525
+ rooted_tag = method_type_to_tag(type)
526
+ return_type = ComplexType.try_parse(rooted_tag).force_rooted
527
+ [parameters, return_type]
528
+ end
529
+
530
+ # @param decl [RBS::AST::Members::AttrReader,RBS::AST::Members::AttrAccessor]
531
+ # @param closure [Pin::Namespace]
532
+ # @param context [Context]
533
+ # @return [void]
534
+ def attr_reader_to_pin(decl, closure, context)
535
+ name = decl.name.to_s
536
+ final_scope = decl.kind == :instance ? :instance : :class
537
+ visibility = calculate_method_visibility(decl, context, closure, final_scope, name)
538
+ pin = Solargraph::Pin::Method.new(
539
+ name: name,
540
+ type_location: location_decl_to_pin_location(decl.location),
541
+ closure: closure,
542
+ comments: decl.comment&.string,
543
+ scope: final_scope,
544
+ attribute: true,
545
+ visibility: visibility,
546
+ source: :rbs
547
+ )
548
+ rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
549
+ pin.docstring.add_tag(YARD::Tags::Tag.new(:return, '', rooted_tag))
550
+ logger.debug { "Conversions#attr_reader_to_pin(name=#{name.inspect}, visibility=#{visibility.inspect}) => #{pin.inspect}" }
551
+ pins.push pin
552
+ end
553
+
554
+ # @param decl [RBS::AST::Members::AttrWriter, RBS::AST::Members::AttrAccessor]
555
+ # @param closure [Pin::Namespace]
556
+ # @param context [Context]
557
+ # @return [void]
558
+ def attr_writer_to_pin(decl, closure, context)
559
+ final_scope = decl.kind == :instance ? :instance : :class
560
+ name = "#{decl.name.to_s}="
561
+ visibility = calculate_method_visibility(decl, context, closure, final_scope, name)
562
+ type_location = location_decl_to_pin_location(decl.location)
563
+ pin = Solargraph::Pin::Method.new(
564
+ name: name,
565
+ type_location: type_location,
566
+ closure: closure,
567
+ parameters: [],
568
+ comments: decl.comment&.string,
569
+ scope: final_scope,
570
+ attribute: true,
571
+ visibility: visibility,
572
+ source: :rbs
573
+ )
574
+ pin.parameters <<
575
+ Solargraph::Pin::Parameter.new(
576
+ name: 'value',
577
+ return_type: ComplexType.try_parse(other_type_to_tag(decl.type)).force_rooted,
578
+ source: :rbs,
579
+ closure: pin,
580
+ type_location: type_location
581
+ )
582
+ rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
583
+ pin.docstring.add_tag(YARD::Tags::Tag.new(:return, '', rooted_tag))
584
+ pins.push pin
585
+ end
586
+
587
+ # @param decl [RBS::AST::Members::AttrAccessor]
588
+ # @param closure [Pin::Namespace]
589
+ # @param context [Context]
590
+ # @return [void]
591
+ def attr_accessor_to_pin(decl, closure, context)
592
+ attr_reader_to_pin(decl, closure, context)
593
+ attr_writer_to_pin(decl, closure, context)
594
+ end
595
+
596
+ # @param decl [RBS::AST::Members::InstanceVariable]
597
+ # @param closure [Pin::Namespace]
598
+ # @return [void]
599
+ def ivar_to_pin(decl, closure)
600
+ pin = Solargraph::Pin::InstanceVariable.new(
601
+ name: decl.name.to_s,
602
+ closure: closure,
603
+ type_location: location_decl_to_pin_location(decl.location),
604
+ comments: decl.comment&.string,
605
+ source: :rbs
606
+ )
607
+ rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
608
+ pin.docstring.add_tag(YARD::Tags::Tag.new(:type, '', rooted_tag))
609
+ pins.push pin
610
+ end
611
+
612
+ # @param decl [RBS::AST::Members::ClassVariable]
613
+ # @param closure [Pin::Namespace]
614
+ # @return [void]
615
+ def cvar_to_pin(decl, closure)
616
+ name = decl.name.to_s
617
+ pin = Solargraph::Pin::ClassVariable.new(
618
+ name: name,
619
+ closure: closure,
620
+ comments: decl.comment&.string,
621
+ type_location: location_decl_to_pin_location(decl.location),
622
+ source: :rbs
623
+ )
624
+ rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
625
+ pin.docstring.add_tag(YARD::Tags::Tag.new(:type, '', rooted_tag))
626
+ pins.push pin
627
+ end
628
+
629
+ # @param decl [RBS::AST::Members::ClassInstanceVariable]
630
+ # @param closure [Pin::Namespace]
631
+ # @return [void]
632
+ def civar_to_pin(decl, closure)
633
+ name = decl.name.to_s
634
+ pin = Solargraph::Pin::InstanceVariable.new(
635
+ name: name,
636
+ closure: closure,
637
+ comments: decl.comment&.string,
638
+ type_location: location_decl_to_pin_location(decl.location),
639
+ source: :rbs
640
+ )
641
+ rooted_tag = ComplexType.parse(other_type_to_tag(decl.type)).force_rooted.rooted_tags
642
+ pin.docstring.add_tag(YARD::Tags::Tag.new(:type, '', rooted_tag))
643
+ pins.push pin
644
+ end
645
+
646
+ # @param decl [RBS::AST::Members::Include]
647
+ # @param closure [Pin::Namespace]
648
+ # @return [void]
649
+ def include_to_pin decl, closure
650
+ type = build_type(decl.name, decl.args)
651
+ generic_values = type.all_params.map(&:to_s)
652
+ pins.push Solargraph::Pin::Reference::Include.new(
653
+ name: decl.name.relative!.to_s,
654
+ type_location: location_decl_to_pin_location(decl.location),
655
+ generic_values: generic_values,
656
+ closure: closure,
657
+ source: :rbs
658
+ )
659
+ end
660
+
661
+ # @param decl [RBS::AST::Members::Prepend]
662
+ # @param closure [Pin::Namespace]
663
+ # @return [void]
664
+ def prepend_to_pin decl, closure
665
+ pins.push Solargraph::Pin::Reference::Prepend.new(
666
+ name: decl.name.relative!.to_s,
667
+ type_location: location_decl_to_pin_location(decl.location),
668
+ closure: closure,
669
+ source: :rbs
670
+ )
671
+ end
672
+
673
+ # @param decl [RBS::AST::Members::Extend]
674
+ # @param closure [Pin::Namespace]
675
+ # @return [void]
676
+ def extend_to_pin decl, closure
677
+ pins.push Solargraph::Pin::Reference::Extend.new(
678
+ name: decl.name.relative!.to_s,
679
+ type_location: location_decl_to_pin_location(decl.location),
680
+ closure: closure,
681
+ source: :rbs
682
+ )
683
+ end
684
+
685
+ # @param decl [RBS::AST::Members::Alias]
686
+ # @param closure [Pin::Namespace]
687
+ # @return [void]
688
+ def alias_to_pin decl, closure
689
+ final_scope = decl.singleton? ? :class : :instance
690
+ pins.push Solargraph::Pin::MethodAlias.new(
691
+ name: decl.new_name.to_s,
692
+ type_location: location_decl_to_pin_location(decl.location),
693
+ original: decl.old_name.to_s,
694
+ closure: closure,
695
+ scope: final_scope,
696
+ source: :rbs,
697
+ )
698
+ end
699
+
700
+ RBS_TO_YARD_TYPE = {
701
+ 'bool' => 'Boolean',
702
+ 'string' => 'String',
703
+ 'int' => 'Integer',
704
+ 'untyped' => '',
705
+ 'NilClass' => 'nil'
706
+ }
707
+
708
+ # @param type [RBS::MethodType]
709
+ # @return [String]
710
+ def method_type_to_tag type
711
+ if type_aliases.key?(type.type.return_type.to_s)
712
+ other_type_to_tag(type_aliases[type.type.return_type.to_s].type)
713
+ else
714
+ other_type_to_tag type.type.return_type
715
+ end
716
+ end
717
+
718
+ # @param type_name [RBS::TypeName]
719
+ # @param type_args [Enumerable<RBS::Types::Bases::Base>]
720
+ # @return [ComplexType::UniqueType]
721
+ def build_type(type_name, type_args = [])
722
+ base = RBS_TO_YARD_TYPE[type_name.relative!.to_s] || type_name.relative!.to_s
723
+ params = type_args.map { |a| other_type_to_tag(a) }.map do |t|
724
+ ComplexType.try_parse(t).force_rooted
725
+ end
726
+ if base == 'Hash' && params.length == 2
727
+ ComplexType::UniqueType.new(base, [params.first], [params.last], rooted: true, parameters_type: :hash)
728
+ else
729
+ ComplexType::UniqueType.new(base, [], params.reject(&:undefined?), rooted: true, parameters_type: :list)
730
+ end
731
+ end
732
+
733
+ # @param type_name [RBS::TypeName]
734
+ # @param type_args [Enumerable<RBS::Types::Bases::Base>]
735
+ # @return [String]
736
+ def type_tag(type_name, type_args = [])
737
+ build_type(type_name, type_args).tags
738
+ end
739
+
740
+ # @param type [RBS::Types::Bases::Base]
741
+ # @return [String]
742
+ def other_type_to_tag type
743
+ if type.is_a?(RBS::Types::Optional)
744
+ "#{other_type_to_tag(type.type)}, nil"
745
+ elsif type.is_a?(RBS::Types::Bases::Any)
746
+ 'undefined'
747
+ elsif type.is_a?(RBS::Types::Bases::Bool)
748
+ 'Boolean'
749
+ elsif type.is_a?(RBS::Types::Tuple)
750
+ "Array(#{type.types.map { |t| other_type_to_tag(t) }.join(', ')})"
751
+ elsif type.is_a?(RBS::Types::Literal)
752
+ type.literal.inspect
753
+ elsif type.is_a?(RBS::Types::Union)
754
+ type.types.map { |t| other_type_to_tag(t) }.join(', ')
755
+ elsif type.is_a?(RBS::Types::Record)
756
+ # @todo Better record support
757
+ 'Hash'
758
+ elsif type.is_a?(RBS::Types::Bases::Nil)
759
+ 'nil'
760
+ elsif type.is_a?(RBS::Types::Bases::Self)
761
+ 'self'
762
+ elsif type.is_a?(RBS::Types::Bases::Void)
763
+ 'void'
764
+ elsif type.is_a?(RBS::Types::Variable)
765
+ "#{Solargraph::ComplexType::GENERIC_TAG_NAME}<#{type.name}>"
766
+ elsif type.is_a?(RBS::Types::ClassInstance) #&& !type.args.empty?
767
+ type_tag(type.name, type.args)
768
+ elsif type.is_a?(RBS::Types::Bases::Instance)
769
+ 'self'
770
+ elsif type.is_a?(RBS::Types::Bases::Top)
771
+ # top is the most super superclass
772
+ 'BasicObject'
773
+ elsif type.is_a?(RBS::Types::Bases::Bottom)
774
+ # bottom is used in contexts where nothing will ever return
775
+ # - e.g., it could be the return type of 'exit()' or 'raise'
776
+ #
777
+ # @todo define a specific bottom type and use it to
778
+ # determine dead code
779
+ 'undefined'
780
+ elsif type.is_a?(RBS::Types::Intersection)
781
+ type.types.map { |member| other_type_to_tag(member) }.join(', ')
782
+ elsif type.is_a?(RBS::Types::Proc)
783
+ 'Proc'
784
+ elsif type.is_a?(RBS::Types::Alias)
785
+ # type-level alias use - e.g., 'bool' in "type bool = true | false"
786
+ # @todo ensure these get resolved after processing all aliases
787
+ # @todo handle recursive aliases
788
+ type_tag(type.name, type.args)
789
+ elsif type.is_a?(RBS::Types::Interface)
790
+ # represents a mix-in module which can be considered a
791
+ # subtype of a consumer of it
792
+ type_tag(type.name, type.args)
793
+ elsif type.is_a?(RBS::Types::ClassSingleton)
794
+ # e.g., singleton(String)
795
+ type_tag(type.name)
796
+ else
797
+ Solargraph.logger.warn "Unrecognized RBS type: #{type.class} at #{type.location}"
798
+ 'undefined'
799
+ end
800
+ end
801
+
802
+ # @param decl [RBS::AST::Declarations::Class, RBS::AST::Declarations::Module]
803
+ # @param namespace [Pin::Namespace]
804
+ # @return [void]
805
+ def add_mixins decl, namespace
806
+ # @param mixin [RBS::AST::Members::Include, RBS::AST::Members::Members::Extend, RBS::AST::Members::Members::Prepend]
807
+ decl.each_mixin do |mixin|
808
+ # @todo are we handling prepend correctly?
809
+ klass = mixin.is_a?(RBS::AST::Members::Include) ? Pin::Reference::Include : Pin::Reference::Extend
810
+ type = build_type(mixin.name, mixin.args)
811
+ generic_values = type.all_params.map(&:to_s)
812
+ pins.push klass.new(
813
+ name: mixin.name.relative!.to_s,
814
+ type_location: location_decl_to_pin_location(mixin.location),
815
+ generic_values: generic_values,
816
+ closure: namespace,
817
+ source: :rbs
818
+ )
819
+ end
820
+ end
821
+ end
822
+ end
823
+ end