Passed
Pull Request — master (#1284)
by René
03:29
created

Version0107Date20201210204702::__construct()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 2
Code Lines 1

Duplication

Lines 0
Ratio 0 %

Importance

Changes 2
Bugs 2 Features 0
Metric Value
eloc 1
c 2
b 2
f 0
dl 0
loc 2
rs 10
cc 1
nc 1
nop 1
1
<?php
2
/**
3
 * @copyright Copyright (c) 2017 René Gieling <[email protected]>
4
 *
5
 * @author René Gieling <[email protected]>
6
 *
7
 * @license GNU AGPL version 3 or any later version
8
 *
9
 *  This program is free software: you can redistribute it and/or modify
10
 *  it under the terms of the GNU Affero General Public License as
11
 *  published by the Free Software Foundation, either version 3 of the
12
 *  License, or (at your option) any later version.
13
 *
14
 *  This program is distributed in the hope that it will be useful,
15
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
16
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17
 *  GNU Affero General Public License for more details.
18
 *
19
 *  You should have received a copy of the GNU Affero General Public License
20
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
21
 *
22
 */
23
24
namespace OCA\Polls\Migration;
25
26
use OCP\DB\ISchemaWrapper;
27
use OCP\Migration\IOutput;
28
use OCP\IDBConnection;
29
use OCP\Migration\SimpleMigrationStep;
30
use Doctrine\DBAL\Schema\SchemaException;
31
32
class Version0107Date20201210204702 extends SimpleMigrationStep {
33
34
	/** @var IDBConnection */
35
	protected $connection;
36
37
	public function __construct(IDBConnection $connection) {
38
		$this->connection = $connection;
39
	}
40
41
	/**
42
	 * @return void
43
	 */
44
	public function preSchemaChange(IOutput $output, \Closure $schemaClosure, array $options) {
45
		$schema = $schemaClosure();
46
47
		if (!$schema->hasTable('polls_options')) {
48
			return;
49
		}
50
51
		// remove duplicates from oc_polls_options
52
		// preserve the first entry
53
		$query = $this->connection->getQueryBuilder();
54
		$query->select('id', 'poll_id', 'poll_option_text', 'timestamp')
55
			->from('polls_options');
56
		$foundEntries = $query->execute();
57
58
		$delete = $this->connection->getQueryBuilder();
59
		$delete->delete('polls_options')
60
			->where('id = :id');
61
62
		$entries2Keep = [];
63
64
		while ($row = $foundEntries->fetch()) {
65
			$currentRecord = [
66
				$row['poll_id'],
67
				$row['poll_option_text'],
68
				$row['timestamp']
69
			];
70
			if (in_array($currentRecord, $entries2Keep)) {
71
				$delete->setParameter('id', $row['id']);
72
				$delete->execute();
73
			} else {
74
				$entries2Keep[] = $currentRecord;
75
			}
76
		}
77
	}
78
79
	public function changeSchema(IOutput $output, \Closure $schemaClosure, array $options) {
80
		/** @var ISchemaWrapper $schema */
81
		$schema = $schemaClosure();
82
83
		if ($schema->hasTable('polls_options')) {
84
			$table = $schema->getTable('polls_options');
85
			$table->changeColumn('poll_option_text', [
86
				'notnull' => true,
87
				'default' => ''
88
			]);
89
			$table->changeColumn('timestamp', [
90
				'notnull' => true,
91
				'default' => 0
92
			]);
93
94
			try {
95
				$table->addUniqueIndex(['poll_id', 'poll_option_text', 'timestamp'], 'UNIQ_options');
96
			} catch (SchemaException $e) {
97
				//catch silently, index is already present
98
			}
99
		}
100
		return $schema;
101
	}
102
}
103