openfoodnetwork/app/services/cart_service.rb

require 'open_food_network/scope_variant_to_hub'

class CartService
  attr_accessor :order, :currency
  attr_reader :variants_h
  attr_reader :errors

  def initialize(order)
    @order = order
    @currency = order.currency
    @errors = ActiveModel::Errors.new(self)
  end

  def populate(from_hash, overwrite = false)
    @distributor, @order_cycle = distributor_and_order_cycle

    @order.with_lock do
      variants_data = read_variants from_hash
      attempt_cart_add_variants variants_data
      overwrite_variants variants_data if overwrite
    end
    valid?
  end

  private

  def attempt_cart_add_variants(variants_data)
    loaded_variants = indexed_variants(variants_data)

    variants_data.each do |variant_data|
      loaded_variant = loaded_variants[variant_data[:variant_id].to_i]

      if loaded_variant.deleted?
        remove_deleted_variant(loaded_variant)
        next
      end

      next unless varies_from_cart(variant_data, loaded_variant)

      attempt_cart_add(loaded_variant, variant_data[:quantity], variant_data[:max_quantity])
    end
  end

  def indexed_variants(variants_data)
    @indexed_variants ||= begin
      variant_ids_in_data = variants_data.map{ |v| v[:variant_id] }

      Spree::Variant.with_deleted.where(id: variant_ids_in_data).
        includes(:default_price, :stock_items, :product).
        index_by(&:id)
    end
  end

  def remove_deleted_variant(variant)
    line_item_for_variant(variant).andand.destroy
  end

  def attempt_cart_add(variant, quantity, max_quantity = nil)
    quantity = quantity.to_i
    max_quantity = max_quantity.to_i if max_quantity
    return unless quantity > 0

    scoper.scope(variant)
    return unless valid_variant?(variant)

    cart_add(variant, quantity, max_quantity)
  end

  def cart_add(variant, quantity, max_quantity)
    quantity_to_add, max_quantity_to_add = quantities_to_add(variant, quantity, max_quantity)
    if quantity_to_add > 0
      @order.add_variant(variant, quantity_to_add, max_quantity_to_add, currency)
    else
      @order.remove_variant variant
    end
  end

  def quantities_to_add(variant, quantity, max_quantity)
    # If not enough stock is available, add as much as we can to the cart
    on_hand = variant.on_hand
    on_hand = [quantity, max_quantity].compact.max if variant.on_demand
    quantity_to_add = [quantity, on_hand].min
    max_quantity_to_add = max_quantity # max_quantity is not capped

    [quantity_to_add, max_quantity_to_add]
  end

  def overwrite_variants(variants)
    variants_removed(variants).each do |id|
      cart_remove(id)
    end
  end

  def scoper
    @scoper ||= OpenFoodNetwork::ScopeVariantToHub.new(@distributor)
  end

  def read_variants(data)
    @variants_h = read_variants_hash(data)
  end

  def read_variants_hash(data)
    (data[:variants] || []).map do |variant_id, quantity|
      if quantity.is_a?(Hash)
        { variant_id: variant_id, quantity: quantity[:quantity], max_quantity: quantity[:max_quantity] }
      else
        { variant_id: variant_id, quantity: quantity }
      end
    end
  end

  def valid?
    errors.empty?
  end

  def cart_remove(variant_id)
    variant = Spree::Variant.find(variant_id)
    @order.remove_variant(variant)
  end

  def distributor_and_order_cycle
    [@order.distributor, @order.order_cycle]
  end

  # Returns true if the saved cart differs from what's in the posted data, otherwise false
  def varies_from_cart(variant_data, loaded_variant)
    li = line_item_for_variant loaded_variant

    li_added = li.nil? && (variant_data[:quantity].to_i > 0 || variant_data[:max_quantity].to_i > 0)
    li_quantity_changed = li.present? && li.quantity.to_i != variant_data[:quantity].to_i
    li_max_quantity_changed = li.present? && li.max_quantity.to_i != variant_data[:max_quantity].to_i

    li_added || li_quantity_changed || li_max_quantity_changed
  end

  def variants_removed(variants_data)
    variant_ids_given = variants_data.map { |data| data[:variant_id].to_i }

    (variant_ids_in_cart - variant_ids_given).uniq
  end

  def valid_variant?(variant)
    check_order_cycle_provided && check_variant_available_under_distribution(variant)
  end

  def check_order_cycle_provided
    order_cycle_provided = @order_cycle.present?
    errors.add(:base, "Please choose an order cycle for this order.") unless order_cycle_provided
    order_cycle_provided
  end

  def check_variant_available_under_distribution(variant)
    return true if OrderCycleDistributedVariants.new(@order_cycle, @distributor).available_variants.include? variant

    errors.add(:base, I18n.t(:spree_order_populator_availability_error))
    false
  end

  def line_item_for_variant(variant)
    order.find_line_item_by_variant variant
  end

  def variant_ids_in_cart
    @order.line_items.pluck :variant_id
  end
end