prestashop/modules/ps_accounts/controllers/admin/AdminAjaxV2PsAccountsController.php

<?php
/**
 * Copyright since 2007 PrestaShop SA and Contributors
 * PrestaShop is an International Registered Trademark & Property of PrestaShop SA
 *
 * NOTICE OF LICENSE
 *
 * This source file is subject to the Academic Free License version 3.0
 * that is bundled with this package in the file LICENSE.md.
 * It is also available through the world-wide-web at this URL:
 * https://opensource.org/licenses/AFL-3.0
 * If you did not receive a copy of the license and are unable to
 * obtain it through the world-wide-web, please send an email
 * to license@prestashop.com so we can send you a copy immediately.
 *
 * @author    PrestaShop SA and Contributors <contact@prestashop.com>
 * @copyright Since 2007 PrestaShop SA and Contributors
 * @license   https://opensource.org/licenses/AFL-3.0 Academic Free License version 3.0
 */
require_once __DIR__ . '/../../src/Http/Controller/AbstractAdminAjaxCorsController.php';

use PrestaShop\Module\PsAccounts\Account\Command\CreateIdentityCommand;
use PrestaShop\Module\PsAccounts\Account\Command\MigrateOrCreateIdentityV8Command;
use PrestaShop\Module\PsAccounts\Account\Command\VerifyIdentityCommand;
use PrestaShop\Module\PsAccounts\Account\Exception\RefreshTokenException;
use PrestaShop\Module\PsAccounts\Account\Query\GetContextQuery;
use PrestaShop\Module\PsAccounts\Account\StatusManager;
use PrestaShop\Module\PsAccounts\Context\ShopContext;
use PrestaShop\Module\PsAccounts\Cqrs\CommandBus;
use PrestaShop\Module\PsAccounts\Cqrs\QueryBus;
use PrestaShop\Module\PsAccounts\Http\Controller\AbstractAdminAjaxCorsController;
use PrestaShop\Module\PsAccounts\Log\Logger;
use PrestaShop\Module\PsAccounts\Service\Accounts;
use PrestaShop\Module\PsAccounts\Service\Accounts\AccountsException;
use PrestaShop\Module\PsAccounts\Service\Accounts\AccountsService;
use PrestaShop\Module\PsAccounts\Service\OAuth2;

/**
 * Controller for all ajax calls.
 */
class AdminAjaxV2PsAccountsController extends AbstractAdminAjaxCorsController
{
    /**
     * @var CommandBus
     */
    private $commandBus;

    /**
     * @var QueryBus
     */
    private $queryBus;

    /**
     * @var ShopContext
     */
    private $shopContext;

    /**
     * AdminAjaxV2PsAccountsController constructor.
     *
     * @throws Exception
     */
    public function __construct()
    {
        parent::__construct();

        $this->commandBus = $this->module->getService(CommandBus::class);
        $this->queryBus = $this->module->getService(QueryBus::class);
        $this->shopContext = $this->module->getService(ShopContext::class);
    }

    /**
     * @return void
     *
     * @throws Exception
     */
    public function ajaxProcessGetContext()
    {
        $command = new GetContextQuery(
            Tools::getValue('source', 'ps_accounts'),
            Tools::getValue('context_type', null),
            Tools::getValue('context_id', null),
            filter_var(Tools::getValue('refresh', false), FILTER_VALIDATE_BOOLEAN)
        );

        $this->ajaxRender(
            (string) json_encode($this->queryBus->handle($command))
        );
    }

    /**
     * @return void
     *
     * @throws Exception
     */
    public function ajaxProcessFallbackCreateIdentity()
    {
        $shopId = Tools::getValue('shop_id', null);
        $source = Tools::getValue('source', 'ps_accounts');

        if (!$shopId) {
            throw new Exception('Shop ID is required for migration or creation.');
        }

        $this->shopContext->execInShopContext($shopId, function () use ($shopId, $source) {
            /** @var StatusManager $statusManager */
            $statusManager = $this->module->getService(StatusManager::class);
            $statusManager->withThrowException(true);

            $command = (new MigrateOrCreateIdentityV8Command($shopId))
                ->withOrigin(AccountsService::ORIGIN_FALLBACK)
                ->withSource($source);

            $this->commandBus->handle($command);

            $statusManager->resetThrowException();
        });

        $this->ajaxRender(
            (string) json_encode([
                'success' => true,
            ])
        );
    }

    /**
     * @return void
     *
     * @throws Exception
     */
    public function ajaxProcessRenewIdentity()
    {
        $shopId = Tools::getValue('shop_id', null);
        $source = Tools::getValue('source', 'ps_accounts');

        if (!$shopId) {
            throw new Exception('Shop ID is required for renew.');
        }

        $this->shopContext->execInShopContext($shopId, function () use ($shopId, $source) {
            $command = (new CreateIdentityCommand($shopId, true))
                ->withOrigin(AccountsService::ORIGIN_MISMATCH_CREATE)
                ->withSource($source);

            $this->commandBus->handle($command);
        });

        $this->ajaxRender(
            (string) json_encode([
                'success' => true,
            ])
        );
    }

    /**
     * @return void
     *
     * @throws Exception
     */
    public function ajaxProcessUpdateIdentity()
    {
        $shopId = Tools::getValue('shop_id', null);
        $source = Tools::getValue('source', 'ps_accounts');

        if (!$shopId) {
            throw new Exception('Shop ID is required for update.');
        }

        $this->shopContext->execInShopContext($shopId, function () use ($shopId, $source) {
            $command = (new VerifyIdentityCommand($shopId, true))
                ->withOrigin(AccountsService::ORIGIN_MISMATCH_UPDATE)
                ->withSource($source);

            $this->commandBus->handle($command);
        });

        $this->ajaxRender(
            (string) json_encode([
                'success' => true,
            ])
        );
    }

    /**
     * @param \Throwable|\Exception $e
     *
     * @return void
     */
    protected function handleError($e)
    {
        Logger::getInstance()->error($e);

        if ($e instanceof RefreshTokenException) {
            $e = $e->getPrevious();
        }

        if ($e instanceof OAuth2\Exception\ConnectException) {
            http_response_code(400);

            $this->ajaxRender(
                (string) json_encode([
                    'message' => $e->getMessage(),
                    'code' => 'oauth-server/connect-error',
                    //'details' => $e->getDetails(),
                ])
            );

            return;
        }

        if ($e instanceof Accounts\Exception\ConnectException) {
            http_response_code(400);

            $this->ajaxRender(
                (string) json_encode([
                    'message' => $e->getMessage(),
                    'code' => 'accounts-api/connect-error',
                    //'details' => $e->getDetails(),
                ])
            );

            return;
        }

        if ($e instanceof AccountsException) {
            http_response_code(400);

            $this->ajaxRender(
                (string) json_encode([
                    'message' => $e->getMessage(),
                    'code' => $e->getErrorCode(),
                    'details' => $e->getDetails(),
                ])
            );

            return;
        }

        parent::handleError($e);
    }
}