digidentity/xmlenc

require "xmlmapper"

module Xmlenc
  module Builder
    module Base
      extend ActiveSupport::Concern

      included do
        include ::XmlMapper
        include ::ActiveModel::Validations

        extend XmlMapperClassMethods
        include XmlMapperInstanceMethods
      end

      module XmlMapperInstanceMethods
        def initialize(attributes = {})
          attributes.each do |key, value|
            send("#{key}=", value) if respond_to?("#{key}=") && value.present?
          end
        end

        def from_xml=(bool)
          @from_xml = bool
        end

        def from_xml?
          @from_xml
        end
      end

      module XmlMapperClassMethods
        def parse(xml, options = {})
          raise Xmlenc::UnparseableMessage.new("Unable to parse nil document") if xml.nil?

          object = super
          if object.is_a?(Array)
            object.map { |x| x.from_xml = true }
          elsif object
            object.from_xml = true
          end
          object
        rescue Nokogiri::XML::SyntaxError => e
          raise Xmlenc::UnparseableMessage.new(e.message)
        rescue NoMethodError => e
          raise Xmlenc::UnparseableMessage.new(e.message)
        end
      end
    end
  end
end