forked from Alex/Pterodactyl-Panel
111 lines
3.6 KiB
PHP
111 lines
3.6 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Services\Servers;
|
|
|
|
use Illuminate\Support\Arr;
|
|
use Pterodactyl\Models\Egg;
|
|
use Pterodactyl\Models\User;
|
|
use Pterodactyl\Models\Server;
|
|
use Pterodactyl\Models\ServerVariable;
|
|
use Illuminate\Database\ConnectionInterface;
|
|
use Pterodactyl\Traits\Services\HasUserLevels;
|
|
|
|
class StartupModificationService
|
|
{
|
|
use HasUserLevels;
|
|
|
|
/**
|
|
* @var \Illuminate\Database\ConnectionInterface
|
|
*/
|
|
private $connection;
|
|
|
|
/**
|
|
* @var \Pterodactyl\Services\Servers\VariableValidatorService
|
|
*/
|
|
private $validatorService;
|
|
|
|
/**
|
|
* StartupModificationService constructor.
|
|
*
|
|
* @param \Illuminate\Database\ConnectionInterface $connection
|
|
* @param \Pterodactyl\Services\Servers\VariableValidatorService $validatorService
|
|
*/
|
|
public function __construct(ConnectionInterface $connection, VariableValidatorService $validatorService)
|
|
{
|
|
$this->connection = $connection;
|
|
$this->validatorService = $validatorService;
|
|
}
|
|
|
|
/**
|
|
* Process startup modification for a server.
|
|
*
|
|
* @param \Pterodactyl\Models\Server $server
|
|
* @param array $data
|
|
* @return \Pterodactyl\Models\Server
|
|
*
|
|
* @throws \Throwable
|
|
*/
|
|
public function handle(Server $server, array $data): Server
|
|
{
|
|
return $this->connection->transaction(function () use ($server, $data) {
|
|
if (! empty($data['environment'])) {
|
|
$egg = $this->isUserLevel(User::USER_LEVEL_ADMIN) ? ($data['egg_id'] ?? $server->egg_id) : $server->egg_id;
|
|
|
|
$results = $this->validatorService
|
|
->setUserLevel($this->getUserLevel())
|
|
->handle($egg, $data['environment']);
|
|
|
|
foreach ($results as $result) {
|
|
ServerVariable::query()->updateOrCreate(
|
|
[
|
|
'server_id' => $server->id,
|
|
'variable_id' => $result->id,
|
|
],
|
|
['variable_value' => $result->value ?? '']
|
|
);
|
|
}
|
|
}
|
|
|
|
if ($this->isUserLevel(User::USER_LEVEL_ADMIN)) {
|
|
$this->updateAdministrativeSettings($data, $server);
|
|
}
|
|
|
|
// Calling ->refresh() rather than ->fresh() here causes it to return the
|
|
// variables as triplicates for some reason? Not entirely sure, should dig
|
|
// in more to figure it out, but luckily we have a test case covering this
|
|
// specific call so we can be assured we're not breaking it _here_ at least.
|
|
//
|
|
// TODO(dane): this seems like a red-flag for the code powering the relationship
|
|
// that should be looked into more.
|
|
return $server->fresh();
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Update certain administrative settings for a server in the DB.
|
|
*
|
|
* @param array $data
|
|
* @param \Pterodactyl\Models\Server $server
|
|
*/
|
|
protected function updateAdministrativeSettings(array $data, Server &$server)
|
|
{
|
|
$eggId = Arr::get($data, 'egg_id');
|
|
|
|
if (is_digit($eggId) && $server->egg_id !== (int)$eggId) {
|
|
/** @var \Pterodactyl\Models\Egg $egg */
|
|
$egg = Egg::query()->findOrFail($data['egg_id']);
|
|
|
|
$server = $server->forceFill([
|
|
'egg_id' => $egg->id,
|
|
'nest_id' => $egg->nest_id,
|
|
]);
|
|
}
|
|
|
|
$server->fill([
|
|
'startup' => $data['startup'] ?? $server->startup,
|
|
'skip_scripts' => $data['skip_scripts'] ?? isset($data['skip_scripts']),
|
|
'image' => $data['docker_image'] ?? $server->image,
|
|
])->save();
|
|
}
|
|
}
|