diff --git a/apps/twofactor_backupcodes/appinfo/database.xml b/apps/twofactor_backupcodes/appinfo/database.xml index 02ca7c93a1..93b0c1ad56 100644 --- a/apps/twofactor_backupcodes/appinfo/database.xml +++ b/apps/twofactor_backupcodes/appinfo/database.xml @@ -5,7 +5,7 @@ false utf8 - *dbprefix*twofactor_backup_codes + *dbprefix*twofactor_backupcodes id @@ -37,7 +37,7 @@ - two_factor_backupcodes_user_id + twofactor_backupcodes_uid user_id ascending diff --git a/apps/twofactor_backupcodes/appinfo/info.xml b/apps/twofactor_backupcodes/appinfo/info.xml index 42985d1548..e5e96eddcf 100644 --- a/apps/twofactor_backupcodes/appinfo/info.xml +++ b/apps/twofactor_backupcodes/appinfo/info.xml @@ -5,7 +5,7 @@ A two-factor auth backup codes provider agpl Christoph Wurst - 1.1.0 + 1.1.1 TwoFactorBackupCodes other @@ -29,4 +29,10 @@ OCA\TwoFactorBackupCodes\Activity\Provider + + + + OCA\TwoFactorBackupCodes\Migration\CopyEntriesFromOldTable + + diff --git a/apps/twofactor_backupcodes/lib/Db/BackupCodeMapper.php b/apps/twofactor_backupcodes/lib/Db/BackupCodeMapper.php index ff993683c5..03fe7bb5ec 100644 --- a/apps/twofactor_backupcodes/lib/Db/BackupCodeMapper.php +++ b/apps/twofactor_backupcodes/lib/Db/BackupCodeMapper.php @@ -22,13 +22,14 @@ namespace OCA\TwoFactorBackupCodes\Db; use OCP\AppFramework\Db\Mapper; +use OCP\DB\QueryBuilder\IQueryBuilder; use OCP\IDBConnection; use OCP\IUser; class BackupCodeMapper extends Mapper { public function __construct(IDBConnection $db) { - parent::__construct($db, 'twofactor_backup_codes'); + parent::__construct($db, 'twofactor_backupcodes'); } /** @@ -40,7 +41,7 @@ class BackupCodeMapper extends Mapper { $qb = $this->db->getQueryBuilder(); $qb->select('id', 'user_id', 'code', 'used') - ->from('twofactor_backup_codes') + ->from('twofactor_backupcodes') ->where($qb->expr()->eq('user_id', $qb->createNamedParameter($user->getUID()))); $result = $qb->execute(); @@ -66,7 +67,7 @@ class BackupCodeMapper extends Mapper { /* @var IQueryBuilder $qb */ $qb = $this->db->getQueryBuilder(); - $qb->delete('twofactor_backup_codes') + $qb->delete('twofactor_backupcodes') ->where($qb->expr()->eq('user_id', $qb->createNamedParameter($uid))); $qb->execute(); } diff --git a/apps/twofactor_backupcodes/lib/Migration/CopyEntriesFromOldTable.php b/apps/twofactor_backupcodes/lib/Migration/CopyEntriesFromOldTable.php new file mode 100644 index 0000000000..6d65a7d0c1 --- /dev/null +++ b/apps/twofactor_backupcodes/lib/Migration/CopyEntriesFromOldTable.php @@ -0,0 +1,108 @@ + + * + * @author Joas Schilling + * + * @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 . + * + */ + +namespace OCA\TwoFactorBackupCodes\Migration; + +use OCP\DB\QueryBuilder\IQueryBuilder; +use OCP\IConfig; +use OCP\IDBConnection; +use OCP\Migration\IOutput; +use OCP\Migration\IRepairStep; + +class CopyEntriesFromOldTable implements IRepairStep { + + /** @var IDBConnection */ + protected $connection; + + /** @var IConfig */ + protected $config; + + /** + * @param IDBConnection $connection + * @param IConfig $config + */ + public function __construct(IDBConnection $connection, IConfig $config) { + $this->connection = $connection; + $this->config = $config; + } + + /** + * Returns the step's name + * + * @return string + * @since 9.1.0 + */ + public function getName() { + return 'Copy twofactor backup codes from legacy table'; + } + + /** + * Run repair step. + * Must throw exception on error. + * + * @since 9.1.0 + * @param IOutput $output + * @throws \Exception in case of failure + */ + public function run(IOutput $output) { + $version = $this->config->getAppValue('twofactor_backupcodes', 'installed_version', '0.0.0'); + if (version_compare($version, '1.1.1', '>=')) { + return; + } + + if (!$this->connection->tableExists('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(); + + $this->connection->dropTable('twofactor_backup_codes'); + } +}