First iteration of rbs command

Author: castwide

lib/solargraph/api_map.rb

@@ -237,7 +237,7 @@ def get_namespace_pins namespace, context
     def qualify tag, context_tag = ''
       return tag if ['self', nil].include?(tag)
       context_type = ComplexType.parse(context_tag)
-      type = ComplexType.parse(tag)
+      type = ComplexType.try_parse(tag)
       fqns = qualify_namespace(type.rooted_namespace, context_type.rooted_namespace)
       return nil if fqns.nil?
       fqns + type.substring

lib/solargraph/doc_map.rb

@@ -87,6 +87,7 @@ def resolve_path_to_gemspec path
       gemspec = Gem::Specification.find_by_path(path)
       if gemspec.nil?
         gem_name_guess = path.split('/').first
+        return nil if gem_name_guess.to_s.empty?
         begin
           # this can happen when the gem is included via a local path in
           # a Gemfile; Gem doesn't try to index the paths in that case.

lib/solargraph/shell.rb

@@ -2,10 +2,12 @@
 
 require 'thor'
 require 'yard'
+require 'sord'
 
 module Solargraph
   class Shell < Thor
     include Solargraph::ServerMethods
+    include ApiMap::SourceToYard
 
     # Tell Thor to ensure the process exits with status 1 if any error happens.
     def self.exit_on_failure?
@@ -263,6 +265,29 @@ def gems *names
       end
     end
 
+    desc 'rbs', 'Generate RBS definitions'
+    option :inference, type: :boolean, desc: 'Enhance definitions with type inference', default: true
+    def rbs
+      api_map = Solargraph::ApiMap.load('.')
+      pins = api_map.source_maps.flat_map(&:pins)
+      store = Solargraph::ApiMap::Store.new(pins)
+      if options[:inference]
+        store.method_pins.each do |pin|
+          if pin.return_type.undefined?
+            type = pin.typify(api_map)
+            type = pin.probe(api_map) if type.undefined?
+            pin.docstring.add_tag YARD::Tags::Tag.new('return', nil, type.items.map(&:to_s))
+            pin.instance_variable_set(:@return_type, type)
+          end
+        end
+      end
+      rake_yard(store)
+      YARD::Registry.save(false, '.yardoc')
+      YARD::Registry.load('.yardoc')
+      FileUtils.mkdir_p('sig')
+      `sord gen sig/solargraph.rbs --rbs --no-regenerate`
+    end
+
     private
 
     # @param pin [Solargraph::Pin::Base]

solargraph.gemspec

@@ -37,6 +37,7 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency 'rbs', '~> 3.3'
   s.add_runtime_dependency 'reverse_markdown', '>= 2.0', '< 4'
   s.add_runtime_dependency 'rubocop', '~> 1.38'
+  s.add_runtime_dependency 'sord', '~> 7.0'
   s.add_runtime_dependency 'thor', '~> 1.0'
   s.add_runtime_dependency 'tilt', '~> 2.0'
   s.add_runtime_dependency 'yard', '~> 0.9', '>= 0.9.24'