YxEnv.php 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164
  1. <?php
  2. class YxEnv
  3. {
  4. /**
  5. * 环境变量数据
  6. * @var array
  7. */
  8. protected $data = [];
  9. protected $filePath = '';
  10. public function __construct()
  11. {
  12. //$this->data = $_ENV;
  13. }
  14. /**
  15. * 读取环境变量定义文件
  16. * @access public
  17. * @param string $file 环境变量定义文件
  18. * @return void
  19. */
  20. public function load($file)
  21. {
  22. $this->filePath = $file;
  23. $env = parse_ini_file($file, true);
  24. $this->set($env);
  25. }
  26. public function makeEnv($file)
  27. {
  28. if(!file_exists($file)){
  29. try{
  30. touch($file);
  31. }catch (Exception $e){
  32. return;
  33. }
  34. }
  35. }
  36. /**
  37. * 获取环境变量值
  38. * @access public
  39. * @param string $name 环境变量名
  40. * @param mixed $default 默认值
  41. * @return mixed
  42. */
  43. public function get($name = null, $default = null, $php_prefix = true)
  44. {
  45. if (is_null($name)) {
  46. return $this->data;
  47. }
  48. if (isset($this->data[$name])) {
  49. return $this->data[$name];
  50. }
  51. return $this->getEnv($name, $default, $php_prefix);
  52. }
  53. protected function getEnv($name, $default = null, $php_prefix = true)
  54. {
  55. if ($php_prefix) {
  56. $name = 'PHP_' . $name;
  57. }
  58. $result = getenv($name);
  59. if (false === $result) {
  60. return $default;
  61. }
  62. if ('false' === $result) {
  63. $result = false;
  64. } elseif ('true' === $result) {
  65. $result = true;
  66. }
  67. if ( !isset($this->data[$name])) {
  68. $this->data[$name] = $result;
  69. }
  70. return $result;
  71. }
  72. /**
  73. * Notes: 写入Env文件
  74. * @author luzg(2020/8/27 18:12)
  75. * @param $envFilePath
  76. * @param array $databaseEnv
  77. */
  78. public function putEnv($envFilePath, array $databaseEnv)
  79. {
  80. $applyDbEnv = [
  81. 'DATABASE.HOSTNAME' => $databaseEnv['host'],
  82. 'DATABASE.DATABASE' => $databaseEnv['name'],
  83. 'DATABASE.USERNAME' => $databaseEnv['user'],
  84. 'DATABASE.PASSWORD' => $databaseEnv['password'],
  85. 'DATABASE.HOSTPORT' => $databaseEnv['port'],
  86. 'DATABASE.PREFIX' => $databaseEnv['prefix'],
  87. ];
  88. $envLine = array_merge($this->data, $applyDbEnv);
  89. $content = '';
  90. $lastPrefix = '';
  91. global $uniqueSalt;
  92. foreach ($envLine as $index => $value) {
  93. if ($index == 'PROJECT.UNIQUE_IDENTIFICATION' && !empty($uniqueSalt)) {
  94. $value = $uniqueSalt;
  95. }
  96. @list($prefix, $key) = explode('.', $index);
  97. if ($prefix != $lastPrefix && $key != null) {
  98. if ($lastPrefix != '')
  99. $content .= "\n";
  100. $content .= "[$prefix]\n";
  101. $lastPrefix = $prefix;
  102. }
  103. if ($prefix != $lastPrefix && $key == null) {
  104. $content .= "$index = \"$value\"\n";
  105. } else {
  106. $content .= "$key = \"$value\"\n";
  107. }
  108. }
  109. if (!empty($content)) {
  110. file_put_contents($envFilePath, $content);
  111. }
  112. }
  113. /**
  114. * 设置环境变量值
  115. * @access public
  116. * @param string|array $env 环境变量
  117. * @param mixed $value 值
  118. * @return void
  119. */
  120. public function set($env, $value = null)
  121. {
  122. if (is_array($env)) {
  123. foreach ($env as $key => $val) {
  124. if (is_array($val)) {
  125. foreach ($val as $k => $v) {
  126. $this->data[$key . '.' . $k] = $v;
  127. }
  128. } else {
  129. $this->data[$key] = $val;
  130. }
  131. }
  132. } else {
  133. $name = strtoupper(str_replace('.', '_', $env));
  134. $this->data[$name] = $value;
  135. }
  136. }
  137. }