custom/plugins/CioBudget/src/Service/SessionService.php line 33

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace CioBudget\Service;
  5. use CioBudget\Definition\Budget\BudgetEntity;
  6. use CioBudget\Definition\BudgetStore\BudgetStoreEntity;
  7. use Psr\Container\ContainerInterface;
  8. use Shopware\Core\Checkout\Customer\CustomerEntity;
  9. use Shopware\Core\Framework\Context;
  10. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  12. use Shopware\Core\System\SalesChannel\SalesChannelContext;
  14. class SessionService
  15. {
  16.     private BudgetInfoService $budgetInfoService;
  17.     private BudgetLoaderService $budgetLoaderService;
  18.     private ContainerInterface $container;
  19.     private StoreLoaderService $storeLoaderService;
  21.     public function __construct(BudgetInfoService $budgetInfoService, BudgetLoaderService $budgetLoaderService, StoreLoaderService $storeLoaderService)
  22.     {
  23.         $this->budgetInfoService = $budgetInfoService;
  24.         $this->budgetLoaderService = $budgetLoaderService;
  25.         $this->storeLoaderService = $storeLoaderService;
  26.     }
  28.     public function setContainer(ContainerInterface $container)
  29.     {
  30.         $this->container = $container;
  31.     }
  33.     public function getCurrentBudgetId()
  34.     {
  35.         return $this->container->get('session')->get('cio_current_budget');
  36.     }
  38.     public function setCustomerBudgetInSession(CustomerEntity $customer, SalesChannelContext $context)
  39.     {
  40.         if ($customerStores = $this->storeLoaderService->getCustomerStores($customer)) {
  41.             $firstStoreId = $customerStores[0]['id'];
  42.             $budgetId = $this->budgetLoaderService->getBudgetIdByStoreId($firstStoreId);
  44.             if ($budgetId) {
  45.                 $this->setBudgetIdInSession($budgetId, $context);
  46.             }
  47.         }
  48.     }
  50.     public function setBudgetIdInSession(string $budgetId, SalesChannelContext $context)
  51.     {
  52.         if (!$budgetId) {
  53.             return;
  54.         }
  56.         if ($this->budgetInfoService->isBudgetActive($budgetId, $context->getContext())) {
  57.             $this->container->get('session')->set('cio_current_budget', $budgetId);
  58.         }
  59.     }
  61.     public function getBudgetIdFromSession()
  62.     {
  63.         return $this->container->get('session')->get('cio_current_budget');
  64.     }
  66.     /**
  67.      * @param Context $context
  68.      * @return BudgetEntity|null
  69.      */
  70.     public function getBudgetEntityFromSession(Context $context)
  71.     {
  72.         if ($this->getCurrentBudgetId()) {
  73.             /** @var EntityRepositoryInterface $budgetRepository */
  74.             $budgetRepository = $this->container->get('cio_budget.repository');
  76.             return $budgetRepository->search(new Criteria([$this->getCurrentBudgetId()]), $context)->first();
  78.         }
  80.         return null;
  81.     }
  83.     /**
  84.      * @param Context $context
  85.      * @return BudgetStoreEntity|null
  86.      */
  87.     public function getBudgetStoreEntityFromSession(Context $context)
  88.     {
  89.         /** @var BudgetEntity $budget */
  90.         if ($budget = $this->getBudgetEntityFromSession($context)) {
  91.             /** @var EntityRepositoryInterface $budgetStoreRepository */
  92.             $budgetStoreRepository = $this->container->get('cio_budget_store.repository');
  94.             return $budgetStoreRepository->search((new Criteria([$budget->getStoreId()]))->addAssociation('region')->addAssociation('division')->addAssociation('addressbook'), $context)->first();
  96.         }
  98.         return null;
  99.     }
  101.     public function removeOldActiveBudget()
  102.     {
  103.         $this->container->get('session')->set('cio_current_budget', null);
  104.     }
  105. }