digidentity/libsaml

require 'xmlmapper'

module Saml
  module Base
    extend ActiveSupport::Concern

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

      def use_parsed
        @use_parsed = true
        self
      end

      def use_parsed?
        @use_parsed
      end

      extend XmlMapperClassMethods
      include XmlMapperInstanceMethods
    end

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

      attr_writer :from_xml

      def from_xml?
        @from_xml
      end

      def use_original(object)
        registered_namespaces.merge!(Saml::Util.collect_extra_namespaces(object.xml_value))
        object.use_parsed
      end
    end

    module XmlMapperClassMethods
      def parse(xml, options = {})
        if xml.is_a?(String)
          ActiveSupport::XmlMini_REXML.parse(xml)
        end

        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, REXML::ParseException => e
        raise Saml::Errors::UnparseableMessage.new(e.message)
      rescue TypeError => e
        raise Saml::Errors::UnparseableMessage.new(e.message)
      rescue NoMethodError => e
        raise Saml::Errors::UnparseableMessage.new(e.message)
      end
    end
  end
end