```ruby
# This class is responsible for saving payment sources in the user's "wallet"
# for future use.  You can substitute your own class via
# `Spree::Config.add_payment_sources_to_wallet_class`.
class Spree::Wallet::AddPaymentSourcesToWallet
  def initialize(order)
    @order = order
  end

  # This is called after an order transistions to complete and should save the
  # order's payment source/s in the user's "wallet" for future use.
  #
  # @return [void]
  def add_to_wallet
    if !order.temporary_payment_source && order.user
      # select valid sources
      sources = order.payments.valid.map(&:source).uniq.compact.select(&:reusable?)

      # add valid sources to wallet and optionally set a default
      if sources.any?
        sources.each do |source|
          order.user.wallet.add(source)
        end

        # arbitrarily pick the last one for the default
        default_source = sources.sort_by(&:id).last
        order.user.wallet.default = default_source
      end
    end
  end

  private

  attr_reader :order
end
```

Update add_after_order_complete for non-PaymentSource sources
user
```ruby
# This class is responsible for saving payment sources in the user's "wallet"
# for future use.  You can substitute your own class via
# `Spree::Config.add_payment_sources_to_wallet_class`.
class Spree::Wallet::AddPaymentSourcesToWallet
  def initialize(order)
    @order = order
  end

  # This is called after an order transistions to complete and should save the
  # order's payment source/s in the user's "wallet" for future use.
  #
  # @return [void]
  def add_to_wallet
    if !order.temporary_payment_source && order.user
      # select valid sources
      payments = order.payments.valid
      sources = payments.map(&:source).
        uniq.
        compact.
        select { |p| p.try(:reusable?) }

      # add valid sources to wallet and optionally set a default
      if sources.any?
        sources.each do |source|
          order.user.wallet.add(source)
        end

        # arbitrarily pick the last one for the default
        default_source = sources.sort_by(&:id).last
        order.user.wallet.default = default_source
      end
    end
  end

  private

  attr_reader :order
end
```
assistant