RemoveInactiveTokens.php

Go to the documentation of this file.

<?php
namespace Magento\Integration\Setup\Patch\Data;

use Magento\Framework\App\ResourceConnection;
use Magento\Framework\Setup\Patch\DataPatchInterface;
use Magento\Framework\Setup\Patch\PatchVersionInterface;

class RemoveInactiveTokens implements DataPatchInterface, PatchVersionInterface
{
    private $moduleDataSetup;

    public function __construct(
        \Magento\Framework\Setup\ModuleDataSetupInterface $moduleDataSetup
    ) {
        $this->moduleDataSetup = $moduleDataSetup;
    }

    public function apply()
    {
        $this->moduleDataSetup->getConnection()->startSetup();

        $this->removeRevokedTokens();
        $this->removeTokensFromInactiveAdmins();
        $this->removeTokensFromInactiveCustomers();

        $this->moduleDataSetup->getConnection()->endSetup();
    }

    public static function getDependencies()
    {
        return [];
    }

    public static function getVersion()
    {
        return '2.2.0';
    }

    public function getAliases()
    {
        return [];
    }

    private function removeRevokedTokens()
    {
        $oauthTokenTable = $this->moduleDataSetup->getTable('oauth_token');

        $where = ['revoked = ?' => 1];
        $this->moduleDataSetup->getConnection()->delete($oauthTokenTable, $where);
    }

    private function removeTokensFromInactiveAdmins()
    {
        $oauthTokenTable = $this->moduleDataSetup->getTable('oauth_token');
        $adminUserTable = $this->moduleDataSetup->getTable('admin_user');

        $select = $this->moduleDataSetup->getConnection()->select()->from(
            $adminUserTable,
            ['user_id', 'is_active']
        );

        $admins = $this->moduleDataSetup->getConnection()->fetchAll($select);
        foreach ($admins as $admin) {
            if ($admin['is_active'] == 0) {
                $where = ['admin_id = ?' => (int)$admin['user_id']];
                $this->moduleDataSetup->getConnection()->delete($oauthTokenTable, $where);
            }
        }
    }

    private function removeTokensFromInactiveCustomers()
    {
        $oauthTokenTable = $this->moduleDataSetup->getTable('oauth_token');
        $adminUserTable = $this->moduleDataSetup->getTable('customer_entity');

        $select = $this->moduleDataSetup->getConnection()->select()->from(
            $adminUserTable,
            ['entity_id', 'is_active']
        );

        $admins = $this->moduleDataSetup->getConnection()->fetchAll($select);
        foreach ($admins as $admin) {
            if ($admin['is_active'] == 0) {
                $where = ['customer_id = ?' => (int)$admin['entity_id']];
                $this->moduleDataSetup->getConnection()->delete($oauthTokenTable, $where);
            }
        }
    }
}