misc_pterodactyl-panel/app/Console/Commands/UpdateEnvironment.php

<?php
/**
 * Pterodactyl - Panel
 * Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

namespace Pterodactyl\Console\Commands;

use Uuid;
use Illuminate\Console\Command;

class UpdateEnvironment extends Command
{
    /**
     * The name and signature of the console command.
     *
     * @var string
     */
    protected $signature = 'pterodactyl:env
                            {--dbhost=}
                            {--dbport=}
                            {--dbname=}
                            {--dbuser=}
                            {--dbpass=}
                            {--url=}
                            {--driver=}
                            {--session-driver=}
                            {--queue-driver=}
                            {--timezone=}';

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = 'Update environment settings automatically.';

    /**
     * Create a new command instance.
     *
     * @return void
     */
    public function __construct()
    {
        parent::__construct();
    }

    /**
     * Execute the console command.
     *
     * @return mixed
     */
    public function handle()
    {
        $variables = [];
        $file = base_path() . '/.env';
        if (! file_exists($file)) {
            $this->error('Missing environment file! It appears that you have not installed this panel correctly.');
            exit();
        }

        $envContents = file_get_contents($file);

        $this->info('Simply leave blank and press enter to fields that you do not wish to update.');
        if (is_null(config('pterodactyl.service.author', null))) {
            $this->info('No service author set, setting one now.');
            $variables['SERVICE_AUTHOR'] = (string) Uuid::generate(4);
        }

        if (is_null($this->option('dbhost'))) {
            $variables['DB_HOST'] = $this->anticipate('Database Host', ['localhost', '127.0.0.1', config('database.connections.mysql.host')], config('database.connections.mysql.host'));
        } else {
            $variables['DB_HOST'] = $this->option('dbhost');
        }

        if (is_null($this->option('dbport'))) {
            $variables['DB_PORT'] = $this->anticipate('Database Port', [3306, config('database.connections.mysql.port')], config('database.connections.mysql.port'));
        } else {
            $variables['DB_PORT'] = $this->option('dbport');
        }

        if (is_null($this->option('dbname'))) {
            $variables['DB_DATABASE'] = $this->anticipate('Database Name', ['pterodactyl', 'homestead', config('database.connections.mysql.database')], config('database.connections.mysql.database'));
        } else {
            $variables['DB_DATABASE'] = $this->option('dbname');
        }

        if (is_null($this->option('dbuser'))) {
            $variables['DB_USERNAME'] = $this->anticipate('Database Username', [config('database.connections.mysql.username')], config('database.connections.mysql.username'));
        } else {
            $variables['DB_USERNAME'] = $this->option('dbuser');
        }

        if (is_null($this->option('dbpass'))) {
            $this->line('The Database Password field is required; you cannot hit enter and use a default value.');
            $variables['DB_PASSWORD'] = $this->secret('Database User Password');
        } else {
            $variables['DB_PASSWORD'] = $this->option('dbpass');
        }

        if (is_null($this->option('url'))) {
            $variables['APP_URL'] = $this->ask('Panel URL (include http(s)://)', config('app.url'));
        } else {
            $variables['APP_URL'] = $this->option('url');
        }

        if (is_null($this->option('timezone'))) {
            $this->line('The timezone should match one of the supported timezones according to http://php.net/manual/en/timezones.php');
            $variables['APP_TIMEZONE'] = $this->anticipate('Panel Timezone', \DateTimeZone::listIdentifiers(\DateTimeZone::ALL), config('app.timezone'));
        } else {
            $variables['APP_TIMEZONE'] = $this->option('timezone');
        }

        if (is_null($this->option('driver'))) {
            $this->line('If you chose redis as your cache driver backend, you *must* have a redis server configured already.');
            $variables['CACHE_DRIVER'] = $this->choice('Which cache driver backend would you like to use?', [
                'memcached' => 'Memcache',
                'redis' => 'Redis (recommended)',
                'apc' => 'APC',
                'array' => 'PHP Array',
            ], config('cache.default', 'memcached'));
        } else {
            $variables['CACHE_DRIVER'] = $this->option('driver');
        }

        if (is_null($this->option('session-driver'))) {
            $this->line('If you chose redis as your cache driver backend, you *must* have a redis server configured already.');
            $variables['SESSION_DRIVER'] = $this->choice('Which session driver backend would you like to use?', [
                'database' => 'MySQL (recommended)',
                'redis' => 'Redis',
                'file' => 'File',
                'cookie' => 'Cookie',
                'apc' => 'APC',
                'array' => 'PHP Array',
            ], config('session.driver', 'database'));
        } else {
            $variables['SESSION_DRIVER'] = $this->option('session-driver');
        }

        if (is_null($this->option('queue-driver'))) {
            $this->line('If you chose redis as your cache driver backend, you *must* have a redis server configured already.');
            $variables['QUEUE_DRIVER'] = $this->choice('Which cache driver backend would you like to use?', [
                'database' => 'Database (recommended)',
                'redis' => 'Redis',
                'sqs' => 'Amazon SQS',
                'sync' => 'Sync',
                'null' => 'None',
            ], config('queue.driver', 'database'));
        } else {
            $variables['QUEUE_DRIVER'] = $this->option('queue-driver');
        }

        $bar = $this->output->createProgressBar(count($variables));

        foreach ($variables as $key => $value) {
            $newValue = $key . '=' . $value;

            if (preg_match_all('/^' . $key . '=(.*)$/m', $envContents) < 1) {
                $envContents = $envContents . "\n" . $newValue;
            } else {
                $envContents = preg_replace('/^' . $key . '=(.*)$/m', $newValue, $envContents);
            }
            $bar->advance();
        }

        file_put_contents($file, $envContents);
        $bar->finish();

        $this->call('config:cache');
        $this->line("\n");
    }
}