require "nokogiri"
require "sax-machine/ns_stack"

module SAXMachine
  class SAXHandler < Nokogiri::XML::SAX::Document
    attr_reader :object

    def initialize(object, nsstack=nil)
      @object = object
      @nsstack = nsstack || NSStack.new
    end

    def characters(string)
      if parsing_collection?
        @collection_handler.characters(string)
      elsif @element_config
        @value << string
      end
    end

    def cdata_block(string)
      characters(string)
    end

    def start_element(name, attrs = [])

      @name   = name
      @attrs  = attrs.map { |a| SAXHandler.decode_xml(a) }
      @nsstack  = NSStack.new(@nsstack, @attrs)

      if parsing_collection?
        @collection_handler.start_element(@name, @attrs)

      elsif @collection_config = sax_config.collection_config(@name, @nsstack)
        @collection_handler = @collection_config.handler(@nsstack)
        @collection_handler.start_element(@name, @attrs)

      elsif (element_configs = sax_config.element_configs_for_attribute(@name, @attrs)).any?
        parse_element_attributes(element_configs)
        set_element_config_for_element_value

      else
        set_element_config_for_element_value
      end
    end

    def end_element(name)
      if parsing_collection? && @collection_config.name == name.split(':').last
        @collection_handler.end_element(name)
        @object.send(@collection_config.accessor) << @collection_handler.object
        reset_current_collection

      elsif parsing_collection?
        @collection_handler.end_element(name)

      elsif characaters_captured?
        @object.send(@element_config.setter, @value)
      end

      reset_current_tag
      @nsstack = @nsstack.pop
    end

    def characaters_captured?
      !@value.nil? && !@value.empty?
    end

    def parsing_collection?
      !@collection_handler.nil?
    end

    def parse_collection_instance_attributes
      instance = @collection_handler.object
      @attrs.each_with_index do |attr_name,index|
        instance.send("#{attr_name}=", @attrs[index + 1]) if index % 2 == 0 && instance.methods.include?("#{attr_name}=")
      end
    end

    def parse_element_attributes(element_configs)
      element_configs.each do |ec|
        @object.send(ec.setter, ec.value_from_attrs(@attrs))
      end
      @element_config = nil
    end

    def set_element_config_for_element_value
      @value = ""
      @element_config = sax_config.element_config_for_tag(@name, @attrs, @nsstack)
    end

    def reset_current_collection
      @collection_handler = nil
      @collection_config  = nil
    end

    def reset_current_tag
      @name   = nil
      @attrs  = nil
      @value  = nil
      @element_config = nil
    end

    def sax_config
      @object.class.sax_config
    end
  
    ##
    # Decodes XML special characters.
    def self.decode_xml(str)
      return str.map &method(:decode_xml) if str.kind_of?(Array)

      # entities = {
      #         '#38'   => '&amp;',
      #         '#13'   => "\r",
      #       } 
      #       entities.keys.inject(str) { |string, key|
      #         string.gsub(/&#{key};/, entities[key])
      #       } 
      CGI.unescapeHTML(str)
    end
  
  end
end