Add database management back to front-end and begin some refactoring

Here we go again boys...

app/Contracts/Repository/DatabaseRepositoryInterface.php

@@ -9,8 +9,35 @@
 
 namespace Pterodactyl\Contracts\Repository;
 
+use Illuminate\Support\Collection;
+
 interface DatabaseRepositoryInterface extends RepositoryInterface
 {
+    const DEFAULT_CONNECTION_NAME = 'dynamic';
+
+    /**
+     * Set the connection name to execute statements against.
+     *
+     * @param string $connection
+     * @return $this
+     */
+    public function setConnection(string $connection);
+
+    /**
+     * Return the connection to execute statements aganist.
+     *
+     * @return string
+     */
+    public function getConnection(): string;
+
+    /**
+     * Return all of the databases belonging to a server.
+     *
+     * @param int $server
+     * @return \Illuminate\Support\Collection
+     */
+    public function getDatabasesForServer(int $server): Collection;
+
     /**
      * Create a new database if it does not already exist on the host with
      * the provided details.
@@ -26,58 +53,52 @@ interface DatabaseRepositoryInterface extends RepositoryInterface
     /**
      * Create a new database on a given connection.
      *
-     * @param string      $database
-     * @param null|string $connection
+     * @param string $database
      * @return bool
      */
-    public function createDatabase($database, $connection = null);
+    public function createDatabase($database);
 
     /**
      * Create a new database user on a given connection.
      *
-     * @param string      $username
-     * @param string      $remote
-     * @param string      $password
-     * @param null|string $connection
+     * @param string $username
+     * @param string $remote
+     * @param string $password
      * @return bool
      */
-    public function createUser($username, $remote, $password, $connection = null);
+    public function createUser($username, $remote, $password);
 
     /**
      * Give a specific user access to a given database.
      *
-     * @param string      $database
-     * @param string      $username
-     * @param string      $remote
-     * @param null|string $connection
+     * @param string $database
+     * @param string $username
+     * @param string $remote
      * @return bool
      */
-    public function assignUserToDatabase($database, $username, $remote, $connection = null);
+    public function assignUserToDatabase($database, $username, $remote);
 
     /**
      * Flush the privileges for a given connection.
      *
-     * @param null|string $connection
      * @return mixed
      */
-    public function flush($connection = null);
+    public function flush();
 
     /**
      * Drop a given database on a specific connection.
      *
-     * @param string      $database
-     * @param null|string $connection
+     * @param string $database
      * @return bool
      */
-    public function dropDatabase($database, $connection = null);
+    public function dropDatabase($database);
 
     /**
      * Drop a given user on a specific connection.
      *
-     * @param string      $username
-     * @param string      $remote
-     * @param null|string $connection
+     * @param string $username
+     * @param string $remote
      * @return mixed
      */
-    public function dropUser($username, $remote, $connection = null);
+    public function dropUser($username, $remote);
 }

app/Http/Controllers/Admin/DatabaseController.php

@@ -9,11 +9,15 @@
 
 namespace Pterodactyl\Http\Controllers\Admin;
 
-use Pterodactyl\Models\DatabaseHost;
+use PDOException;
+use Illuminate\View\View;
+use Illuminate\Http\RedirectResponse;
 use Prologue\Alerts\AlertsMessageBag;
 use Pterodactyl\Http\Controllers\Controller;
-use Pterodactyl\Services\Database\DatabaseHostService;
+use Pterodactyl\Services\Databases\Hosts\HostUpdateService;
 use Pterodactyl\Http\Requests\Admin\DatabaseHostFormRequest;
+use Pterodactyl\Services\Databases\Hosts\HostCreationService;
+use Pterodactyl\Services\Databases\Hosts\HostDeletionService;
 use Pterodactyl\Contracts\Repository\LocationRepositoryInterface;
 use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
 
@@ -22,41 +26,57 @@ class DatabaseController extends Controller
     /**
      * @var \Prologue\Alerts\AlertsMessageBag
      */
-    protected $alert;
+    private $alert;
+
+    /**
+     * @var \Pterodactyl\Services\Databases\Hosts\HostCreationService
+     */
+    private $creationService;
+
+    /**
+     * @var \Pterodactyl\Services\Databases\Hosts\HostDeletionService
+     */
+    private $deletionService;
 
     /**
      * @var \Pterodactyl\Contracts\Repository\LocationRepositoryInterface
      */
-    protected $locationRepository;
+    private $locationRepository;
 
     /**
      * @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface
      */
-    protected $repository;
+    private $repository;
 
     /**
-     * @var \Pterodactyl\Services\Database\DatabaseHostService
+     * @var \Pterodactyl\Services\Databases\Hosts\HostUpdateService
      */
-    protected $service;
+    private $updateService;
 
     /**
      * DatabaseController constructor.
      *
      * @param \Prologue\Alerts\AlertsMessageBag                                 $alert
      * @param \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface $repository
-     * @param \Pterodactyl\Services\Database\DatabaseHostService                $service
+     * @param \Pterodactyl\Services\Databases\Hosts\HostCreationService         $creationService
+     * @param \Pterodactyl\Services\Databases\Hosts\HostDeletionService         $deletionService
+     * @param \Pterodactyl\Services\Databases\Hosts\HostUpdateService           $updateService
      * @param \Pterodactyl\Contracts\Repository\LocationRepositoryInterface     $locationRepository
      */
     public function __construct(
         AlertsMessageBag $alert,
         DatabaseHostRepositoryInterface $repository,
-        DatabaseHostService $service,
+        HostCreationService $creationService,
+        HostDeletionService $deletionService,
+        HostUpdateService $updateService,
         LocationRepositoryInterface $locationRepository
     ) {
         $this->alert = $alert;
+        $this->creationService = $creationService;
+        $this->deletionService = $deletionService;
         $this->repository = $repository;
-        $this->service = $service;
         $this->locationRepository = $locationRepository;
+        $this->updateService = $updateService;
     }
 
     /**
@@ -64,7 +84,7 @@ class DatabaseController extends Controller
      *
      * @return \Illuminate\View\View
      */
-    public function index()
+    public function index(): View
     {
         return view('admin.databases.index', [
             'locations' => $this->locationRepository->getAllWithNodes(),
@@ -80,7 +100,7 @@ class DatabaseController extends Controller
      *
      * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
-    public function view($host)
+    public function view($host): View
     {
         return view('admin.databases.view', [
             'locations' => $this->locationRepository->getAllWithNodes(),
@@ -94,42 +114,41 @@ class DatabaseController extends Controller
      * @param \Pterodactyl\Http\Requests\Admin\DatabaseHostFormRequest $request
      * @return \Illuminate\Http\RedirectResponse
      *
-     * @throws \Throwable
+     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
-    public function create(DatabaseHostFormRequest $request)
+    public function create(DatabaseHostFormRequest $request): RedirectResponse
     {
         try {
-            $host = $this->service->create($request->normalize());
-            $this->alert->success('Successfully created a new database host on the system.')->flash();
-
-            return redirect()->route('admin.databases.view', $host->id);
-        } catch (\PDOException $ex) {
+            $host = $this->creationService->handle($request->normalize());
+        } catch (PDOException $ex) {
             $this->alert->danger($ex->getMessage())->flash();
+
+            return redirect()->route('admin.databases');
         }
 
-        return redirect()->route('admin.databases');
+        $this->alert->success('Successfully created a new database host on the system.')->flash();
+
+        return redirect()->route('admin.databases.view', $host->id);
     }
 
     /**
      * Handle updating database host.
      *
      * @param \Pterodactyl\Http\Requests\Admin\DatabaseHostFormRequest $request
-     * @param \Pterodactyl\Models\DatabaseHost                         $host
+     * @param int                                                      $host
      * @return \Illuminate\Http\RedirectResponse
      *
      * @throws \Pterodactyl\Exceptions\DisplayException
      * @throws \Pterodactyl\Exceptions\Model\DataValidationException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
-    public function update(DatabaseHostFormRequest $request, DatabaseHost $host)
+    public function update(DatabaseHostFormRequest $request, int $host): RedirectResponse
     {
-        if ($request->input('action') === 'delete') {
-            return $this->delete($host);
-        }
-
         try {
-            $host = $this->service->update($host->id, $request->normalize());
+            $host = $this->updateService->handle($host, $request->normalize());
             $this->alert->success('Database host was updated successfully.')->flash();
-        } catch (\PDOException $ex) {
+        } catch (PDOException $ex) {
             $this->alert->danger($ex->getMessage())->flash();
         }
 
@@ -139,14 +158,14 @@ class DatabaseController extends Controller
     /**
      * Handle request to delete a database host.
      *
-     * @param \Pterodactyl\Models\DatabaseHost $host
+     * @param int $host
      * @return \Illuminate\Http\RedirectResponse
      *
-     * @throws \Pterodactyl\Exceptions\DisplayException
+     * @throws \Pterodactyl\Exceptions\Service\HasActiveServersException
      */
-    public function delete(DatabaseHost $host)
+    public function delete(int $host): RedirectResponse
     {
-        $this->service->delete($host->id);
+        $this->deletionService->handle($host);
         $this->alert->success('The requested database host has been deleted from the system.')->flash();
 
         return redirect()->route('admin.databases');

app/Http/Controllers/Admin/ServersController.php

@@ -22,12 +22,13 @@ use Pterodactyl\Services\Servers\ServerDeletionService;
 use Pterodactyl\Services\Servers\ReinstallServerService;
 use Pterodactyl\Services\Servers\ContainerRebuildService;
 use Pterodactyl\Services\Servers\BuildModificationService;
-use Pterodactyl\Services\Database\DatabaseManagementService;
+use Pterodactyl\Services\Databases\DatabasePasswordService;
 use Pterodactyl\Services\Servers\DetailsModificationService;
 use Pterodactyl\Services\Servers\StartupModificationService;
 use Pterodactyl\Contracts\Repository\NestRepositoryInterface;
 use Pterodactyl\Contracts\Repository\NodeRepositoryInterface;
 use Pterodactyl\Repositories\Eloquent\DatabaseHostRepository;
+use Pterodactyl\Services\Databases\DatabaseManagementService;
 use Illuminate\Contracts\Config\Repository as ConfigRepository;
 use Pterodactyl\Contracts\Repository\ServerRepositoryInterface;
 use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
@@ -67,10 +68,15 @@ class ServersController extends Controller
     protected $databaseRepository;
 
     /**
-     * @var \Pterodactyl\Services\Database\DatabaseManagementService
+     * @var \Pterodactyl\Services\Databases\DatabaseManagementService
      */
     protected $databaseManagementService;
 
+    /**
+     * @var \Pterodactyl\Services\Databases\DatabasePasswordService
+     */
+    protected $databasePasswordService;
+
     /**
      * @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface
      */
@@ -135,7 +141,8 @@ class ServersController extends Controller
      * @param \Illuminate\Contracts\Config\Repository                         $config
      * @param \Pterodactyl\Services\Servers\ContainerRebuildService           $containerRebuildService
      * @param \Pterodactyl\Services\Servers\ServerCreationService             $service
-     * @param \Pterodactyl\Services\Database\DatabaseManagementService        $databaseManagementService
+     * @param \Pterodactyl\Services\Databases\DatabaseManagementService       $databaseManagementService
+     * @param \Pterodactyl\Services\Databases\DatabasePasswordService         $databasePasswordService
      * @param \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface   $databaseRepository
      * @param \Pterodactyl\Repositories\Eloquent\DatabaseHostRepository       $databaseHostRepository
      * @param \Pterodactyl\Services\Servers\ServerDeletionService             $deletionService
@@ -156,6 +163,7 @@ class ServersController extends Controller
         ContainerRebuildService $containerRebuildService,
         ServerCreationService $service,
         DatabaseManagementService $databaseManagementService,
+        DatabasePasswordService $databasePasswordService,
         DatabaseRepositoryInterface $databaseRepository,
         DatabaseHostRepository $databaseHostRepository,
         ServerDeletionService $deletionService,
@@ -173,9 +181,10 @@ class ServersController extends Controller
         $this->buildModificationService = $buildModificationService;
         $this->config = $config;
         $this->containerRebuildService = $containerRebuildService;
-        $this->databaseManagementService = $databaseManagementService;
-        $this->databaseRepository = $databaseRepository;
         $this->databaseHostRepository = $databaseHostRepository;
+        $this->databaseManagementService = $databaseManagementService;
+        $this->databasePasswordService = $databasePasswordService;
+        $this->databaseRepository = $databaseRepository;
         $this->detailsModificationService = $detailsModificationService;
         $this->deletionService = $deletionService;
         $this->locationRepository = $locationRepository;
@@ -609,7 +618,7 @@ class ServersController extends Controller
             ['id', '=', $request->input('database')],
         ]);
 
-        $this->databaseManagementService->changePassword($database->id, str_random(20));
+        $this->databasePasswordService->handle($database, str_random(20));
 
         return response('', 204);
     }

app/Http/Controllers/Server/DatabaseController.php

@@ -0,0 +1,71 @@
+<?php
+
+namespace Pterodactyl\Http\Controllers\Server;
+
+use Illuminate\View\View;
+use Illuminate\Http\Request;
+use Illuminate\Http\JsonResponse;
+use Pterodactyl\Http\Controllers\Controller;
+use Pterodactyl\Traits\Controllers\JavascriptInjection;
+use Pterodactyl\Services\Databases\DatabasePasswordService;
+use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
+
+class DatabaseController extends Controller
+{
+    use JavascriptInjection;
+
+    /**
+     * @var \Pterodactyl\Services\Databases\DatabasePasswordService
+     */
+    protected $passwordService;
+
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface
+     */
+    protected $repository;
+
+    /**
+     * DatabaseController constructor.
+     *
+     * @param \Pterodactyl\Services\Databases\DatabasePasswordService       $passwordService
+     * @param \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface $repository
+     */
+    public function __construct(DatabasePasswordService $passwordService, DatabaseRepositoryInterface $repository)
+    {
+        $this->passwordService = $passwordService;
+        $this->repository = $repository;
+    }
+
+    /**
+     * Render the database listing for a server.
+     *
+     * @param \Illuminate\Http\Request $request
+     * @return \Illuminate\View\View
+     */
+    public function index(Request $request): View
+    {
+        $server = $request->attributes->get('server');
+        $this->injectJavascript();
+
+        return view('server.databases.index', [
+            'databases' => $this->repository->getDatabasesForServer($server->id),
+        ]);
+    }
+
+    /**
+     * Handle a request to update the password for a specific database.
+     *
+     * @param \Illuminate\Http\Request $request
+     * @return \Illuminate\Http\JsonResponse
+     *
+     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
+     */
+    public function update(Request $request): JsonResponse
+    {
+        $password = str_random(20);
+        $this->passwordService->handle($request->attributes->get('database'), $password);
+
+        return response()->json(['password' => $password]);
+    }
+}

app/Http/Kernel.php

@@ -5,6 +5,9 @@ namespace Pterodactyl\Http;
 use Pterodactyl\Http\Middleware\DaemonAuthenticate;
 use Illuminate\Foundation\Http\Kernel as HttpKernel;
 use Illuminate\Routing\Middleware\SubstituteBindings;
+use Pterodactyl\Http\Middleware\Server\SubuserBelongsToServer;
+use Pterodactyl\Http\Middleware\Server\DatabaseBelongsToServer;
+use Pterodactyl\Http\Middleware\Server\ScheduleBelongsToServer;
 
 class Kernel extends HttpKernel
 {
@@ -63,7 +66,6 @@ class Kernel extends HttpKernel
         'guest' => \Pterodactyl\Http\Middleware\RedirectIfAuthenticated::class,
         'server' => \Pterodactyl\Http\Middleware\ServerAuthenticate::class,
         'subuser.auth' => \Pterodactyl\Http\Middleware\SubuserAccessAuthenticate::class,
-        'subuser' => \Pterodactyl\Http\Middleware\Server\SubuserAccess::class,
         'admin' => \Pterodactyl\Http\Middleware\AdminAuthenticate::class,
         'daemon-old' => DaemonAuthenticate::class,
         'csrf' => \Pterodactyl\Http\Middleware\VerifyCsrfToken::class,
@@ -71,6 +73,13 @@ class Kernel extends HttpKernel
         'can' => \Illuminate\Auth\Middleware\Authorize::class,
         'bindings' => \Illuminate\Routing\Middleware\SubstituteBindings::class,
         'recaptcha' => \Pterodactyl\Http\Middleware\VerifyReCaptcha::class,
-        'schedule' => \Pterodactyl\Http\Middleware\Server\ScheduleAccess::class,
+
+        // Server specific middleware (used for authenticating access to resources)
+        //
+        // These are only used for individual server authentication, and not gloabl
+        // actions from other resources. They are defined in the route files.
+        'server..database' => DatabaseBelongsToServer::class,
+        'server..subuser' => SubuserBelongsToServer::class,
+        'server..schedule' => ScheduleBelongsToServer::class,
     ];
 }

app/Http/Middleware/Server/DatabaseBelongsToServer.php

@@ -0,0 +1,51 @@
+<?php
+
+namespace Pterodactyl\Http\Middleware\Server;
+
+use Closure;
+use Illuminate\Http\Request;
+use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
+use Symfony\Component\HttpKernel\Exception\NotFoundHttpException;
+
+class DatabaseBelongsToServer
+{
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface
+     */
+    protected $repository;
+
+    /**
+     * DatabaseAccess constructor.
+     *
+     * @param \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface $repository
+     */
+    public function __construct(DatabaseRepositoryInterface $repository)
+    {
+        $this->repository = $repository;
+    }
+
+    /**
+     * Check if a database being requested belongs to the currently loaded server.
+     * If it does not, throw a 404 error, otherwise continue on with the request
+     * and set an attribute with the database.
+     *
+     * @param \Illuminate\Http\Request $request
+     * @param \Closure                 $next
+     * @return mixed
+     *
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
+     */
+    public function handle(Request $request, Closure $next)
+    {
+        $server = $request->attributes->get('server');
+
+        $database = $this->repository->find($request->input('database'));
+        if ($database->server_id !== $server->id) {
+            throw new NotFoundHttpException;
+        }
+
+        $request->attributes->set('database', $database);
+
+        return $next($request);
+    }
+}

app/Http/Middleware/Server/ScheduleBelongsToServer.php

@@ -14,7 +14,7 @@ use Illuminate\Contracts\Session\Session;
 use Pterodactyl\Contracts\Extensions\HashidsInterface;
 use Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface;
 
-class ScheduleAccess
+class ScheduleBelongsToServer
 {
     /**
      * @var \Pterodactyl\Contracts\Extensions\HashidsInterface

app/Http/Middleware/Server/SubuserBelongsToServer.php

@@ -15,7 +15,7 @@ use Pterodactyl\Exceptions\DisplayException;
 use Pterodactyl\Contracts\Repository\SubuserRepositoryInterface;
 use Symfony\Component\HttpKernel\Exception\NotFoundHttpException;
 
-class SubuserAccess
+class SubuserBelongsToServer
 {
     /**
      * @var \Pterodactyl\Contracts\Repository\SubuserRepositoryInterface

app/Http/Middleware/ServerAuthenticate.php

@@ -105,8 +105,13 @@ class ServerAuthenticate
         }
 
         // Store the server in the session.
+        // @todo remove from session. use request attributes.
         $this->session->now('server_data.model', $server);
 
+        // Add server to the request attributes. This will replace sessions
+        // as files are updated.
+        $request->attributes->set('server', $server);
+
         return $next($request);
     }
 }

app/Http/Middleware/SubuserAccessAuthenticate.php

@@ -60,6 +60,7 @@ class SubuserAccessAuthenticate
         try {
             $token = $this->keyProviderService->handle($server->id, $request->user()->id);
             $this->session->now('server_data.token', $token);
+            $request->attributes->set('server_token', $token);
         } catch (RecordNotFoundException $exception) {
             throw new AuthenticationException('This account does not have permission to access this server.');
         }

app/Http/ViewComposers/Server/ServerDataComposer.php

@@ -1,32 +1,25 @@
 <?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\ViewComposers\Server;
 
 use Illuminate\View\View;
-use Illuminate\Contracts\Session\Session;
+use Illuminate\Http\Request;
 
 class ServerDataComposer
 {
     /**
-     * @var \Illuminate\Contracts\Session\Session
+     * @var \Illuminate\Http\Request
      */
-    protected $session;
+    protected $request;
 
     /**
      * ServerDataComposer constructor.
      *
-     * @param \Illuminate\Contracts\Session\Session $session
+     * @param \Illuminate\Http\Request $request
      */
-    public function __construct(Session $session)
+    public function __construct(Request $request)
     {
-        $this->session = $session;
+        $this->request = $request;
     }
 
     /**
@@ -36,10 +29,10 @@ class ServerDataComposer
      */
     public function compose(View $view)
     {
-        $data = $this->session->get('server_data');
+        $server = $this->request->get('server');
 
-        $view->with('server', array_get($data, 'model'));
-        $view->with('node', object_get($data['model'], 'node'));
-        $view->with('daemon_token', array_get($data, 'token'));
+        $view->with('server', $server);
+        $view->with('node', object_get($server, 'node'));
+        $view->with('daemon_token', $this->request->get('server_token'));
     }
 }

app/Repositories/Eloquent/DatabaseRepository.php

@@ -10,6 +10,7 @@
 namespace Pterodactyl\Repositories\Eloquent;
 
 use Pterodactyl\Models\Database;
+use Illuminate\Support\Collection;
 use Illuminate\Foundation\Application;
 use Illuminate\Database\DatabaseManager;
 use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
@@ -17,6 +18,11 @@ use Pterodactyl\Exceptions\Repository\DuplicateDatabaseNameException;
 
 class DatabaseRepository extends EloquentRepository implements DatabaseRepositoryInterface
 {
+    /**
+     * @var string
+     */
+    protected $connection = self::DEFAULT_CONNECTION_NAME;
+
     /**
      * @var \Illuminate\Database\DatabaseManager
      */
@@ -45,6 +51,40 @@ class DatabaseRepository extends EloquentRepository implements DatabaseRepositor
         return Database::class;
     }
 
+    /**
+     * Set the connection name to execute statements against.
+     *
+     * @param string $connection
+     * @return $this
+     */
+    public function setConnection(string $connection)
+    {
+        $this->connection = $connection;
+
+        return $this;
+    }
+
+    /**
+     * Return the connection to execute statements aganist.
+     *
+     * @return string
+     */
+    public function getConnection(): string
+    {
+        return $this->connection;
+    }
+
+    /**
+     * Return all of the databases belonging to a server.
+     *
+     * @param int $server
+     * @return \Illuminate\Support\Collection
+     */
+    public function getDatabasesForServer(int $server): Collection
+    {
+        return $this->getBuilder()->where('server_id', $server)->get($this->getColumns());
+    }
+
     /**
      * {@inheritdoc}
      * @return bool|\Illuminate\Database\Eloquent\Model
@@ -67,80 +107,64 @@ class DatabaseRepository extends EloquentRepository implements DatabaseRepositor
     /**
      * {@inheritdoc}
      */
-    public function createDatabase($database, $connection = null)
+    public function createDatabase($database)
     {
-        return $this->runStatement(
-            sprintf('CREATE DATABASE IF NOT EXISTS `%s`', $database),
-            $connection
-        );
+        return $this->runStatement(sprintf('CREATE DATABASE IF NOT EXISTS `%s`', $database));
     }
 
     /**
      * {@inheritdoc}
      */
-    public function createUser($username, $remote, $password, $connection = null)
+    public function createUser($username, $remote, $password)
     {
-        return $this->runStatement(
-            sprintf('CREATE USER `%s`@`%s` IDENTIFIED BY \'%s\'', $username, $remote, $password),
-            $connection
-        );
+        return $this->runStatement(sprintf('CREATE USER `%s`@`%s` IDENTIFIED BY \'%s\'', $username, $remote, $password));
     }
 
     /**
      * {@inheritdoc}
      */
-    public function assignUserToDatabase($database, $username, $remote, $connection = null)
+    public function assignUserToDatabase($database, $username, $remote)
     {
-        return $this->runStatement(
-            sprintf(
-                'GRANT SELECT, INSERT, UPDATE, DELETE, CREATE, DROP, ALTER, INDEX, EXECUTE ON `%s`.* TO `%s`@`%s`',
-                $database,
-                $username,
-                $remote
-            ),
-            $connection
-        );
+        return $this->runStatement(sprintf(
+            'GRANT SELECT, INSERT, UPDATE, DELETE, CREATE, DROP, ALTER, INDEX, EXECUTE ON `%s`.* TO `%s`@`%s`',
+            $database,
+            $username,
+            $remote
+        ));
     }
 
     /**
      * {@inheritdoc}
      */
-    public function flush($connection = null)
+    public function flush()
     {
-        return $this->runStatement('FLUSH PRIVILEGES', $connection);
+        return $this->runStatement('FLUSH PRIVILEGES');
     }
 
     /**
      * {@inheritdoc}
      */
-    public function dropDatabase($database, $connection = null)
+    public function dropDatabase($database)
     {
-        return $this->runStatement(
-            sprintf('DROP DATABASE IF EXISTS `%s`', $database),
-            $connection
-        );
+        return $this->runStatement(sprintf('DROP DATABASE IF EXISTS `%s`', $database));
     }
 
     /**
      * {@inheritdoc}
      */
-    public function dropUser($username, $remote, $connection = null)
+    public function dropUser($username, $remote)
     {
-        return $this->runStatement(
-            sprintf('DROP USER IF EXISTS `%s`@`%s`', $username, $remote),
-            $connection
-        );
+        return $this->runStatement(sprintf('DROP USER IF EXISTS `%s`@`%s`', $username, $remote));
     }
 
     /**
      * Run the provided statement against the database on a given connection.
      *
-     * @param string      $statement
-     * @param null|string $connection
+     * @param string $statement
      * @return bool
      */
-    protected function runStatement($statement, $connection = null)
+    protected function runStatement($statement)
     {
-        return $this->database->connection($connection)->statement($statement);
+        return $this->database->connection($this->getConnection())->statement($statement);
     }
 }

app/Services/Database/DatabaseHostService.php

@@ -1,148 +0,0 @@
-<?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\Services\Database;
-
-use Illuminate\Database\DatabaseManager;
-use Pterodactyl\Exceptions\DisplayException;
-use Illuminate\Contracts\Encryption\Encrypter;
-use Pterodactyl\Extensions\DynamicDatabaseConnection;
-use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
-use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
-
-class DatabaseHostService
-{
-    /**
-     * @var \Illuminate\Database\DatabaseManager
-     */
-    protected $database;
-
-    /**
-     * @var \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface
-     */
-    protected $databaseRepository;
-
-    /**
-     * @var \Pterodactyl\Extensions\DynamicDatabaseConnection
-     */
-    protected $dynamic;
-
-    /**
-     * @var \Illuminate\Contracts\Encryption\Encrypter
-     */
-    protected $encrypter;
-
-    /**
-     * @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface
-     */
-    protected $repository;
-
-    /**
-     * DatabaseHostService constructor.
-     *
-     * @param \Illuminate\Database\DatabaseManager                              $database
-     * @param \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface     $databaseRepository
-     * @param \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface $repository
-     * @param \Pterodactyl\Extensions\DynamicDatabaseConnection                 $dynamic
-     * @param \Illuminate\Contracts\Encryption\Encrypter                        $encrypter
-     */
-    public function __construct(
-        DatabaseManager $database,
-        DatabaseRepositoryInterface $databaseRepository,
-        DatabaseHostRepositoryInterface $repository,
-        DynamicDatabaseConnection $dynamic,
-        Encrypter $encrypter
-    ) {
-        $this->database = $database;
-        $this->databaseRepository = $databaseRepository;
-        $this->dynamic = $dynamic;
-        $this->encrypter = $encrypter;
-        $this->repository = $repository;
-    }
-
-    /**
-     * Create a new database host and persist it to the database.
-     *
-     * @param array $data
-     * @return \Pterodactyl\Models\DatabaseHost
-     *
-     * @throws \Throwable
-     * @throws \PDOException
-     */
-    public function create(array $data)
-    {
-        $this->database->beginTransaction();
-
-        $host = $this->repository->create([
-            'password' => $this->encrypter->encrypt(array_get($data, 'password')),
-            'name' => array_get($data, 'name'),
-            'host' => array_get($data, 'host'),
-            'port' => array_get($data, 'port'),
-            'username' => array_get($data, 'username'),
-            'max_databases' => null,
-            'node_id' => array_get($data, 'node_id'),
-        ]);
-
-        // Check Access
-        $this->dynamic->set('dynamic', $host);
-        $this->database->connection('dynamic')->select('SELECT 1 FROM dual');
-
-        $this->database->commit();
-
-        return $host;
-    }
-
-    /**
-     * Update a database host and persist to the database.
-     *
-     * @param int   $id
-     * @param array $data
-     * @return mixed
-     *
-     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
-     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
-     */
-    public function update($id, array $data)
-    {
-        $this->database->beginTransaction();
-
-        if (! empty(array_get($data, 'password'))) {
-            $data['password'] = $this->encrypter->encrypt($data['password']);
-        } else {
-            unset($data['password']);
-        }
-
-        $host = $this->repository->update($id, $data);
-
-        $this->dynamic->set('dynamic', $host);
-        $this->database->connection('dynamic')->select('SELECT 1 FROM dual');
-
-        $this->database->commit();
-
-        return $host;
-    }
-
-    /**
-     * Delete a database host if it has no active databases attached to it.
-     *
-     * @param int $id
-     * @return bool|null
-     *
-     * @throws \Pterodactyl\Exceptions\DisplayException
-     */
-    public function delete($id)
-    {
-        $count = $this->databaseRepository->findCountWhere([['database_host_id', '=', $id]]);
-        if ($count > 0) {
-            throw new DisplayException(trans('exceptions.databases.delete_has_databases'));
-        }
-
-        return $this->repository->delete($id);
-    }
-}

app/Services/Databases/DatabaseManagementService.php

@@ -7,7 +7,7 @@
  * https://opensource.org/licenses/MIT
  */
 
-namespace Pterodactyl\Services\Database;
+namespace Pterodactyl\Services\Databases;
 
 use Illuminate\Database\DatabaseManager;
 use Illuminate\Contracts\Encryption\Encrypter;
@@ -79,28 +79,26 @@ class DatabaseManagementService
             $database = $this->repository->createIfNotExists($data);
             $this->dynamic->set('dynamic', $data['database_host_id']);
 
-            $this->repository->createDatabase($database->database, 'dynamic');
+            $this->repository->createDatabase($database->database);
             $this->repository->createUser(
                 $database->username,
                 $database->remote,
-                $this->encrypter->decrypt($database->password),
-                'dynamic'
+                $this->encrypter->decrypt($database->password)
             );
             $this->repository->assignUserToDatabase(
                 $database->database,
                 $database->username,
-                $database->remote,
-                'dynamic'
+                $database->remote
             );
-            $this->repository->flush('dynamic');
+            $this->repository->flush();
 
             $this->database->commit();
         } catch (\Exception $ex) {
             try {
                 if (isset($database)) {
-                    $this->repository->dropDatabase($database->database, 'dynamic');
-                    $this->repository->dropUser($database->username, $database->remote, 'dynamic');
-                    $this->repository->flush('dynamic');
+                    $this->repository->dropDatabase($database->database);
+                    $this->repository->dropUser($database->username, $database->remote);
+                    $this->repository->flush();
                 }
             } catch (\Exception $exTwo) {
                 // ignore an exception
@@ -113,62 +111,22 @@ class DatabaseManagementService
         return $database;
     }
 
-    /**
-     * Change the password for a specific user and database combination.
-     *
-     * @param int    $id
-     * @param string $password
-     * @return bool
-     *
-     * @throws \Exception
-     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
-     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
-     */
-    public function changePassword($id, $password)
-    {
-        $database = $this->repository->find($id);
-        $this->dynamic->set('dynamic', $database->database_host_id);
-
-        $this->database->beginTransaction();
-
-        try {
-            $updated = $this->repository->withoutFresh()->update($id, [
-                'password' => $this->encrypter->encrypt($password),
-            ]);
-
-            $this->repository->dropUser($database->username, $database->remote, 'dynamic');
-            $this->repository->createUser($database->username, $database->remote, $password, 'dynamic');
-            $this->repository->assignUserToDatabase(
-                $database->database,
-                $database->username,
-                $database->remote,
-                'dynamic'
-            );
-            $this->repository->flush('dynamic');
-
-            $this->database->commit();
-        } catch (\Exception $ex) {
-            $this->database->rollBack();
-            throw $ex;
-        }
-
-        return $updated;
-    }
-
     /**
      * Delete a database from the given host server.
      *
      * @param int $id
      * @return bool|null
+     *
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
     public function delete($id)
     {
         $database = $this->repository->find($id);
         $this->dynamic->set('dynamic', $database->database_host_id);
 
-        $this->repository->dropDatabase($database->database, 'dynamic');
-        $this->repository->dropUser($database->username, $database->remote, 'dynamic');
-        $this->repository->flush('dynamic');
+        $this->repository->dropDatabase($database->database);
+        $this->repository->dropUser($database->username, $database->remote);
+        $this->repository->flush();
 
         return $this->repository->delete($id);
     }

app/Services/Databases/DatabasePasswordService.php

@@ -0,0 +1,86 @@
+<?php
+
+namespace Pterodactyl\Services\Databases;
+
+use Pterodactyl\Models\Database;
+use Illuminate\Database\ConnectionInterface;
+use Illuminate\Contracts\Encryption\Encrypter;
+use Pterodactyl\Extensions\DynamicDatabaseConnection;
+use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
+
+class DatabasePasswordService
+{
+    /**
+     * @var \Illuminate\Database\ConnectionInterface
+     */
+    private $connection;
+
+    /**
+     * @var \Pterodactyl\Extensions\DynamicDatabaseConnection
+     */
+    private $dynamic;
+
+    /**
+     * @var \Illuminate\Contracts\Encryption\Encrypter
+     */
+    private $encrypter;
+
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface
+     */
+    private $repository;
+
+    /**
+     * DatabasePasswordService constructor.
+     *
+     * @param \Illuminate\Database\ConnectionInterface                      $connection
+     * @param \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface $repository
+     * @param \Pterodactyl\Extensions\DynamicDatabaseConnection             $dynamic
+     * @param \Illuminate\Contracts\Encryption\Encrypter                    $encrypter
+     */
+    public function __construct(
+        ConnectionInterface $connection,
+        DatabaseRepositoryInterface $repository,
+        DynamicDatabaseConnection $dynamic,
+        Encrypter $encrypter
+    ) {
+        $this->connection = $connection;
+        $this->dynamic = $dynamic;
+        $this->encrypter = $encrypter;
+        $this->repository = $repository;
+    }
+
+    /**
+     * Updates a password for a given database.
+     *
+     * @param \Pterodactyl\Models\Database|int $database
+     * @param string                           $password
+     * @return bool
+     *
+     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
+     */
+    public function handle($database, string $password): bool
+    {
+        if (! $database instanceof Database) {
+            $database = $this->repository->find($database);
+        }
+
+        $this->dynamic->set('dynamic', $database->database_host_id);
+        $this->connection->beginTransaction();
+
+        $updated = $this->repository->withoutFresh()->update($database->id, [
+            'password' => $this->encrypter->encrypt($password),
+        ]);
+
+        $this->repository->dropUser($database->username, $database->remote);
+        $this->repository->createUser($database->username, $database->remote, $password);
+        $this->repository->assignUserToDatabase($database->database, $database->username, $database->remote);
+        $this->repository->flush();
+
+        unset($password);
+        $this->connection->commit();
+
+        return $updated;
+    }
+}

app/Services/Databases/Hosts/HostCreationService.php

@@ -0,0 +1,92 @@
+<?php
+
+namespace Pterodactyl\Services\Databases\Hosts;
+
+use Pterodactyl\Models\DatabaseHost;
+use Illuminate\Database\DatabaseManager;
+use Illuminate\Database\ConnectionInterface;
+use Illuminate\Contracts\Encryption\Encrypter;
+use Pterodactyl\Extensions\DynamicDatabaseConnection;
+use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
+
+class HostCreationService
+{
+    /**
+     * @var \Illuminate\Database\ConnectionInterface
+     */
+    private $connection;
+
+    /**
+     * @var \Illuminate\Database\DatabaseManager
+     */
+    private $databaseManager;
+
+    /**
+     * @var \Pterodactyl\Extensions\DynamicDatabaseConnection
+     */
+    private $dynamic;
+
+    /**
+     * @var \Illuminate\Contracts\Encryption\Encrypter
+     */
+    private $encrypter;
+
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface
+     */
+    private $repository;
+
+    /**
+     * HostCreationService constructor.
+     *
+     * @param \Illuminate\Database\ConnectionInterface                          $connection
+     * @param \Illuminate\Database\DatabaseManager                              $databaseManager
+     * @param \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface $repository
+     * @param \Pterodactyl\Extensions\DynamicDatabaseConnection                 $dynamic
+     * @param \Illuminate\Contracts\Encryption\Encrypter                        $encrypter
+     */
+    public function __construct(
+        ConnectionInterface $connection,
+        DatabaseManager $databaseManager,
+        DatabaseHostRepositoryInterface $repository,
+        DynamicDatabaseConnection $dynamic,
+        Encrypter $encrypter
+    ) {
+        $this->connection = $connection;
+        $this->databaseManager = $databaseManager;
+        $this->dynamic = $dynamic;
+        $this->encrypter = $encrypter;
+        $this->repository = $repository;
+    }
+
+    /**
+     * Create a new database host on the Panel.
+     *
+     * @param array $data
+     * @return \Pterodactyl\Models\DatabaseHost
+     *
+     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
+     */
+    public function handle(array $data): DatabaseHost
+    {
+        $this->connection->beginTransaction();
+
+        $host = $this->repository->create([
+            'password' => $this->encrypter->encrypt(array_get($data, 'password')),
+            'name' => array_get($data, 'name'),
+            'host' => array_get($data, 'host'),
+            'port' => array_get($data, 'port'),
+            'username' => array_get($data, 'username'),
+            'max_databases' => null,
+            'node_id' => array_get($data, 'node_id'),
+        ]);
+
+        // Confirm access using the provided credentials before saving data.
+        $this->dynamic->set('dynamic', $host);
+        $this->databaseManager->connection('dynamic')->select('SELECT 1 FROM dual');
+        $this->connection->commit();
+
+        return $host;
+    }
+}

app/Services/Databases/Hosts/HostDeletionService.php

@@ -0,0 +1,53 @@
+<?php
+
+namespace Pterodactyl\Services\Databases\Hosts;
+
+use Pterodactyl\Exceptions\Service\HasActiveServersException;
+use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
+use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
+
+class HostDeletionService
+{
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface
+     */
+    private $databaseRepository;
+
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface
+     */
+    private $repository;
+
+    /**
+     * HostDeletionService constructor.
+     *
+     * @param \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface     $databaseRepository
+     * @param \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface $repository
+     */
+    public function __construct(
+        DatabaseRepositoryInterface $databaseRepository,
+        DatabaseHostRepositoryInterface $repository
+    ) {
+        $this->databaseRepository = $databaseRepository;
+        $this->repository = $repository;
+    }
+
+    /**
+     * Delete a specified host from the Panel if no databases are
+     * attached to it.
+     *
+     * @param int $host
+     * @return int
+     *
+     * @throws \Pterodactyl\Exceptions\Service\HasActiveServersException
+     */
+    public function handle(int $host): int
+    {
+        $count = $this->databaseRepository->findCountWhere([['database_host_id', '=', $host]]);
+        if ($count > 0) {
+            throw new HasActiveServersException(trans('exceptions.databases.delete_has_databases'));
+        }
+
+        return $this->repository->delete($host);
+    }
+}

app/Services/Databases/Hosts/HostUpdateService.php

@@ -0,0 +1,96 @@
+<?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\Services\Databases\Hosts;
+
+use Pterodactyl\Models\DatabaseHost;
+use Illuminate\Database\DatabaseManager;
+use Illuminate\Database\ConnectionInterface;
+use Illuminate\Contracts\Encryption\Encrypter;
+use Pterodactyl\Extensions\DynamicDatabaseConnection;
+use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
+
+class HostUpdateService
+{
+    /**
+     * @var \Illuminate\Database\ConnectionInterface
+     */
+    private $connection;
+
+    /**
+     * @var \Illuminate\Database\DatabaseManager
+     */
+    private $databaseManager;
+
+    /**
+     * @var \Pterodactyl\Extensions\DynamicDatabaseConnection
+     */
+    private $dynamic;
+
+    /**
+     * @var \Illuminate\Contracts\Encryption\Encrypter
+     */
+    private $encrypter;
+
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface
+     */
+    private $repository;
+
+    /**
+     * DatabaseHostService constructor.
+     *
+     * @param \Illuminate\Database\ConnectionInterface                          $connection
+     * @param \Illuminate\Database\DatabaseManager                              $databaseManager
+     * @param \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface $repository
+     * @param \Pterodactyl\Extensions\DynamicDatabaseConnection                 $dynamic
+     * @param \Illuminate\Contracts\Encryption\Encrypter                        $encrypter
+     */
+    public function __construct(
+        ConnectionInterface $connection,
+        DatabaseManager $databaseManager,
+        DatabaseHostRepositoryInterface $repository,
+        DynamicDatabaseConnection $dynamic,
+        Encrypter $encrypter
+    ) {
+        $this->connection = $connection;
+        $this->databaseManager = $databaseManager;
+        $this->dynamic = $dynamic;
+        $this->encrypter = $encrypter;
+        $this->repository = $repository;
+    }
+
+    /**
+     * Update a database host and persist to the database.
+     *
+     * @param int   $hostId
+     * @param array $data
+     * @return mixed
+     *
+     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
+     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
+     */
+    public function handle(int $hostId, array $data): DatabaseHost
+    {
+        if (! empty(array_get($data, 'password'))) {
+            $data['password'] = $this->encrypter->encrypt($data['password']);
+        } else {
+            unset($data['password']);
+        }
+
+        $this->connection->beginTransaction();
+        $host = $this->repository->update($hostId, $data);
+
+        $this->dynamic->set('dynamic', $host);
+        $this->databaseManager->connection('dynamic')->select('SELECT 1 FROM dual');
+        $this->connection->commit();
+
+        return $host;
+    }
+}

app/Services/Servers/ServerDeletionService.php

@@ -14,7 +14,7 @@ use Pterodactyl\Models\Server;
 use GuzzleHttp\Exception\RequestException;
 use Illuminate\Database\ConnectionInterface;
 use Pterodactyl\Exceptions\DisplayException;
-use Pterodactyl\Services\Database\DatabaseManagementService;
+use Pterodactyl\Services\Databases\DatabaseManagementService;
 use Pterodactyl\Contracts\Repository\ServerRepositoryInterface;
 use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
 use Pterodactyl\Contracts\Repository\Daemon\ServerRepositoryInterface as DaemonServerRepositoryInterface;
@@ -32,7 +32,7 @@ class ServerDeletionService
     protected $daemonServerRepository;
 
     /**
-     * @var \Pterodactyl\Services\Database\DatabaseManagementService
+     * @var \Pterodactyl\Services\Databases\DatabaseManagementService
      */
     protected $databaseManagementService;
 
@@ -62,7 +62,7 @@ class ServerDeletionService
      * @param \Illuminate\Database\ConnectionInterface                           $connection
      * @param \Pterodactyl\Contracts\Repository\Daemon\ServerRepositoryInterface $daemonServerRepository
      * @param \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface      $databaseRepository
-     * @param \Pterodactyl\Services\Database\DatabaseManagementService           $databaseManagementService
+     * @param \Pterodactyl\Services\Databases\DatabaseManagementService          $databaseManagementService
      * @param \Pterodactyl\Contracts\Repository\ServerRepositoryInterface        $repository
      * @param \Illuminate\Log\Writer                                             $writer
      */

app/Traits/Controllers/JavascriptInjection.php

@@ -10,25 +10,23 @@
 namespace Pterodactyl\Traits\Controllers;
 
 use Javascript;
+use Illuminate\Http\Request;
 
 trait JavascriptInjection
 {
-    /**
-     * @var \Illuminate\Contracts\Session\Session
-     */
-    protected $session;
-
     /**
      * Injects server javascript into the page to be used by other services.
      *
-     * @param array $args
-     * @param bool  $overwrite
-     * @return mixed
+     * @param array                         $args
+     * @param bool                          $overwrite
+     * @param \Illuminate\Http\Request|null $request
+     * @return array
      */
-    public function injectJavascript($args = [], $overwrite = false)
+    public function injectJavascript($args = [], $overwrite = false, Request $request = null)
     {
-        $server = $this->session->get('server_data.model');
-        $token = $this->session->get('server_data.token');
+        $request = $request ?? app()->make(Request::class);
+        $server = $request->attributes->get('server');
+        $token = $request->attributes->get('server_token');
 
         $response = array_merge([
             'server' => [