storeManager = $storeManager; $this->productRepository = $productRepository; parent::__construct($context, $customerSession); } /** * Check if URL is internal * * @param string $url * @return bool */ protected function isInternal($url) { if (strpos($url, 'http') === false) { return false; } $currentStore = $this->storeManager->getStore(); return strpos($url, $currentStore->getBaseUrl()) === 0 || strpos($url, $currentStore->getBaseUrl(UrlInterface::URL_TYPE_LINK, true)) === 0; } /** * Method for adding info about product alert price. * * @return \Magento\Framework\Controller\Result\Redirect */ public function execute() { $backUrl = $this->getRequest()->getParam(Action::PARAM_NAME_URL_ENCODED); $productId = (int)$this->getRequest()->getParam('product_id'); /** @var \Magento\Framework\Controller\Result\Redirect $resultRedirect */ $resultRedirect = $this->resultFactory->create(ResultFactory::TYPE_REDIRECT); if (!$backUrl || !$productId) { $resultRedirect->setPath('/'); return $resultRedirect; } $store = $this->storeManager->getStore(); try { /* @var $product \Magento\Catalog\Model\Product */ $product = $this->productRepository->getById($productId); /** @var \Magento\ProductAlert\Model\Price $model */ $model = $this->_objectManager->create(\Magento\ProductAlert\Model\Price::class) ->setCustomerId($this->customerSession->getCustomerId()) ->setProductId($product->getId()) ->setPrice($product->getFinalPrice()) ->setWebsiteId($store->getWebsiteId()) ->setStoreId($store->getId()); $model->save(); $this->messageManager->addSuccess(__('You saved the alert subscription.')); } catch (NoSuchEntityException $noEntityException) { $this->messageManager->addError(__('There are not enough parameters.')); if ($this->isInternal($backUrl)) { $resultRedirect->setUrl($backUrl); } else { $resultRedirect->setPath('/'); } return $resultRedirect; } catch (\Exception $e) { $this->messageManager->addException( $e, __("The alert subscription couldn't update at this time. Please try again later.") ); } $resultRedirect->setUrl($this->_redirect->getRedirectUrl()); return $resultRedirect; } }