Get to the point where we can start notifying the other daemon, remove TransferJob.php, add DaemonTransferRepository.php

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

@@ -6,11 +6,12 @@ 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;
+use Pterodactyl\Services\Servers\SuspensionService;
+use Pterodactyl\Services\Servers\TransferService;
 
 class ServerTransferController extends Controller
 {
@@ -39,6 +40,16 @@ class ServerTransferController extends Controller
      */
     private $nodeRepository;
 
+    /**
+     * @var \Pterodactyl\Services\Servers\SuspensionService
+     */
+    private $suspensionService;
+
+    /**
+     * @var \Pterodactyl\Services\Servers\TransferService
+     */
+    private $transferService;
+
     /**
      * ServerTransferController constructor.
      *
@@ -47,19 +58,25 @@ class ServerTransferController extends Controller
      * @param \Pterodactyl\Repositories\Eloquent\ServerRepository $repository
      * @param \Pterodactyl\Repositories\Eloquent\LocationRepository $locationRepository
      * @param \Pterodactyl\Repositories\Eloquent\NodeRepository $nodeRepository
+     * @param \Pterodactyl\Services\Servers\SuspensionService $suspensionService
+     * @param \Pterodactyl\Services\Servers\TransferService $transferService
      */
     public function __construct(
         AlertsMessageBag $alert,
         Dispatcher $dispatcher,
         ServerRepository $repository,
         LocationRepository $locationRepository,
-        NodeRepository $nodeRepository
+        NodeRepository $nodeRepository,
+        SuspensionService $suspensionService,
+        TransferService $transferService
     ) {
         $this->alert = $alert;
         $this->dispatcher = $dispatcher;
         $this->repository = $repository;
         $this->locationRepository = $locationRepository;
         $this->nodeRepository = $nodeRepository;
+        $this->suspensionService = $suspensionService;
+        $this->transferService = $transferService;
     }
 
     /**
@@ -68,6 +85,8 @@ class ServerTransferController extends Controller
      * @param \Illuminate\Http\Request $request
      * @param \Pterodactyl\Models\Server $server
      * @return \Illuminate\Http\RedirectResponse
+     *
+     * @throws \Throwable
      */
     public function transfer(Request $request, Server $server)
     {
@@ -84,8 +103,9 @@ class ServerTransferController extends Controller
         // 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));
+            // Suspend the server and request an archive to be created.
+            // $this->suspensionService->toggle($server, 'suspend');
+            $this->transferService->requestArchive($server);
 
             $this->alert->success(trans('admin/server.alerts.transfer_started'))->flash();
         } else {

app/Http/Controllers/Api/Remote/Servers/ServerTransferController.php

@@ -0,0 +1,85 @@
+<?php
+
+namespace Pterodactyl\Http\Controllers\Api\Remote\Servers;
+
+use Cake\Chronos\Chronos;
+use Illuminate\Http\JsonResponse;
+use Illuminate\Http\Request;
+use Illuminate\Http\Response;
+use Lcobucci\JWT\Builder;
+use Lcobucci\JWT\Signer\Hmac\Sha256;
+use Lcobucci\JWT\Signer\Key;
+use Pterodactyl\Http\Controllers\Controller;
+use Pterodactyl\Models\Server;
+use Pterodactyl\Repositories\Eloquent\ServerRepository;
+use Pterodactyl\Repositories\Eloquent\NodeRepository;
+use Pterodactyl\Repositories\Wings\DaemonTransferRepository;
+
+class ServerTransferController extends Controller
+{
+    /**
+     * @var \Pterodactyl\Repositories\Eloquent\ServerRepository
+     */
+    private $repository;
+
+    /**
+     * @var \Pterodactyl\Repositories\Eloquent\NodeRepository
+     */
+    private $nodeRepository;
+
+    /**
+     * @var \Pterodactyl\Repositories\Wings\DaemonTransferRepository
+     */
+    private $daemonTransferRepository;
+
+    /**
+     * ServerTransferController constructor.
+     *
+     * @param \Pterodactyl\Repositories\Eloquent\ServerRepository $repository
+     * @param \Pterodactyl\Repositories\Eloquent\NodeRepository $nodeRepository
+     * @param DaemonTransferRepository $daemonTransferRepository
+     */
+    public function __construct(
+        ServerRepository $repository,
+        NodeRepository $nodeRepository,
+        DaemonTransferRepository $daemonTransferRepository
+    ) {
+        $this->repository = $repository;
+        $this->nodeRepository = $nodeRepository;
+        $this->daemonTransferRepository = $daemonTransferRepository;
+    }
+
+    /**
+     * The daemon notifies us about the archive status.
+     *
+     * @param \Illuminate\Http\Request $request
+     * @param \Pterodactyl\Models\Server $server
+     * @return \Illuminate\Http\JsonResponse
+     *
+     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
+     */
+    public function archive(Request $request, Server $server)
+    {
+        // Unsuspend the server and don't continue the transfer.
+        if (!$request->input('successful')) {
+            // $this->suspensionService->toggle($server, 'unsuspend');
+            return JsonResponse::create([], Response::HTTP_NO_CONTENT);
+        }
+
+        $now = Chronos::now();
+        $signer = new Sha256;
+
+        $token = (new Builder)->issuedBy(config('app.url'))
+            ->permittedFor($server->node->getConnectionAddress())
+            ->identifiedBy(hash('sha256', $server->uuid), true)
+            ->issuedAt($now->getTimestamp())
+            ->canOnlyBeUsedAfter($now->getTimestamp())
+            ->expiresAt($now->addMinutes(15)->getTimestamp())
+            ->relatedTo($server->id, true)
+            ->getToken($signer, new Key($server->node->daemonSecret));
+
+        $this->daemonTransferRepository->notify($server, $token->__toString());
+
+        return JsonResponse::create([], Response::HTTP_NO_CONTENT);
+    }
+}