Add method definition AST

Author: soutaro

lib/rbs.rb

@@ -28,6 +28,7 @@
 require "rbs/ast/ruby/helpers/constant_helper"
 require "rbs/ast/ruby/helpers/location_helper"
 require "rbs/ast/ruby/declarations"
+require "rbs/ast/ruby/members"
 require "rbs/source"
 require "rbs/inline_parser"
 require "rbs/environment"

lib/rbs/ast/ruby/members.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module RBS
+  module AST
+    module Ruby
+      module Members
+        class Base
+          attr_reader :buffer
+
+          def initialize(buffer)
+            @buffer = buffer
+          end
+
+          include Helpers::LocationHelper
+        end
+
+        class DefMember < Base
+          Overload = AST::Members::MethodDefinition::Overload
+
+          attr_reader :name
+          attr_reader :node
+
+          def initialize(buffer, name, node)
+            super(buffer)
+            @name = name
+            @node = node
+          end
+
+          def location
+            rbs_location(node.location)
+          end
+
+          def overloads
+            method_type = MethodType.new(
+              type_params: [],
+              type: Types::UntypedFunction.new(return_type: Types::Bases::Any.new(location: nil)),
+              block: nil,
+              location: nil
+            )
+            
+            [
+              Overload.new(method_type: method_type, annotations: [])
+            ]
+          end
+        end
+      end
+    end
+  end
+end

lib/rbs/environment.rb

@@ -672,6 +672,8 @@ def resolve_ruby_decl(resolver, decl, context:, prefix:)
             case member
             when AST::Ruby::Declarations::Base
               resolved.members << resolve_ruby_decl(resolver, member, context: inner_context, prefix: inner_prefix)
+            when AST::Ruby::Members::Base
+              resolved.members << member
             else
               raise "Unknown member type: #{member.class}"
             end

sig/ast/ruby/declarations.rbs

@@ -14,7 +14,7 @@ module RBS
         end
 
         class ClassDecl < Base
-          type member = t
+          type member = t | Members::t
 
           attr_reader class_name: TypeName
 
@@ -35,7 +35,7 @@ module RBS
         end
 
         class ModuleDecl < Base
-          type member = t
+          type member = t | Members::t
 
           attr_reader module_name: TypeName
 

sig/ast/ruby/members.rbs

@@ -0,0 +1,30 @@
+module RBS
+  module AST
+    module Ruby
+      module Members
+        class Base
+          attr_reader buffer: Buffer
+
+          def initialize: (Buffer) -> void
+
+          include Helpers::LocationHelper
+        end
+
+        type t = DefMember
+
+        class DefMember < Base
+          class Overload = AST::Members::MethodDefinition::Overload
+
+          attr_reader name: Symbol
+          attr_reader node: Prism::DefNode
+
+          def initialize: (Buffer, Symbol name, Prism::DefNode node) -> void
+
+          def location: () -> Location
+
+          def overloads: () -> Array[Overload]
+        end
+      end
+    end
+  end
+end