forked from Alex/Pterodactyl-Panel
82 lines
2.3 KiB
PHP
82 lines
2.3 KiB
PHP
<?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\Extensions;
|
|
|
|
use Pterodactyl\Models\DatabaseHost;
|
|
use Illuminate\Contracts\Encryption\Encrypter;
|
|
use Illuminate\Config\Repository as ConfigRepository;
|
|
use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
|
|
|
|
class DynamicDatabaseConnection
|
|
{
|
|
const DB_CHARSET = 'utf8';
|
|
const DB_COLLATION = 'utf8_unicode_ci';
|
|
const DB_DRIVER = 'mysql';
|
|
|
|
/**
|
|
* @var \Illuminate\Config\Repository
|
|
*/
|
|
protected $config;
|
|
|
|
/**
|
|
* @var \Illuminate\Contracts\Encryption\Encrypter
|
|
*/
|
|
protected $encrypter;
|
|
|
|
/**
|
|
* @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface
|
|
*/
|
|
protected $repository;
|
|
|
|
/**
|
|
* DynamicDatabaseConnection constructor.
|
|
*
|
|
* @param \Illuminate\Config\Repository $config
|
|
* @param \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface $repository
|
|
* @param \Illuminate\Contracts\Encryption\Encrypter $encrypter
|
|
*/
|
|
public function __construct(
|
|
ConfigRepository $config,
|
|
DatabaseHostRepositoryInterface $repository,
|
|
Encrypter $encrypter
|
|
) {
|
|
$this->config = $config;
|
|
$this->encrypter = $encrypter;
|
|
$this->repository = $repository;
|
|
}
|
|
|
|
/**
|
|
* Adds a dynamic database connection entry to the runtime config.
|
|
*
|
|
* @param string $connection
|
|
* @param \Pterodactyl\Models\DatabaseHost|int $host
|
|
* @param string $database
|
|
*
|
|
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
|
|
*/
|
|
public function set($connection, $host, $database = 'mysql')
|
|
{
|
|
if (! $host instanceof DatabaseHost) {
|
|
$host = $this->repository->find($host);
|
|
}
|
|
|
|
$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,
|
|
]);
|
|
}
|
|
}
|