зеркало из https://github.com/nextcloud/spreed.git
421 строка
12 KiB
PHP
421 строка
12 KiB
PHP
<?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\Spreed\Migration;
|
|
|
|
use Doctrine\DBAL\Exception\InvalidFieldNameException;
|
|
use Doctrine\DBAL\Exception\TableNotFoundException;
|
|
use Doctrine\DBAL\Platforms\PostgreSqlPlatform;
|
|
use Doctrine\DBAL\Types\Type;
|
|
use OCP\DB\ISchemaWrapper;
|
|
use OCP\DB\QueryBuilder\IQueryBuilder;
|
|
use OCP\IConfig;
|
|
use OCP\IDBConnection;
|
|
use OCP\Migration\SimpleMigrationStep;
|
|
use OCP\Migration\IOutput;
|
|
|
|
class Version2001Date20171026134605 extends SimpleMigrationStep {
|
|
|
|
/** @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;
|
|
}
|
|
|
|
/**
|
|
* @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('talk_signaling')) {
|
|
$table = $schema->createTable('talk_signaling');
|
|
|
|
$table->addColumn('sender', Type::STRING, [
|
|
'notnull' => true,
|
|
'length' => 255,
|
|
]);
|
|
$table->addColumn('recipient', Type::STRING, [
|
|
'notnull' => true,
|
|
'length' => 255,
|
|
]);
|
|
$table->addColumn('message', Type::TEXT, [
|
|
'notnull' => true,
|
|
]);
|
|
$table->addColumn('timestamp', Type::INTEGER, [
|
|
'notnull' => true,
|
|
'length' => 11,
|
|
]);
|
|
|
|
$table->addIndex(['recipient', 'timestamp'], 'ts_recipient_time');
|
|
}
|
|
|
|
if (!$schema->hasTable('talk_rooms')) {
|
|
$table = $schema->createTable('talk_rooms');
|
|
|
|
$table->addColumn('id', Type::INTEGER, [
|
|
'autoincrement' => true,
|
|
'notnull' => true,
|
|
'length' => 11,
|
|
]);
|
|
$table->addColumn('name', Type::STRING, [
|
|
'notnull' => false,
|
|
'length' => 255,
|
|
'default' => '',
|
|
]);
|
|
$table->addColumn('token', Type::STRING, [
|
|
'notnull' => false,
|
|
'length' => 32,
|
|
'default' => '',
|
|
]);
|
|
$table->addColumn('type', Type::INTEGER, [
|
|
'notnull' => true,
|
|
'length' => 11,
|
|
]);
|
|
$table->addColumn('password', Type::STRING, [
|
|
'notnull' => false,
|
|
'length' => 255,
|
|
'default' => '',
|
|
]);
|
|
$table->addColumn('activeSince', Type::DATETIME, [
|
|
'notnull' => false,
|
|
]);
|
|
$table->addColumn('activeGuests', Type::INTEGER, [
|
|
'notnull' => true,
|
|
'length' => 4,
|
|
'default' => 0,
|
|
'unsigned' => true,
|
|
]);
|
|
|
|
$table->setPrimaryKey(['id']);
|
|
$table->addUniqueIndex(['token'], 'tr_room_token');
|
|
}
|
|
|
|
if (!$schema->hasTable('talk_participants')) {
|
|
$table = $schema->createTable('talk_participants');
|
|
|
|
$table->addColumn('userId', Type::STRING, [
|
|
'notnull' => false,
|
|
'length' => 255,
|
|
]);
|
|
$table->addColumn('roomId', Type::INTEGER, [
|
|
'notnull' => true,
|
|
'length' => 11,
|
|
]);
|
|
$table->addColumn('lastPing', Type::INTEGER, [
|
|
'notnull' => true,
|
|
'length' => 11,
|
|
]);
|
|
$table->addColumn('sessionId', Type::STRING, [
|
|
'notnull' => true,
|
|
'length' => 255,
|
|
]);
|
|
$table->addColumn('participantType', Type::SMALLINT, [
|
|
'notnull' => true,
|
|
'length' => 6,
|
|
'default' => 0,
|
|
]);
|
|
}
|
|
|
|
return $schema;
|
|
}
|
|
|
|
/**
|
|
* @param IOutput $output
|
|
* @param \Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper`
|
|
* @param array $options
|
|
* @since 13.0.0
|
|
*/
|
|
public function postSchemaChange(IOutput $output, \Closure $schemaClosure, array $options) {
|
|
|
|
if (version_compare($this->config->getAppValue('spreed', 'installed_version', '0.0.0'), '2.0.0', '<')) {
|
|
// Migrations only work after 2.0.0
|
|
return;
|
|
}
|
|
|
|
$roomIdMap = $this->copyRooms();
|
|
$this->copyParticipants($roomIdMap);
|
|
$this->fixNotifications($roomIdMap);
|
|
$this->fixActivities($roomIdMap);
|
|
$this->fixActivityMails($roomIdMap);
|
|
}
|
|
|
|
/**
|
|
* @return int[]
|
|
*/
|
|
protected function copyRooms() {
|
|
$roomIdMap = [];
|
|
|
|
$insert = $this->connection->getQueryBuilder();
|
|
$insert->insert('talk_rooms')
|
|
->values([
|
|
'name' => $insert->createParameter('name'),
|
|
'token' => $insert->createParameter('token'),
|
|
'type' => $insert->createParameter('type'),
|
|
'password' => $insert->createParameter('password'),
|
|
]);
|
|
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select('*')
|
|
->from('spreedme_rooms');
|
|
|
|
$result = $query->execute();
|
|
while ($row = $result->fetch()) {
|
|
$insert
|
|
->setParameter('name', $row['name'])
|
|
->setParameter('token', $row['token'])
|
|
->setParameter('type', (int) $row['type'], IQueryBuilder::PARAM_INT)
|
|
->setParameter('password', $row['password']);
|
|
$insert->execute();
|
|
|
|
$roomIdMap[(int)$row['id']] = $insert->getLastInsertId();
|
|
}
|
|
$result->closeCursor();
|
|
|
|
return $roomIdMap;
|
|
}
|
|
|
|
/**
|
|
* @param int[] $roomIdMap
|
|
*/
|
|
protected function copyParticipants(array $roomIdMap) {
|
|
|
|
$insert = $this->connection->getQueryBuilder();
|
|
if (!$this->connection->getDatabasePlatform() instanceof PostgreSqlPlatform) {
|
|
$insert->insert('talk_participants')
|
|
->values([
|
|
'userId' => $insert->createParameter('userId'),
|
|
'roomId' => $insert->createParameter('roomId'),
|
|
'lastPing' => $insert->createParameter('lastPing'),
|
|
'sessionId' => $insert->createParameter('sessionId'),
|
|
'participantType' => $insert->createParameter('participantType'),
|
|
]);
|
|
} else {
|
|
$insert->insert('talk_participants')
|
|
->values([
|
|
'userid' => $insert->createParameter('userId'),
|
|
'roomid' => $insert->createParameter('roomId'),
|
|
'lastping' => $insert->createParameter('lastPing'),
|
|
'sessionid' => $insert->createParameter('sessionId'),
|
|
'participanttype' => $insert->createParameter('participantType'),
|
|
]);
|
|
}
|
|
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select('*')
|
|
->from('spreedme_room_participants');
|
|
|
|
$result = $query->execute();
|
|
while ($row = $result->fetch()) {
|
|
if (!isset($roomIdMap[(int) $row['roomId']])) {
|
|
continue;
|
|
}
|
|
|
|
$insert
|
|
->setParameter('userId', $row['userId'])
|
|
->setParameter('roomId', $roomIdMap[(int) $row['roomId']], IQueryBuilder::PARAM_INT)
|
|
->setParameter('lastPing', (int) $row['lastPing'], IQueryBuilder::PARAM_INT)
|
|
->setParameter('sessionId', $row['sessionId'])
|
|
;
|
|
if (!$this->connection->getDatabasePlatform() instanceof PostgreSqlPlatform) {
|
|
$insert->setParameter('participantType', (int) $row['participantType'], IQueryBuilder::PARAM_INT);
|
|
} else {
|
|
$insert->setParameter('participantType', (int) $row['participanttype'], IQueryBuilder::PARAM_INT);
|
|
}
|
|
$insert->execute();
|
|
}
|
|
$result->closeCursor();
|
|
|
|
}
|
|
|
|
/**
|
|
* @param int[] $roomIdMap
|
|
*/
|
|
protected function fixNotifications(array $roomIdMap) {
|
|
|
|
$update = $this->connection->getQueryBuilder();
|
|
$update->update('notifications')
|
|
->set('object_id', $update->createParameter('newId'))
|
|
->where($update->expr()->eq('notification_id', $update->createParameter('id')));
|
|
|
|
$delete = $this->connection->getQueryBuilder();
|
|
$delete->delete('notifications')
|
|
->where($delete->expr()->eq('notification_id', $delete->createParameter('id')));
|
|
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select(['notification_id', 'object_id'])
|
|
->from('notifications')
|
|
->where($query->expr()->eq('app', $query->createNamedParameter('spreed')))
|
|
->andWhere($query->expr()->eq('object_type', $query->createNamedParameter('room')));
|
|
|
|
try {
|
|
$result = $query->execute();
|
|
} catch (TableNotFoundException $e) {
|
|
return;
|
|
}
|
|
|
|
while ($row = $result->fetch()) {
|
|
if (!isset($roomIdMap[(int) $row['object_id']])) {
|
|
$delete
|
|
->setParameter('id', (int) $row['notification_id'])
|
|
;
|
|
$delete->execute();
|
|
continue;
|
|
}
|
|
|
|
$update
|
|
->setParameter('id', (int) $row['notification_id'])
|
|
->setParameter('newId', $roomIdMap[(int) $row['object_id']])
|
|
;
|
|
$update->execute();
|
|
}
|
|
$result->closeCursor();
|
|
|
|
}
|
|
|
|
/**
|
|
* @param int[] $roomIdMap
|
|
*/
|
|
protected function fixActivities(array $roomIdMap) {
|
|
|
|
$update = $this->connection->getQueryBuilder();
|
|
$update->update('activity')
|
|
->set('object_id', $update->createParameter('newId'))
|
|
->set('subjectparams', $update->createParameter('subjectParams'))
|
|
->where($update->expr()->eq('activity_id', $update->createParameter('id')));
|
|
|
|
$delete = $this->connection->getQueryBuilder();
|
|
$delete->delete('activity')
|
|
->where($delete->expr()->eq('activity_id', $delete->createParameter('id')));
|
|
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select(['activity_id', 'object_id', 'subjectparams'])
|
|
->from('activity')
|
|
->where($query->expr()->eq('app', $query->createNamedParameter('spreed')))
|
|
->andWhere($query->expr()->eq('type', $query->createNamedParameter('spreed')))
|
|
->andWhere($query->expr()->eq('object_type', $query->createNamedParameter('room')));
|
|
|
|
try {
|
|
$result = $query->execute();
|
|
} catch (TableNotFoundException $e) {
|
|
return;
|
|
} catch (InvalidFieldNameException $e) {
|
|
return;
|
|
}
|
|
|
|
while ($row = $result->fetch()) {
|
|
if (!isset($roomIdMap[(int) $row['object_id']])) {
|
|
$delete
|
|
->setParameter('id', (int) $row['activity_id'])
|
|
;
|
|
$delete->execute();
|
|
continue;
|
|
}
|
|
|
|
$params = json_decode($row['subjectparams'], true);
|
|
|
|
if (!isset($params['room'])) {
|
|
$delete
|
|
->setParameter('id', (int) $row['activity_id'])
|
|
;
|
|
$delete->execute();
|
|
continue;
|
|
}
|
|
|
|
$params['room'] = $roomIdMap[(int) $row['object_id']];
|
|
|
|
$update
|
|
->setParameter('id', (int) $row['activity_id'])
|
|
->setParameter('newId', $roomIdMap[(int) $row['object_id']])
|
|
->setParameter('subjectParams', json_encode($params))
|
|
;
|
|
$update->execute();
|
|
}
|
|
$result->closeCursor();
|
|
|
|
}
|
|
|
|
/**
|
|
* @param int[] $roomIdMap
|
|
*/
|
|
protected function fixActivityMails(array $roomIdMap) {
|
|
|
|
$update = $this->connection->getQueryBuilder();
|
|
$update->update('activity_mq')
|
|
->set('amq_subjectparams', $update->createParameter('subjectParams'))
|
|
->where($update->expr()->eq('mail_id', $update->createParameter('id')));
|
|
|
|
$delete = $this->connection->getQueryBuilder();
|
|
$delete->delete('activity_mq')
|
|
->where($delete->expr()->eq('mail_id', $delete->createParameter('id')));
|
|
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select(['mail_id', 'amq_subjectparams'])
|
|
->from('activity_mq')
|
|
->where($query->expr()->eq('amq_appid', $query->createNamedParameter('spreed')))
|
|
->andWhere($query->expr()->eq('amq_type', $query->createNamedParameter('spreed')));
|
|
|
|
try {
|
|
$result = $query->execute();
|
|
} catch (TableNotFoundException $e) {
|
|
return;
|
|
} catch (InvalidFieldNameException $e) {
|
|
return;
|
|
}
|
|
|
|
while ($row = $result->fetch()) {
|
|
$params = json_decode($row['subjectparams'], true);
|
|
|
|
if (!isset($params['room']) || !isset($roomIdMap[(int) $params['room']])) {
|
|
$delete
|
|
->setParameter('id', (int) $row['mail_id'])
|
|
;
|
|
$delete->execute();
|
|
continue;
|
|
}
|
|
|
|
$params['room'] = $roomIdMap[(int) $params['room']];
|
|
|
|
$update
|
|
->setParameter('id', (int) $row['mail_id'])
|
|
->setParameter('subjectParams', json_encode($params))
|
|
;
|
|
$update->execute();
|
|
}
|
|
$result->closeCursor();
|
|
}
|
|
}
|