Finish first round of User/Node API additions

DaneEveritt · DaneEveritt · commit 15289b76a7f3 · 2018-01-01T15:11:44.000-06:00
Will still need some tweaking and improvements to allow everything to be used.
diff --git a/app/Exceptions/DisplayException.php b/app/Exceptions/DisplayException.php
@@ -11,6 +11,7 @@
 
 use Log;
 use Throwable;
+use Illuminate\Http\Response;
 use Prologue\Alerts\AlertsMessageBag;
 
 class DisplayException extends PterodactylException
@@ -65,7 +66,7 @@ public function render($request)
         if ($request->expectsJson()) {
             return response()->json(Handler::convertToArray($this, [
                 'detail' => $this->getMessage(),
-            ]), method_exists($this, 'getStatusCode') ? $this->getStatusCode() : 500);
+            ]), method_exists($this, 'getStatusCode') ? $this->getStatusCode() : Response::HTTP_INTERNAL_SERVER_ERROR);
         }
 
         app()->make(AlertsMessageBag::class)->danger($this->getMessage())->flash();
diff --git a/app/Exceptions/Service/HasActiveServersException.php b/app/Exceptions/Service/HasActiveServersException.php
@@ -1,16 +1,17 @@
 <?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\Exceptions\Service;
 
+use Illuminate\Http\Response;
 use Pterodactyl\Exceptions\DisplayException;
 
 class HasActiveServersException extends DisplayException
 {
+    /**
+     * @return int
+     */
+    public function getStatusCode()
+    {
+        return Response::HTTP_BAD_REQUEST;
+    }
 }
diff --git a/app/Exceptions/Service/Location/HasActiveNodesException.php b/app/Exceptions/Service/Location/HasActiveNodesException.php
@@ -1,16 +1,17 @@
 <?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\Exceptions\Service\Location;
 
+use Illuminate\Http\Response;
 use Pterodactyl\Exceptions\DisplayException;
 
 class HasActiveNodesException extends DisplayException
 {
+    /**
+     * @return int
+     */
+    public function getStatusCode()
+    {
+        return Response::HTTP_BAD_REQUEST;
+    }
 }
diff --git a/app/Http/Controllers/API/Admin/Nodes/NodeController.php b/app/Http/Controllers/API/Admin/Nodes/NodeController.php
@@ -5,13 +5,29 @@
 use Spatie\Fractal\Fractal;
 use Illuminate\Http\Request;
 use Pterodactyl\Models\Node;
+use Illuminate\Http\Response;
+use Illuminate\Http\JsonResponse;
 use Pterodactyl\Http\Controllers\Controller;
+use Pterodactyl\Services\Nodes\NodeUpdateService;
+use Pterodactyl\Services\Nodes\NodeCreationService;
+use Pterodactyl\Services\Nodes\NodeDeletionService;
 use Pterodactyl\Transformers\Api\Admin\NodeTransformer;
 use League\Fractal\Pagination\IlluminatePaginatorAdapter;
+use Pterodactyl\Http\Requests\Admin\Node\NodeFormRequest;
 use Pterodactyl\Contracts\Repository\NodeRepositoryInterface;
 
 class NodeController extends Controller
 {
+    /**
+     * @var \Pterodactyl\Services\Nodes\NodeCreationService
+     */
+    private $creationService;
+
+    /**
+     * @var \Pterodactyl\Services\Nodes\NodeDeletionService
+     */
+    private $deletionService;
+
     /**
      * @var \Spatie\Fractal\Fractal
      */
@@ -22,16 +38,32 @@ class NodeController extends Controller
      */
     private $repository;
 
+    /**
+     * @var \Pterodactyl\Services\Nodes\NodeUpdateService
+     */
+    private $updateService;
+
     /**
      * NodeController constructor.
      *
      * @param \Spatie\Fractal\Fractal                                   $fractal
+     * @param \Pterodactyl\Services\Nodes\NodeCreationService           $creationService
+     * @param \Pterodactyl\Services\Nodes\NodeDeletionService           $deletionService
+     * @param \Pterodactyl\Services\Nodes\NodeUpdateService             $updateService
      * @param \Pterodactyl\Contracts\Repository\NodeRepositoryInterface $repository
      */
-    public function __construct(Fractal $fractal, NodeRepositoryInterface $repository)
-    {
+    public function __construct(
+        Fractal $fractal,
+        NodeCreationService $creationService,
+        NodeDeletionService $deletionService,
+        NodeUpdateService $updateService,
+        NodeRepositoryInterface $repository
+    ) {
         $this->fractal = $fractal;
         $this->repository = $repository;
+        $this->creationService = $creationService;
+        $this->deletionService = $deletionService;
+        $this->updateService = $updateService;
     }
 
     /**
@@ -67,4 +99,63 @@ public function view(Request $request, Node $node): array
 
         return $fractal->toArray();
     }
+
+    /**
+     * Create a new node on the Panel. Returns the created node and a HTTP/201
+     * status response on success.
+     *
+     * @param \Pterodactyl\Http\Requests\Admin\Node\NodeFormRequest $request
+     * @return \Illuminate\Http\JsonResponse
+     *
+     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
+     */
+    public function store(NodeFormRequest $request): JsonResponse
+    {
+        $node = $this->creationService->handle($request->normalize());
+
+        return $this->fractal->item($node)
+            ->transformWith(new NodeTransformer($request))
+            ->withResourceName('node')
+            ->addMeta([
+                'link' => route('api.admin.node.view', ['node' => $node->id]),
+            ])
+            ->respond(201);
+    }
+
+    /**
+     * Update an existing node on the Panel.
+     *
+     * @param \Pterodactyl\Http\Requests\Admin\Node\NodeFormRequest $request
+     * @param \Pterodactyl\Models\Node                              $node
+     * @return array
+     *
+     * @throws \Pterodactyl\Exceptions\DisplayException
+     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
+     */
+    public function update(NodeFormRequest $request, Node $node): array
+    {
+        $node = $this->updateService->returnUpdatedModel()->handle($node, $request->normalize());
+
+        return $this->fractal->item($node)
+            ->transformWith(new NodeTransformer($request))
+            ->withResourceName('node')
+            ->toArray();
+    }
+
+    /**
+     * Deletes a given node from the Panel as long as there are no servers
+     * currently attached to it.
+     *
+     * @param \Pterodactyl\Models\Node $node
+     * @return \Illuminate\Http\Response
+     *
+     * @throws \Pterodactyl\Exceptions\Service\HasActiveServersException
+     */
+    public function delete(Node $node): Response
+    {
+        $this->deletionService->handle($node);
+
+        return response('', 201);
+    }
 }
diff --git a/app/Http/Controllers/API/Admin/Users/UserController.php b/app/Http/Controllers/API/Admin/Users/UserController.php
@@ -146,13 +146,17 @@ public function update(UserFormRequest $request, User $user): array
             }
         }
 
-        return $this->fractal->item($collection->get('user'))
+        $response = $this->fractal->item($collection->get('model'))
             ->transformWith(new UserTransformer($request))
-            ->withResourceName('user')
-            ->addMeta([
+            ->withResourceName('user');
+
+        if (count($errors) > 0) {
+            $response->addMeta([
                 'revocation_errors' => $errors,
-            ])
-            ->toArray();
+            ]);
+        }
+
+        return $response->toArray();
     }
 
     /**
diff --git a/app/Models/User.php b/app/Models/User.php
@@ -4,11 +4,13 @@
 
 use Sofa\Eloquence\Eloquence;
 use Sofa\Eloquence\Validable;
+use Illuminate\Validation\Rules\In;
 use Illuminate\Auth\Authenticatable;
 use Illuminate\Database\Eloquent\Model;
 use Illuminate\Notifications\Notifiable;
 use Sofa\Eloquence\Contracts\CleansAttributes;
 use Illuminate\Auth\Passwords\CanResetPassword;
+use Pterodactyl\Traits\Helpers\AvailableLanguages;
 use Illuminate\Foundation\Auth\Access\Authorizable;
 use Sofa\Eloquence\Contracts\Validable as ValidableContract;
 use Illuminate\Contracts\Auth\Authenticatable as AuthenticatableContract;
@@ -23,7 +25,9 @@ class User extends Model implements
     CleansAttributes,
     ValidableContract
 {
-    use Authenticatable, Authorizable, CanResetPassword, Eloquence, Notifiable, Validable;
+    use Authenticatable, Authorizable, AvailableLanguages, CanResetPassword, Eloquence, Notifiable, Validable {
+        gatherRules as eloquenceGatherRules;
+    }
 
     const USER_LEVEL_USER = 0;
     const USER_LEVEL_ADMIN = 1;
@@ -138,11 +142,23 @@ class User extends Model implements
         'name_last' => 'string|between:1,255',
         'password' => 'nullable|string',
         'root_admin' => 'boolean',
-        'language' => 'string|between:2,5',
+        'language' => 'string',
         'use_totp' => 'boolean',
         'totp_secret' => 'nullable|string',
     ];
 
+    /**
+     * Implement language verification by overriding Eloquence's gather
+     * rules function.
+     */
+    protected static function gatherRules()
+    {
+        $rules = self::eloquenceGatherRules();
+        $rules['language'][] = new In(array_keys((new self)->getAvailableLanguages()));
+
+        return $rules;
+    }
+
     /**
      * Send the password reset notification.
      *
diff --git a/app/Services/Nodes/NodeUpdateService.php b/app/Services/Nodes/NodeUpdateService.php
@@ -9,82 +9,71 @@
 
 namespace Pterodactyl\Services\Nodes;
 
-use Illuminate\Log\Writer;
 use Pterodactyl\Models\Node;
 use GuzzleHttp\Exception\RequestException;
-use Pterodactyl\Exceptions\DisplayException;
+use Pterodactyl\Traits\Services\ReturnsUpdatedModels;
 use Pterodactyl\Contracts\Repository\NodeRepositoryInterface;
+use Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException;
 use Pterodactyl\Contracts\Repository\Daemon\ConfigurationRepositoryInterface;
 
 class NodeUpdateService
 {
+    use ReturnsUpdatedModels;
+
     /**
      * @var \Pterodactyl\Contracts\Repository\Daemon\ConfigurationRepositoryInterface
      */
-    protected $configRepository;
+    private $configRepository;
 
     /**
      * @var \Pterodactyl\Contracts\Repository\NodeRepositoryInterface
      */
-    protected $repository;
-
-    /**
-     * @var \Illuminate\Log\Writer
-     */
-    protected $writer;
+    private $repository;
 
     /**
      * UpdateService constructor.
      *
      * @param \Pterodactyl\Contracts\Repository\Daemon\ConfigurationRepositoryInterface $configurationRepository
      * @param \Pterodactyl\Contracts\Repository\NodeRepositoryInterface                 $repository
-     * @param \Illuminate\Log\Writer                                                    $writer
      */
     public function __construct(
         ConfigurationRepositoryInterface $configurationRepository,
-        NodeRepositoryInterface $repository,
-        Writer $writer
+        NodeRepositoryInterface $repository
     ) {
         $this->configRepository = $configurationRepository;
         $this->repository = $repository;
-        $this->writer = $writer;
     }
 
     /**
      * Update the configuration values for a given node on the machine.
      *
-     * @param int|\Pterodactyl\Models\Node $node
-     * @param array                        $data
-     * @return mixed
+     * @param \Pterodactyl\Models\Node $node
+     * @param array                    $data
+     * @return \Pterodactyl\Models\Node|mixed
      *
      * @throws \Pterodactyl\Exceptions\DisplayException
      * @throws \Pterodactyl\Exceptions\Model\DataValidationException
      * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
-    public function handle($node, array $data)
+    public function handle(Node $node, array $data)
     {
-        if (! $node instanceof Node) {
-            $node = $this->repository->find($node);
-        }
-
         if (! is_null(array_get($data, 'reset_secret'))) {
-            $data['daemonSecret'] = str_random(NodeCreationService::DAEMON_SECRET_LENGTH);
+            $data['daemonSecret'] = str_random(Node::DAEMON_SECRET_LENGTH);
             unset($data['reset_secret']);
         }
 
-        $updateResponse = $this->repository->withoutFresh()->update($node->id, $data);
+        if ($this->getUpdatedModel()) {
+            $response = $this->repository->update($node->id, $data);
+        } else {
+            $response = $this->repository->withoutFresh()->update($node->id, $data);
+        }
 
         try {
             $this->configRepository->setNode($node->id)->update();
         } catch (RequestException $exception) {
-            $response = $exception->getResponse();
-            $this->writer->warning($exception);
-
-            throw new DisplayException(trans('exceptions.node.daemon_off_config_updated', [
-                'code' => is_null($response) ? 'E_CONN_REFUSED' : $response->getStatusCode(),
-            ]));
+            throw new DaemonConnectionException($exception);
         }
 
-        return $updateResponse;
+        return $response;
     }
 }
diff --git a/app/Traits/Services/ReturnsUpdatedModels.php b/app/Traits/Services/ReturnsUpdatedModels.php
diff --git a/routes/api-admin.php b/routes/api-admin.php
diff --git a/tests/Unit/Services/Nodes/NodeUpdateServiceTest.php b/tests/Unit/Services/Nodes/NodeUpdateServiceTest.php
