Skip to content

Commit 59efdd5

Browse files
author
Oleksii Korshenko
authored
MAGETWO-85960: magento#12817: Coupon code with canceled order magento#1095 
2 parents 754a30e + bcff65e commit 59efdd5

File tree

14 files changed

+508
-298
lines changed

14 files changed

+508
-298
lines changed

‎app/code/Magento/SalesRule/Model/Coupon/UpdateCouponUsages.php

+152
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,152 @@
1+
<?php
2+
/**
3+
* Copyright © Magento, Inc. All rights reserved.
4+
* See COPYING.txt for license details.
5+
*/
6+
7+
namespace Magento\SalesRule\Model\Coupon;
8+
9+
use Magento\Sales\Model\Order;
10+
use Magento\SalesRule\Model\Coupon;
11+
use Magento\SalesRule\Model\ResourceModel\Coupon\Usage;
12+
use Magento\SalesRule\Model\Rule\CustomerFactory;
13+
use Magento\SalesRule\Model\RuleFactory;
14+
15+
/**
16+
* Updates the coupon usages.
17+
*/
18+
class UpdateCouponUsages
19+
{
20+
/**
21+
* @var RuleFactory
22+
*/
23+
private $ruleFactory;
24+
25+
/**
26+
* @var RuleFactory
27+
*/
28+
private $ruleCustomerFactory;
29+
30+
/**
31+
* @var Coupon
32+
*/
33+
private $coupon;
34+
35+
/**
36+
* @var Usage
37+
*/
38+
private $couponUsage;
39+
40+
/**
41+
* @param RuleFactory $ruleFactory
42+
* @param CustomerFactory $ruleCustomerFactory
43+
* @param Coupon $coupon
44+
* @param Usage $couponUsage
45+
*/
46+
public function __construct(
47+
RuleFactory $ruleFactory,
48+
CustomerFactory $ruleCustomerFactory,
49+
Coupon $coupon,
50+
Usage $couponUsage
51+
) {
52+
$this->ruleFactory = $ruleFactory;
53+
$this->ruleCustomerFactory = $ruleCustomerFactory;
54+
$this->coupon = $coupon;
55+
$this->couponUsage = $couponUsage;
56+
}
57+
58+
/**
59+
* Executes the current command.
60+
*
61+
* @param Order $subject
62+
* @param bool $increment
63+
* @return Order
64+
*/
65+
public function execute(Order $subject, bool $increment)
66+
{
67+
if (!$subject || !$subject->getAppliedRuleIds()) {
68+
return $subject;
69+
}
70+
// lookup rule ids
71+
$ruleIds = explode(',', $subject->getAppliedRuleIds());
72+
$ruleIds = array_unique($ruleIds);
73+
$customerId = (int)$subject->getCustomerId();
74+
// use each rule (and apply to customer, if applicable)
75+
foreach ($ruleIds as $ruleId) {
76+
if (!$ruleId) {
77+
continue;
78+
}
79+
$this->updateRuleUsages($increment, (int)$ruleId, $customerId);
80+
}
81+
$this->updateCouponUsages($subject, $increment, $customerId);
82+
83+
return $subject;
84+
}
85+
86+
/**
87+
* Update the number of rule usages.
88+
*
89+
* @param bool $increment
90+
* @param int $ruleId
91+
* @param int $customerId
92+
*/
93+
private function updateRuleUsages(bool $increment, int $ruleId, int $customerId)
94+
{
95+
/** @var \Magento\SalesRule\Model\Rule $rule */
96+
$rule = $this->ruleFactory->create();
97+
$rule->load($ruleId);
98+
if ($rule->getId()) {
99+
$rule->loadCouponCode();
100+
if ($increment || $rule->getTimesUsed() > 0) {
101+
$rule->setTimesUsed($rule->getTimesUsed() + ($increment ? 1 : -1));
102+
$rule->save();
103+
}
104+
if ($customerId) {
105+
$this->updateCustomerRuleUsages($increment, $ruleId, $customerId);
106+
}
107+
}
108+
}
109+
110+
/**
111+
* Update the number of rule usages per customer.
112+
*
113+
* @param bool $increment
114+
* @param int $ruleId
115+
* @param int $customerId
116+
*/
117+
private function updateCustomerRuleUsages(bool $increment, int $ruleId, int $customerId)
118+
{
119+
/** @var \Magento\SalesRule\Model\Rule\Customer $ruleCustomer */
120+
$ruleCustomer = $this->ruleCustomerFactory->create();
121+
$ruleCustomer->loadByCustomerRule($customerId, $ruleId);
122+
if ($ruleCustomer->getId()) {
123+
if ($increment || $ruleCustomer->getTimesUsed() > 0) {
124+
$ruleCustomer->setTimesUsed($ruleCustomer->getTimesUsed() + ($increment ? 1 : -1));
125+
}
126+
} elseif ($increment) {
127+
$ruleCustomer->setCustomerId($customerId)->setRuleId($ruleId)->setTimesUsed(1);
128+
}
129+
$ruleCustomer->save();
130+
}
131+
132+
/**
133+
* Update the number of coupon usages.
134+
*
135+
* @param Order $subject
136+
* @param bool $increment
137+
* @param int $customerId
138+
*/
139+
private function updateCouponUsages(Order $subject, bool $increment, int $customerId)
140+
{
141+
$this->coupon->load($subject->getCouponCode(), 'code');
142+
if ($this->coupon->getId()) {
143+
if ($increment || $this->coupon->getTimesUsed() > 0) {
144+
$this->coupon->setTimesUsed($this->coupon->getTimesUsed() + ($increment ? 1 : -1));
145+
$this->coupon->save();
146+
}
147+
if ($customerId) {
148+
$this->couponUsage->updateCustomerCouponTimesUsed($customerId, $this->coupon->getId(), $increment);
149+
}
150+
}
151+
}
152+
}

‎app/code/Magento/SalesRule/Model/ResourceModel/Coupon/Usage.php

+5-4
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,10 @@ protected function _construct()
2727
*
2828
* @param int $customerId
2929
* @param mixed $couponId
30+
* @param bool $increment
3031
* @return void
3132
*/
32-
public function updateCustomerCouponTimesUsed($customerId, $couponId)
33+
public function updateCustomerCouponTimesUsed($customerId, $couponId, $increment = true)
3334
{
3435
$connection = $this->getConnection();
3536
$select = $connection->select();
@@ -44,13 +45,13 @@ public function updateCustomerCouponTimesUsed($customerId, $couponId)
4445

4546
$timesUsed = $connection->fetchOne($select, [':coupon_id' => $couponId, ':customer_id' => $customerId]);
4647

47-
if ($timesUsed > 0) {
48+
if ($timesUsed !== false) {
4849
$this->getConnection()->update(
4950
$this->getMainTable(),
50-
['times_used' => $timesUsed + 1],
51+
['times_used' => $timesUsed + ($increment ? 1 : -1)],
5152
['coupon_id = ?' => $couponId, 'customer_id = ?' => $customerId]
5253
);
53-
} else {
54+
} elseif ($increment) {
5455
$this->getConnection()->insert(
5556
$this->getMainTable(),
5657
['coupon_id' => $couponId, 'customer_id' => $customerId, 'times_used' => 1]

‎app/code/Magento/SalesRule/Observer/SalesOrderAfterPlaceObserver.php

-110
This file was deleted.

‎app/code/Magento/SalesRule/Plugin/CouponUsagesDecrement.php

+42
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
<?php
2+
/**
3+
* Copyright © Magento, Inc. All rights reserved.
4+
* See COPYING.txt for license details.
5+
*/
6+
7+
namespace Magento\SalesRule\Plugin;
8+
9+
use Magento\Sales\Model\Order;
10+
use Magento\SalesRule\Model\Coupon\UpdateCouponUsages;
11+
12+
class CouponUsagesDecrement
13+
{
14+
/**
15+
* @var UpdateCouponUsages
16+
*/
17+
private $updateCouponUsages;
18+
19+
public function __construct(
20+
UpdateCouponUsages $updateCouponUsages
21+
) {
22+
$this->updateCouponUsages = $updateCouponUsages;
23+
}
24+
25+
/**
26+
* Decrements number of coupon usages after cancelling order.
27+
*
28+
* @param Order $subject
29+
* @param callable $proceed
30+
* @return Order
31+
*/
32+
public function aroundCancel(Order $subject, callable $proceed)
33+
{
34+
$canCancel = $subject->canCancel();
35+
$returnValue = $proceed();
36+
if ($canCancel) {
37+
$returnValue = $this->updateCouponUsages->execute($returnValue, false);
38+
}
39+
40+
return $returnValue;
41+
}
42+
}

‎app/code/Magento/SalesRule/Plugin/CouponUsagesIncrement.php

+39
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
<?php
2+
/**
3+
* Copyright © Magento, Inc. All rights reserved.
4+
* See COPYING.txt for license details.
5+
*/
6+
7+
namespace Magento\SalesRule\Plugin;
8+
9+
use Magento\Sales\Model\Order;
10+
use Magento\SalesRule\Model\Coupon\UpdateCouponUsages;
11+
12+
class CouponUsagesIncrement
13+
{
14+
/**
15+
* @var UpdateCouponUsages
16+
*/
17+
private $updateCouponUsages;
18+
19+
public function __construct(
20+
UpdateCouponUsages $updateCouponUsages
21+
) {
22+
$this->updateCouponUsages = $updateCouponUsages;
23+
}
24+
25+
/**
26+
* Increments number of coupon usages after placing order.
27+
*
28+
* @param Order $subject
29+
* @param Order $result
30+
* @return Order
31+
* @SuppressWarnings(PHPMD.UnusedFormalParameter)
32+
*/
33+
public function afterPlace(Order $subject, Order $result)
34+
{
35+
$this->updateCouponUsages->execute($subject, true);
36+
37+
return $subject;
38+
}
39+
}

0 commit comments

Comments
 (0)