Refactor documents comment translation

Author: soutaro

lib/rbs/ast/ruby/comment_block.rb

@@ -222,6 +222,23 @@ def leading_annotation?(index)
 
           false
         end
+
+        def as_comment
+          lines = [] #: Array[String]
+
+          each_paragraph([]) do |paragraph|
+            case paragraph
+            when Location
+              lines << paragraph.local_source
+            end
+          end
+
+          string = lines.join("\n")
+
+          unless string.strip.empty?
+            AST::Comment.new(string: string, location: location)
+          end
+        end
       end
     end
   end

lib/rbs/ast/ruby/declarations.rb

@@ -177,6 +177,10 @@ def type
               Types::Bases::Any.new(location: nil)
             end
           end
+
+          def comment
+            leading_comment&.as_comment
+          end
         end
       end
     end

lib/rbs/definition.rb

@@ -66,25 +66,7 @@ def comment
           when AST::Members::Base
             member.comment
           when AST::Ruby::Members::Base
-            if member.leading_comment
-              lines = [] #: Array[String]
-
-              member.leading_comment.each_paragraph([]) do |paragraph|
-                case paragraph
-                when Location
-                  lines << paragraph.local_source
-                end
-              end
-
-              string = lines.join("\n")
-              
-              unless string.strip.empty?
-                AST::Comment.new(
-                  string: string,
-                  location: member.leading_comment.location
-                )
-              end
-            end
+            member.leading_comment&.as_comment
           end
         end
 

sig/ast/ruby/comment_block.rbs

@@ -115,6 +115,12 @@ module RBS
         def location: () -> Location
 
         private def leading_annotation?: (Integer index) -> bool
+
+        # Returns an comment object that contains the docs of from the comment block
+        #
+        # It ignores type annotations and syntax errors.
+        # 
+        def as_comment: () -> AST::Comment?
       end
     end
   end

sig/ast/ruby/declarations.rbs

@@ -101,7 +101,12 @@ module RBS
           # - When type_anntoation is given, it returns the type from the annotation.
           # - When type_annotation is not given, it returns infered type from the right-hand-side of the constant assignment
           # - Or it returns `untyped` type
+          #
           def type: () -> Types::t
+
+          # Returns the comment content extracted from the leading comment block
+          # 
+          def comment: () -> AST::Comment?
         end
       end
     end