78 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			78 lines
		
	
	
		
			2.1 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
 | |
| {
 | |
|     public const DB_CHARSET = 'utf8';
 | |
|     public const DB_COLLATION = 'utf8_unicode_ci';
 | |
|     public 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.
 | |
|      */
 | |
|     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,
 | |
|         ]);
 | |
|     }
 | |
| }
 | 
