Add ui elements for handling server transfers, add TransferJob.php and TransferService.php

app/Http/Controllers/Admin/Servers/ServerTransferController.php

@@ -0,0 +1,97 @@
+<?php
+
+namespace Pterodactyl\Http\Controllers\Admin\Servers;
+
+use Illuminate\Bus\Dispatcher;
+use Illuminate\Http\Request;
+use Prologue\Alerts\AlertsMessageBag;
+use Pterodactyl\Http\Controllers\Controller;
+use Pterodactyl\Jobs\Server\TransferJob;
+use Pterodactyl\Models\Server;
+use Pterodactyl\Repositories\Eloquent\ServerRepository;
+use Pterodactyl\Repositories\Eloquent\LocationRepository;
+use Pterodactyl\Repositories\Eloquent\NodeRepository;
+
+class ServerTransferController extends Controller
+{
+    /**
+     * @var \Prologue\Alerts\AlertsMessageBag
+     */
+    private $alert;
+
+    /**
+     * @var \Illuminate\Bus\Dispatcher
+     */
+    private $dispatcher;
+
+    /**
+     * @var \Pterodactyl\Repositories\Eloquent\ServerRepository
+     */
+    private $repository;
+
+    /**
+     * @var \Pterodactyl\Repositories\Eloquent\LocationRepository
+     */
+    private $locationRepository;
+
+    /**
+     * @var \Pterodactyl\Repositories\Eloquent\NodeRepository
+     */
+    private $nodeRepository;
+
+    /**
+     * ServerTransferController constructor.
+     *
+     * @param \Prologue\Alerts\AlertsMessageBag $alert
+     * @param \Illuminate\Bus\Dispatcher $dispatcher
+     * @param \Pterodactyl\Repositories\Eloquent\ServerRepository $repository
+     * @param \Pterodactyl\Repositories\Eloquent\LocationRepository $locationRepository
+     * @param \Pterodactyl\Repositories\Eloquent\NodeRepository $nodeRepository
+     */
+    public function __construct(
+        AlertsMessageBag $alert,
+        Dispatcher $dispatcher,
+        ServerRepository $repository,
+        LocationRepository $locationRepository,
+        NodeRepository $nodeRepository
+    ) {
+        $this->alert = $alert;
+        $this->dispatcher = $dispatcher;
+        $this->repository = $repository;
+        $this->locationRepository = $locationRepository;
+        $this->nodeRepository = $nodeRepository;
+    }
+
+    /**
+     * Starts a transfer of a server to a new node.
+     *
+     * @param \Illuminate\Http\Request $request
+     * @param \Pterodactyl\Models\Server $server
+     * @return \Illuminate\Http\RedirectResponse
+     */
+    public function transfer(Request $request, Server $server)
+    {
+        $validatedData = $request->validate([
+            'node_id' => 'required|exists:nodes,id',
+            'allocation_id' => 'required|bail|unique:servers|exists:allocations,id',
+            'allocation_additional' => 'nullable',
+        ]);
+
+        $node_id = $validatedData['node_id'];
+        $allocation_id = $validatedData['allocation_id'];
+        $additional_allocations = $validatedData['allocation_additional'] ?? [];
+
+        // Check if the node is viable for the transfer.
+        $node = $this->nodeRepository->getNodeWithResourceUsage($node_id);
+        if ($node->isViable($server->memory, $server->disk)) {
+            // TODO: Run TransferJob.
+            $this->dispatcher->dispatch(new TransferJob($server, $node, $allocation_id, $additional_allocations));
+
+            $this->alert->success(trans('admin/server.alerts.transfer_started'))->flash();
+        } else {
+            $this->alert->danger(trans('admin/server.alerts.transfer_not_viable'))->flash();
+        }
+
+        return redirect()->route('admin.servers.view.manage', $server->id);
+    }
+}

app/Http/Controllers/Admin/Servers/ServerViewController.php

@@ -2,6 +2,7 @@
 
 namespace Pterodactyl\Http\Controllers\Admin\Servers;
 
+use JavaScript;
 use Illuminate\Http\Request;
 use Pterodactyl\Models\Nest;
 use Pterodactyl\Models\Server;
@@ -9,6 +10,8 @@ use Illuminate\Contracts\View\Factory;
 use Pterodactyl\Exceptions\DisplayException;
 use Pterodactyl\Http\Controllers\Controller;
 use Pterodactyl\Repositories\Eloquent\NestRepository;
+use Pterodactyl\Repositories\Eloquent\LocationRepository;
+use Pterodactyl\Repositories\Eloquent\NodeRepository;
 use Pterodactyl\Repositories\Eloquent\ServerRepository;
 use Pterodactyl\Traits\Controllers\JavascriptInjection;
 use Pterodactyl\Repositories\Eloquent\DatabaseHostRepository;
@@ -37,17 +40,31 @@ class ServerViewController extends Controller
      */
     private $nestRepository;
 
+    /**
+     * @var \Pterodactyl\Repositories\Eloquent\LocationRepository
+     */
+    private $locationRepository;
+
+    /**
+     * @var \Pterodactyl\Repositories\Eloquent\NodeRepository
+     */
+    private $nodeRepository;
+
     /**
      * ServerViewController constructor.
      *
      * @param \Pterodactyl\Repositories\Eloquent\DatabaseHostRepository $databaseHostRepository
      * @param \Pterodactyl\Repositories\Eloquent\NestRepository $nestRepository
+     * @param \Pterodactyl\Repositories\Eloquent\LocationRepository $locationRepository
+     * @param \Pterodactyl\Repositories\Eloquent\NodeRepository $nodeRepository
      * @param \Pterodactyl\Repositories\Eloquent\ServerRepository $repository
      * @param \Illuminate\Contracts\View\Factory $view
      */
     public function __construct(
         DatabaseHostRepository $databaseHostRepository,
         NestRepository $nestRepository,
+        LocationRepository $locationRepository,
+        NodeRepository $nodeRepository,
         ServerRepository $repository,
         Factory $view
     ) {
@@ -55,6 +72,8 @@ class ServerViewController extends Controller
         $this->databaseHostRepository = $databaseHostRepository;
         $this->repository = $repository;
         $this->nestRepository = $nestRepository;
+        $this->nodeRepository = $nodeRepository;
+        $this->locationRepository = $locationRepository;
     }
 
     /**
@@ -150,6 +169,7 @@ class ServerViewController extends Controller
      * @return \Illuminate\Contracts\View\View
      *
      * @throws \Pterodactyl\Exceptions\DisplayException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
     public function manage(Request $request, Server $server)
     {
@@ -159,7 +179,22 @@ class ServerViewController extends Controller
             );
         }
 
-        return $this->view->make('admin.servers.view.manage', compact('server'));
+        // Check if the panel doesn't have at least 2 nodes configured.
+        $nodes = $this->nodeRepository->all();
+        $canTransfer = false;
+        if (count($nodes) >= 2) {
+            $canTransfer = true;
+        }
+
+        Javascript::put([
+            'nodeData' => $this->nodeRepository->getNodesForServerCreation(),
+        ]);
+
+        return $this->view->make('admin.servers.view.manage', [
+            'server' => $server,
+            'locations' => $this->locationRepository->all(),
+            'canTransfer' => $canTransfer,
+        ]);
     }
 
     /**

app/Jobs/Server/TransferJob.php

@@ -0,0 +1,91 @@
+<?php
+
+namespace Pterodactyl\Jobs\Server;
+
+use Illuminate\Bus\Queueable;
+use Illuminate\Queue\SerializesModels;
+use Illuminate\Queue\InteractsWithQueue;
+use Illuminate\Contracts\Queue\ShouldQueue;
+use Illuminate\Foundation\Bus\Dispatchable;
+use Pterodactyl\Models\Node;
+use Pterodactyl\Models\Server;
+use Pterodactyl\Services\Servers\ServerCreationService;
+use Pterodactyl\Services\Servers\ServerDeletionService;
+use Pterodactyl\Services\Servers\SuspensionService;
+use Pterodactyl\Services\Servers\TransferService;
+
+class TransferJob implements ShouldQueue
+{
+    use Dispatchable, InteractsWithQueue, Queueable, SerializesModels;
+
+    private $server, $node, $allocation_id, $additional_allocations;
+
+    /**
+     * Create a new job instance.
+     *
+     * @param Server $serverToTransfer
+     * @param Node $newNode
+     */
+    public function __construct(Server $serverToTransfer, Node $newNode, int $allocation_id, array $additional_allocations)
+    {
+        $this->server = $serverToTransfer;
+        $this->node = $newNode;
+        $this->allocation_id = $allocation_id;
+        $this->additional_allocations = $additional_allocations;
+    }
+
+    /**
+     * Execute the job.
+     *
+     * @param ServerCreationService $creationService
+     * @param ServerDeletionService $deletionService
+     * @param SuspensionService $suspensionService
+     * @param TransferService $transferService
+     * @return void
+     *
+     * @throws \Pterodactyl\Exceptions\DisplayException
+     * @throws \Throwable
+     */
+    public function handle(
+        ServerCreationService $creationService,
+        ServerDeletionService $deletionService,
+        SuspensionService $suspensionService,
+        TransferService $transferService
+    ) {
+        //$server = $this->server;
+        //$newNode = $this->node;
+
+        // 1. Suspend Old Server
+        //$suspensionService->toggle($server, 'suspend');
+
+        // 2. Zip Folder
+        //$backup = $server->generateBackup();
+
+        // 3. Transfer Zip File
+        //$archive = $newNode->transfer($backup);
+
+        // 4. Verify File Hash
+        /*if ($backup->hash !== $archive->hash) {
+            $archive->delete();
+            abort(500, 'File transfer corrupted, please try again.');
+        }*/
+
+        // 5. Unzip File
+        //$archive->extract();
+
+        // 6. Update Settings on New Node
+        //$newServerDetails = $server->toArray();
+        //$newServerDetails['node_id'] = $newNode->id;
+        //$newServer = $creationService->create($newServerDetails);
+
+        // 7. Verify Server Status
+        /*if (!$newServer->isWorking()) {
+            $deletionService->withForce()->handle($newServer);
+            abort(500, 'Server failed to startup, please try again.');
+        }*/
+
+        // 8. Unsuspend Old Server
+        //$deletionService->withForce()->handle($server);
+        //$suspensionService->toggle($server, 'unsuspend');
+    }
+}

app/Models/Node.php

@@ -235,4 +235,18 @@ class Node extends Validable
     {
         return $this->hasMany(Allocation::class);
     }
+
+    /**
+     * Returns a boolean if the node is viable for an additional server to be placed on it.
+     *
+     * @param int $memory
+     * @param int $disk
+     * @return bool
+     */
+    public function isViable(int $memory, int $disk): bool {
+        $memoryLimit = $this->memory * (1 + ($this->memory_overallocate / 100));
+        $diskLimit = $this->disk * (1 + ($this->disk_overallocate / 100));
+
+        return ($this->sum_memory + $memory) <= $memoryLimit && ($this->sum_disk + $disk) <= $diskLimit;
+    }
 }

app/Repositories/Eloquent/NodeRepository.php

@@ -174,6 +174,23 @@ class NodeRepository extends EloquentRepository implements NodeRepositoryInterfa
         })->values();
     }
 
+    /**
+     * Returns a node with the given id with the Node's resource usage.
+     *
+     * @param int $node_id
+     * @return Node
+     */
+    public function getNodeWithResourceUsage(int $node_id): Node
+    {
+        $instance = $this->getBuilder()
+            ->select(['nodes.id', 'nodes.memory', 'nodes.disk', 'nodes.memory_overallocate', 'nodes.disk_overallocate'])
+            ->selectRaw('IFNULL(SUM(servers.memory), 0) as sum_memory, IFNULL(SUM(servers.disk), 0) as sum_disk')
+            ->leftJoin('servers', 'servers.node_id', '=', 'nodes.id')
+            ->where('nodes.id', $node_id);
+
+        return $instance->first();
+    }
+
     /**
      * Return the IDs of all nodes that exist in the provided locations and have the space
      * available to support the additional disk and memory provided.

app/Services/Servers/ServerCreationService.php

@@ -4,7 +4,6 @@ namespace Pterodactyl\Services\Servers;
 
 use Ramsey\Uuid\Uuid;
 use Illuminate\Support\Arr;
-use Pterodactyl\Models\Node;
 use Pterodactyl\Models\User;
 use Pterodactyl\Models\Server;
 use Illuminate\Support\Collection;

app/Services/Servers/TransferService.php

@@ -0,0 +1,58 @@
+<?php
+
+namespace Pterodactyl\Services\Servers;
+
+use Illuminate\Database\ConnectionInterface;
+use Psr\Log\LoggerInterface;
+use Pterodactyl\Contracts\Repository\ServerRepositoryInterface;
+use Pterodactyl\Models\Server;
+use Pterodactyl\Repositories\Wings\DaemonServerRepository;
+
+class TransferService
+{
+
+    /**
+     * @var \Illuminate\Database\ConnectionInterface
+     */
+    private $connection;
+
+    /**
+     * @var \Pterodactyl\Contracts\Repository\ServerRepositoryInterface
+     */
+    private $repository;
+
+    /**
+     * @var \Pterodactyl\Repositories\Wings\DaemonServerRepository
+     */
+    private $daemonServerRepository;
+
+    /**
+     * @var \Psr\Log\LoggerInterface
+     */
+    private $writer;
+
+    /**
+     * TransferService constructor.
+     *
+     * @param \Illuminate\Database\ConnectionInterface $connection
+     * @param \Pterodactyl\Repositories\Wings\DaemonServerRepository $daemonServerRepository
+     * @param \Pterodactyl\Contracts\Repository\ServerRepositoryInterface $repository
+     * @param \Psr\Log\LoggerInterface $writer
+     */
+    public function __construct(
+        ConnectionInterface $connection,
+        DaemonServerRepository $daemonServerRepository,
+        ServerRepositoryInterface $repository,
+        LoggerInterface $writer
+    ) {
+        $this->connection = $connection;
+        $this->repository = $repository;
+        $this->daemonServerRepository = $daemonServerRepository;
+        $this->writer = $writer;
+    }
+
+    public function handle(Server $server)
+    {
+
+    }
+}