killbill/killbill-client-ruby

module KillBillClient
  module Model
    class ComboTransaction < ComboPaymentTransactionAttributes

      def auth(user = nil, reason = nil, comment = nil, options = {}, refresh_options = nil)
        @transaction.transaction_type = 'AUTHORIZE'

        combo_payment(user, reason, comment, options, refresh_options)
      end

      def purchase(user = nil, reason = nil, comment = nil, options = {}, refresh_options = nil)
        @transaction.transaction_type = 'PURCHASE'

        combo_payment(user, reason, comment, options, refresh_options)
      end

      def credit(user = nil, reason = nil, comment = nil, options = {}, refresh_options = nil)
        @transaction.transaction_type = 'CREDIT'

        combo_payment(user, reason, comment, options, refresh_options)
      end

      private

      def combo_payment(user, reason, comment, options, refresh_options = nil)
        follow_location = true
        follow_location = options.delete(:follow_location) if options.has_key?(:follow_location)
        begin
          created_transaction = self.class.post "#{Payment::KILLBILL_API_PAYMENTS_PREFIX}/combo",
                                                to_json,
                                                {},
                                                {
                                                    :user => user,
                                                    :reason => reason,
                                                    :comment => comment,
                                                }.merge(options)
        rescue KillBillClient::API::ResponseError => error
          response = error.response
          if follow_location && response.header['location']
            created_transaction = ComboTransaction.new
            created_transaction.uri = response.header['location']
          else
            raise error
          end
        end

        if follow_location
          return created_transaction.refresh(refresh_options || options, Payment)
        end

        created_payment = Payment.new
        created_payment.uri = created_transaction.uri
        created_payment
      end
    end
  end
end