diff options
| -rwxr-xr-x | yarp/templates/template.rb | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/yarp/templates/template.rb b/yarp/templates/template.rb index c661dd04e7..5880f11957 100755 --- a/yarp/templates/template.rb +++ b/yarp/templates/template.rb @@ -266,7 +266,6 @@ module YARP template = File.expand_path("../#{filepath}", __dir__) erb = read_template(template) - erb.filename = template non_ruby_heading = <<~HEADING /******************************************************************************/ @@ -305,9 +304,17 @@ module YARP def read_template(filepath) template = File.read(filepath, encoding: Encoding::UTF_8) - if ERB.instance_method(:initialize).parameters.assoc(:key) # Ruby 2.6+ + erb = erb(template) + erb.filename = filepath + erb + end + + if ERB.instance_method(:initialize).parameters.assoc(:key) # Ruby 2.6+ + def erb(template) ERB.new(template, trim_mode: "-") - else + end + else + def erb ERB.new(template, nil, "-") end end |