custom/plugins/SwagB2bPlatform/components/Budget/BridgePlatform/BudgetPreferenceSubscriber.php line 83

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Shopware\B2B\Budget\BridgePlatform;
  5. use Shopware\B2B\Budget\Framework\BudgetRepository;
  6. use Shopware\B2B\Budget\Framework\BudgetService;
  7. use Shopware\B2B\Common\IdValue;
  8. use Shopware\B2B\Common\Repository\NotFoundException;
  9. use Shopware\B2B\Currency\Framework\CurrencyService;
  10. use Shopware\B2B\Order\BridgePlatform\OrderContextCreatedEvent;
  11. use Shopware\B2B\StoreFrontAuthentication\Framework\AuthenticationService;
  12. use Shopware\B2B\StoreFrontAuthentication\Framework\OwnershipContext;
  13. use Shopware\Storefront\Page\Checkout\Confirm\CheckoutConfirmPageLoadedEvent;
  14. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  16. class BudgetPreferenceSubscriber implements EventSubscriberInterface
  17. {
  18.     /**
  19.      * @var CurrencyService
  20.      */
  21.     private $currencyService;
  23.     /**
  24.      * @var BudgetService
  25.      */
  26.     private $budgetService;
  28.     /**
  29.      * @var AuthenticationService
  30.      */
  31.     private $authenticationService;
  33.     /**
  34.      * @var BudgetRepository
  35.      */
  36.     private $budgetRepository;
  38.     public static function getSubscribedEvents(): array
  39.     {
  40.         return [
  41.             CheckoutConfirmPageLoadedEvent::class => 'updateCartPreference',
  42.             OrderContextCreatedEvent::class => 'addBudgetPreference',
  43.         ];
  44.     }
  46.     public function __construct(
  47.         AuthenticationService $authenticationService,
  48.         BudgetRepository $budgetRepository,
  49.         CurrencyService $currencyService,
  50.         BudgetService $budgetService
  51.     ) {
  52.         $this->authenticationService = $authenticationService;
  53.         $this->budgetRepository = $budgetRepository;
  54.         $this->currencyService = $currencyService;
  55.         $this->budgetService = $budgetService;
  56.     }
  58.     public function updateCartPreference(CheckoutConfirmPageLoadedEvent $event): void
  59.     {
  60.         if (!$this->authenticationService->isB2b()) {
  61.             return;
  62.         }
  64.         $ownershipContext = $this->authenticationService->getIdentity()->getOwnershipContext();
  66.         try {
  67.             $this->budgetRepository->fetchOrderBudgetPreferenceByCart($ownershipContext);
  69.             return;
  70.         } catch (NotFoundException $e) {
  71.             //nth
  72.         }
  74.         $amount = $event->getPage()->getCart()->getPrice()->getNetPrice();
  75.         $context = $this->authenticationService->getIdentity()->getOwnershipContext();
  77.         $budgetId = $this->findPreferredBudgetId($context, $amount);
  79.         $this->budgetRepository
  80.             ->setOrderBudgetPreferenceByCart($budgetId);
  81.     }
  83.     public function addBudgetPreference(OrderContextCreatedEvent $event): void
  84.     {
  85.         $orderContext = $event->getOrderContext();
  86.         $ownershipContext = $this->authenticationService->getIdentity()->getOwnershipContext();
  88.         try {
  89.             $budgetId = $this->budgetRepository
  90.                 ->fetchOrderBudgetPreferenceByCart($ownershipContext);
  91.         } catch (NotFoundException $e) {
  92.             return;
  93.         }
  95.         $this->budgetRepository
  96.             ->setOrderBudgetPreferenceByOrderContextId($orderContext->id, $budgetId);
  98.         $this->budgetRepository
  99.             ->setOrderBudgetPreferenceByCart(IdValue::null());
  100.     }
  102.     /**
  103.      * @internal
  104.      */
  105.     protected function findPreferredBudgetId(OwnershipContext $context, float $amount): IdValue
  106.     {
  107.         $currencyContext = $this->currencyService->createCurrencyContext();
  109.         $budgets = $this->budgetService
  110.             ->getUserSelectableBudgetsWithStatus($context, $amount, $currencyContext);
  112.         $budgetId = IdValue::null();
  113.         foreach ($budgets as $budget) {
  114.             if ($budget->currentStatus->isSufficient) {
  115.                 $budgetId = $budget->id;
  116.                 break;
  117.             }
  118.         }
  120.         return $budgetId;
  121.     }
  122. }