99 lines
3.3 KiB
PHP
99 lines
3.3 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Http\Controllers\Api\Remote\Servers;
|
|
|
|
use Illuminate\Http\Response;
|
|
use Illuminate\Http\JsonResponse;
|
|
use Pterodactyl\Models\Allocation;
|
|
use Illuminate\Support\Facades\Log;
|
|
use Pterodactyl\Models\ServerTransfer;
|
|
use Illuminate\Database\ConnectionInterface;
|
|
use Pterodactyl\Http\Controllers\Controller;
|
|
use Pterodactyl\Repositories\Eloquent\ServerRepository;
|
|
use Pterodactyl\Repositories\Wings\DaemonServerRepository;
|
|
use Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException;
|
|
|
|
class ServerTransferController extends Controller
|
|
{
|
|
/**
|
|
* ServerTransferController constructor.
|
|
*/
|
|
public function __construct(
|
|
private ConnectionInterface $connection,
|
|
private ServerRepository $repository,
|
|
private DaemonServerRepository $daemonServerRepository
|
|
) {
|
|
}
|
|
|
|
/**
|
|
* The daemon notifies us about a transfer failure.
|
|
*
|
|
* @throws \Throwable
|
|
*/
|
|
public function failure(string $uuid): JsonResponse
|
|
{
|
|
$server = $this->repository->getByUuid($uuid);
|
|
|
|
return $this->processFailedTransfer($server->transfer);
|
|
}
|
|
|
|
/**
|
|
* The daemon notifies us about a transfer success.
|
|
*
|
|
* @throws \Throwable
|
|
*/
|
|
public function success(string $uuid): JsonResponse
|
|
{
|
|
$server = $this->repository->getByUuid($uuid);
|
|
$transfer = $server->transfer;
|
|
|
|
/** @var \Pterodactyl\Models\Server $server */
|
|
$server = $this->connection->transaction(function () use ($server, $transfer) {
|
|
$allocations = array_merge([$transfer->old_allocation], $transfer->old_additional_allocations);
|
|
|
|
// Remove the old allocations for the server and re-assign the server to the new
|
|
// primary allocation and node.
|
|
Allocation::query()->whereIn('id', $allocations)->update(['server_id' => null]);
|
|
$server->update([
|
|
'allocation_id' => $transfer->new_allocation,
|
|
'node_id' => $transfer->new_node,
|
|
]);
|
|
|
|
$server = $server->fresh();
|
|
$server->transfer->update(['successful' => true]);
|
|
|
|
return $server;
|
|
});
|
|
|
|
// Delete the server from the old node making sure to point it to the old node so
|
|
// that we do not delete it from the new node the server was transferred to.
|
|
try {
|
|
$this->daemonServerRepository
|
|
->setServer($server)
|
|
->setNode($transfer->oldNode)
|
|
->delete();
|
|
} catch (DaemonConnectionException $exception) {
|
|
Log::warning($exception, ['transfer_id' => $server->transfer->id]);
|
|
}
|
|
|
|
return new JsonResponse([], Response::HTTP_NO_CONTENT);
|
|
}
|
|
|
|
/**
|
|
* Release all the reserved allocations for this transfer and mark it as failed in
|
|
* the database.
|
|
*
|
|
* @throws \Throwable
|
|
*/
|
|
protected function processFailedTransfer(ServerTransfer $transfer): JsonResponse
|
|
{
|
|
$this->connection->transaction(function () use (&$transfer) {
|
|
$transfer->forceFill(['successful' => false])->saveOrFail();
|
|
|
|
$allocations = array_merge([$transfer->new_allocation], $transfer->new_additional_allocations);
|
|
Allocation::query()->whereIn('id', $allocations)->update(['server_id' => null]);
|
|
});
|
|
|
|
return new JsonResponse([], Response::HTTP_NO_CONTENT);
|
|
}
|
|
}
|