vendor/store.shopware.com/swagb2bplatform/components/Budget/BridgePlatform/BudgetPreferenceSubscriber.php line 71

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.     private CurrencyService $currencyService;
  20.     private BudgetService $budgetService;
  22.     private AuthenticationService $authenticationService;
  24.     private BudgetRepository $budgetRepository;
  26.     public static function getSubscribedEvents(): array
  27.     {
  28.         return [
  29.             CheckoutConfirmPageLoadedEvent::class => 'updateCartPreference',
  30.             OrderContextCreatedEvent::class => 'addBudgetPreference',
  31.         ];
  32.     }
  34.     public function __construct(
  35.         AuthenticationService $authenticationService,
  36.         BudgetRepository $budgetRepository,
  37.         CurrencyService $currencyService,
  38.         BudgetService $budgetService
  39.     ) {
  40.         $this->authenticationService = $authenticationService;
  41.         $this->budgetRepository = $budgetRepository;
  42.         $this->currencyService = $currencyService;
  43.         $this->budgetService = $budgetService;
  44.     }
  46.     public function updateCartPreference(CheckoutConfirmPageLoadedEvent $event): void
  47.     {
  48.         if (!$this->authenticationService->isB2b()) {
  49.             return;
  50.         }
  52.         $ownershipContext = $this->authenticationService->getIdentity()->getOwnershipContext();
  54.         try {
  55.             $this->budgetRepository->fetchOrderBudgetPreferenceByCart($ownershipContext);
  57.             return;
  58.         } catch (NotFoundException $e) {
  59.             //nth
  60.         }
  62.         $amount = $event->getPage()->getCart()->getPrice()->getNetPrice();
  63.         $context = $this->authenticationService->getIdentity()->getOwnershipContext();
  65.         $budgetId = $this->findPreferredBudgetId($context, $amount);
  67.         $this->budgetRepository
  68.             ->setOrderBudgetPreferenceByCart($budgetId);
  69.     }
  71.     public function addBudgetPreference(OrderContextCreatedEvent $event): void
  72.     {
  73.         $orderContext = $event->getOrderContext();
  74.         $ownershipContext = $this->authenticationService->getIdentity()->getOwnershipContext();
  76.         try {
  77.             $budgetId = $this->budgetRepository
  78.                 ->fetchOrderBudgetPreferenceByCart($ownershipContext);
  79.         } catch (NotFoundException $e) {
  80.             return;
  81.         }
  83.         $this->budgetRepository
  84.             ->setOrderBudgetPreferenceByOrderContextId($orderContext->id, $budgetId);
  86.         $this->budgetRepository
  87.             ->setOrderBudgetPreferenceByCart(IdValue::null());
  88.     }
  90.     /**
  91.      * @internal
  92.      */
  93.     protected function findPreferredBudgetId(OwnershipContext $context, float $amount): IdValue
  94.     {
  95.         $currencyContext = $this->currencyService->createCurrencyContext();
  97.         $budgets = $this->budgetService
  98.             ->getUserSelectableBudgetsWithStatus($context, $amount, $currencyContext);
  100.         $budgetId = IdValue::null();
  101.         foreach ($budgets as $budget) {
  102.             if ($budget->currentStatus->isSufficient) {
  103.                 $budgetId = $budget->id;
  104.                 break;
  105.             }
  106.         }
  108.         return $budgetId;
  109.     }
  110. }