UpgradeData.php 2.79 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
namespace Magento\Integration\Setup;

use Magento\Framework\Setup\UpgradeDataInterface;
use Magento\Framework\Setup\ModuleContextInterface;
use Magento\Framework\Setup\ModuleDataSetupInterface;

/**
 * Upgrade data script for Integration module
 */
class UpgradeData implements UpgradeDataInterface
{
    /**
     * @inheritdoc
     */
    public function upgrade(ModuleDataSetupInterface $setup, ModuleContextInterface $context)
    {
        $setup->startSetup();

        if (version_compare($context->getVersion(), '2.2.0', '<')) {
            $this->removeRevokedTokens($setup);
            $this->removeTokensFromInactiveAdmins($setup);
            $this->removeTokensFromInactiveCustomers($setup);
        }

        $setup->endSetup();
    }

    /**
     * Remove any revoked tokens from oauth_token table
     *
     * @param ModuleDataSetupInterface $setup
     * @return void
     */
    private function removeRevokedTokens($setup)
    {
        $oauthTokenTable = $setup->getTable('oauth_token');

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

    /**
     * Remove any tokens from oauth_token table where admin is inactive
     *
     * @param ModuleDataSetupInterface $setup
     * @return void
     */
    private function removeTokensFromInactiveAdmins($setup)
    {
        $oauthTokenTable = $setup->getTable('oauth_token');
        $adminUserTable = $setup->getTable('admin_user');

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

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

    /**
     * Remove any tokens from oauth_token table where customer is inactive
     *
     * @param ModuleDataSetupInterface $setup
     * @return void
     */
    private function removeTokensFromInactiveCustomers($setup)
    {
        $oauthTokenTable = $setup->getTable('oauth_token');
        $adminUserTable = $setup->getTable('customer_entity');

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

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