bring back wrap! method, TODO: benchmark against String#sub

korny · korny · commit 196b7296666c · 2013-06-17T12:51:49.000+02:00
diff --git a/lib/coderay/encoders/html/output.rb b/lib/coderay/encoders/html/output.rb
@@ -17,6 +17,16 @@ def self.wrap_string_in string, element, css = nil
         end.sub('<%CONTENT%>', string)
       end
       
+      def self.wrap! str, template, target
+        target = Regexp.new(Regexp.escape("<%#{target}%>"))
+        if template =~ target
+          str[0,0] = $`
+          str << $'
+        else
+          raise "Template target <%%%p%%> not found" % target
+        end
+      end
+      
       SPAN = '<span class="CodeRay"><%CONTENT%></span>'
       
       DIV = <<-DIV
