mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
9ecaa80300
Conflicts: actionpack/lib/action_view/template/handlers/erb.rb
124 lines
3.5 KiB
Ruby
124 lines
3.5 KiB
Ruby
require 'active_support/core_ext/class/attribute_accessors'
|
|
require 'active_support/core_ext/string/output_safety'
|
|
require 'action_view/template'
|
|
require 'action_view/template/handler'
|
|
require 'erubis'
|
|
|
|
module ActionView
|
|
class OutputBuffer < ActiveSupport::SafeBuffer
|
|
def initialize(*)
|
|
super
|
|
encode! if encoding_aware?
|
|
end
|
|
|
|
def <<(value)
|
|
super(value.to_s)
|
|
end
|
|
alias :append= :<<
|
|
alias :safe_append= :safe_concat
|
|
end
|
|
|
|
class Template
|
|
module Handlers
|
|
class Erubis < ::Erubis::Eruby
|
|
def add_preamble(src)
|
|
src << "@output_buffer = ActionView::OutputBuffer.new;"
|
|
end
|
|
|
|
def add_text(src, text)
|
|
return if text.empty?
|
|
src << "@output_buffer.safe_concat('" << escape_text(text) << "');"
|
|
end
|
|
|
|
BLOCK_EXPR = /\s+(do|\{)(\s*\|[^|]*\|)?\s*\Z/
|
|
|
|
def add_expr_literal(src, code)
|
|
if code =~ BLOCK_EXPR
|
|
src << '@output_buffer.append= ' << code
|
|
else
|
|
src << '@output_buffer.append= (' << code << ');'
|
|
end
|
|
end
|
|
|
|
def add_expr_escaped(src, code)
|
|
if code =~ BLOCK_EXPR
|
|
src << "@output_buffer.safe_append= " << code
|
|
else
|
|
src << "@output_buffer.safe_concat((" << code << ").to_s);"
|
|
end
|
|
end
|
|
|
|
def add_postamble(src)
|
|
src << '@output_buffer.to_s'
|
|
end
|
|
end
|
|
|
|
class ERB
|
|
# Specify trim mode for the ERB compiler. Defaults to '-'.
|
|
# See ERb documentation for suitable values.
|
|
class_attribute :erb_trim_mode
|
|
self.erb_trim_mode = '-'
|
|
|
|
# Default format used by ERB.
|
|
class_attribute :default_format
|
|
self.default_format = Mime::HTML
|
|
|
|
# Default implemenation used.
|
|
class_attribute :erb_implementation
|
|
self.erb_implementation = Erubis
|
|
|
|
ENCODING_TAG = Regexp.new("\\A(<%#{ENCODING_FLAG}-?%>)[ \\t]*")
|
|
|
|
def self.call(template)
|
|
new.call(template)
|
|
end
|
|
|
|
def handles_encoding?
|
|
true
|
|
end
|
|
|
|
def call(template)
|
|
if template.source.encoding_aware?
|
|
# First, convert to BINARY, so in case the encoding is
|
|
# wrong, we can still find an encoding tag
|
|
# (<%# encoding %>) inside the String using a regular
|
|
# expression
|
|
template_source = template.source.dup.force_encoding("BINARY")
|
|
|
|
erb = template_source.gsub(ENCODING_TAG, '')
|
|
encoding = $2
|
|
|
|
erb.force_encoding valid_encoding(template.source.dup, encoding)
|
|
|
|
# Always make sure we return a String in the default_internal
|
|
erb.encode!
|
|
else
|
|
erb = template.source.dup
|
|
end
|
|
|
|
self.class.erb_implementation.new(
|
|
erb,
|
|
:trim => (self.class.erb_trim_mode == "-")
|
|
).src
|
|
end
|
|
|
|
private
|
|
|
|
def valid_encoding(string, encoding)
|
|
# If a magic encoding comment was found, tag the
|
|
# String with this encoding. This is for a case
|
|
# where the original String was assumed to be,
|
|
# for instance, UTF-8, but a magic comment
|
|
# proved otherwise
|
|
string.force_encoding(encoding) if encoding
|
|
|
|
# If the String is valid, return the encoding we found
|
|
return string.encoding if string.valid_encoding?
|
|
|
|
# Otherwise, raise an exception
|
|
raise WrongEncodingError.new(string, string.encoding)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|