You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.1 KiB
105 lines
3.1 KiB
<?php |
|
/** |
|
* @copyright Copyright (c) 2017 Joas Schilling <coding@schilljs.com> |
|
* |
|
* @author Joas Schilling <coding@schilljs.com> |
|
* |
|
* @license GNU AGPL version 3 or any later version |
|
* |
|
* This program is free software: you can redistribute it and/or modify |
|
* it under the terms of the GNU Affero General Public License as |
|
* published by the Free Software Foundation, either version 3 of the |
|
* License, or (at your option) any later version. |
|
* |
|
* This program is distributed in the hope that it will be useful, |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
* GNU Affero General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU Affero General Public License |
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
* |
|
*/ |
|
|
|
namespace OCA\TwoFactorBackupCodes\Migration; |
|
|
|
use OCP\DB\ISchemaWrapper; |
|
use OCP\DB\QueryBuilder\IQueryBuilder; |
|
use OCP\IDBConnection; |
|
use OCP\Migration\SimpleMigrationStep; |
|
use OCP\Migration\IOutput; |
|
|
|
class Version1002Date20170607113030 extends SimpleMigrationStep { |
|
|
|
/** @var IDBConnection */ |
|
protected $connection; |
|
|
|
/** |
|
* @param IDBConnection $connection |
|
*/ |
|
public function __construct(IDBConnection $connection) { |
|
$this->connection = $connection; |
|
} |
|
|
|
/** |
|
* @param IOutput $output |
|
* @param \Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper` |
|
* @param array $options |
|
* @since 13.0.0 |
|
*/ |
|
public function preSchemaChange(IOutput $output, \Closure $schemaClosure, array $options) { |
|
/** @var ISchemaWrapper $schema */ |
|
$schema = $schemaClosure(); |
|
|
|
if (!$schema->hasTable('twofactor_backup_codes')) { |
|
// Legacy table does not exist |
|
return; |
|
} |
|
|
|
$insert = $this->connection->getQueryBuilder(); |
|
$insert->insert('twofactor_backupcodes') |
|
->values([ |
|
// Inserting with id might fail: 'id' => $insert->createParameter('id'), |
|
'user_id' => $insert->createParameter('user_id'), |
|
'code' => $insert->createParameter('code'), |
|
'used' => $insert->createParameter('used'), |
|
]); |
|
|
|
$query = $this->connection->getQueryBuilder(); |
|
$query->select('*') |
|
->from('twofactor_backup_codes') |
|
->orderBy('id', 'ASC'); |
|
$result = $query->execute(); |
|
|
|
$output->startProgress(); |
|
while ($row = $result->fetch()) { |
|
$output->advance(); |
|
|
|
$insert |
|
// Inserting with id might fail: ->setParameter('id', $row['id'], IQueryBuilder::PARAM_INT) |
|
->setParameter('user_id', $row['user_id'], IQueryBuilder::PARAM_STR) |
|
->setParameter('code', $row['code'], IQueryBuilder::PARAM_STR) |
|
->setParameter('used', $row['used'], IQueryBuilder::PARAM_INT) |
|
->execute(); |
|
} |
|
$output->finishProgress(); |
|
} |
|
|
|
/** |
|
* @param IOutput $output |
|
* @param \Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper` |
|
* @param array $options |
|
* @return null|ISchemaWrapper |
|
* @since 13.0.0 |
|
*/ |
|
public function changeSchema(IOutput $output, \Closure $schemaClosure, array $options) { |
|
/** @var ISchemaWrapper $schema */ |
|
$schema = $schemaClosure(); |
|
|
|
if ($schema->hasTable('twofactor_backup_codes')) { |
|
$schema->dropTable('twofactor_backup_codes'); |
|
return $schema; |
|
} |
|
return null; |
|
} |
|
}
|
|
|