Class: Spree::Calculator::RelatedProductDiscount

Inherits:
Spree::Calculator show all
Defined in:
app/models/spree/calculator/related_product_discount.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.descriptionObject



8
9
10
# File 'app/models/spree/calculator/related_product_discount.rb', line 8

def self.description
  Spree.t(:related_product_discount)
end

Instance Method Details

#compute(object) ⇒ Object



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# File 'app/models/spree/calculator/related_product_discount.rb', line 12

def compute(object)
  if object.is_a?(Array)
    return if object.empty?

    order = object.first.order
  else
    order = object
  end

  total = 0
  if eligible?(order)
    total = order.line_items.inject(0) do |sum, line_item|
      relations = Spree::Relation.where(*discount_query(line_item))
      discount_applies_to = relations.map { |rel| rel.related_to.master.product }
      order.line_items.each do |li|
        next unless discount_applies_to.include? li.variant.product

        discount = relations.detect { |rel| rel.related_to.master.product == li.variant.product }.discount_amount
        sum += if li.quantity < line_item.quantity
                 (discount * li.quantity)
               else
                 (discount * line_item.quantity)
        end
      end

      sum
    end
  end

  total
end

#discount_query(line_item) ⇒ Object



50
51
52
53
54
55
56
# File 'app/models/spree/calculator/related_product_discount.rb', line 50

def discount_query(line_item)
  [
    'discount_amount <> 0.0 AND relatable_type = ? AND relatable_id = ?',
    'Spree::Product',
    line_item.variant.product.id
  ]
end

#eligible?(order) ⇒ Boolean

Returns:

  • (Boolean)


44
45
46
47
48
# File 'app/models/spree/calculator/related_product_discount.rb', line 44

def eligible?(order)
  order.line_items.any? do |line_item|
    Spree::Relation.exists?(discount_query(line_item))
  end
end