|
| 1 | +<?php |
| 2 | +/** |
| 3 | + * @copyright 2021 Louis Chemineau <[email protected]> |
| 4 | + * |
| 5 | + * @license GNU AGPL version 3 or any later version |
| 6 | + * |
| 7 | + * This program is free software: you can redistribute it and/or modify |
| 8 | + * it under the terms of the GNU Affero General Public License as |
| 9 | + * published by the Free Software Foundation, either version 3 of the |
| 10 | + * License, or (at your option) any later version. |
| 11 | + * |
| 12 | + * This program is distributed in the hope that it will be useful, |
| 13 | + * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 15 | + * GNU Affero General Public License for more details. |
| 16 | + * |
| 17 | + * You should have received a copy of the GNU Affero General Public License |
| 18 | + * along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 19 | + * |
| 20 | + */ |
| 21 | +namespace OC\Repair\Owncloud; |
| 22 | + |
| 23 | +use OCP\Migration\IOutput; |
| 24 | +use OCP\Migration\IRepairStep; |
| 25 | +use OC\DB\Connection; |
| 26 | +use OC\DB\SchemaWrapper; |
| 27 | +use OCP\DB\QueryBuilder\IQueryBuilder; |
| 28 | + |
| 29 | +class MigrateOauthTables implements IRepairStep { |
| 30 | + |
| 31 | + /** @var Connection */ |
| 32 | + protected $db; |
| 33 | + |
| 34 | + /** |
| 35 | + * @param Connection $db |
| 36 | + */ |
| 37 | + public function __construct(Connection $db) { |
| 38 | + $this->db = $db; |
| 39 | + } |
| 40 | + |
| 41 | + /** |
| 42 | + * @return string |
| 43 | + */ |
| 44 | + public function getName() { |
| 45 | + return 'Migrate oauth2_clients table to nextcloud schema'; |
| 46 | + } |
| 47 | + |
| 48 | + public function run(IOutput $output) { |
| 49 | + $schema = new SchemaWrapper($this->db); |
| 50 | + if (!$schema->hasTable('oauth2_clients')) { |
| 51 | + $output->info("oauth2_clients table does not exist."); |
| 52 | + return; |
| 53 | + } |
| 54 | + |
| 55 | + $output->info("Update the oauth2_access_tokens table schema."); |
| 56 | + $schema = new SchemaWrapper($this->db); |
| 57 | + $table = $schema->getTable('oauth2_access_tokens'); |
| 58 | + $table->addColumn('hashed_code', 'string', [ |
| 59 | + 'notnull' => true, |
| 60 | + 'length' => 128, |
| 61 | + ]); |
| 62 | + $table->addColumn('encrypted_token', 'string', [ |
| 63 | + 'notnull' => true, |
| 64 | + 'length' => 786, |
| 65 | + ]); |
| 66 | + $table->addUniqueIndex(['hashed_code'], 'oauth2_access_hash_idx'); |
| 67 | + $table->addIndex(['client_id'], 'oauth2_access_client_id_idx'); |
| 68 | + |
| 69 | + |
| 70 | + $output->info("Update the oauth2_clients table schema."); |
| 71 | + $schema = new SchemaWrapper($this->db); |
| 72 | + $table = $schema->getTable('oauth2_clients'); |
| 73 | + $table->getColumn('name')->setLength(64); |
| 74 | + $table->dropColumn('allow_subdomains'); |
| 75 | + |
| 76 | + if (!$schema->getTable('oauth2_clients')->hasColumn('client_identifier')) { |
| 77 | + $table->addColumn('client_identifier', 'string', [ |
| 78 | + 'notnull' => true, |
| 79 | + 'length' => 64, |
| 80 | + 'default' => '' |
| 81 | + ]); |
| 82 | + $table->addIndex(['client_identifier'], 'oauth2_client_id_idx'); |
| 83 | + } |
| 84 | + |
| 85 | + $this->db->migrateToSchema($schema->getWrappedSchema()); |
| 86 | + |
| 87 | + |
| 88 | + if ($schema->getTable('oauth2_clients')->hasColumn('identifier')) { |
| 89 | + $output->info("Move identifier column's data to the new client_identifier column."); |
| 90 | + // 1. Fetch all [id, identifier] couple. |
| 91 | + $selectQuery = $this->db->getQueryBuilder(); |
| 92 | + $selectQuery->select('id', 'identifier')->from('oauth2_clients'); |
| 93 | + $result = $selectQuery->executeQuery(); |
| 94 | + $identifiers = $result->fetchAll(); |
| 95 | + $result->closeCursor(); |
| 96 | + |
| 97 | + // 2. Insert them into the client_identifier column. |
| 98 | + foreach ($identifiers as ["id" => $id, "identifier" => $clientIdentifier]) { |
| 99 | + $insertQuery = $this->db->getQueryBuilder(); |
| 100 | + $insertQuery->update('oauth2_clients') |
| 101 | + ->set('client_identifier', $insertQuery->createNamedParameter($clientIdentifier, IQueryBuilder::PARAM_STR)) |
| 102 | + ->where($insertQuery->expr()->eq('id', $insertQuery->createNamedParameter($id, IQueryBuilder::PARAM_INT))) |
| 103 | + ->executeStatement(); |
| 104 | + } |
| 105 | + |
| 106 | + $output->info("Drop the identifier column."); |
| 107 | + $schema = new SchemaWrapper($this->db); |
| 108 | + $table = $schema->getTable('oauth2_clients'); |
| 109 | + $table->dropColumn('identifier'); |
| 110 | + $this->db->migrateToSchema($schema->getWrappedSchema()); |
| 111 | + } |
| 112 | + } |
| 113 | +} |
0 commit comments