Fix errors with mounts; closes pterodactyl#2374

Author: DaneEveritt

app/Http/Controllers/Admin/MountController.php

@@ -2,10 +2,12 @@
 
 namespace Pterodactyl\Http\Controllers\Admin;
 
+use Ramsey\Uuid\Uuid;
 use Illuminate\Http\Request;
+use Pterodactyl\Models\Nest;
 use Pterodactyl\Models\Mount;
+use Pterodactyl\Models\Location;
 use Prologue\Alerts\AlertsMessageBag;
-use Pterodactyl\Exceptions\DisplayException;
 use Pterodactyl\Http\Controllers\Controller;
 use Pterodactyl\Services\Mounts\MountUpdateService;
 use Pterodactyl\Http\Requests\Admin\MountFormRequest;
@@ -37,48 +39,24 @@ class MountController extends Controller
      */
     protected $repository;
 
-    /**
-     * @var \Pterodactyl\Services\Mounts\MountCreationService
-     */
-    protected $creationService;
-
-    /**
-     * @var \Pterodactyl\Services\Mounts\MountDeletionService
-     */
-    protected $deletionService;
-
-    /**
-     * @var \Pterodactyl\Services\Mounts\MountUpdateService
-     */
-    protected $updateService;
-
     /**
      * MountController constructor.
      *
      * @param \Prologue\Alerts\AlertsMessageBag $alert
      * @param \Pterodactyl\Contracts\Repository\NestRepositoryInterface $nestRepository
      * @param \Pterodactyl\Contracts\Repository\LocationRepositoryInterface $locationRepository
      * @param \Pterodactyl\Repositories\Eloquent\MountRepository $repository
-     * @param \Pterodactyl\Services\Mounts\MountCreationService $creationService
-     * @param \Pterodactyl\Services\Mounts\MountDeletionService $deletionService
-     * @param \Pterodactyl\Services\Mounts\MountUpdateService $updateService
      */
     public function __construct(
         AlertsMessageBag $alert,
         NestRepositoryInterface $nestRepository,
         LocationRepositoryInterface $locationRepository,
-        MountRepository $repository,
-        MountCreationService $creationService,
-        MountDeletionService $deletionService,
-        MountUpdateService $updateService
+        MountRepository $repository
     ) {
         $this->alert = $alert;
         $this->nestRepository = $nestRepository;
         $this->locationRepository = $locationRepository;
         $this->repository = $repository;
-        $this->creationService = $creationService;
-        $this->deletionService = $deletionService;
-        $this->updateService = $updateService;
     }
 
     /**
@@ -103,11 +81,8 @@ public function index()
      */
     public function view($id)
     {
-        $nests = $this->nestRepository->all();
-        $nests->load('eggs');
-
-        $locations = $this->locationRepository->all();
-        $locations->load('nodes');
+        $nests = Nest::query()->with('eggs')->get();
+        $locations = Location::query()->with('nodes')->get();
 
         return view('admin.mounts.view', [
             'mount' => $this->repository->getWithRelations($id),
@@ -126,7 +101,11 @@ public function view($id)
      */
     public function create(MountFormRequest $request)
     {
-        $mount = $this->creationService->handle($request->normalize());
+        /** @var \Pterodactyl\Models\Mount $mount */
+        $mount = Mount::query()->create(array_merge($request->validated(), [
+            'uuid' => Uuid::uuid4()->toString(),
+        ]));
+
         $this->alert->success('Mount was created successfully.')->flash();
 
         return redirect()->route('admin.mounts.view', $mount->id);
@@ -147,7 +126,8 @@ public function update(MountFormRequest $request, Mount $mount)
             return $this->delete($mount);
         }
 
-        $this->updateService->handle($mount->id, $request->normalize());
+        $mount->forceFill($request->validated())->save();
+
         $this->alert->success('Mount was updated successfully.')->flash();
 
         return redirect()->route('admin.mounts.view', $mount->id);
@@ -163,15 +143,9 @@ public function update(MountFormRequest $request, Mount $mount)
      */
     public function delete(Mount $mount)
     {
-        try {
-            $this->deletionService->handle($mount->id);
-
-            return redirect()->route('admin.mounts');
-        } catch (DisplayException $ex) {
-            $this->alert->danger($ex->getMessage())->flash();
-        }
+        $mount->delete();
 
-        return redirect()->route('admin.mounts.view', $mount->id);
+        return redirect()->route('admin.mounts');
     }
 
     /**
@@ -188,11 +162,12 @@ public function addEggs(Request $request, Mount $mount)
         ]);
 
         $eggs = $validatedData['eggs'] ?? [];
-        if (sizeof($eggs) > 0) {
-            $mount->eggs()->attach(array_map('intval', $eggs));
-            $this->alert->success('Mount was updated successfully.')->flash();
+        if (count($eggs) > 0) {
+            $mount->eggs()->attach($eggs);
         }
 
+        $this->alert->success('Mount was updated successfully.')->flash();
+
         return redirect()->route('admin.mounts.view', $mount->id);
     }
 
@@ -205,16 +180,15 @@ public function addEggs(Request $request, Mount $mount)
      */
     public function addNodes(Request $request, Mount $mount)
     {
-        $validatedData = $request->validate([
-            'nodes' => 'required|exists:nodes,id',
-        ]);
+        $data = $request->validate(['nodes' => 'required|exists:nodes,id']);
 
-        $nodes = $validatedData['nodes'] ?? [];
-        if (sizeof($nodes) > 0) {
-            $mount->nodes()->attach(array_map('intval', $nodes));
-            $this->alert->success('Mount was updated successfully.')->flash();
+        $nodes = $data['nodes'] ?? [];
+        if (count($nodes) > 0) {
+            $mount->nodes()->attach($nodes);
         }
 
+        $this->alert->success('Mount was updated successfully.')->flash();
+
         return redirect()->route('admin.mounts.view', $mount->id);
     }
 

app/Http/Controllers/Admin/ServersController.php

@@ -12,6 +12,7 @@
 use Illuminate\Support\Arr;
 use Illuminate\Http\Request;
 use Pterodactyl\Models\User;
+use Pterodactyl\Models\Mount;
 use Pterodactyl\Models\Server;
 use Prologue\Alerts\AlertsMessageBag;
 use GuzzleHttp\Exception\RequestException;
@@ -412,12 +413,18 @@ public function deleteDatabase($server, $database)
      * Add a mount to a server.
      *
      * @param Server $server
-     * @param int $mount_id
+     * @param \Pterodactyl\Models\Mount $mount
      * @return \Illuminate\Http\RedirectResponse
+     *
+     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
-    public function addMount(Server $server, int $mount_id)
+    public function addMount(Server $server, Mount $mount)
     {
-        $server->mounts()->attach($mount_id);
+        $server->mounts()->updateOrCreate([
+            'mount_id' => $mount->id,
+            'server_id' => $server->id,
+        ]);
 
         $data = $this->serverConfigurationStructureService->handle($server);
 
@@ -438,15 +445,18 @@ public function addMount(Server $server, int $mount_id)
      * Remove a mount from a server.
      *
      * @param Server $server
-     * @param int $mount_id
+     * @param \Pterodactyl\Models\Mount $mount
      * @return \Illuminate\Http\RedirectResponse
      *
-     * @throws DaemonConnectionException
+     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
      * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
-    public function deleteMount(Server $server, int $mount_id)
+    public function deleteMount(Server $server, Mount $mount)
     {
-        $server->mounts()->detach($mount_id);
+        $server->mounts()
+            ->where('mount_id', $mount->id)
+            ->where('server_id', $server->id)
+            ->delete();
 
         $data = $this->serverConfigurationStructureService->handle($server);
 

app/Http/Requests/Admin/MountFormRequest.php

@@ -1,11 +1,4 @@
 <?php
-/**
- * Pterodactyl - Panel
- * Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
- *
- * This software is licensed under the terms of the MIT license.
- * https://opensource.org/licenses/MIT
- */
 
 namespace Pterodactyl\Http\Requests\Admin;
 

app/Models/EggMount.php

@@ -0,0 +1,21 @@
+<?php
+
+namespace Pterodactyl\Models;
+
+class EggMount extends Model
+{
+    /**
+     * @var string
+     */
+    protected $table = 'egg_mount';
+
+    /**
+     * @var null
+     */
+    protected $primaryKey = null;
+
+    /**
+     * @var bool
+     */
+    public $incrementing = false;
+}