ruby-avm-library/lib/avm/creator.rb

require 'avm/contact'
require 'nokogiri'

module AVM
  class Creator
    attr_reader :contacts, :image

    IPTC_CORE_FIELDS = [ :address, :city, :state, :zip, :country ]
    PRIMARY_CONTACT_FIELDS = IPTC_CORE_FIELDS + [ :province, :postal_code ]

    def initialize(image)
      @options = {}
      @contacts = []
      @image = image
    end

    def merge!(hash)
      @options.merge!(hash)
    end

    def method_missing(key, *opts)
      if key.to_s[-1..-1] == '='
        @options[key.to_s[0..-2].to_sym] = opts.first
      else
        if PRIMARY_CONTACT_FIELDS.include?(key)
          primary_contact_field key
        else
          @options[key]
        end
      end
    end

    def add_to_document(document)
      document.add_to_doc do |refs|
        creator = refs[:dublin_core].add_child('<dc:creator><rdf:Seq></rdf:Seq></dc:creator>')

        list = creator.at_xpath('.//rdf:Seq')
        contact_info = refs[:iptc].add_child('<Iptc4xmpCore:CreatorContactInfo rdf:parseType="Resource" />').first

        contacts.sort.each do |contact| 
          list.add_child(contact.to_creator_list_element) 
        end

        if primary_contact
          [ [ :telephone, 'CiTelWork' ], [ :email, 'CiEmailWork' ] ].each do |key, element_name|
            contact_info.add_child "<Iptc4xmpCore:#{element_name}>#{contacts.sort.collect(&key).join(',')}</Iptc4xmpCore:#{element_name}>"
          end

          iptc_namespace = document.doc.root.namespace_scopes.find { |ns| ns.prefix == 'Iptc4xmpCore' }

          IPTC_CORE_FIELDS.zip(%w{CiAdrExtadr CiAdrCity CiAdrRegion CiAdrPcode CiAdrCtry}).each do |key, element_name|
            node = contact_info.document.create_element(element_name, primary_contact.send(key))
            node.namespace = iptc_namespace
            contact_info.add_child node
          end
        end
      end
    end

    def primary_contact
      @contacts.find(&:primary) || @contacts.sort.first
    end

    def create_contact(info)
      contact = Contact.new(info)
      contacts << contact
      contact
    end

    private
      def primary_contact_field(field)
        if contact = primary_contact
          contact.send(field)
        else
          nil
        end
      end
  end
end