2017-06-18 00:48:31 +00:00
|
|
|
<?php
|
|
|
|
|
|
|
|
namespace Pterodactyl\Extensions;
|
|
|
|
|
|
|
|
use Pterodactyl\Models\DatabaseHost;
|
2017-06-18 01:52:32 +00:00
|
|
|
use Illuminate\Contracts\Encryption\Encrypter;
|
|
|
|
use Illuminate\Config\Repository as ConfigRepository;
|
2017-08-05 22:26:30 +00:00
|
|
|
use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
|
2017-06-18 00:48:31 +00:00
|
|
|
|
|
|
|
class DynamicDatabaseConnection
|
|
|
|
{
|
2021-01-23 20:33:34 +00:00
|
|
|
public const DB_CHARSET = 'utf8';
|
|
|
|
public const DB_COLLATION = 'utf8_unicode_ci';
|
|
|
|
public const DB_DRIVER = 'mysql';
|
2017-06-18 00:48:31 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* DynamicDatabaseConnection constructor.
|
|
|
|
*/
|
|
|
|
public function __construct(
|
2022-10-14 16:59:20 +00:00
|
|
|
protected ConfigRepository $config,
|
|
|
|
protected Encrypter $encrypter,
|
|
|
|
protected DatabaseHostRepositoryInterface $repository
|
2017-06-18 00:48:31 +00:00
|
|
|
) {
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Adds a dynamic database connection entry to the runtime config.
|
|
|
|
*
|
2017-09-10 04:55:21 +00:00
|
|
|
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
|
2017-06-18 00:48:31 +00:00
|
|
|
*/
|
2022-10-14 16:59:20 +00:00
|
|
|
public function set(string $connection, DatabaseHost|int $host, string $database = 'mysql'): void
|
2017-06-18 00:48:31 +00:00
|
|
|
{
|
2021-01-23 20:33:34 +00:00
|
|
|
if (!$host instanceof DatabaseHost) {
|
2017-07-03 02:29:58 +00:00
|
|
|
$host = $this->repository->find($host);
|
2017-06-18 00:48:31 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
$this->config->set('database.connections.' . $connection, [
|
|
|
|
'driver' => self::DB_DRIVER,
|
|
|
|
'host' => $host->host,
|
|
|
|
'port' => $host->port,
|
|
|
|
'database' => $database,
|
|
|
|
'username' => $host->username,
|
|
|
|
'password' => $this->encrypter->decrypt($host->password),
|
|
|
|
'charset' => self::DB_CHARSET,
|
|
|
|
'collation' => self::DB_COLLATION,
|
|
|
|
]);
|
|
|
|
}
|
|
|
|
}
|