Add initial go at user created databases for servers, still needs cleaning

app/Services/Databases/DatabaseManagementService.php

@@ -13,22 +13,27 @@ class DatabaseManagementService
     /**
      * @var \Illuminate\Database\DatabaseManager
      */
-    protected $database;
+    private $database;
 
     /**
      * @var \Pterodactyl\Extensions\DynamicDatabaseConnection
      */
-    protected $dynamic;
+    private $dynamic;
 
     /**
      * @var \Illuminate\Contracts\Encryption\Encrypter
      */
-    protected $encrypter;
+    private $encrypter;
 
     /**
      * @var \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface
      */
-    protected $repository;
+    private $repository;
+
+    /**
+     * @var bool
+     */
+    protected $useRandomHost = false;
 
     /**
      * CreationService constructor.
@@ -55,7 +60,7 @@ class DatabaseManagementService
      *
      * @param int   $server
      * @param array $data
-     * @return \Illuminate\Database\Eloquent\Model
+     * @return \Pterodactyl\Models\Database
      *
      * @throws \Exception
      */

app/Services/Databases/DeployServerDatabaseService.php

@@ -0,0 +1,87 @@
+<?php
+
+namespace Pterodactyl\Services\Databases;
+
+use Pterodactyl\Models\Server;
+use Pterodactyl\Models\Database;
+use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
+use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
+use Pterodactyl\Exceptions\Service\Database\TooManyDatabasesException;
+use Pterodactyl\Exceptions\Service\Database\NoSuitableDatabaseHostException;
+use Pterodactyl\Exceptions\Service\Database\DatabaseClientFeatureNotEnabledException;
+
+class DeployServerDatabaseService
+{
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface
+     */
+    private $databaseHostRepository;
+
+    /**
+     * @var \Pterodactyl\Services\Databases\DatabaseManagementService
+     */
+    private $managementService;
+    /**
+     * @var \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface
+     */
+    private $repository;
+
+    /**
+     * ServerDatabaseCreationService constructor.
+     *
+     * @param \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface     $repository
+     * @param \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface $databaseHostRepository
+     * @param \Pterodactyl\Services\Databases\DatabaseManagementService         $managementService
+     */
+    public function __construct(
+        DatabaseRepositoryInterface $repository,
+        DatabaseHostRepositoryInterface $databaseHostRepository,
+        DatabaseManagementService $managementService
+    ) {
+        $this->databaseHostRepository = $databaseHostRepository;
+        $this->managementService = $managementService;
+        $this->repository = $repository;
+    }
+
+    /**
+     * @param \Pterodactyl\Models\Server $server
+     * @param array                      $data
+     * @return \Pterodactyl\Models\Database
+     *
+     * @throws \Pterodactyl\Exceptions\Service\Database\DatabaseClientFeatureNotEnabledException
+     * @throws \Exception
+     */
+    public function handle(Server $server, array $data): Database
+    {
+        if (! config('pterodactyl.client_features.databases.enabled')) {
+            throw new DatabaseClientFeatureNotEnabledException;
+        }
+
+        $databases = $this->repository->findCountWhere([['server_id', '=', $server->id]]);
+        if (! is_null($server->database_limit) && $databases >= $server->database_limit) {
+            throw new TooManyDatabasesException;
+        }
+
+        $allowRandom = config('pterodactyl.client_features.databases.allow_random');
+        $host = $this->databaseHostRepository->setColumns(['id'])->findWhere([
+            ['node_id', '=', $server->node_id],
+        ])->random();
+
+        if (empty($host) && ! $allowRandom) {
+            throw new NoSuitableDatabaseHostException;
+        }
+
+        if (empty($host)) {
+            $host = $this->databaseHostRepository->setColumns(['id'])->all()->random();
+            if (empty($host)) {
+                throw new NoSuitableDatabaseHostException;
+            }
+        }
+
+        return $this->managementService->create($server->id, [
+            'database_host_id' => $host->id,
+            'database' => array_get($data, 'database'),
+            'remote' => array_get($data, 'remote'),
+        ]);
+    }
+}