Passed
Pull Request — master (#1284)
by René
04:02
created

Version0106Date20201031080745::preSchemaChange()   A

Complexity

Conditions 4
Paths 4

Size

Total Lines 31
Code Lines 21

Duplication

Lines 0
Ratio 0 %

Importance

Changes 2
Bugs 0 Features 0
Metric Value
eloc 21
c 2
b 0
f 0
dl 0
loc 31
rs 9.584
cc 4
nc 4
nop 3
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\IConfig;
28
use OCP\IDBConnection;
29
use OCP\Migration\SimpleMigrationStep;
30
use OCP\Migration\IOutput;
31
32
class Version0106Date20201031080745 extends SimpleMigrationStep {
33
34
	/** @var IDBConnection */
35
	protected $connection;
36
37
	/** @var IConfig */
38
	protected $config;
39
40
	public function __construct(IDBConnection $connection, IConfig $config) {
41
		$this->connection = $connection;
42
		$this->config = $config;
43
	}
44
45
	public function changeSchema(IOutput $output, \Closure $schemaClosure, array $options) {
46
		/** @var ISchemaWrapper $schema */
47
		$schema = $schemaClosure();
48
		if ($schema->hasTable('polls_share')) {
49
			$table = $schema->getTable('polls_share');
50
51
			if (!$table->hasColumn('display_name')) {
52
				$table->addColumn('display_name', 'string', [
53
					'notnull' => false,
54
					'length' => 64,
55
					'default' => ''
56
				]);
57
			}
58
59
			if (!$table->hasColumn('email_address')) {
60
				$table->addColumn('email_address', 'string', [
61
					'notnull' => false,
62
					'length' => 254,
63
					'default' => ''
64
				]);
65
			}
66
		}
67
68
		if ($schema->hasTable('polls_preferences')) {
69
			$table = $schema->getTable('polls_preferences');
70
			$table->changeColumn('user_id', [
71
				'default' => ''
72
			]);
73
			$table->changeColumn('preferences', [
74
				'notnull' => false
75
			]);
76
			try {
77
				$table->addUniqueIndex(['user_id']);
78
			} catch (\Exception $e) {
79
				//catch silently, index is already present
80
			}
81
		}
82
83
		return $schema;
84
	}
85
86
	/**
87
	 * @return void
88
	 */
89
	public function postSchemaChange(IOutput $output, \Closure $schemaClosure, array $options) {
90
		$schema = $schemaClosure();
91
		if ($schema->hasTable('polls_share')) {
92
			$table = $schema->getTable('polls_share');
93
			if ($table->hasColumn('email_address') && $table->hasColumn('user_email')) {
94
				$query = $this->connection->getQueryBuilder();
95
				$query->update('polls_share')
96
					->set('email_address', 'user_email');
97
				$query->execute();
98
			}
99
		}
100
	}
101
}
102