Let def syntaxes have comments

Author: soutaro

lib/rbs/ast/ruby/members.rb

@@ -183,12 +183,14 @@ class DefMember < Base
           attr_reader :name
           attr_reader :node
           attr_reader :method_type
+          attr_reader :leading_comment
 
-          def initialize(buffer, name, node, method_type)
+          def initialize(buffer, name, node, method_type, leading_comment)
             super(buffer)
             @name = name
             @node = node
             @method_type = method_type
+            @leading_comment = leading_comment
           end
 
           def location

lib/rbs/definition.rb

@@ -65,8 +65,7 @@ def comment
           case member
           when AST::Members::Base
             member.comment
-          when AST::Ruby::Members::AttributeMember
-            # For Ruby attr_* members, extract comment from leading_comment
+          when AST::Ruby::Members::Base
             if member.leading_comment
               lines = [] #: Array[String]
 
@@ -77,19 +76,15 @@ def comment
                 end
               end
 
-              AST::Comment.new(
-                string: lines.join("\n"),
-                location: member.leading_comment.location
-              )
-            else
-              nil
+              string = lines.join("\n")
+              
+              unless string.strip.empty?
+                AST::Comment.new(
+                  string: string,
+                  location: member.leading_comment.location
+                )
+              end
             end
-          when AST::Ruby::Members::DefMember
-            # DefMember doesn't currently store leading comments directly
-            # The comments are processed for type annotations only
-            nil
-          when AST::Ruby::Members::Base
-            nil
           end
         end
 

lib/rbs/environment.rb

@@ -722,7 +722,8 @@ def resolve_ruby_member(resolver, member, context:)
           member.buffer,
           member.name,
           member.node,
-          member.method_type.map_type_name {|name, _, _| absolute_type_name(resolver, nil, name, context: context) }
+          member.method_type.map_type_name {|name, _, _| absolute_type_name(resolver, nil, name, context: context) },
+          member.leading_comment
         )
       when AST::Ruby::Members::IncludeMember
         resolved_annotation = member.annotation&.map_type_name {|name, _, _| absolute_type_name(resolver, nil, name, context: context) }

lib/rbs/inline_parser.rb

@@ -157,7 +157,7 @@ def visit_def_node(node)
           method_type, leading_unuseds, trailing_unused = AST::Ruby::Members::MethodTypeAnnotation.build(leading_block, trailing_block, [])
           report_unused_annotation(trailing_unused, *leading_unuseds)
 
-          defn = AST::Ruby::Members::DefMember.new(buffer, node.name, node, method_type)
+          defn = AST::Ruby::Members::DefMember.new(buffer, node.name, node, method_type, leading_block)
           current.members << defn
 
           # Skip other comments in `def` node

sig/ast/ruby/members.rbs

@@ -57,8 +57,9 @@ module RBS
           attr_reader name: Symbol
           attr_reader node: Prism::DefNode
           attr_reader method_type: MethodTypeAnnotation
+          attr_reader leading_comment: CommentBlock?
 
-          def initialize: (Buffer, Symbol name, Prism::DefNode node, MethodTypeAnnotation) -> void
+          def initialize: (Buffer, Symbol name, Prism::DefNode node, MethodTypeAnnotation, CommentBlock? leading_comment) -> void
 
           def location: () -> Location
 

test/rbs/definition_builder_test.rb

@@ -3439,4 +3439,46 @@ class MultipleAttributeTest
       end
     end
   end
+
+  def test_ruby_def_members_docs
+    SignatureManager.new do |manager|
+      manager.add_ruby_file("a.rb", <<~RUBY)
+        class MultipleAttributeTest
+          # This is a document for foo method
+          #
+          # @rbs return: String?
+          def foo = nil
+
+          # Line 1
+          #
+          # Line 2
+          # @rbs () -> Integer
+          #
+          # Line 3
+          def bar = 123
+
+          # @rbs return: untyped
+          def baz = nil
+        end
+      RUBY
+
+      manager.build do |env|
+        builder = DefinitionBuilder.new(env: env)
+
+        builder.build_instance(type_name("::MultipleAttributeTest")).tap do |definition|
+          definition.methods[:foo].tap do |method|
+            assert_equal ["This is a document for foo method\n"], method.comments.map(&:string)
+          end
+
+          definition.methods[:bar].tap do |method|
+            assert_equal ["Line 1\n\nLine 2\n\nLine 3"], method.comments.map(&:string)
+          end
+
+          definition.methods[:baz].tap do |method|
+            assert_equal [], method.comments
+          end
+        end
+      end
+    end
+  end
 end