openfoodnetwork/lib/tasks/sample_data/payment_method_factory.rb

require "tasks/sample_data/addressing"
require "tasks/sample_data/logging"

class PaymentMethodFactory
  include Logging
  include Addressing

  def create_samples(enterprises)
    log "Creating payment methods:"
    distributors = enterprises.select(&:is_distributor)
    distributors.each do |enterprise|
      create_payment_methods(enterprise)
    end
  end

  private

  def create_payment_methods(enterprise)
    return if enterprise.payment_methods.present?
    log "- #{enterprise.name}"
    create_cash_method(enterprise)
    create_card_method(enterprise)
  end

  def create_cash_method(enterprise)
    create_payment_method(
      enterprise,
      "Cash on collection",
      "Pay on collection!",
      Spree::Calculator::FlatRate.new
    )
  end

  def create_card_method(enterprise)
    create_payment_method(
      enterprise,
      "Credit card (fake)",
      "We charge 1%, but won't ask for your details. ;-)",
      Spree::Calculator::FlatPercentItemTotal.new(preferred_flat_percent: 1)
    )
  end

  def create_payment_method(enterprise, name, description, calculator)
    card = enterprise.payment_methods.new(
      name: name,
      description: description,
      distributor_ids: [enterprise.id]
    )
    card.calculator = calculator
    card.save!
  end
end