From 2dc79efb32ed268637489a5411ca36aed5da9b45 Mon Sep 17 00:00:00 2001 From: Joas Schilling Date: Tue, 9 May 2017 16:59:18 +0200 Subject: [PATCH 1/3] Rename the table to a valid name Signed-off-by: Joas Schilling --- apps/twofactor_backupcodes/appinfo/database.xml | 2 +- apps/twofactor_backupcodes/lib/Db/BackupCodeMapper.php | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/apps/twofactor_backupcodes/appinfo/database.xml b/apps/twofactor_backupcodes/appinfo/database.xml index 02ca7c93a1..90a7730768 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 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(); } From 74a6a6c839101be9b849194db186fad85edc4d35 Mon Sep 17 00:00:00 2001 From: Joas Schilling Date: Tue, 9 May 2017 17:16:03 +0200 Subject: [PATCH 2/3] Add a update step that copies all the backup codes Signed-off-by: Joas Schilling --- apps/twofactor_backupcodes/appinfo/info.xml | 8 +- .../lib/Migration/CopyEntriesFromOldTable.php | 108 ++++++++++++++++++ 2 files changed, 115 insertions(+), 1 deletion(-) create mode 100644 apps/twofactor_backupcodes/lib/Migration/CopyEntriesFromOldTable.php 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/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'); + } +} From d89d96203affc8ae1cb0dd7da061bd2e853811b7 Mon Sep 17 00:00:00 2001 From: Joas Schilling Date: Mon, 15 May 2017 12:42:04 +0200 Subject: [PATCH 3/3] Rename the index so the name is unique until the old table is deleted Signed-off-by: Joas Schilling --- apps/twofactor_backupcodes/appinfo/database.xml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apps/twofactor_backupcodes/appinfo/database.xml b/apps/twofactor_backupcodes/appinfo/database.xml index 90a7730768..93b0c1ad56 100644 --- a/apps/twofactor_backupcodes/appinfo/database.xml +++ b/apps/twofactor_backupcodes/appinfo/database.xml @@ -37,7 +37,7 @@ - two_factor_backupcodes_user_id + twofactor_backupcodes_uid user_id ascending