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);
}
}
}
}