erector_template.rb

module ActionView #:nodoc:
  module TemplateHandlers #:nodoc:
    class ErectorTemplate < TemplateHandler
      include Compilable
      def self.line_offset
        2
      end

      ActionView::Template.instance_eval do
        register_template_handler :ert, ActionView::TemplateHandlers::ErectorTemplate
      end

      def compile(template)
        source = <<-ERT
          <%=begin
            extend ::Erector::Mixin
            memoized_instance_variables = instance_variables.inject({}) do |all, instance_variable|
              all[instance_variable] = instance_variable_get(instance_variable)
              all
            end
            Erector.inline do
              memoized_instance_variables.each do |instance_variable, value|
                instance_variable_set(instance_variable, value)
              end
              eval(<<-RUBY, binding, __FILE__, 1)
                #{template.source}
              RUBY
            end
          end%>
        ERT
        ::ERB.new(source, nil, "-", '@output_buffer').src
      end
    end
  end
end