module Erubis::PrintEnabledEnhancer

def self.desc   # :nodoc:
  "enable to use print function in '<% %>'"
end

def add_preamble(src)
  src << "@_buf = "
  super
end

def evaluate(context=nil)
  _src = @src
  if context.is_a?(Hash)
    context.each do |key, val| instance_variable_set("@#{key}", val) end
  elsif context
    context.instance_variables.each do |name|
      instance_variable_set(name, context.instance_variable_get(name))
    end
  end
  return instance_eval(_src, (@filename || '(erubis)'))
end

def print(*args)
  args.each do |arg|
    @_buf << arg.to_s
  end
end