scotdalton/exlibris-primo

module Exlibris
  module Primo
    #
    # Utility for parsing and building XML
    #
    module XmlUtil
      require 'nokogiri'

      def self.included(klass)
        klass.class_eval do
          extend ClassAttributes
        end
      end

      module ClassAttributes
        def xml_options
          @xml_options ||= {
            :encoding => 'UTF-8',
            :indent => 0,
            :save_with => Nokogiri::XML::Node::SaveOptions::AS_XML | Nokogiri::XML::Node::SaveOptions::NO_DECLARATION
          }
        end
      end

      attr_reader :raw_xml
      protected :raw_xml

      # Returns an XML string and takes any args that are
      # understood by Nokogiri::XML::Builder.
      def build_xml options={}, &block
        Nokogiri::XML::Builder.new(options.merge(:encoding => 'UTF-8'), &block).to_xml(xml_options).strip
      end
      protected :build_xml

      def xml_options
        @xml_options ||= self.class.xml_options
      end
      protected :xml_options

      def xml
        @xml ||= Nokogiri::XML(raw_xml, nil, 'UTF-8')
      end
      protected :xml

      def xml_without_namespaces
        xml.clone.remove_namespaces!
      end
      protected :xml_without_namespaces

      def remove_namespaces_from_raw_xml(raw_xml, namespaces)
        tmp_xml_with_namespaces = build_xml do |xml|
          xml.root(namespaces) do
            xml << raw_xml
          end
        end
        tmp_xml = Nokogiri::XML(tmp_xml_with_namespaces)
        tmp_xml.remove_namespaces!
        tmp_xml.root.children.first.to_xml(xml_options)
      end
      protected :remove_namespaces_from_raw_xml

      def to_hash
        Hash.from_xml(to_xml)
      end

      def to_xml
        xml.to_xml(xml_options).strip
      end

      def to_json
        to_hash.to_json
      end
    end
  end
end