From b0a2373e14d475cfcf6ce21d25cbde2518c0dddc Mon Sep 17 00:00:00 2001
From: Kevin Newton <kddnewton@gmail.com>
Date: Mon, 30 Oct 2023 10:33:27 -0400
Subject: [PATCH] [ruby/prism] Share comment generation logic

https://github.com/ruby/prism/commit/6e5556dbdf
---
 prism/templates/lib/prism/node.rb.erb | 4 ++--
 prism/templates/template.rb           | 4 ++++
 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