forked from Alex/Pterodactyl-Panel
50 lines
1.6 KiB
PHP
50 lines
1.6 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Traits\Commands;
|
|
|
|
use Pterodactyl\Exceptions\PterodactylException;
|
|
|
|
trait EnvironmentWriterTrait
|
|
{
|
|
/**
|
|
* Escapes an environment value by looking for any characters that could
|
|
* reasonablly cause environment parsing issues. Those values are then wrapped
|
|
* in quotes before being returned.
|
|
*/
|
|
public function escapeEnvironmentValue(string $value): string
|
|
{
|
|
if (!preg_match('/^\"(.*)\"$/', $value) && preg_match('/([^\w.\-+\/])+/', $value)) {
|
|
return sprintf('"%s"', addslashes($value));
|
|
}
|
|
|
|
return $value;
|
|
}
|
|
|
|
/**
|
|
* Update the .env file for the application using the passed in values.
|
|
*
|
|
* @throws \Pterodactyl\Exceptions\PterodactylException
|
|
*/
|
|
public function writeToEnvironment(array $values = [])
|
|
{
|
|
$path = base_path('.env');
|
|
if (!file_exists($path)) {
|
|
throw new PterodactylException('Cannot locate .env file, was this software installed correctly?');
|
|
}
|
|
|
|
$saveContents = file_get_contents($path);
|
|
collect($values)->each(function ($value, $key) use (&$saveContents) {
|
|
$key = strtoupper($key);
|
|
$saveValue = sprintf('%s=%s', $key, $this->escapeEnvironmentValue($value));
|
|
|
|
if (preg_match_all('/^' . $key . '=(.*)$/m', $saveContents) < 1) {
|
|
$saveContents = $saveContents . PHP_EOL . $saveValue;
|
|
} else {
|
|
$saveContents = preg_replace('/^' . $key . '=(.*)$/m', $saveValue, $saveContents);
|
|
}
|
|
});
|
|
|
|
file_put_contents($path, $saveContents);
|
|
}
|
|
}
|