Lib Or Model? You DECIDE! Today on Who Want's To Be A Programmer?

shortlist.rb

class Shortlist

  def initialize(variants)
    @list_storage = variants
    if @list_storage.nil?
      @list_storage = []
    end
  end

  def add_variant(variant_id)
    if variant_id.is_a?(Integer)
      @list_storage << variant_id unless @list_storage.include?(variant_id)
    end
  end

  def remove_variant(variant_id)
    if variant_id.is_a?(Integer)
      @list_storage.delete(variant_id)
    end
  end

  def get_variants_by_ids
    ProductVariant.where(:id => @list_storage)
  end


end