Merge branch 'develop' into v2

This commit is contained in:
Matthew Penner 2021-08-04 21:40:38 -06:00
commit 4b32828423
No known key found for this signature in database
GPG key ID: 5396CC4C3C1C9704
7 changed files with 34 additions and 14 deletions

View file

@ -13,6 +13,7 @@ use Illuminate\Auth\Access\AuthorizationException;
use Pterodactyl\Services\Backups\DeleteBackupService; use Pterodactyl\Services\Backups\DeleteBackupService;
use Pterodactyl\Services\Backups\DownloadLinkService; use Pterodactyl\Services\Backups\DownloadLinkService;
use Pterodactyl\Services\Backups\InitiateBackupService; use Pterodactyl\Services\Backups\InitiateBackupService;
use Pterodactyl\Repositories\Eloquent\BackupRepository;
use Pterodactyl\Repositories\Wings\DaemonBackupRepository; use Pterodactyl\Repositories\Wings\DaemonBackupRepository;
use Pterodactyl\Transformers\Api\Client\BackupTransformer; use Pterodactyl\Transformers\Api\Client\BackupTransformer;
use Pterodactyl\Http\Controllers\Api\Client\ClientApiController; use Pterodactyl\Http\Controllers\Api\Client\ClientApiController;
@ -24,16 +25,18 @@ class BackupController extends ClientApiController
private InitiateBackupService $initiateBackupService; private InitiateBackupService $initiateBackupService;
private DeleteBackupService $deleteBackupService; private DeleteBackupService $deleteBackupService;
private DownloadLinkService $downloadLinkService; private DownloadLinkService $downloadLinkService;
private DaemonBackupRepository $repository; private DaemonBackupRepository $daemonRepository;
private BackupRepository $repository;
/** /**
* BackupController constructor. * BackupController constructor.
*/ */
public function __construct( public function __construct(
DaemonBackupRepository $repository, DaemonBackupRepository $daemonRepository,
DeleteBackupService $deleteBackupService, DeleteBackupService $deleteBackupService,
InitiateBackupService $initiateBackupService, InitiateBackupService $initiateBackupService,
DownloadLinkService $downloadLinkService DownloadLinkService $downloadLinkService,
BackupRepository $repository
) { ) {
parent::__construct(); parent::__construct();
@ -41,6 +44,7 @@ class BackupController extends ClientApiController
$this->initiateBackupService = $initiateBackupService; $this->initiateBackupService = $initiateBackupService;
$this->deleteBackupService = $deleteBackupService; $this->deleteBackupService = $deleteBackupService;
$this->downloadLinkService = $downloadLinkService; $this->downloadLinkService = $downloadLinkService;
$this->daemonRepository = $daemonRepository;
} }
/** /**
@ -61,7 +65,7 @@ class BackupController extends ClientApiController
return $this->fractal->collection($server->backups()->paginate($limit)) return $this->fractal->collection($server->backups()->paginate($limit))
->transformWith($this->getTransformer(BackupTransformer::class)) ->transformWith($this->getTransformer(BackupTransformer::class))
->addMeta([ ->addMeta([
'used_backup_count' => $this->initiateBackupService->getNonFailedBackups($server)->count(), 'backup_count' => $this->repository->getNonFailedBackups($server)->count(),
]) ])
->toArray(); ->toArray();
} }
@ -69,6 +73,8 @@ class BackupController extends ClientApiController
/** /**
* Starts the backup process for a server. * Starts the backup process for a server.
* *
* @throws \Spatie\Fractalistic\Exceptions\InvalidTransformation
* @throws \Spatie\Fractalistic\Exceptions\NoTransformerSpecified
* @throws \Throwable * @throws \Throwable
*/ */
public function store(StoreBackupRequest $request, Server $server): array public function store(StoreBackupRequest $request, Server $server): array
@ -128,7 +134,6 @@ class BackupController extends ClientApiController
* Returns information about a single backup. * Returns information about a single backup.
* *
* @throws \Illuminate\Auth\Access\AuthorizationException * @throws \Illuminate\Auth\Access\AuthorizationException
* @throws \Illuminate\Contracts\Container\BindingResolutionException
*/ */
public function view(Request $request, Server $server, Backup $backup): array public function view(Request $request, Server $server, Backup $backup): array
{ {
@ -231,7 +236,7 @@ class BackupController extends ClientApiController
// actions against it via the Panel API. // actions against it via the Panel API.
$server->update(['status' => Server::STATUS_RESTORING_BACKUP]); $server->update(['status' => Server::STATUS_RESTORING_BACKUP]);
$this->repository->setServer($server)->restore($backup, $url ?? null, $request->input('truncate')); $this->daemonRepository->setServer($server)->restore($backup, $url ?? null, $request->input('truncate'));
}); });
return $this->returnNoContent(); return $this->returnNoContent();

View file

@ -4,6 +4,8 @@ namespace Pterodactyl\Repositories\Eloquent;
use Carbon\Carbon; use Carbon\Carbon;
use Pterodactyl\Models\Backup; use Pterodactyl\Models\Backup;
use Pterodactyl\Models\Server;
use Illuminate\Database\Eloquent\Relations\HasMany;
class BackupRepository extends EloquentRepository class BackupRepository extends EloquentRepository
{ {
@ -33,4 +35,17 @@ class BackupRepository extends EloquentRepository
->get() ->get()
->toBase(); ->toBase();
} }
/**
* Returns a query filtering only non-failed backups for a specific server.
*
* @return \Illuminate\Database\Eloquent\Relations\HasMany
*/
public function getNonFailedBackups(Server $server): HasMany
{
return $server->backups()->where(function ($query) {
$query->whereNull('completed_at')
->orWhere('is_successful', true);
});
}
} }

View file

@ -135,7 +135,7 @@ class InitiateBackupService
// Check if the server has reached or exceeded its backup limit. // Check if the server has reached or exceeded its backup limit.
// completed_at == null will cover any ongoing backups, while is_successful == true will cover any completed backups. // completed_at == null will cover any ongoing backups, while is_successful == true will cover any completed backups.
$successful = $this->getNonFailedBackups($server); $successful = $this->repository->getNonFailedBackups($server);
if (!$server->backup_limit || $successful->count() >= $server->backup_limit) { if (!$server->backup_limit || $successful->count() >= $server->backup_limit) {
// Do not allow the user to continue if this server is already at its limit and can't override. // Do not allow the user to continue if this server is already at its limit and can't override.
if (!$override || $server->backup_limit <= 0) { if (!$override || $server->backup_limit <= 0) {

View file

@ -12,7 +12,7 @@ interface ctx {
export const Context = createContext<ctx>({ page: 1, setPage: () => 1 }); export const Context = createContext<ctx>({ page: 1, setPage: () => 1 });
type BackupResponse = PaginatedResult<ServerBackup> & { usedBackupCount: number }; type BackupResponse = PaginatedResult<ServerBackup> & { backupCount: number };
export default () => { export default () => {
const { page } = useContext(Context); const { page } = useContext(Context);
@ -24,7 +24,7 @@ export default () => {
return ({ return ({
items: (data.data || []).map(rawDataToServerBackup), items: (data.data || []).map(rawDataToServerBackup),
pagination: getPaginationSet(data.meta.pagination), pagination: getPaginationSet(data.meta.pagination),
usedBackupCount: data.meta.used_backup_count, backupCount: data.meta.backup_count,
}); });
}); });
}; };

View file

@ -65,12 +65,12 @@ const BackupContainer = () => {
} }
<Can action={'backup.create'}> <Can action={'backup.create'}>
<div css={tw`mt-6 sm:flex items-center justify-end`}> <div css={tw`mt-6 sm:flex items-center justify-end`}>
{(backupLimit > 0 && backups.usedBackupCount > 0) && {(backupLimit > 0 && backups.backupCount > 0) &&
<p css={tw`text-sm text-neutral-300 mb-4 sm:mr-6 sm:mb-0`}> <p css={tw`text-sm text-neutral-300 mb-4 sm:mr-6 sm:mb-0`}>
{backups.usedBackupCount} of {backupLimit} backups have been created for this server. {backups.backupCount} of {backupLimit} backups have been created for this server.
</p> </p>
} }
{backupLimit > 0 && backupLimit > backups.usedBackupCount && {backupLimit > 0 && backupLimit > backups.backupCount &&
<CreateBackupButton css={tw`w-full sm:w-auto`}/> <CreateBackupButton css={tw`w-full sm:w-auto`}/>
} }
</div> </div>

View file

@ -51,7 +51,7 @@ export default ({ backup }: Props) => {
.then(() => mutate(data => ({ .then(() => mutate(data => ({
...data!, ...data!,
items: data!.items.filter(b => b.uuid !== backup.uuid), items: data!.items.filter(b => b.uuid !== backup.uuid),
usedBackupCount: data!.usedBackupCount - 1, backupCount: data!.backupCount - 1,
}), false)) }), false))
.catch(error => { .catch(error => {
console.error(error); console.error(error);

View file

@ -81,7 +81,7 @@ export default () => {
clearFlashes('backups:create'); clearFlashes('backups:create');
createServerBackup(uuid, values) createServerBackup(uuid, values)
.then(backup => { .then(backup => {
mutate(data => ({ ...data!, items: data!.items.concat(backup), usedBackupCount: data!.usedBackupCount + 1 }), false); mutate(data => ({ ...data!, items: data!.items.concat(backup), backupCount: data!.backupCount + 1 }), false);
setVisible(false); setVisible(false);
}) })
.catch(error => { .catch(error => {