helpers.php 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300
  1. <?php
  2. /**
  3. * 助手函数
  4. *
  5. * @author mybsdc <[email protected]>
  6. * @date 2019/3/3
  7. * @time 16:34
  8. */
  9. use Luolongfei\App\Exceptions\LlfException;
  10. use Luolongfei\Libs\Argv;
  11. use Luolongfei\Libs\Config;
  12. use Luolongfei\Libs\Log;
  13. use Luolongfei\Libs\Env;
  14. use Luolongfei\Libs\Lang;
  15. use Luolongfei\Libs\PhpColor;
  16. use Luolongfei\App\Console\MigrateEnvFile;
  17. use Luolongfei\App\Console\Upgrade;
  18. if (!function_exists('config')) {
  19. /**
  20. * 获取配置
  21. *
  22. * @param string $key 键,支持点式访问
  23. * @param string $default 默认值
  24. *
  25. * @return array|mixed
  26. */
  27. function config($key = '', $default = null)
  28. {
  29. return Config::getInstance()->get($key, $default);
  30. }
  31. }
  32. if (!function_exists('lang')) {
  33. /**
  34. * 读取语言包
  35. *
  36. * @param string $key 键,支持点式访问
  37. *
  38. * @return array|mixed|null
  39. */
  40. function lang($key = '')
  41. {
  42. return Lang::getInstance()->get($key);
  43. }
  44. }
  45. if (!function_exists('system_log')) {
  46. /**
  47. * 写日志
  48. *
  49. * @param $content
  50. * @param array $response
  51. * @param string $fileName
  52. * @description 受支持的着色标签
  53. * 'reset', 'bold', 'dark', 'italic', 'underline', 'blink', 'reverse', 'concealed', 'default', 'black', 'red',
  54. * 'green', 'yellow', 'blue', 'magenta', 'cyan', 'light_gray', 'dark_gray', 'light_red', 'light_green',
  55. * 'light_yellow', 'light_blue', 'light_magenta', 'light_cyan', 'white', 'bg_default', 'bg_black', 'bg_red',
  56. * 'bg_green', 'bg_yellow', 'bg_blue', 'bg_magenta', 'bg_cyan', 'bg_light_gray', 'bg_dark_gray', 'bg_light_red',
  57. * 'bg_light_green','bg_light_yellow', 'bg_light_blue', 'bg_light_magenta', 'bg_light_cyan', 'bg_white'
  58. *
  59. * system_log('<light_magenta>颜色 light_magenta</light_magenta>');
  60. */
  61. function system_log($content, array $response = [], $fileName = '')
  62. {
  63. try {
  64. $path = sprintf('%s/logs/%s/', ROOT_PATH, date('Y-m'));
  65. $file = $path . ($fileName ?: date('d')) . '.log';
  66. if (!is_dir($path)) {
  67. mkdir($path, 0777, true);
  68. chmod($path, 0777);
  69. }
  70. $handle = fopen($file, 'a'); // 追加而非覆盖
  71. if (!filesize($file)) {
  72. chmod($file, 0666);
  73. }
  74. $msg = sprintf(
  75. "[%s] %s %s\n",
  76. date('Y-m-d H:i:s'),
  77. is_string($content) ? $content : json_encode($content),
  78. $response ? json_encode($response, JSON_UNESCAPED_UNICODE) : '');
  79. // 在 Github Actions 上运行,过滤敏感信息
  80. if (env('ON_GITHUB_ACTIONS')) {
  81. $msg = preg_replace_callback('/(?P<secret>[\w-.]{1,4}?)(?=@[\w-.]+)/i', function ($m) {
  82. return str_ireplace($m['secret'], str_repeat('*', strlen($m['secret'])), $m['secret']);
  83. }, $msg);
  84. }
  85. // 尝试为消息着色
  86. $c = PhpColor::getInstance()->getColorInstance();
  87. echo $c($msg)->colorize();
  88. // 干掉着色标签
  89. $msg = strip_tags($msg); // 不完整或者破损标签将导致更多的数据被删除
  90. fwrite($handle, $msg);
  91. fclose($handle);
  92. flush();
  93. } catch (\Exception $e) {
  94. // do nothing
  95. }
  96. }
  97. }
  98. if (!function_exists('is_locked')) {
  99. /**
  100. * 检查任务是否已被锁定
  101. *
  102. * @param string $taskName
  103. * @param bool $always 是否被永久锁定
  104. *
  105. * @return bool
  106. * @throws Exception
  107. */
  108. function is_locked($taskName = '', $always = false)
  109. {
  110. try {
  111. $lock = sprintf(
  112. '%s/num_limit/%s/%s.lock',
  113. APP_PATH,
  114. $always ? 'always' : date('Y-m-d'),
  115. $taskName
  116. );
  117. return file_exists($lock);
  118. } catch (\Exception $e) {
  119. system_log(sprintf('检查任务%s是否锁定时出错,错误原因:%s', $taskName, $e->getMessage()));
  120. }
  121. return false;
  122. }
  123. }
  124. if (!function_exists('lock_task')) {
  125. /**
  126. * 锁定任务
  127. *
  128. * 防止重复执行
  129. *
  130. * @param string $taskName
  131. * @param bool $always 是否永久锁定
  132. *
  133. * @return bool
  134. */
  135. function lock_task($taskName = '', $always = false)
  136. {
  137. try {
  138. $lock = sprintf(
  139. '%s/num_limit/%s/%s.lock',
  140. APP_PATH,
  141. $always ? 'always' : date('Y-m-d'),
  142. $taskName
  143. );
  144. $path = dirname($lock);
  145. if (!is_dir($path)) {
  146. mkdir($path, 0777, true);
  147. chmod($path, 0777);
  148. }
  149. if (file_exists($lock)) {
  150. return true;
  151. }
  152. $handle = fopen($lock, 'a'); // 追加而非覆盖
  153. if (!filesize($lock)) {
  154. chmod($lock, 0666);
  155. }
  156. fwrite($handle, sprintf(
  157. "Locked at %s.\n",
  158. date('Y-m-d H:i:s')
  159. )
  160. );
  161. fclose($handle);
  162. Log::info(sprintf('%s已被锁定,此任务%s已不会再执行,请知悉', $taskName, $always ? '' : '今天内'));
  163. } catch (\Exception $e) {
  164. system_log(sprintf('创建锁定任务文件%s时出错,错误原因:%s', $lock, $e->getMessage()));
  165. return false;
  166. }
  167. return true;
  168. }
  169. }
  170. if (!function_exists('env')) {
  171. /**
  172. * 获取环境变量值
  173. *
  174. * @param string $key
  175. * @param string $default 默认值
  176. *
  177. * @return array | bool | false | null | string
  178. */
  179. function env($key = '', $default = null)
  180. {
  181. return Env::getInstance()->get($key, $default);
  182. }
  183. }
  184. if (!function_exists('get_argv')) {
  185. /**
  186. * 获取命令行传参
  187. *
  188. * @param string $name
  189. * @param string $default 默认值
  190. *
  191. * @return mixed|string
  192. */
  193. function get_argv(string $name, string $default = '')
  194. {
  195. return Argv::getInstance()->get($name, $default);
  196. }
  197. }
  198. if (!function_exists('system_check')) {
  199. /**
  200. * 检查环境是否满足要求
  201. *
  202. * @throws LlfException
  203. */
  204. function system_check()
  205. {
  206. if (version_compare(PHP_VERSION, '7.3.0') < 0) {
  207. throw new LlfException(34520006, ['7.3', PHP_VERSION]);
  208. }
  209. // 如果是在云函数部署,则不需要检查这几项
  210. if (IS_SCF) {
  211. system_log(lang('100009'));
  212. system_log(lang('100010'));
  213. system_log(lang('100011'));
  214. } else {
  215. if (!function_exists('putenv')) {
  216. throw new LlfException(34520005);
  217. }
  218. $envFile = ROOT_PATH . '/.env';
  219. if (!file_exists($envFile)) {
  220. throw new LlfException(copy(ROOT_PATH . '/.env.example', $envFile) ? 34520007 : 34520008);
  221. }
  222. // 检查当前 .env 文件版本是否过低,过低自动升级
  223. MigrateEnvFile::getInstance()->handle();
  224. }
  225. // 是否有新版可用
  226. if (config('new_version_detection')) {
  227. Upgrade::getInstance()->handle();
  228. } else {
  229. system_log(lang('100012'));
  230. }
  231. if (!extension_loaded('curl')) {
  232. throw new LlfException(34520010);
  233. }
  234. }
  235. }
  236. if (!function_exists('get_local_num')) {
  237. /**
  238. * 获取当地数字
  239. *
  240. * @param string|int $num
  241. *
  242. * @return string
  243. */
  244. function get_local_num($num)
  245. {
  246. $num = (string)$num;
  247. if (\config('language') === 'zh') {
  248. return $num;
  249. }
  250. // 英文数字规则
  251. $lastDigit = substr($num, -1);
  252. switch ($lastDigit) {
  253. case '1':
  254. return $num . 'st';
  255. case '2':
  256. return $num . 'nd';
  257. case '3':
  258. return $num . 'rd';
  259. default:
  260. return $num . 'th';
  261. }
  262. }
  263. }