How to rewrite customer controller in Magento 2?

We can do it by using preference if we need to add or edit some information in createPost controller.

To do that we will need a module : Magemonkeys_General

We have need two files.

Create di.xml file in this folder app/code/Magemonkeys/General/etc/

<?xml version="1.0"?>
<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:ObjectManager/etc/config.xsd">
    <preference for="Magento\Customer\Controller\Account\CreatePost" type="Magemonkeys\General\Controller\Magento\Customer\Account\CreatePost" />
</config>

<?xml version="1.0"?>

</config>

Create CreatePost.php in this folder app/code/Magemonkeys/General/Controller/Magento/Customer/Account/

<?php

namespace Magemonkeys\General\Controller\Magento\Customer\Account;
use Magento\Customer\Api\CustomerRepositoryInterface as CustomerRepository;
use Magento\Framework\App\Action\HttpPostActionInterface as HttpPostActionInterface;
use Magento\Customer\Model\Account\Redirect as AccountRedirect;
use Magento\Customer\Api\Data\AddressInterface;
use Magento\Framework\Api\DataObjectHelper;
use Magento\Framework\App\Action\Context;
use Magento\Customer\Model\Session;
use Magento\Framework\App\Config\ScopeConfigInterface;
use Magento\Framework\App\CsrfAwareActionInterface;
use Magento\Framework\App\ObjectManager;
use Magento\Framework\App\Request\InvalidRequestException;
use Magento\Framework\App\RequestInterface;
use Magento\Framework\Controller\Result\Redirect;
use Magento\Framework\Exception\LocalizedException;
use Magento\Framework\Phrase;
use Magento\Store\Model\StoreManagerInterface;
use Magento\Customer\Api\AccountManagementInterface;
use Magento\Customer\Helper\Address;
use Magento\Framework\UrlFactory;
use Magento\Customer\Model\Metadata\FormFactory;
use Magento\Newsletter\Model\SubscriberFactory;
use Magento\Customer\Api\Data\RegionInterfaceFactory;
use Magento\Customer\Api\Data\AddressInterfaceFactory;
use Magento\Customer\Api\Data\CustomerInterfaceFactory;
use Magento\Customer\Model\Url as CustomerUrl;
use Magento\Customer\Model\Registration;
use Magento\Framework\Escaper;
use Magento\Customer\Model\CustomerExtractor;
use Magento\Framework\Exception\StateException;
use Magento\Framework\Exception\InputException;
use Magento\Framework\Data\Form\FormKey\Validator;
use Magento\Customer\Controller\AbstractAccount;

class CreatePost extends \Magento\Customer\Controller\Account\CreatePost
{
	/**
     * @var Validator
     */
    private $formKeyValidator;
    protected $inlineTranslation;

    /**
     * @var \Magento\Framework\Stdlib\Cookie\PhpCookieManager
     */
    private $cookieMetadataManager;

    /**
     * @var \Magento\Framework\Stdlib\Cookie\CookieMetadataFactory
     */
    private $cookieMetadataFactory;

    public function __construct(
    	Context $context,
        Session $customerSession,
        ScopeConfigInterface $scopeConfig,
        StoreManagerInterface $storeManager,
        AccountManagementInterface $accountManagement,
        Address $addressHelper,
        UrlFactory $urlFactory,
        FormFactory $formFactory,
        SubscriberFactory $subscriberFactory,
        RegionInterfaceFactory $regionDataFactory,
        AddressInterfaceFactory $addressDataFactory,
        CustomerInterfaceFactory $customerDataFactory,
        CustomerUrl $customerUrl,
        Registration $registration,
        Escaper $escaper,
        CustomerExtractor $customerExtractor,
        DataObjectHelper $dataObjectHelper,
        AccountRedirect $accountRedirect,
        CustomerRepository $customerRepository,
        Validator $formKeyValidator = null
    ) {
    	$this->formKeyValidator = $formKeyValidator ?: ObjectManager::getInstance()->get(Validator::class);
        $this->accountRedirect = $accountRedirect;
    	return parent::__construct($context,$customerSession,$scopeConfig,$storeManager,$accountManagement,$addressHelper,$urlFactory,$formFactory,$subscriberFactory,$regionDataFactory,$addressDataFactory,$customerDataFactory,$customerUrl,$registration,$escaper,$customerExtractor,$dataObjectHelper,$accountRedirect,$customerRepository,$formKeyValidator);
    }

	public function execute()
    {          
        /** @var Redirect $resultRedirect */
        $resultRedirect = $this->resultRedirectFactory->create();
        if ($this->session->isLoggedIn() || !$this->registration->isAllowed()) {
            $resultRedirect->setPath('*/*/');
            return $resultRedirect;
        }

        // We can write our login here
    }
}

<?php

namespace Magemonkeys\General\Controller\Magento\Customer\Account;

use Magento\Customer\Api\CustomerRepositoryInterface as CustomerRepository;

use Magento\Framework\App\Action\HttpPostActionInterface as HttpPostActionInterface;

use Magento\Customer\Model\Account\Redirect as AccountRedirect;

use Magento\Customer\Api\Data\AddressInterface;

use Magento\Framework\Api\DataObjectHelper;

use Magento\Framework\App\Action\Context;

use Magento\Customer\Model\Session;

use Magento\Framework\App\Config\ScopeConfigInterface;

use Magento\Framework\App\CsrfAwareActionInterface;

use Magento\Framework\App\ObjectManager;

use Magento\Framework\App\Request\InvalidRequestException;

use Magento\Framework\App\RequestInterface;

use Magento\Framework\Controller\Result\Redirect;

use Magento\Framework\Exception\LocalizedException;

use Magento\Framework\Phrase;

use Magento\Store\Model\StoreManagerInterface;

use Magento\Customer\Api\AccountManagementInterface;

use Magento\Customer\Helper\Address;

use Magento\Framework\UrlFactory;

use Magento\Customer\Model\Metadata\FormFactory;

use Magento\Newsletter\Model\SubscriberFactory;

use Magento\Customer\Api\Data\RegionInterfaceFactory;

use Magento\Customer\Api\Data\AddressInterfaceFactory;

use Magento\Customer\Api\Data\CustomerInterfaceFactory;

use Magento\Customer\Model\Url as CustomerUrl;

use Magento\Customer\Model\Registration;

use Magento\Framework\Escaper;

use Magento\Customer\Model\CustomerExtractor;

use Magento\Framework\Exception\StateException;

use Magento\Framework\Exception\InputException;

use Magento\Framework\Data\Form\FormKey\Validator;

use Magento\Customer\Controller\AbstractAccount;

class CreatePost extends \Magento\Customer\Controller\Account\CreatePost

{

/**

* @var Validator

private $formKeyValidator;

protected $inlineTranslation;

/**

* @var \Magento\Framework\Stdlib\Cookie\PhpCookieManager

private $cookieMetadataManager;

/**

* @var \Magento\Framework\Stdlib\Cookie\CookieMetadataFactory

private $cookieMetadataFactory;

public function __construct(

Context $context,

Session $customerSession,

ScopeConfigInterface $scopeConfig,

StoreManagerInterface $storeManager,

AccountManagementInterface $accountManagement,

Address $addressHelper,

UrlFactory $urlFactory,

FormFactory $formFactory,

SubscriberFactory $subscriberFactory,

RegionInterfaceFactory $regionDataFactory,

AddressInterfaceFactory $addressDataFactory,

CustomerInterfaceFactory $customerDataFactory,

CustomerUrl $customerUrl,

Registration $registration,

Escaper $escaper,

CustomerExtractor $customerExtractor,

DataObjectHelper $dataObjectHelper,

AccountRedirect $accountRedirect,

CustomerRepository $customerRepository,

Validator $formKeyValidator = null

) {

$this->formKeyValidator = $formKeyValidator ?: ObjectManager::getInstance()->get(Validator::class);

$this->accountRedirect = $accountRedirect;

return parent::__construct($context,$customerSession,$scopeConfig,$storeManager,$accountManagement,$addressHelper,$urlFactory,$formFactory,$subscriberFactory,$regionDataFactory,$addressDataFactory,$customerDataFactory,$customerUrl,$registration,$escaper,$customerExtractor,$dataObjectHelper,$accountRedirect,$customerRepository,$formKeyValidator);

}

public function execute()

{

/** @var Redirect $resultRedirect */

$resultRedirect = $this->resultRedirectFactory->create();

if ($this->session->isLoggedIn() || !$this->registration->isAllowed()) {

$resultRedirect->setPath('*/*/');

return $resultRedirect;

}

// We can write our login here

}

we will also need to add private function in this file as we used execute method.

/**
     * Retrieve cookie manager
     *
     * @deprecated 100.1.0
     * @return \Magento\Framework\Stdlib\Cookie\PhpCookieManager
     */
    private function getCookieManager()
    {
        if (!$this->cookieMetadataManager) {
            $this->cookieMetadataManager = ObjectManager::getInstance()->get(
                \Magento\Framework\Stdlib\Cookie\PhpCookieManager::class
            );
        }
        return $this->cookieMetadataManager;
    }

/**

* Retrieve cookie manager

* @deprecated 100.1.0

* @return \Magento\Framework\Stdlib\Cookie\PhpCookieManager

private function getCookieManager()

{

if (!$this->cookieMetadataManager) {

$this->cookieMetadataManager = ObjectManager::getInstance()->get(

\Magento\Framework\Stdlib\Cookie\PhpCookieManager::class

);

}

return $this->cookieMetadataManager;

}

That’s it.

Fill the below form if you have any Magento programming need.

Add an image programmatically from any url in Magento 2
[crayon-6387d0f7c853a390958181/] Using above fucntion Images can be imported directly from...
Magento 2 : Override the Category view block file
Override view block using di.xml and add the below code...
How to check list of Layout XML called for a specific page in Magento 2?
You can check a list of called layout XML for...
Magento 2 : Setup PWA from Scratch
Follow the below steps to install and set up PWA...
How to remove leading zero’s from the order, invoice and shipment number in Magento 2?
If you want to remove all leading zero's from order,...

Author Info

Magento 2 Developer

Hire P. Garchar

Let’s have a call

Fill the form & Get Installation Support