| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164 |
- <?php
- class YxEnv
- {
- /**
- * 环境变量数据
- * @var array
- */
- protected $data = [];
- protected $filePath = '';
- public function __construct()
- {
- //$this->data = $_ENV;
- }
- /**
- * 读取环境变量定义文件
- * @access public
- * @param string $file 环境变量定义文件
- * @return void
- */
- public function load($file)
- {
- $this->filePath = $file;
- $env = parse_ini_file($file, true);
- $this->set($env);
- }
- public function makeEnv($file)
- {
- if(!file_exists($file)){
- try{
- touch($file);
- }catch (Exception $e){
- return;
- }
- }
- }
- /**
- * 获取环境变量值
- * @access public
- * @param string $name 环境变量名
- * @param mixed $default 默认值
- * @return mixed
- */
- public function get($name = null, $default = null, $php_prefix = true)
- {
- if (is_null($name)) {
- return $this->data;
- }
- if (isset($this->data[$name])) {
- return $this->data[$name];
- }
- return $this->getEnv($name, $default, $php_prefix);
- }
- protected function getEnv($name, $default = null, $php_prefix = true)
- {
- if ($php_prefix) {
- $name = 'PHP_' . $name;
- }
- $result = getenv($name);
- if (false === $result) {
- return $default;
- }
- if ('false' === $result) {
- $result = false;
- } elseif ('true' === $result) {
- $result = true;
- }
- if ( !isset($this->data[$name])) {
- $this->data[$name] = $result;
- }
- return $result;
- }
- /**
- * Notes: 写入Env文件
- * @author luzg(2020/8/27 18:12)
- * @param $envFilePath
- * @param array $databaseEnv
- */
- public function putEnv($envFilePath, array $databaseEnv)
- {
- $applyDbEnv = [
- 'DATABASE.HOSTNAME' => $databaseEnv['host'],
- 'DATABASE.DATABASE' => $databaseEnv['name'],
- 'DATABASE.USERNAME' => $databaseEnv['user'],
- 'DATABASE.PASSWORD' => $databaseEnv['password'],
- 'DATABASE.HOSTPORT' => $databaseEnv['port'],
- 'DATABASE.PREFIX' => $databaseEnv['prefix'],
- ];
- $envLine = array_merge($this->data, $applyDbEnv);
- $content = '';
- $lastPrefix = '';
- global $uniqueSalt;
- foreach ($envLine as $index => $value) {
- if ($index == 'PROJECT.UNIQUE_IDENTIFICATION' && !empty($uniqueSalt)) {
- $value = $uniqueSalt;
- }
- @list($prefix, $key) = explode('.', $index);
- if ($prefix != $lastPrefix && $key != null) {
- if ($lastPrefix != '')
- $content .= "\n";
- $content .= "[$prefix]\n";
- $lastPrefix = $prefix;
- }
- if ($prefix != $lastPrefix && $key == null) {
- $content .= "$index = \"$value\"\n";
- } else {
- $content .= "$key = \"$value\"\n";
- }
- }
- if (!empty($content)) {
- file_put_contents($envFilePath, $content);
- }
- }
- /**
- * 设置环境变量值
- * @access public
- * @param string|array $env 环境变量
- * @param mixed $value 值
- * @return void
- */
- public function set($env, $value = null)
- {
- if (is_array($env)) {
- foreach ($env as $key => $val) {
- if (is_array($val)) {
- foreach ($val as $k => $v) {
- $this->data[$key . '.' . $k] = $v;
- }
- } else {
- $this->data[$key] = $val;
- }
- }
- } else {
- $name = strtoupper(str_replace('.', '_', $env));
- $this->data[$name] = $value;
- }
- }
- }
|