503 lines
17 KiB
PHP
503 lines
17 KiB
PHP
<?php
|
|
/**
|
|
* @copyright Copyright (c) 2016, ownCloud, Inc.
|
|
*
|
|
* @author Joas Schilling <coding@schilljs.com>
|
|
* @author Lukas Reschke <lukas@statuscode.ch>
|
|
* @author Robin Appelman <robin@icewind.nl>
|
|
* @author Robin McCorkell <robin@mccorkell.me.uk>
|
|
*
|
|
* @license AGPL-3.0
|
|
*
|
|
* This code is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3,
|
|
* as published by the Free Software Foundation.
|
|
*
|
|
* 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, version 3,
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*
|
|
*/
|
|
|
|
namespace OCA\Files_External\Service;
|
|
|
|
use Doctrine\DBAL\Exception\UniqueConstraintViolationException;
|
|
use OCP\DB\QueryBuilder\IQueryBuilder;
|
|
use OCP\IDBConnection;
|
|
use OCP\Security\ICrypto;
|
|
|
|
/**
|
|
* Stores the mount config in the database
|
|
*/
|
|
class DBConfigService {
|
|
const MOUNT_TYPE_ADMIN = 1;
|
|
const MOUNT_TYPE_PERSONAl = 2;
|
|
|
|
const APPLICABLE_TYPE_GLOBAL = 1;
|
|
const APPLICABLE_TYPE_GROUP = 2;
|
|
const APPLICABLE_TYPE_USER = 3;
|
|
|
|
/**
|
|
* @var IDBConnection
|
|
*/
|
|
private $connection;
|
|
|
|
/**
|
|
* @var ICrypto
|
|
*/
|
|
private $crypto;
|
|
|
|
/**
|
|
* DBConfigService constructor.
|
|
*
|
|
* @param IDBConnection $connection
|
|
* @param ICrypto $crypto
|
|
*/
|
|
public function __construct(IDBConnection $connection, ICrypto $crypto) {
|
|
$this->connection = $connection;
|
|
$this->crypto = $crypto;
|
|
}
|
|
|
|
/**
|
|
* @param int $mountId
|
|
* @return array
|
|
*/
|
|
public function getMountById($mountId) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->select(['mount_id', 'mount_point', 'storage_backend', 'auth_backend', 'priority', 'type'])
|
|
->from('external_mounts', 'm')
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)));
|
|
$mounts = $this->getMountsFromQuery($query);
|
|
if (count($mounts) > 0) {
|
|
return $mounts[0];
|
|
} else {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get all configured mounts
|
|
*
|
|
* @return array
|
|
*/
|
|
public function getAllMounts() {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->select(['mount_id', 'mount_point', 'storage_backend', 'auth_backend', 'priority', 'type'])
|
|
->from('external_mounts');
|
|
return $this->getMountsFromQuery($query);
|
|
}
|
|
|
|
public function getMountsForUser($userId, $groupIds) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->select(['m.mount_id', 'mount_point', 'storage_backend', 'auth_backend', 'priority', 'm.type'])
|
|
->from('external_mounts', 'm')
|
|
->innerJoin('m', 'external_applicable', 'a', $builder->expr()->eq('m.mount_id', 'a.mount_id'))
|
|
->where($builder->expr()->orX(
|
|
$builder->expr()->andX( // global mounts
|
|
$builder->expr()->eq('a.type', $builder->createNamedParameter(self::APPLICABLE_TYPE_GLOBAL, IQueryBuilder::PARAM_INT)),
|
|
$builder->expr()->isNull('a.value')
|
|
),
|
|
$builder->expr()->andX( // mounts for user
|
|
$builder->expr()->eq('a.type', $builder->createNamedParameter(self::APPLICABLE_TYPE_USER, IQueryBuilder::PARAM_INT)),
|
|
$builder->expr()->eq('a.value', $builder->createNamedParameter($userId))
|
|
),
|
|
$builder->expr()->andX( // mounts for group
|
|
$builder->expr()->eq('a.type', $builder->createNamedParameter(self::APPLICABLE_TYPE_GROUP, IQueryBuilder::PARAM_INT)),
|
|
$builder->expr()->in('a.value', $builder->createNamedParameter($groupIds, IQueryBuilder::PARAM_STR_ARRAY))
|
|
)
|
|
));
|
|
|
|
return $this->getMountsFromQuery($query);
|
|
}
|
|
|
|
/**
|
|
* Get admin defined mounts
|
|
*
|
|
* @return array
|
|
* @suppress SqlInjectionChecker
|
|
*/
|
|
public function getAdminMounts() {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->select(['mount_id', 'mount_point', 'storage_backend', 'auth_backend', 'priority', 'type'])
|
|
->from('external_mounts')
|
|
->where($builder->expr()->eq('type', $builder->expr()->literal(self::MOUNT_TYPE_ADMIN, IQueryBuilder::PARAM_INT)));
|
|
return $this->getMountsFromQuery($query);
|
|
}
|
|
|
|
protected function getForQuery(IQueryBuilder $builder, $type, $value) {
|
|
$query = $builder->select(['m.mount_id', 'mount_point', 'storage_backend', 'auth_backend', 'priority', 'm.type'])
|
|
->from('external_mounts', 'm')
|
|
->innerJoin('m', 'external_applicable', 'a', $builder->expr()->eq('m.mount_id', 'a.mount_id'))
|
|
->where($builder->expr()->eq('a.type', $builder->createNamedParameter($type, IQueryBuilder::PARAM_INT)));
|
|
|
|
if (is_null($value)) {
|
|
$query = $query->andWhere($builder->expr()->isNull('a.value'));
|
|
} else {
|
|
$query = $query->andWhere($builder->expr()->eq('a.value', $builder->createNamedParameter($value)));
|
|
}
|
|
|
|
return $query;
|
|
}
|
|
|
|
/**
|
|
* Get mounts by applicable
|
|
*
|
|
* @param int $type any of the self::APPLICABLE_TYPE_ constants
|
|
* @param string|null $value user_id, group_id or null for global mounts
|
|
* @return array
|
|
*/
|
|
public function getMountsFor($type, $value) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $this->getForQuery($builder, $type, $value);
|
|
|
|
return $this->getMountsFromQuery($query);
|
|
}
|
|
|
|
/**
|
|
* Get admin defined mounts by applicable
|
|
*
|
|
* @param int $type any of the self::APPLICABLE_TYPE_ constants
|
|
* @param string|null $value user_id, group_id or null for global mounts
|
|
* @return array
|
|
* @suppress SqlInjectionChecker
|
|
*/
|
|
public function getAdminMountsFor($type, $value) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $this->getForQuery($builder, $type, $value);
|
|
$query->andWhere($builder->expr()->eq('m.type', $builder->expr()->literal(self::MOUNT_TYPE_ADMIN, IQueryBuilder::PARAM_INT)));
|
|
|
|
return $this->getMountsFromQuery($query);
|
|
}
|
|
|
|
/**
|
|
* Get admin defined mounts for multiple applicable
|
|
*
|
|
* @param int $type any of the self::APPLICABLE_TYPE_ constants
|
|
* @param string[] $values user_ids or group_ids
|
|
* @return array
|
|
* @suppress SqlInjectionChecker
|
|
*/
|
|
public function getAdminMountsForMultiple($type, array $values) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$params = array_map(function ($value) use ($builder) {
|
|
return $builder->createNamedParameter($value, IQueryBuilder::PARAM_STR);
|
|
}, $values);
|
|
|
|
$query = $builder->select(['m.mount_id', 'mount_point', 'storage_backend', 'auth_backend', 'priority', 'm.type'])
|
|
->from('external_mounts', 'm')
|
|
->innerJoin('m', 'external_applicable', 'a', $builder->expr()->eq('m.mount_id', 'a.mount_id'))
|
|
->where($builder->expr()->eq('a.type', $builder->createNamedParameter($type, IQueryBuilder::PARAM_INT)))
|
|
->andWhere($builder->expr()->in('a.value', $params));
|
|
$query->andWhere($builder->expr()->eq('m.type', $builder->expr()->literal(self::MOUNT_TYPE_ADMIN, IQueryBuilder::PARAM_INT)));
|
|
|
|
return $this->getMountsFromQuery($query);
|
|
}
|
|
|
|
/**
|
|
* Get user defined mounts by applicable
|
|
*
|
|
* @param int $type any of the self::APPLICABLE_TYPE_ constants
|
|
* @param string|null $value user_id, group_id or null for global mounts
|
|
* @return array
|
|
* @suppress SqlInjectionChecker
|
|
*/
|
|
public function getUserMountsFor($type, $value) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $this->getForQuery($builder, $type, $value);
|
|
$query->andWhere($builder->expr()->eq('m.type', $builder->expr()->literal(self::MOUNT_TYPE_PERSONAl, IQueryBuilder::PARAM_INT)));
|
|
|
|
return $this->getMountsFromQuery($query);
|
|
}
|
|
|
|
/**
|
|
* Add a mount to the database
|
|
*
|
|
* @param string $mountPoint
|
|
* @param string $storageBackend
|
|
* @param string $authBackend
|
|
* @param int $priority
|
|
* @param int $type self::MOUNT_TYPE_ADMIN or self::MOUNT_TYPE_PERSONAL
|
|
* @return int the id of the new mount
|
|
*/
|
|
public function addMount($mountPoint, $storageBackend, $authBackend, $priority, $type) {
|
|
if (!$priority) {
|
|
$priority = 100;
|
|
}
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->insert('external_mounts')
|
|
->values([
|
|
'mount_point' => $builder->createNamedParameter($mountPoint, IQueryBuilder::PARAM_STR),
|
|
'storage_backend' => $builder->createNamedParameter($storageBackend, IQueryBuilder::PARAM_STR),
|
|
'auth_backend' => $builder->createNamedParameter($authBackend, IQueryBuilder::PARAM_STR),
|
|
'priority' => $builder->createNamedParameter($priority, IQueryBuilder::PARAM_INT),
|
|
'type' => $builder->createNamedParameter($type, IQueryBuilder::PARAM_INT)
|
|
]);
|
|
$query->execute();
|
|
return (int)$this->connection->lastInsertId('*PREFIX*external_mounts');
|
|
}
|
|
|
|
/**
|
|
* Remove a mount from the database
|
|
*
|
|
* @param int $mountId
|
|
*/
|
|
public function removeMount($mountId) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->delete('external_mounts')
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)));
|
|
$query->execute();
|
|
|
|
$query = $builder->delete('external_applicable')
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)));
|
|
$query->execute();
|
|
|
|
$query = $builder->delete('external_config')
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)));
|
|
$query->execute();
|
|
|
|
$query = $builder->delete('external_options')
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)));
|
|
$query->execute();
|
|
}
|
|
|
|
/**
|
|
* @param int $mountId
|
|
* @param string $newMountPoint
|
|
*/
|
|
public function setMountPoint($mountId, $newMountPoint) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
|
|
$query = $builder->update('external_mounts')
|
|
->set('mount_point', $builder->createNamedParameter($newMountPoint))
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)));
|
|
|
|
$query->execute();
|
|
}
|
|
|
|
/**
|
|
* @param int $mountId
|
|
* @param string $newAuthBackend
|
|
*/
|
|
public function setAuthBackend($mountId, $newAuthBackend) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
|
|
$query = $builder->update('external_mounts')
|
|
->set('auth_backend', $builder->createNamedParameter($newAuthBackend))
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)));
|
|
|
|
$query->execute();
|
|
}
|
|
|
|
/**
|
|
* @param int $mountId
|
|
* @param string $key
|
|
* @param string $value
|
|
*/
|
|
public function setConfig($mountId, $key, $value) {
|
|
if ($key === 'password') {
|
|
$value = $this->encryptValue($value);
|
|
}
|
|
|
|
try {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$builder->insert('external_config')
|
|
->setValue('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT))
|
|
->setValue('key', $builder->createNamedParameter($key, IQueryBuilder::PARAM_STR))
|
|
->setValue('value', $builder->createNamedParameter($value, IQueryBuilder::PARAM_STR))
|
|
->execute();
|
|
} catch(UniqueConstraintViolationException $e) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->update('external_config')
|
|
->set('value', $builder->createNamedParameter($value, IQueryBuilder::PARAM_STR))
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)))
|
|
->andWhere($builder->expr()->eq('key', $builder->createNamedParameter($key, IQueryBuilder::PARAM_STR)));
|
|
$query->execute();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param int $mountId
|
|
* @param string $key
|
|
* @param string $value
|
|
*/
|
|
public function setOption($mountId, $key, $value) {
|
|
try {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$builder->insert('external_options')
|
|
->setValue('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT))
|
|
->setValue('key', $builder->createNamedParameter($key, IQueryBuilder::PARAM_STR))
|
|
->setValue('value', $builder->createNamedParameter(json_encode($value), IQueryBuilder::PARAM_STR))
|
|
->execute();
|
|
} catch(UniqueConstraintViolationException $e) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->update('external_options')
|
|
->set('value', $builder->createNamedParameter(json_encode($value), IQueryBuilder::PARAM_STR))
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)))
|
|
->andWhere($builder->expr()->eq('key', $builder->createNamedParameter($key, IQueryBuilder::PARAM_STR)));
|
|
$query->execute();
|
|
}
|
|
}
|
|
|
|
public function addApplicable($mountId, $type, $value) {
|
|
try {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$builder->insert('external_applicable')
|
|
->setValue('mount_id', $builder->createNamedParameter($mountId))
|
|
->setValue('type', $builder->createNamedParameter($type))
|
|
->setValue('value', $builder->createNamedParameter($value))
|
|
->execute();
|
|
} catch(UniqueConstraintViolationException $e) {
|
|
// applicable exists already
|
|
}
|
|
}
|
|
|
|
public function removeApplicable($mountId, $type, $value) {
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$query = $builder->delete('external_applicable')
|
|
->where($builder->expr()->eq('mount_id', $builder->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)))
|
|
->andWhere($builder->expr()->eq('type', $builder->createNamedParameter($type, IQueryBuilder::PARAM_INT)));
|
|
|
|
if (is_null($value)) {
|
|
$query = $query->andWhere($builder->expr()->isNull('value'));
|
|
} else {
|
|
$query = $query->andWhere($builder->expr()->eq('value', $builder->createNamedParameter($value, IQueryBuilder::PARAM_STR)));
|
|
}
|
|
|
|
$query->execute();
|
|
}
|
|
|
|
private function getMountsFromQuery(IQueryBuilder $query) {
|
|
$result = $query->execute();
|
|
$mounts = $result->fetchAll();
|
|
$uniqueMounts = [];
|
|
foreach ($mounts as $mount) {
|
|
$id = $mount['mount_id'];
|
|
if (!isset($uniqueMounts[$id])) {
|
|
$uniqueMounts[$id] = $mount;
|
|
}
|
|
}
|
|
$uniqueMounts = array_values($uniqueMounts);
|
|
|
|
$mountIds = array_map(function ($mount) {
|
|
return $mount['mount_id'];
|
|
}, $uniqueMounts);
|
|
$mountIds = array_values(array_unique($mountIds));
|
|
|
|
$applicable = $this->getApplicableForMounts($mountIds);
|
|
$config = $this->getConfigForMounts($mountIds);
|
|
$options = $this->getOptionsForMounts($mountIds);
|
|
|
|
return array_map(function ($mount, $applicable, $config, $options) {
|
|
$mount['type'] = (int)$mount['type'];
|
|
$mount['priority'] = (int)$mount['priority'];
|
|
$mount['applicable'] = $applicable;
|
|
$mount['config'] = $config;
|
|
$mount['options'] = $options;
|
|
return $mount;
|
|
}, $uniqueMounts, $applicable, $config, $options);
|
|
}
|
|
|
|
/**
|
|
* Get mount options from a table grouped by mount id
|
|
*
|
|
* @param string $table
|
|
* @param string[] $fields
|
|
* @param int[] $mountIds
|
|
* @return array [$mountId => [['field1' => $value1, ...], ...], ...]
|
|
*/
|
|
private function selectForMounts($table, array $fields, array $mountIds) {
|
|
if (count($mountIds) === 0) {
|
|
return [];
|
|
}
|
|
$builder = $this->connection->getQueryBuilder();
|
|
$fields[] = 'mount_id';
|
|
$placeHolders = array_map(function ($id) use ($builder) {
|
|
return $builder->createPositionalParameter($id, IQueryBuilder::PARAM_INT);
|
|
}, $mountIds);
|
|
$query = $builder->select($fields)
|
|
->from($table)
|
|
->where($builder->expr()->in('mount_id', $placeHolders));
|
|
$rows = $query->execute()->fetchAll();
|
|
|
|
$result = [];
|
|
foreach ($mountIds as $mountId) {
|
|
$result[$mountId] = [];
|
|
}
|
|
foreach ($rows as $row) {
|
|
if (isset($row['type'])) {
|
|
$row['type'] = (int)$row['type'];
|
|
}
|
|
$result[$row['mount_id']][] = $row;
|
|
}
|
|
return $result;
|
|
}
|
|
|
|
/**
|
|
* @param int[] $mountIds
|
|
* @return array [$id => [['type' => $type, 'value' => $value], ...], ...]
|
|
*/
|
|
public function getApplicableForMounts($mountIds) {
|
|
return $this->selectForMounts('external_applicable', ['type', 'value'], $mountIds);
|
|
}
|
|
|
|
/**
|
|
* @param int[] $mountIds
|
|
* @return array [$id => ['key1' => $value1, ...], ...]
|
|
*/
|
|
public function getConfigForMounts($mountIds) {
|
|
$mountConfigs = $this->selectForMounts('external_config', ['key', 'value'], $mountIds);
|
|
return array_map([$this, 'createKeyValueMap'], $mountConfigs);
|
|
}
|
|
|
|
/**
|
|
* @param int[] $mountIds
|
|
* @return array [$id => ['key1' => $value1, ...], ...]
|
|
*/
|
|
public function getOptionsForMounts($mountIds) {
|
|
$mountOptions = $this->selectForMounts('external_options', ['key', 'value'], $mountIds);
|
|
$optionsMap = array_map([$this, 'createKeyValueMap'], $mountOptions);
|
|
return array_map(function (array $options) {
|
|
return array_map(function ($option) {
|
|
return json_decode($option);
|
|
}, $options);
|
|
}, $optionsMap);
|
|
}
|
|
|
|
/**
|
|
* @param array $keyValuePairs [['key'=>$key, 'value=>$value], ...]
|
|
* @return array ['key1' => $value1, ...]
|
|
*/
|
|
private function createKeyValueMap(array $keyValuePairs) {
|
|
$decryptedPairts = array_map(function ($pair) {
|
|
if ($pair['key'] === 'password') {
|
|
$pair['value'] = $this->decryptValue($pair['value']);
|
|
}
|
|
return $pair;
|
|
}, $keyValuePairs);
|
|
$keys = array_map(function ($pair) {
|
|
return $pair['key'];
|
|
}, $decryptedPairts);
|
|
$values = array_map(function ($pair) {
|
|
return $pair['value'];
|
|
}, $decryptedPairts);
|
|
|
|
return array_combine($keys, $values);
|
|
}
|
|
|
|
private function encryptValue($value) {
|
|
return $this->crypto->encrypt($value);
|
|
}
|
|
|
|
private function decryptValue($value) {
|
|
try {
|
|
return $this->crypto->decrypt($value);
|
|
} catch (\Exception $e) {
|
|
return $value;
|
|
}
|
|
}
|
|
}
|