relationLoaded('egg') || $refresh) { $server->load('egg.scriptFrom'); } return $server; } /** * Return a collection of servers with their associated data for rebuild operations. */ public function getDataForRebuild(int $server = null, int $node = null): Collection { $instance = $this->getBuilder()->with(['allocation', 'allocations', 'egg', 'node']); if (!is_null($server) && is_null($node)) { $instance = $instance->where('id', '=', $server); } elseif (is_null($server) && !is_null($node)) { $instance = $instance->where('node_id', '=', $node); } return $instance->get($this->getColumns()); } /** * Return a collection of servers with their associated data for reinstall operations. */ public function getDataForReinstall(int $server = null, int $node = null): Collection { $instance = $this->getBuilder()->with(['allocation', 'allocations', 'egg', 'node']); if (!is_null($server) && is_null($node)) { $instance = $instance->where('id', '=', $server); } elseif (is_null($server) && !is_null($node)) { $instance = $instance->where('node_id', '=', $node); } return $instance->get($this->getColumns()); } /** * Return a server model and all variables associated with the server. * * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException */ public function findWithVariables(int $id): Server { try { /* @noinspection PhpIncompatibleReturnTypeInspection */ // @phpstan-ignore-next-line return $this->getBuilder()->with('egg.variables', 'variables') ->where($this->getModel()->getKeyName(), '=', $id) ->firstOrFail($this->getColumns()); } catch (ModelNotFoundException $exception) { throw new RecordNotFoundException(); } } /** * Get the primary allocation for a given server. If a model is passed into * the function, load the allocation relationship onto it. Otherwise, find and * return the server from the database. */ public function getPrimaryAllocation(Server $server, bool $refresh = false): Server { if (!$server->relationLoaded('allocation') || $refresh) { $server->load('allocation'); } return $server; } /** * Return enough data to be used for the creation of a server via the daemon. */ public function getDataForCreation(Server $server, bool $refresh = false): Server { foreach (['allocation', 'allocations', 'egg'] as $relation) { if (!$server->relationLoaded($relation) || $refresh) { $server->load($relation); } } return $server; } /** * Load associated databases onto the server model. */ public function loadDatabaseRelations(Server $server, bool $refresh = false): Server { if (!$server->relationLoaded('databases') || $refresh) { $server->load('databases.host'); } return $server; } /** * Get data for use when updating a server on the Daemon. Returns an array of * the egg which is used for build and rebuild. Only loads relations * if they are missing, or refresh is set to true. */ public function getDaemonServiceData(Server $server, bool $refresh = false): array { if (!$server->relationLoaded('egg') || $refresh) { $server->load('egg'); } return [ 'egg' => $server->getRelation('egg')->uuid, ]; } /** * Return a server by UUID. * * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException */ public function getByUuid(string $uuid): Server { try { /** @var \Pterodactyl\Models\Server $model */ $model = $this->getBuilder() ->with('nest', 'node') ->where(function (Builder $query) use ($uuid) { $query->where('uuidShort', $uuid)->orWhere('uuid', $uuid); }) ->firstOrFail($this->getColumns()); return $model; } catch (ModelNotFoundException $exception) { throw new RecordNotFoundException(); } } /** * Check if a given UUID and UUID-Short string are unique to a server. */ public function isUniqueUuidCombo(string $uuid, string $short): bool { return !$this->getBuilder()->where('uuid', '=', $uuid)->orWhere('uuidShort', '=', $short)->exists(); } /** * Returns all of the servers that exist for a given node in a paginated response. */ public function loadAllServersForNode(int $node, int $limit): LengthAwarePaginator { return $this->getBuilder() ->with(['user', 'nest', 'egg']) ->where('node_id', '=', $node) ->paginate($limit); } }