61 lines
1.7 KiB
PHP
61 lines
1.7 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Repositories\Eloquent;
|
|
|
|
use Pterodactyl\Models\Mount;
|
|
use Pterodactyl\Models\Server;
|
|
use Illuminate\Support\Collection;
|
|
use Illuminate\Database\Eloquent\ModelNotFoundException;
|
|
use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
|
|
|
|
class MountRepository extends EloquentRepository
|
|
{
|
|
/**
|
|
* Return the model backing this repository.
|
|
*
|
|
* @return string
|
|
*/
|
|
public function model()
|
|
{
|
|
return Mount::class;
|
|
}
|
|
|
|
/**
|
|
* Return mounts with a count of eggs, nodes, and servers attached to it.
|
|
*/
|
|
public function getAllWithDetails(): Collection
|
|
{
|
|
return $this->getBuilder()->withCount('eggs', 'nodes')->get($this->getColumns());
|
|
}
|
|
|
|
/**
|
|
* Return all of the mounts and their respective relations.
|
|
*
|
|
* @return mixed
|
|
*
|
|
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
|
|
*/
|
|
public function getWithRelations(string $id): Mount
|
|
{
|
|
try {
|
|
return $this->getBuilder()->with('eggs', 'nodes')->findOrFail($id, $this->getColumns());
|
|
} catch (ModelNotFoundException $exception) {
|
|
throw new RecordNotFoundException();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Return mounts available to a server (ignoring if they are or are not mounted).
|
|
*/
|
|
public function getMountListForServer(Server $server): Collection
|
|
{
|
|
return $this->getBuilder()
|
|
->whereHas('eggs', function ($q) use ($server) {
|
|
$q->where('id', '=', $server->egg_id);
|
|
})
|
|
->whereHas('nodes', function ($q) use ($server) {
|
|
$q->where('id', '=', $server->node_id);
|
|
})
|
|
->get($this->getColumns());
|
|
}
|
|
}
|