<?php
namespace SR\Inspiration\Controller\Index;
class Index extends \Magento\Framework\App\Action\Action
{
protected $_pageFactory;
public function __construct(
\Magento\Framework\App\Action\Context $context,
\Magento\Framework\View\Result\PageFactory $pageFactory,
\Magento\Customer\Model\Session $customerSession,
\SR\Inspiration\Model\FavouriteFactory $favouriteFactory,
\SR\Inspiration\Model\ResourceModel\Favourite\CollectionFactory $favouriteCollection
)
{
$this->_pageFactory = $pageFactory;
$this->customerSession = $customerSession;
$this->_favouriteFactory = $favouriteFactory;
$this->_favouriteCollection = $favouriteCollection;
return parent::__construct($context);
}

public function execute()
{
$pid = $this->getRequest()->getParam('id');
if(!$pid){
$this->messageManager->addError(
__('Item not found.'));
$this->_redirect('couponnenbak');
return;
}
$favourites = $this->customerSession->getFavourites();
if(!is_array($favourites)){
$favourites = array();
}
if(in_array($pid,$favourites)){
// already exist
$this->messageManager->addError(
__('Item is already added in favourite list.'));
$this->_redirect('couponnenbak');
return;
}
$newfavourites = array_merge($favourites,[$pid]);
try{
if($this->customerSession->isLoggedIn()){
$customer = $this->customerSession->getCustomer();
$collection = $this->_favouriteCollection->create();
$collection->addFieldToFilter('customer_id',$customer->getId());
$collection->addFieldToFilter('product_id',$pid);
$recordid = '';
if($collection->count()){
foreach($collection as $item){
$recordid = $item->getId();
break;
}
}
$favouritemodel = $this->_favouriteFactory->create();
if($recordid){
$favouritemodel->load($recordid);
}
$favouritemodel->setCustomerId($customer->getId());
$favouritemodel->setProductId($pid);
$favouritemodel->save();
}

$this->customerSession->setFavourites($newfavourites);
$this->messageManager->addSuccess(
__('Item is added successfully in favourite list.'));
}catch (\Exception $e) {
echo 'Error ='.$e->getMessage();
exit;
$this->messageManager->addError(
__('We can\'t process your request right now. Sorry, that\'s all we know.'));
}
$this->_redirect('couponnenbak');
return;
//return $this->_pageFactory->create();
}
}