app/Plugin/PrivateURL42/Doctrine/EventSubscriber/PrivateSubscriber.php line 82

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of PrivateURL42
  5.  *
  6.  * Copyright(c) Akira Kurozumi <info@a-zumi.net>
  7.  *
  8.  * https://a-zumi.net
  9.  *
  10.  * For the full copyright and license information, please view the LICENSE
  11.  * file that was distributed with this source code.
  12.  */
  14. namespace Plugin\PrivateURL42\Doctrine\EventSubscriber;
  16. use Doctrine\Common\EventSubscriber;
  17. use Doctrine\ORM\EntityManagerInterface;
  18. use Doctrine\ORM\Event\PreUpdateEventArgs;
  19. use Doctrine\ORM\Events;
  20. use Doctrine\Persistence\Event\LifecycleEventArgs;
  21. use Eccube\Entity\Master\ProductStatus;
  22. use Eccube\Entity\Product;
  23. use Eccube\Request\Context;
  24. use Eccube\Util\StringUtil;
  25. use Plugin\PrivateURL42\Common\Constant;
  26. use Symfony\Component\HttpFoundation\RequestStack;
  28. class PrivateSubscriber implements EventSubscriber
  29. {
  30.     /**
  31.      * @var Context
  32.      */
  33.     protected Context $requestContext;
  35.     /**
  36.      * @var EntityManagerInterface
  37.      */
  38.     protected EntityManagerInterface $entityManager;
  40.     /**
  41.      * @var RequestStack
  42.      */
  43.     private RequestStack $requestStack;
  45.     public function __construct(
  46.         Context $requestContext,
  47.         EntityManagerInterface $entityManager,
  48.         RequestStack $requestStack
  49.     ) {
  50.         $this->requestContext = $requestContext;
  51.         $this->entityManager = $entityManager;
  52.         $this->requestStack = $requestStack;
  53.     }
  55.     /**
  56.      * @return array|string[]
  57.      */
  58.     public function getSubscribedEvents(): array
  59.     {
  60.         return [
  61.             Events::postLoad,
  62.             Events::preUpdate,
  63.             Events::prePersist,
  64.         ];
  65.     }
  67.     /**
  68.      * Product読み込み時の処理
  69.      *
  70.      * @param LifecycleEventArgs $args
  71.      *
  72.      * @return void
  73.      */
  74.     public function postLoad(LifecycleEventArgs $args): void
  75.     {
  76.         $entity = $args->getObject();
  77.         $entityManager = $args->getObjectManager();
  79.         if ($entity instanceof Product) {
  80.             // フロント処理
  81.             if ($this->requestContext->isFront()) {
  82.                 $private_key = $this->requestStack->getSession()->get(Constant::SESSION_PLG_PRIVATE_URL_PRIVATE_KEY.'_'.$entity->getId());
  84.                 // セッションに保持されているプライベートキーがDBと一致したら、商品ステータスを一時的に公開に変更
  85.                 if ($private_key && $entity->getPlgPuPrivateKey() === $private_key) {
  86.                     $productStatus = $entityManager->find(ProductStatus::class, ProductStatus::DISPLAY_SHOW);
  87.                     $entity->setStatus($productStatus);
  88.                 }
  89.             }
  90.         }
  91.     }
  93.     /**
  94.      * Product更新前処理
  95.      *
  96.      * @param PreUpdateEventArgs $args
  97.      *
  98.      * @return void
  99.      */
  100.     public function preUpdate(PreUpdateEventArgs $args): void
  101.     {
  102.         $entity = $args->getObject();
  104.         if ($entity instanceof Product) {
  105.             // 管理画面処理
  106.             if ($this->requestContext->isAdmin()) {
  107.                 // 限定公開URL生成に変化があった場合
  108.                 if ($args->hasChangedField('plg_pu_private')) {
  109.                     // 有効化された場合プライベートキー発行
  110.                     if ($entity->isPlgPuPrivate()) {
  111.                         $key = $this->getUniquePrivateKey();
  112.                         $entity->setPlgPuPrivateKey($key);
  113.                     } else {
  114.                         // 無効化されたらプライベートキー削除
  115.                         $entity->setPlgPuPrivateKey(null);
  116.                         // セッションから削除
  117.                         $this->requestStack->getSession()->remove(Constant::SESSION_PLG_PRIVATE_URL_PRIVATE_KEY.'_'.$entity->getId());
  118.                     }
  119.                 }
  120.             }
  122.             // フロント処理
  123.             if ($this->requestContext->isFront()) {
  124.                 // postLoadで公開に変更した場合、商品ステータスを非公開に戻す
  125.                 if ($args->hasChangedField('Status')) {
  126.                     $entity->setStatus($args->getOldValue('Status'));
  127.                 }
  128.             }
  129.         }
  130.     }
  132.     /**
  133.      * 新規作成したときに限定公開URL生成が有効だったら秘密キーを生成
  134.      *
  135.      * @param LifecycleEventArgs $args
  136.      *
  137.      * @return void
  138.      */
  139.     public function prePersist(LifecycleEventArgs $args): void
  140.     {
  141.         $entity = $args->getObject();
  143.         if ($entity instanceof Product) {
  144.             // 管理画面処理
  145.             if ($this->requestContext->isAdmin()) {
  146.                 // 有効化された場合プライベートキー発行
  147.                 if ($entity->isPlgPuPrivate()) {
  148.                     $key = $this->getUniquePrivateKey();
  149.                     $entity->setPlgPuPrivateKey($key);
  150.                 }
  151.             }
  152.         }
  153.     }
  155.     /**
  156.      * @return string
  157.      */
  158.     private function getUniquePrivateKey(): string
  159.     {
  160.         do {
  161.             $key = StringUtil::random(32);
  162.             $Product = $this->entityManager->getRepository(Product::class)->findOneBy(['plg_pu_private_key' => $key]);
  163.         } while ($Product);
  165.         return $key;
  166.     }
  167. }