$:.unshift File.dirname(__FILE__)

require 'nokogiri'
require 'ruhl/errors'

module Ruhl
  class Engine
    attr_reader :document, :scope, :layout, :layout_source, :local_object

    def initialize(html, options = {})
      @local_object   = options[:local_object]
      @layout_source  = options[:layout_source]

      if @layout = options[:layout]
        raise LayoutNotFoundError.new(@layout) unless File.exists?(@layout)
      end

      if @layout || @local_object
        @document = Nokogiri::HTML.fragment(html)
      else
        @document = Nokogiri::HTML(html)
      end
    end

    def render(current_scope)
      set_scope(current_scope)

      parse_doc(@document)

      if @layout
        render_with_layout 
      else
        document.to_s
      end
    end

    # The _render_ method is used within a layout to inject
    # the results of the template render.
    #
    # Ruhl::Engine.new(html, :layout => path_to_layout).render(self)
    def _render_
      document.to_s
    end

    private

    def render_with_layout
      render_file( @layout_source || File.read(@layout) ) 
    end

    def render_partial(tag, code)
      file = execute_ruby(tag, code)
      raise PartialNotFoundError.new(file) unless File.exists?(file)
      render_file( File.read(file) )
    end

    def render_collection(tag, code)
      results = execute_ruby(tag, code)

      html = tag.to_html

      new_content = results.collect do |item|
        Ruhl::Engine.new(html, :local_object => item).render(scope)
      end.to_s

      tag.swap(new_content)
    end

    def render_file(contents)
      doc = Nokogiri::HTML( contents ) 
      parse_doc(doc)
      doc.to_s
    end

    def parse_doc(doc)
      if (nodes = doc.xpath('//*[@data-ruhl][1]')).empty?
        nodes = doc.search('*[@data-ruhl]')
      end

      return if nodes.empty?

      tag = nodes.first
      code = tag.remove_attribute('data-ruhl') 
      process_attribute(tag, code.value)
      parse_doc(doc)
    end

    def process_attribute(tag, code)
      code.split(',').each do |pair|
        attribute, value = pair.split(':')
        
        if value.nil?
          tag.inner_html = execute_ruby(tag, attribute)
        else
          value.strip!
          if attribute  == "_partial"
            tag.inner_html = render_partial(tag, value)
          elsif attribute  == "_collection"
            render_collection(tag, value)
          elsif attribute  == "_if"
            contents = execute_ruby(tag, value)
            if contents
              tag.inner_html = contents
            else
              tag.remove
            end
          elsif attribute  == "_render_if"
            unless execute_ruby(tag, value)
              tag.remove
            end
          else
            tag[attribute] = execute_ruby(tag, value)
          end
        end
      end
    end
     
    def execute_ruby(tag, code)
      unless code == '_render_'
        if local_object && local_object.respond_to?(code)
          local_object.send(code)
        else
          scope.send(code, tag)
        end
      else
        _render_
      end
    end

    def set_scope(current_scope)
      raise Ruhl::NoScopeError unless current_scope
      @scope = current_scope 
    end

  end
end