YxEnv.php 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
  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. 'project.file_domain' => $_SERVER['HTTP_HOST'],
  87. ];
  88. $envLine = array_merge($this->data, $applyDbEnv);
  89. ksort($envLine);
  90. $content = '';
  91. $lastPrefix = '';
  92. foreach ($envLine as $index => $value) {
  93. list($prefix, $key) = explode('.', $index);
  94. if ($prefix != $lastPrefix && $key != null) {
  95. if ($lastPrefix != '')
  96. $content .= "\n";
  97. $content .= "[$prefix]\n";
  98. $lastPrefix = $prefix;
  99. }
  100. if ($prefix != $lastPrefix && $key == null) {
  101. $content .= "$index = \"$value\"\n";
  102. } else {
  103. $content .= "$key = \"$value\"\n";
  104. }
  105. }
  106. if ( !empty($content)) {
  107. file_put_contents($envFilePath, $content);
  108. }
  109. }
  110. /**
  111. * 设置环境变量值
  112. * @access public
  113. * @param string|array $env 环境变量
  114. * @param mixed $value 值
  115. * @return void
  116. */
  117. public function set($env, $value = null)
  118. {
  119. if (is_array($env)) {
  120. foreach ($env as $key => $val) {
  121. if (is_array($val)) {
  122. foreach ($val as $k => $v) {
  123. $this->data[$key . '.' . $k] = $v;
  124. }
  125. } else {
  126. $this->data[$key] = $val;
  127. }
  128. }
  129. } else {
  130. $name = strtoupper(str_replace('.', '_', $env));
  131. $this->data[$name] = $value;
  132. }
  133. }
  134. }