require 'temple'
require 'hamlit/compiler'
require 'hamlit/parser'

module Hamlit
  class Engine < Temple::Engine
    define_options(
      generator:   Temple::Generators::ArrayBuffer,
      format:      :html,
      html_type:   nil,
      attr_quote:  "'",
      escape_html: true,
      pretty:      false,
      autoclose:   %w(area base basefont br col command embed frame
                      hr img input isindex keygen link menuitem meta
                      param source track wbr),
      filename:    "",
      indent_tags: [],
    )

    use Parser
    use Compiler
    html :Pretty
    filter :Escapable
    filter :ControlFlow
    filter :MultiFlattener
    filter :StaticMerger
    use :Generator, -> { options[:generator] }
  end

  class HamlEngine
    def initialize(template, options = {})
      @template = template
      @options  = options
    end

    def precompiled
      Engine.new(temple_options).call(@template)
    end

    def render(scope = Object.new, locals = {}, &block)
      scope = wrap_binding(scope)
      set_locals(locals, scope)
      eval(precompiled, scope)
    end
    alias_method :to_html, :render

    private

    def temple_options
      @options.dup.tap do |options|
        options[:pretty] = !options.delete(:ugly)
        case options[:format]
        when :html5
          options[:format] = :html
        when :html4
          options[:format] = :html
          options[:html_type] = 'transitional'
        end
      end
    end

    def wrap_binding(scope)
      return scope if scope.is_a?(Binding)
      scope.instance_eval { binding }
    end

    def set_locals(locals, scope)
      set_locals = locals.map { |k, v| "#{k} = #{v.inspect}" }.join("\n")
      eval(set_locals, scope)
    end
  end
end