[ruby/prism] Share comment generation logic

https://github.com/ruby/prism/commit/6e5556dbdf

2 files changed, 6 insertions, 2 deletions

diff --git a/prism/templates/lib/prism/node.rb.erb b/prism/templates/lib/prism/node.rb.erb
index affd2875ea..b7efe33523 100644
--- a/prism/templates/lib/prism/node.rb.erb
+++ b/prism/templates/lib/prism/node.rb.erb
@@ -32,8 +32,8 @@ module Prism
   end
   <%- nodes.each do |node| -%>
 
-  <%- node.comment.each_line do |line| -%>
-  #<%= line.prepend(" ").rstrip %>
+  <%- node.each_comment_line do |line| -%>
+  #<%= line %>
   <%- end -%>
   class <%= node.name -%> < Node
     <%- node.fields.each do |field| -%>
diff --git a/prism/templates/template.rb b/prism/templates/template.rb
index 46dbf88f4f..abd0976c41 100755
--- a/prism/templates/template.rb
+++ b/prism/templates/template.rb
@@ -255,6 +255,10 @@ module Prism
       @comment = config.fetch("comment")
     end
 
+    def each_comment_line
+      comment.each_line { |line| yield line.prepend(" ").rstrip }
+    end
+
     def semantic_fields
       @semantic_fields ||= @fields.select(&:semantic_field?)
     end