lib/rake/funnel/support/template_engine.rb in rake-funnel-0.22.2 vs lib/rake/funnel/support/template_engine.rb in rake-funnel-0.22.3

@@ -1,18 +1,29 @@
+# frozen_string_literal: true
+
 require 'erb'
 
 module Rake
   module Funnel
     module Support
       class TemplateEngine
         class << self
           def render(template, filename = nil, binding = nil)
-            render = ERB.new(replace_at_markers(template), nil, '%<>')
+            render = erb(template)
             render.filename = filename
             render.result(binding || top_level_binding)
           end
 
           private
+
+          def erb(template)
+            template = replace_at_markers(template)
+            trim_mode = '%<>'
+
+            return ERB.new(template, trim_mode: trim_mode) if RUBY_VERSION >= '2.6'
+
+            ERB.new(template, nil, trim_mode)
+          end
 
           def replace_at_markers(template)
             tags = /(@\w[\w\.]+\w@)/
 
             (template || '').gsub(tags) do |match|