forked from Alex/Pterodactyl-Panel
58 lines
1.9 KiB
PHP
58 lines
1.9 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Services\Databases;
|
|
|
|
use Webmozart\Assert\Assert;
|
|
use Pterodactyl\Models\Server;
|
|
use Pterodactyl\Models\Database;
|
|
use Pterodactyl\Models\DatabaseHost;
|
|
use Pterodactyl\Exceptions\Service\Database\NoSuitableDatabaseHostException;
|
|
|
|
class DeployServerDatabaseService
|
|
{
|
|
/**
|
|
* @var \Pterodactyl\Services\Databases\DatabaseManagementService
|
|
*/
|
|
private $managementService;
|
|
|
|
/**
|
|
* ServerDatabaseCreationService constructor.
|
|
*
|
|
* @param \Pterodactyl\Services\Databases\DatabaseManagementService $managementService
|
|
*/
|
|
public function __construct(DatabaseManagementService $managementService)
|
|
{
|
|
$this->managementService = $managementService;
|
|
}
|
|
|
|
/**
|
|
* @throws \Throwable
|
|
* @throws \Pterodactyl\Exceptions\Service\Database\TooManyDatabasesException
|
|
* @throws \Pterodactyl\Exceptions\Service\Database\DatabaseClientFeatureNotEnabledException
|
|
*/
|
|
public function handle(Server $server, array $data): Database
|
|
{
|
|
Assert::notEmpty($data['database'] ?? null);
|
|
Assert::notEmpty($data['remote'] ?? null);
|
|
|
|
$hosts = DatabaseHost::query()->get()->toBase();
|
|
if ($hosts->isEmpty()) {
|
|
throw new NoSuitableDatabaseHostException();
|
|
} else {
|
|
$nodeHosts = $hosts->where('node_id', $server->node_id)->toBase();
|
|
|
|
if ($nodeHosts->isEmpty() && !config('pterodactyl.client_features.databases.allow_random')) {
|
|
throw new NoSuitableDatabaseHostException();
|
|
}
|
|
}
|
|
|
|
return $this->managementService->create($server, [
|
|
'database_host_id' => $nodeHosts->isEmpty()
|
|
? $hosts->random()->id
|
|
: $nodeHosts->random()->id,
|
|
'database' => DatabaseManagementService::generateUniqueDatabaseName($data['database'], $server->id),
|
|
'remote' => $data['remote'],
|
|
]);
|
|
}
|
|
}
|