internetee-registry/spec/support/epp_contact_xml_builder.rb

module EppContactXmlBuilder
  
  def contact_check_xml(xml_params={})

    xml_params[:ids] = xml_params[:ids] || [ { id: 'check-1234' }, { id: 'check-4321' } ]  

    xml = Builder::XmlMarkup.new

    xml.instruct!(:xml, standalone: 'no')
    xml.epp('xmlns' => 'urn:ietf:params:xml:ns:epp-1.0') do
      xml.command do
        xml.check do
          xml.tag!('contact:check', 'xmlns:contact' => 'urn:ietf:params:xml:ns:contact-1.0') do
            unless xml_params[:ids] == [false]
              xml_params[:ids].each do |x|
                xml.tag!('contact:id', x[:id])
              end
            end
          end
        end
        xml.clTRID 'ABC-12345'
      end
    end
  end

  def contact_create_xml(xml_params={})
    #xml_params[:ids] = xml_params[:ids] || [ { id: 'check-1234' }, { id: 'check-4321' } ]  
    xml = Builder::XmlMarkup.new
    
    xml_params[:addr] = xml_params[:addr] ||  { street: '123 Example Dr.', street2: 'Suite 100', street3: nil,
                                                city: 'Megaton' , sp: 'F3 ' , pc: '201-33' , cc: 'EE' }
    xml_params[:authInfo] = xml_params[:authInfo] || { pw: 'Aas34fq' } 


    xml.instruct!(:xml, standalone: 'no')
    xml.epp('xmlns' => 'urn:ietf:params:xml:ns:epp-1.0') do
      xml.command do
        xml.create do
          xml.tag!('contact:create', 'xmlns:contact' => 'urn:ietf:params:xml:ns:contact-1.0') do
            xml.tag!('contact:id', xml_params[:id], 'sh8013') unless xml_params[:id] == false
            unless xml_params[:postalInfo] == [false]
              xml.tag!('contact:postalInfo') do
                xml.tag!('contact:name', ( xml_params[:name] || 'Sillius Soddus'  ))  unless xml_params[:name] == false
                xml.tag!('contact:org', ( xml_params[:org_name] || 'Example Inc.' ))  unless xml_params[:org_name] == false
                unless xml_params[:addr] == [false]
                  xml.tag!('contact:addr') do
                    xml.tag!('contact:street', xml_params[:addr][:street]   ) unless xml_params[:addr][:street] == false
                    xml.tag!('contact:street', xml_params[:addr][:street2]  ) unless xml_params[:addr][:street2] == false
                    xml.tag!('contact:street', xml_params[:addr][:street3]  ) unless xml_params[:addr][:street3] == false
                    xml.tag!('contact:city'  , xml_params[:addr][:city]     ) unless xml_params[:addr][:city] == false
                    xml.tag!('contact:sp'    , xml_params[:addr][:sp]       ) unless xml_params[:addr][:sp] == false
                    xml.tag!('contact:pc'    , xml_params[:addr][:pc]       ) unless xml_params[:addr][:pc] == false
                    xml.tag!('contact:cc'    , xml_params[:addr][:cc]       ) unless xml_params[:addr][:cc] == false
                  end
                end
              end
            end
            xml.tag!('contact:voice', (xml_params[:voice] || '+372.1234567')) unless xml_params[:voice] == false
            xml.tag!('contact:fax', (xml_params[:fax] || '123123' )) unless xml_params[:fax] == false
            xml.tag!('contact:email', (xml_params[:email] || 'example@test.example')) unless xml_params[:email] == false
            xml.tag!('contact:ident', (xml_params[:ident] || '37605030299'), type: 'op') unless xml_params[:ident] == false
            unless xml_params[:authInfo] == [false]
              xml.tag!('contact:authInfo') do
                xml.tag!('contact:pw', xml_params[:authInfo][:pw] ) unless xml_params[:authInfo][:pw] == false
              end
            end
            #Disclosure logic
          end
        end
        xml.clTRID 'ABC-12345'
      end
    end
  end

  #CONTACT UPDATE NEEDS WORK USE ON YOUR OWN RISK
  def contact_update_xml(xml_params={})
    xml = Builder::XmlMarkup.new

    #postalInfo = xml_params.try(:chg).try(:postalInfo)
    #addr = postalInfo.try(:addr)
    postalInfo = xml_params[:chg][:postalInfo] rescue nil
    addr = postalInfo[:addr] rescue nil

    if !addr
      addr = { street: 'Downtown', city: 'Stockholm', cc: 'SE' }
    end
    
    if !postalInfo
      postalInfo = { name: 'Jane Doe', org: 'Fake Inc.', voice: '+321.12345', fax: '12312312', addr: addr  }
    end

    xml_params[:chg] = xml_params[:chg] || { postalInfo: postalInfo }

    xml_params[:chg][:postalInfo] = postalInfo
    xml_params[:chg][:postalInfo][:addr] = addr

    xml_params[:chg][:authInfo] = xml_params[:chg][:authInfo] || { pw: 'ccds4324pok' }


    xml.instruct!(:xml, standalone: 'no')
     xml.epp('xmlns' => 'urn:ietf:params:xml:ns:epp-1.0') do
      xml.command do
        xml.update do
          xml.tag!('contact:update', 'xmlns:contact' => 'urn:ietf:params:xml:ns:contact-1.0') do
            xml.tag!('contact:id', (xml_params[:id] || 'sh8013') ) unless xml_params[:id] == false
            unless xml_params[:chg] == [ false ]
              xml.tag!('contact:chg') do
                xml.tag!('contact:voice', xml_params[:chg][:phone] || '+123.321123' ) unless xml_params[:chg][:phone] == false
                xml.tag!('contact:email', xml_params[:chg][:email] || 'jane@doe.com' ) unless xml_params[:chg][:email] == false
                unless xml_params[:chg][:postalInfo] == false
                  xml.tag!('contact:postalInfo') do
                    xml.tag!('contact:name', xml_params[:chg][:postalInfo][:name] ) unless xml_params[:chg][:postalInfo][:name] == false
                    xml.tag!('contact:org', xml_params[:chg][:postalInfo][:org] ) unless xml_params[:chg][:postalInfo][:org] == false
                    unless xml_params[:chg][:postalInfo][:addr] == false
                      xml.tag!('contact:addr') do
                        xml.tag!('contact:street', xml_params[:chg][:postalInfo][:addr][:street]   ) unless xml_params[:chg][:postalInfo][:addr][:street] == false
                        xml.tag!('contact:street', xml_params[:chg][:postalInfo][:addr][:street2]  ) unless xml_params[:chg][:postalInfo][:addr][:street2] == false
                        xml.tag!('contact:street', xml_params[:chg][:postalInfo][:addr][:street3]  ) unless xml_params[:chg][:postalInfo][:addr][:street3] == false
                        xml.tag!('contact:city'  , xml_params[:chg][:postalInfo][:addr][:city]     ) unless xml_params[:chg][:postalInfo][:addr][:city] == false
                        xml.tag!('contact:sp'    , xml_params[:chg][:postalInfo][:addr][:sp]       ) unless xml_params[:chg][:postalInfo][:addr][:sp] == false
                        xml.tag!('contact:pc'    , xml_params[:chg][:postalInfo][:addr][:pc]       ) unless xml_params[:chg][:postalInfo][:addr][:pc] == false
                        xml.tag!('contact:cc'    , xml_params[:chg][:postalInfo][:addr][:cc]       ) unless xml_params[:chg][:postalInfo][:addr][:cc] == false
                      end
                    end
                  end
                end
                unless xml_params[:chg][:authInfo] == [false]
                  xml.tag!('contact:authInfo') do
                    xml.tag!('contact:pw', xml_params[:chg][:authInfo][:pw] ) unless xml_params[:chg][:authInfo][:pw] == false
                  end
                end
              end
            end
          end
        end
        xml.clTRID 'ABC-12345'
      end
    end
  end
end

RSpec.configure do |c|
  c.include EppContactXmlBuilder
end