class ActiveSupport::SafeBuffer

def %(args)
  case args
  when Hash
    escaped_args = Hash[args.map { |k,arg| [k, html_escape_interpolated_argument(arg)] }]
  else
    escaped_args = Array(args).map { |arg| html_escape_interpolated_argument(arg) }
  end
  self.class.new(super(escaped_args))
end

def +(other)
  dup.concat(other)
end

def [](*args)
  if args.size < 2
    super
  else
    if html_safe?
      new_safe_buffer = super
      if new_safe_buffer
        new_safe_buffer.instance_variable_set :@html_safe, true
      end
      new_safe_buffer
    else
      to_str[*args]
    end
  end
end

def clone_empty
  self[0, 0]
end

def concat(value)
  super(html_escape_interpolated_argument(value))
end

def encode_with(coder)
  coder.represent_object nil, to_str
end

def html_escape_interpolated_argument(arg)
  (!html_safe? || arg.html_safe?) ? arg : CGI.escapeHTML(arg.to_s)
end

def html_safe?
  defined?(@html_safe) && @html_safe
end

def initialize(str = '')
  @html_safe = true
  super
end

def initialize_copy(other)
  super
  @html_safe = other.html_safe?
end

def prepend(value)
  super(html_escape_interpolated_argument(value))
end

def safe_concat(value)
  raise SafeConcatError unless html_safe?
  original_concat(value)
end

def to_param
  to_str
end

def to_s
  self
end