123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268 |
- <?php
- namespace OCA\Federation;
- use OC\Files\Filesystem;
- use OCP\DB\Exception as DBException;
- use OCP\DB\QueryBuilder\IQueryBuilder;
- use OCP\HintException;
- use OCP\IDBConnection;
- use OCP\IL10N;
- class DbHandler {
- private string $dbTable = 'trusted_servers';
- public function __construct(
- private IDBConnection $connection,
- private IL10N $IL10N,
- ) {
- }
-
- public function addServer(string $url): int {
- $hash = $this->hash($url);
- $url = rtrim($url, '/');
- $query = $this->connection->getQueryBuilder();
- $query->insert($this->dbTable)
- ->values([
- 'url' => $query->createParameter('url'),
- 'url_hash' => $query->createParameter('url_hash'),
- ])
- ->setParameter('url', $url)
- ->setParameter('url_hash', $hash);
- $result = $query->executeStatement();
- if ($result) {
- return $query->getLastInsertId();
- }
- $message = 'Internal failure, Could not add trusted server: ' . $url;
- $message_t = $this->IL10N->t('Could not add server');
- throw new HintException($message, $message_t);
- return -1;
- }
-
- public function removeServer(int $id): void {
- $query = $this->connection->getQueryBuilder();
- $query->delete($this->dbTable)
- ->where($query->expr()->eq('id', $query->createParameter('id')))
- ->setParameter('id', $id);
- $query->executeStatement();
- }
-
- public function getServerById(int $id): array {
- $query = $this->connection->getQueryBuilder();
- $query->select('*')->from($this->dbTable)
- ->where($query->expr()->eq('id', $query->createParameter('id')))
- ->setParameter('id', $id, IQueryBuilder::PARAM_INT);
- $qResult = $query->executeQuery();
- $result = $qResult->fetchAll();
- $qResult->closeCursor();
- if (empty($result)) {
- throw new \Exception('No Server found with ID: ' . $id);
- }
- return $result[0];
- }
-
- public function getAllServer(): array {
- $query = $this->connection->getQueryBuilder();
- $query->select(['url', 'url_hash', 'id', 'status', 'shared_secret', 'sync_token'])
- ->from($this->dbTable);
- $statement = $query->executeQuery();
- $result = $statement->fetchAll();
- $statement->closeCursor();
- return $result;
- }
-
- public function serverExists(string $url): bool {
- $hash = $this->hash($url);
- $query = $this->connection->getQueryBuilder();
- $query->select('url')
- ->from($this->dbTable)
- ->where($query->expr()->eq('url_hash', $query->createParameter('url_hash')))
- ->setParameter('url_hash', $hash);
- $statement = $query->executeQuery();
- $result = $statement->fetchAll();
- $statement->closeCursor();
- return !empty($result);
- }
-
- public function addToken(string $url, string $token): void {
- $hash = $this->hash($url);
- $query = $this->connection->getQueryBuilder();
- $query->update($this->dbTable)
- ->set('token', $query->createParameter('token'))
- ->where($query->expr()->eq('url_hash', $query->createParameter('url_hash')))
- ->setParameter('url_hash', $hash)
- ->setParameter('token', $token);
- $query->executeStatement();
- }
-
- public function getToken(string $url): string {
- $hash = $this->hash($url);
- $query = $this->connection->getQueryBuilder();
- $query->select('token')->from($this->dbTable)
- ->where($query->expr()->eq('url_hash', $query->createParameter('url_hash')))
- ->setParameter('url_hash', $hash);
- $statement = $query->executeQuery();
- $result = $statement->fetch();
- $statement->closeCursor();
- if (!isset($result['token'])) {
- throw new \Exception('No token found for: ' . $url);
- }
- return $result['token'];
- }
-
- public function addSharedSecret(string $url, string $sharedSecret): void {
- $hash = $this->hash($url);
- $query = $this->connection->getQueryBuilder();
- $query->update($this->dbTable)
- ->set('shared_secret', $query->createParameter('sharedSecret'))
- ->where($query->expr()->eq('url_hash', $query->createParameter('url_hash')))
- ->setParameter('url_hash', $hash)
- ->setParameter('sharedSecret', $sharedSecret);
- $query->executeStatement();
- }
-
- public function getSharedSecret(string $url): string {
- $hash = $this->hash($url);
- $query = $this->connection->getQueryBuilder();
- $query->select('shared_secret')->from($this->dbTable)
- ->where($query->expr()->eq('url_hash', $query->createParameter('url_hash')))
- ->setParameter('url_hash', $hash);
- $statement = $query->executeQuery();
- $result = $statement->fetch();
- $statement->closeCursor();
- return (string)$result['shared_secret'];
- }
-
- public function setServerStatus(string $url, int $status, ?string $token = null): void {
- $hash = $this->hash($url);
- $query = $this->connection->getQueryBuilder();
- $query->update($this->dbTable)
- ->set('status', $query->createNamedParameter($status))
- ->where($query->expr()->eq('url_hash', $query->createNamedParameter($hash)));
- if (!is_null($token)) {
- $query->set('sync_token', $query->createNamedParameter($token));
- }
- $query->executeStatement();
- }
-
- public function getServerStatus(string $url): int {
- $hash = $this->hash($url);
- $query = $this->connection->getQueryBuilder();
- $query->select('status')->from($this->dbTable)
- ->where($query->expr()->eq('url_hash', $query->createParameter('url_hash')))
- ->setParameter('url_hash', $hash);
- $statement = $query->executeQuery();
- $result = $statement->fetch();
- $statement->closeCursor();
- return (int)$result['status'];
- }
-
- protected function hash(string $url): string {
- $normalized = $this->normalizeUrl($url);
- return sha1($normalized);
- }
-
- protected function normalizeUrl(string $url): string {
- $normalized = $url;
- if (strpos($url, 'https://') === 0) {
- $normalized = substr($url, strlen('https://'));
- } elseif (strpos($url, 'http://') === 0) {
- $normalized = substr($url, strlen('http://'));
- }
- $normalized = Filesystem::normalizePath($normalized);
- $normalized = trim($normalized, '/');
- return $normalized;
- }
- public function auth(string $username, string $password): bool {
- if ($username !== 'system') {
- return false;
- }
- $query = $this->connection->getQueryBuilder();
- $query->select('url')->from($this->dbTable)
- ->where($query->expr()->eq('shared_secret', $query->createNamedParameter($password)));
- $statement = $query->executeQuery();
- $result = $statement->fetch();
- $statement->closeCursor();
- return !empty($result);
- }
- }
|