module SimpleShipping::Ups
  # Builds shipment element for UPS SOAP service.
  class ShipmentBuilder < SimpleShipping::Abstract::Builder
    # The type of payment for this shipment.
    PAYMENT_TYPE = '01' # 01 - Transportation, 02 - Duties and Taxes

    # Service codes in UPS terminology.
    SERVICE_TYPES = {:next_day_air            => '01',
                     :second_day_air          => '02',
                     :ground                  => '03',
                     :express                 => '07',
                     :expedited               => '08',
                     :standard                => '11',
                     :three_day_select        => '12',
                     :next_day_air_saver      => '13',
                     :next_day_air_early_am   => '14',
                     :express_plus            => '54',
                     :second_day_air_am       => '59',
                     :saver                   => '65',
                     :today_standard          => '82',
                     :today_dedicated_courier => '83',
                     :today_intercity         => '84',
                     :today_express           => '85',
                     :today_express_saver     => '86'}

    set_default_opts :service_type => :second_day_air

    # Return a hash for Savon representing a shipment model.
    def build
      { 'Shipper'            => PartyBuilder.build(@model.shipper, :shipper => true),
        'ShipTo'             => PartyBuilder.build(@model.recipient),
        'PaymentInformation' => build_payment_info,
        'Service'            => build_service,
        'Package'            => PackageBuilder.build(@model.package),
        :order! => ['Shipper', 'ShipTo', 'PaymentInformation', 'Service', 'Package'] }
    end

    # Return the PaymentInformation hash.
    def build_payment_info
      {'ShipmentCharge' => build_shipment_charge }
    end

    # Return the shipment charge for the PaymentInformation hash.
    def build_shipment_charge
      result = {'Type' => PAYMENT_TYPE, :order! => ['Type']}
      if @model.payor == :shipper
        result['BillShipper'] = {'AccountNumber' => @model.shipper.account_number}
        result[:order!] << 'BillShipper'
      else
        result['BillReceiver'] = {'AccountNumber' => @model.recipient.account_number}
        result[:order!] << 'BillReceiver'
      end
      result
    end

    # Return the hash representing the Service.
    def build_service
      {'Code' => SERVICE_TYPES[@opts[:service_type]]}
    end

    # Validate that the service type is included.
    def validate
      validate_inclusion_of(:service_type, SERVICE_TYPES)
    end
  end
end