<?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); } }