openfoodfoundation/openfoodnetwork

# frozen_string_literal: true

require "tasks/sample_data/logging"

module SampleData
  class ProductFactory
    include Logging

    def create_samples(enterprises)
      log "Creating products:"
      product_data(enterprises).map do |hash|
        create_product(hash)
      end
    end

    private

    def product_data(enterprises)
      vegetables = Spree::Taxon.find_by(name: 'Vegetables')
      fruit = Spree::Taxon.find_by(name: 'Fruit')
      fungi = Spree::Taxon.find_by(name: 'Fungi')
      producers = enterprises.select(&:is_primary_producer)
      [
        {
          name: 'Garlic',
          price: 20.00,
          supplier: producers[0],
          taxons: [vegetables],
        },
        {
          name: 'Fuji Apple',
          price: 5.00,
          supplier: producers[1],
          taxons: [fruit],
        },
        {
          name: 'Mushrooms',
          price: 50.00,
          supplier: producers[1],
          taxons: [fungi],
        },
        {
          name: 'Carrots',
          price: 3.00,
          supplier: producers[2],
          taxons: [vegetables],
        },
        {
          name: 'Potatoes',
          price: 2.00,
          supplier: producers[2],
          taxons: [vegetables],
        },
        {
          name: 'Tomatoes',
          price: 2.00,
          supplier: producers[2],
          taxons: [vegetables],
        }
      ]
    end

    def create_product(hash)
      log "- #{hash[:name]}"
      params = hash.slice(:name, :price).merge(
        supplier_id: hash[:supplier].id,
        primary_taxon_id: hash[:taxons].first.id,
        variant_unit: "weight",
        variant_unit_scale: 1,
        unit_value: 1,
        tax_category_id: find_or_create_tax_category.id
      )
      product = Spree::Product.create_with(params).find_or_create_by!(name: params[:name])
      product.variants.first.update_attribute :on_demand, true
      product
    end

    def find_or_create_tax_category
      tax_category_name = "Tax Category"
      tax_category = Spree::TaxCategory.find_by(name: tax_category_name)
      tax_category ||= Spree::TaxCategory.create!(name: tax_category_name)
      tax_category
    end
  end
end