diff --git a/app/Contracts/Repository/DatabaseRepositoryInterface.php b/app/Contracts/Repository/DatabaseRepositoryInterface.php deleted file mode 100644 index befc2ce4d..000000000 --- a/app/Contracts/Repository/DatabaseRepositoryInterface.php +++ /dev/null @@ -1,61 +0,0 @@ -findOrFail($host); + $databases = $host->databases()->with('server')->paginate(25); + return $this->view->make('admin.databases.view', [ 'locations' => $this->locationRepository->getAllWithNodes(), - 'host' => $this->repository->find($host), - 'databases' => $this->databaseRepository->getDatabasesForHost($host), + 'host' => $host, + 'databases' => $databases, ]); } diff --git a/app/Http/Controllers/Admin/ServersController.php b/app/Http/Controllers/Admin/ServersController.php index 3d2c55ac4..323972cae 100644 --- a/app/Http/Controllers/Admin/ServersController.php +++ b/app/Http/Controllers/Admin/ServersController.php @@ -29,7 +29,6 @@ use Pterodactyl\Repositories\Eloquent\DatabaseHostRepository; use Pterodactyl\Services\Databases\DatabaseManagementService; use Illuminate\Contracts\Config\Repository as ConfigRepository; use Pterodactyl\Contracts\Repository\ServerRepositoryInterface; -use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface; use Pterodactyl\Contracts\Repository\AllocationRepositoryInterface; use Pterodactyl\Services\Servers\ServerConfigurationStructureService; use Pterodactyl\Http\Requests\Admin\Servers\Databases\StoreServerDatabaseRequest; @@ -47,7 +46,6 @@ class ServersController extends Controller protected DaemonServerRepository $daemonServerRepository, protected DatabaseManagementService $databaseManagementService, protected DatabasePasswordService $databasePasswordService, - protected DatabaseRepositoryInterface $databaseRepository, protected DatabaseHostRepository $databaseHostRepository, protected ServerDeletionService $deletionService, protected DetailsModificationService $detailsModificationService, diff --git a/app/Models/Database.php b/app/Models/Database.php index e3856a4e2..e1386caff 100644 --- a/app/Models/Database.php +++ b/app/Models/Database.php @@ -4,6 +4,7 @@ namespace Pterodactyl\Models; use Illuminate\Container\Container; use Illuminate\Database\Eloquent\Relations\BelongsTo; +use Illuminate\Support\Facades\DB; use Pterodactyl\Contracts\Extensions\HashidsInterface; /** @@ -28,10 +29,7 @@ class Database extends Model */ public const RESOURCE_NAME = 'server_database'; - /** - * The table associated with the model. - */ - protected $table = 'databases'; + public const DEFAULT_CONNECTION_NAME = 'dynamic'; /** * The attributes excluded from the model's JSON form. @@ -107,4 +105,73 @@ class Database extends Model { return $this->belongsTo(Server::class); } + + /** + * Run the provided statement against the database on a given connection. + */ + private function run(string $statement): bool + { + return DB::connection($this->connection)->statement($statement); + } + + /** + * Create a new database on a given connection. + */ + public function createDatabase(string $database): bool + { + return $this->run(sprintf('CREATE DATABASE IF NOT EXISTS `%s`', $database)); + } + + /** + * Create a new database user on a given connection. + */ + public function createUser(string $username, string $remote, string $password, ?int $max_connections): bool + { + $args = [$username, $remote, $password]; + $command = 'CREATE USER `%s`@`%s` IDENTIFIED BY \'%s\''; + + if (!empty($max_connections)) { + $args[] = $max_connections; + $command .= ' WITH MAX_USER_CONNECTIONS %s'; + } + + return $this->run(sprintf($command, ...$args)); + } + + /** + * Give a specific user access to a given database. + */ + public function assignUserToDatabase(string $database, string $username, string $remote): bool + { + return $this->run(sprintf( + 'GRANT SELECT, INSERT, UPDATE, DELETE, CREATE, DROP, ALTER, REFERENCES, INDEX, LOCK TABLES, CREATE ROUTINE, ALTER ROUTINE, EXECUTE, CREATE TEMPORARY TABLES, CREATE VIEW, SHOW VIEW, EVENT, TRIGGER ON `%s`.* TO `%s`@`%s`', + $database, + $username, + $remote + )); + } + + /** + * Flush the privileges for a given connection. + */ + public function flush(): bool + { + return $this->run('FLUSH PRIVILEGES'); + } + + /** + * Drop a given database on a specific connection. + */ + public function dropDatabase(string $database): bool + { + return $this->run(sprintf('DROP DATABASE IF EXISTS `%s`', $database)); + } + + /** + * Drop a given user on a specific connection. + */ + public function dropUser(string $username, string $remote): bool + { + return $this->run(sprintf('DROP USER IF EXISTS `%s`@`%s`', $username, $remote)); + } } diff --git a/app/Providers/RepositoryServiceProvider.php b/app/Providers/RepositoryServiceProvider.php index 8a0434f52..1da001aba 100644 --- a/app/Providers/RepositoryServiceProvider.php +++ b/app/Providers/RepositoryServiceProvider.php @@ -12,7 +12,6 @@ use Pterodactyl\Repositories\Eloquent\ApiKeyRepository; use Pterodactyl\Repositories\Eloquent\ServerRepository; use Pterodactyl\Repositories\Eloquent\SessionRepository; use Pterodactyl\Repositories\Eloquent\SubuserRepository; -use Pterodactyl\Repositories\Eloquent\DatabaseRepository; use Pterodactyl\Repositories\Eloquent\LocationRepository; use Pterodactyl\Repositories\Eloquent\ScheduleRepository; use Pterodactyl\Repositories\Eloquent\SettingsRepository; @@ -29,7 +28,6 @@ use Pterodactyl\Contracts\Repository\ServerRepositoryInterface; use Pterodactyl\Repositories\Eloquent\ServerVariableRepository; use Pterodactyl\Contracts\Repository\SessionRepositoryInterface; use Pterodactyl\Contracts\Repository\SubuserRepositoryInterface; -use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface; use Pterodactyl\Contracts\Repository\LocationRepositoryInterface; use Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface; use Pterodactyl\Contracts\Repository\SettingsRepositoryInterface; @@ -48,7 +46,6 @@ class RepositoryServiceProvider extends ServiceProvider // Eloquent Repositories $this->app->bind(AllocationRepositoryInterface::class, AllocationRepository::class); $this->app->bind(ApiKeyRepositoryInterface::class, ApiKeyRepository::class); - $this->app->bind(DatabaseRepositoryInterface::class, DatabaseRepository::class); $this->app->bind(DatabaseHostRepositoryInterface::class, DatabaseHostRepository::class); $this->app->bind(EggRepositoryInterface::class, EggRepository::class); $this->app->bind(EggVariableRepositoryInterface::class, EggVariableRepository::class); diff --git a/app/Repositories/Eloquent/DatabaseRepository.php b/app/Repositories/Eloquent/DatabaseRepository.php deleted file mode 100644 index 9bc33d035..000000000 --- a/app/Repositories/Eloquent/DatabaseRepository.php +++ /dev/null @@ -1,136 +0,0 @@ -connection; - } - - /** - * Set the connection name to execute statements against. - */ - public function setConnection(string $connection): self - { - $this->connection = $connection; - - return $this; - } - - /** - * Return all the databases belonging to a server. - */ - public function getDatabasesForServer(int $server): Collection - { - return $this->getBuilder()->with('host')->where('server_id', $server)->get($this->getColumns()); - } - - /** - * Return all the databases for a given host with the server relationship loaded. - */ - public function getDatabasesForHost(int $host, int $count = 25): LengthAwarePaginator - { - return $this->getBuilder()->with('server') - ->where('database_host_id', $host) - ->paginate($count, $this->getColumns()); - } - - /** - * Create a new database on a given connection. - */ - public function createDatabase(string $database): bool - { - return $this->run(sprintf('CREATE DATABASE IF NOT EXISTS `%s`', $database)); - } - - /** - * Create a new database user on a given connection. - */ - public function createUser(string $username, string $remote, string $password, ?int $max_connections): bool - { - $args = [$username, $remote, $password]; - $command = 'CREATE USER `%s`@`%s` IDENTIFIED BY \'%s\''; - - if (!empty($max_connections)) { - $args[] = $max_connections; - $command .= ' WITH MAX_USER_CONNECTIONS %s'; - } - - return $this->run(sprintf($command, ...$args)); - } - - /** - * Give a specific user access to a given database. - */ - public function assignUserToDatabase(string $database, string $username, string $remote): bool - { - return $this->run(sprintf( - 'GRANT SELECT, INSERT, UPDATE, DELETE, CREATE, DROP, ALTER, REFERENCES, INDEX, LOCK TABLES, CREATE ROUTINE, ALTER ROUTINE, EXECUTE, CREATE TEMPORARY TABLES, CREATE VIEW, SHOW VIEW, EVENT, TRIGGER ON `%s`.* TO `%s`@`%s`', - $database, - $username, - $remote - )); - } - - /** - * Flush the privileges for a given connection. - */ - public function flush(): bool - { - return $this->run('FLUSH PRIVILEGES'); - } - - /** - * Drop a given database on a specific connection. - */ - public function dropDatabase(string $database): bool - { - return $this->run(sprintf('DROP DATABASE IF EXISTS `%s`', $database)); - } - - /** - * Drop a given user on a specific connection. - */ - public function dropUser(string $username, string $remote): bool - { - return $this->run(sprintf('DROP USER IF EXISTS `%s`@`%s`', $username, $remote)); - } - - /** - * Run the provided statement against the database on a given connection. - */ - private function run(string $statement): bool - { - return $this->database->connection($this->getConnection())->statement($statement); - } -} diff --git a/app/Services/Databases/DatabaseManagementService.php b/app/Services/Databases/DatabaseManagementService.php index b70cb8b4d..970e7485a 100644 --- a/app/Services/Databases/DatabaseManagementService.php +++ b/app/Services/Databases/DatabaseManagementService.php @@ -10,7 +10,6 @@ use Pterodactyl\Helpers\Utilities; use Illuminate\Database\ConnectionInterface; use Illuminate\Contracts\Encryption\Encrypter; use Pterodactyl\Extensions\DynamicDatabaseConnection; -use Pterodactyl\Repositories\Eloquent\DatabaseRepository; use Pterodactyl\Exceptions\Repository\DuplicateDatabaseNameException; use Pterodactyl\Exceptions\Service\Database\TooManyDatabasesException; use Pterodactyl\Exceptions\Service\Database\DatabaseClientFeatureNotEnabledException; @@ -36,8 +35,7 @@ class DatabaseManagementService public function __construct( protected ConnectionInterface $connection, protected DynamicDatabaseConnection $dynamic, - protected Encrypter $encrypter, - protected DatabaseRepository $repository + protected Encrypter $encrypter ) { } @@ -105,26 +103,26 @@ class DatabaseManagementService $this->dynamic->set('dynamic', $data['database_host_id']); - $this->repository->createDatabase($database->database); - $this->repository->createUser( + $database->createDatabase($database->database); + $database->createUser( $database->username, $database->remote, $this->encrypter->decrypt($database->password), $database->max_connections ); - $this->repository->assignUserToDatabase($database->database, $database->username, $database->remote); - $this->repository->flush(); + $database->assignUserToDatabase($database->database, $database->username, $database->remote); + $database->flush(); return $database; }); } catch (Exception $exception) { try { if ($database instanceof Database) { - $this->repository->dropDatabase($database->database); - $this->repository->dropUser($database->username, $database->remote); - $this->repository->flush(); + $database->dropDatabase($database->database); + $database->dropUser($database->username, $database->remote); + $database->flush(); } - } catch (Exception $deletionException) { + } catch (Exception) { // Do nothing here. We've already encountered an issue before this point so no // reason to prioritize this error over the initial one. } @@ -142,9 +140,9 @@ class DatabaseManagementService { $this->dynamic->set('dynamic', $database->database_host_id); - $this->repository->dropDatabase($database->database); - $this->repository->dropUser($database->username, $database->remote); - $this->repository->flush(); + $database->dropDatabase($database->database); + $database->dropUser($database->username, $database->remote); + $database->flush(); return $database->delete(); } diff --git a/app/Services/Databases/DatabasePasswordService.php b/app/Services/Databases/DatabasePasswordService.php index 3862b2393..81d8f6e61 100644 --- a/app/Services/Databases/DatabasePasswordService.php +++ b/app/Services/Databases/DatabasePasswordService.php @@ -7,7 +7,6 @@ use Pterodactyl\Helpers\Utilities; use Illuminate\Database\ConnectionInterface; use Illuminate\Contracts\Encryption\Encrypter; use Pterodactyl\Extensions\DynamicDatabaseConnection; -use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface; class DatabasePasswordService { @@ -17,8 +16,7 @@ class DatabasePasswordService public function __construct( private ConnectionInterface $connection, private DynamicDatabaseConnection $dynamic, - private Encrypter $encrypter, - private DatabaseRepositoryInterface $repository + private Encrypter $encrypter ) { } @@ -34,14 +32,14 @@ class DatabasePasswordService $this->connection->transaction(function () use ($database, $password) { $this->dynamic->set('dynamic', $database->database_host_id); - $this->repository->withoutFreshModel()->update($database->id, [ + $database->update([ 'password' => $this->encrypter->encrypt($password), ]); - $this->repository->dropUser($database->username, $database->remote); - $this->repository->createUser($database->username, $database->remote, $password, $database->max_connections); - $this->repository->assignUserToDatabase($database->database, $database->username, $database->remote); - $this->repository->flush(); + $database->dropUser($database->username, $database->remote); + $database->createUser($database->username, $database->remote, $password, $database->max_connections); + $database->assignUserToDatabase($database->database, $database->username, $database->remote); + $database->flush(); }); return $password; diff --git a/app/Services/Databases/Hosts/HostDeletionService.php b/app/Services/Databases/Hosts/HostDeletionService.php index 4a06af8da..eede63bf7 100644 --- a/app/Services/Databases/Hosts/HostDeletionService.php +++ b/app/Services/Databases/Hosts/HostDeletionService.php @@ -3,33 +3,25 @@ namespace Pterodactyl\Services\Databases\Hosts; use Pterodactyl\Exceptions\Service\HasActiveServersException; -use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface; -use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface; +use Pterodactyl\Models\DatabaseHost; class HostDeletionService { - /** - * HostDeletionService constructor. - */ - public function __construct( - private DatabaseRepositoryInterface $databaseRepository, - private DatabaseHostRepositoryInterface $repository - ) { - } - /** * Delete a specified host from the Panel if no databases are * attached to it. * - * @throws \Pterodactyl\Exceptions\Service\HasActiveServersException + * @throws HasActiveServersException */ public function handle(int $host): int { - $count = $this->databaseRepository->findCountWhere([['database_host_id', '=', $host]]); - if ($count > 0) { + /** @var DatabaseHost $host */ + $host = DatabaseHost::query()->findOrFail($host); + + if ($host->databases()->count() > 0) { throw new HasActiveServersException(trans('exceptions.databases.delete_has_databases')); } - return $this->repository->delete($host); + return $host->delete(); } }