Add Scaffold::RBI

Author: soutaro

lib/ruby/signature.rb

@@ -3,6 +3,7 @@
 require "set"
 require "json"
 require "pathname"
+require "pp"
 
 require "ruby/signature/errors"
 require "ruby/signature/buffer"
@@ -23,5 +24,6 @@
 require "ruby/signature/constant"
 require "ruby/signature/constant_table"
 require "ruby/signature/writer"
+require "ruby/signature/scaffold/rbi"
 
 require "ruby/signature/parser"

lib/ruby/signature/builtin_names.rb

@@ -41,6 +41,9 @@ def self.define(name, namespace: Namespace.root)
       Enumerable = Name.define(:Enumerable)
       Class = Name.define(:Class)
       Module = Name.define(:Module)
+      Array = Name.define(:Array)
+      Hash = Name.define(:Hash)
+      Range = Name.define(:Range)
     end
   end
 end

lib/ruby/signature/cli.rb

@@ -37,7 +37,7 @@ def initialize(stdout:, stderr:)
         @stderr = stderr
       end
 
-      COMMANDS = [:ast, :list, :ancestors, :methods, :method, :validate, :constant, :paths, :version]
+      COMMANDS = [:ast, :list, :ancestors, :methods, :method, :validate, :constant, :paths, :scaffold, :version]
 
       def library_parse(opts, options:)
         opts.on("-r LIBRARY") do |lib|
@@ -392,6 +392,22 @@ def run_paths(args, options)
         end
       end
 
+      def run_scaffold(args, options)
+        format = args.shift
+
+        parser = case format
+                 when "rbi"
+                   Scaffold::RBI.new()
+                 end
+
+        args.each do |file|
+          parser.parse Pathname(file).read, filename: file
+        end
+
+        writer = Writer.new(out: stdout)
+        writer.write parser.decls
+      end
+
       def parse_type_name(string)
         Namespace.parse(string).yield_self do |namespace|
           last = namespace.path.last

lib/ruby/signature/method_type.rb

@@ -67,6 +67,15 @@ def sub(s)
         end
       end
 
+      def update(type_params: self.type_params, type: self.type, block: self.block, location: self.location)
+        self.class.new(
+          type_params: type_params,
+          type: type,
+          block: block,
+          location: location
+        )
+      end
+
       def free_variables(set = Set.new)
         type.free_variables(set)
         block&.type&.free_variables(set)

lib/ruby/signature/namespace.rb

@@ -45,6 +45,10 @@ def absolute!
         self.class.new(path: path, absolute: true)
       end
 
+      def relative!
+        self.class.new(path: path, absolute: false)
+      end
+
       def empty?
         path.empty?
       end