YxEnv.php 3.9 KB

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