mirror of
https://gitee.com/RubyMetric/chsrc.git
synced 2025-12-26 05:02:46 +00:00
Move implementation out of xy.h
This commit is contained in:
Aoran Zeng
parent
2d75a48c19
commit
8d35bcd24d
|
|
@ -0,0 +1,741 @@
|
|||
/** ------------------------------------------------------------
|
||||
* Copyright © 2023-2025 Aoran Zeng, Heng Guo
|
||||
* SPDX-License-Identifier: MIT
|
||||
* -------------------------------------------------------------
|
||||
* File Authors : Aoran Zeng <ccmywish@qq.com>
|
||||
* | Heng Guo <2085471348@qq.com>
|
||||
* Contributors : juzeon <skyjuzheng@gmail.com>
|
||||
* |
|
||||
* Created On : <2023-08-28>
|
||||
* Major Revision : 2
|
||||
* Last Modified : <2025-03-06>
|
||||
* ------------------------------------------------------------*/
|
||||
|
||||
#include "xy.h"
|
||||
|
||||
/* Global default */
|
||||
bool xy_enable_color = 1;
|
||||
|
||||
|
||||
void putf (double n) { printf ("%f\n", n); }
|
||||
void puti (long long n) { printf ("%lld\n", n); }
|
||||
void putb (bool n) { if (n) puts ("true"); else puts ("false"); }
|
||||
void print (const char *s) { printf ("%s", s); }
|
||||
void println (const char *s) { printf ("%s\n", s); }
|
||||
void say (const char *s) { printf ("%s\n", s); }
|
||||
void br () { puts (""); }
|
||||
void p (const char *s) { printf ("%s\n", s); }
|
||||
|
||||
|
||||
|
||||
void *
|
||||
xy_malloc0 (size_t size)
|
||||
{
|
||||
void *ptr = malloc (size);
|
||||
memset (ptr, 0, size);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
|
||||
/******************************************************
|
||||
* String
|
||||
******************************************************/
|
||||
|
||||
/**
|
||||
* 将str中所有的pat字符串替换成replace，返回一个全新的字符串
|
||||
*/
|
||||
char *
|
||||
xy_str_gsub (const char *str, const char *pat, const char *replace)
|
||||
{
|
||||
size_t replace_len = strlen (replace);
|
||||
size_t pat_len = strlen (pat);
|
||||
|
||||
int unit = replace_len - pat_len;
|
||||
if (unit <= 0)
|
||||
unit = 0;
|
||||
|
||||
size_t len = strlen (str);
|
||||
|
||||
const char *cur = str;
|
||||
int count = 0;
|
||||
|
||||
while (cur < str + len)
|
||||
{
|
||||
char *fnd = strstr (cur, pat);
|
||||
if (fnd)
|
||||
{
|
||||
count++;
|
||||
cur = fnd + pat_len;
|
||||
}
|
||||
else
|
||||
break;
|
||||
}
|
||||
// puti(count); DEBUG 匹配次数
|
||||
|
||||
char *ret = malloc (unit * count + len + 1);
|
||||
char *retcur = ret;
|
||||
|
||||
cur = str;
|
||||
while (cur < str + len)
|
||||
{
|
||||
char *fnd = strstr (cur, pat);
|
||||
if (fnd)
|
||||
{
|
||||
ptrdiff_t diff = fnd - cur;
|
||||
strncpy (retcur, cur, diff);
|
||||
cur = fnd + pat_len;
|
||||
|
||||
retcur += diff;
|
||||
strcpy (retcur, replace);
|
||||
retcur += replace_len;
|
||||
}
|
||||
else
|
||||
break;
|
||||
}
|
||||
strcpy (retcur, cur);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_2strjoin (const char *str1, const char *str2)
|
||||
{
|
||||
size_t len = strlen (str1);
|
||||
size_t size = len + strlen (str2) + 1;
|
||||
char *ret = malloc (size);
|
||||
strcpy (ret, str1);
|
||||
strcpy (ret + len, str2);
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_strjoin (unsigned int count, ...)
|
||||
{
|
||||
size_t al_fixed = 128;
|
||||
char *ret = calloc (1, al_fixed);
|
||||
// 已分配次数
|
||||
int al_times = 1;
|
||||
// 当前已分配量
|
||||
size_t al_cur = al_fixed;
|
||||
|
||||
const char *str = NULL;
|
||||
// 需要分配的量
|
||||
size_t al_need = 0;
|
||||
// 用于 strcpy() 到 ret 的哪个位置
|
||||
char *cur = ret + 0;
|
||||
|
||||
va_list args;
|
||||
va_start (args, count);
|
||||
|
||||
for (int i = 0; i < count; i++)
|
||||
{
|
||||
// 是否需要重新分配
|
||||
bool need_realloc = false;
|
||||
|
||||
str = va_arg (args, const char *);
|
||||
al_need += strlen (str);
|
||||
while (al_need > al_cur)
|
||||
{
|
||||
al_times += 1;
|
||||
al_cur = al_times * al_fixed;
|
||||
need_realloc = true;
|
||||
}
|
||||
// printf("al_times %d, al_need %zd, al_cur %zd\n", al_times, al_need,
|
||||
// al_cur);
|
||||
if (need_realloc)
|
||||
{
|
||||
ptrdiff_t diff = cur - ret;
|
||||
ret = realloc (ret, al_cur);
|
||||
cur = ret + diff;
|
||||
}
|
||||
if (NULL == ret)
|
||||
{
|
||||
fprintf (stderr, "xy.h: No availble memory!");
|
||||
return NULL;
|
||||
}
|
||||
strcpy (cur, str);
|
||||
// puts(ret);
|
||||
cur += strlen (str);
|
||||
}
|
||||
va_end (args);
|
||||
|
||||
*cur = '\0';
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_strdup (const char *str)
|
||||
{
|
||||
size_t len = strlen (str);
|
||||
char *new = xy_malloc0 (len + 1);
|
||||
strcpy (new, str);
|
||||
return new;
|
||||
}
|
||||
|
||||
#define _XY_Str_Bold 1
|
||||
#define _XY_Str_Faint 2
|
||||
#define _XY_Str_Italic 3
|
||||
#define _XY_Str_Underline 4
|
||||
#define _XY_Str_Blink 5
|
||||
#define _XY_Str_Cross 9
|
||||
|
||||
#define _XY_Str_Red 31
|
||||
#define _XY_Str_Green 32
|
||||
#define _XY_Str_Yellow 33
|
||||
#define _XY_Str_Blue 34
|
||||
#define _XY_Str_Magenta 35
|
||||
#define _XY_Str_Cyan 36
|
||||
|
||||
static char *
|
||||
_xy_str_to_terminal_style (int style, const char *str)
|
||||
{
|
||||
char *color_fmt_str = NULL;
|
||||
|
||||
if (!xy_enable_color)
|
||||
{
|
||||
color_fmt_str = "%s";
|
||||
goto new_str;
|
||||
}
|
||||
|
||||
switch (style)
|
||||
{
|
||||
case _XY_Str_Red:
|
||||
color_fmt_str = "\e[31m%s\e[0m"; break;
|
||||
case _XY_Str_Green:
|
||||
color_fmt_str = "\e[32m%s\e[0m"; break;
|
||||
case _XY_Str_Yellow:
|
||||
color_fmt_str = "\e[33m%s\e[0m"; break;
|
||||
case _XY_Str_Blue:
|
||||
color_fmt_str = "\e[34m%s\e[0m"; break;
|
||||
case _XY_Str_Magenta:
|
||||
color_fmt_str = "\e[35m%s\e[0m"; break;
|
||||
case _XY_Str_Cyan:
|
||||
color_fmt_str = "\e[36m%s\e[0m"; break;
|
||||
case _XY_Str_Bold:
|
||||
color_fmt_str = "\e[1m%s\e[0m"; break;
|
||||
case _XY_Str_Faint:
|
||||
color_fmt_str = "\e[2m%s\e[0m"; break;
|
||||
case _XY_Str_Italic:
|
||||
color_fmt_str = "\e[3m%s\e[0m"; break;
|
||||
case _XY_Str_Underline:
|
||||
color_fmt_str = "\e[4m%s\e[0m"; break;
|
||||
case _XY_Str_Blink:
|
||||
color_fmt_str = "\e[5m%s\e[0m"; break;
|
||||
case _XY_Str_Cross:
|
||||
color_fmt_str = "\e[9m%s\e[0m"; break;
|
||||
}
|
||||
|
||||
|
||||
size_t len = 0;
|
||||
new_str:
|
||||
// -2 把中间%s减掉
|
||||
len = strlen (color_fmt_str) - 2;
|
||||
char *buf = malloc (strlen (str) + len + 1);
|
||||
sprintf (buf, color_fmt_str, str);
|
||||
return buf;
|
||||
}
|
||||
|
||||
char * xy_str_to_bold(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Bold, str); }
|
||||
char * xy_str_to_faint(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Faint, str); }
|
||||
char * xy_str_to_italic(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Italic, str);}
|
||||
char * xy_str_to_underline(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Underline, str);}
|
||||
char * xy_str_to_blink(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Blink, str); }
|
||||
char * xy_str_to_cross(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Cross, str); }
|
||||
|
||||
char * xy_str_to_red(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Red, str); }
|
||||
char * xy_str_to_green(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Green, str); }
|
||||
char * xy_str_to_yellow(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Yellow, str); }
|
||||
char * xy_str_to_blue(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Blue, str); }
|
||||
char * xy_str_to_magenta(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Magenta, str);}
|
||||
char * xy_str_to_purple(const char *str) { return xy_str_to_magenta (str); }
|
||||
char * xy_str_to_cyan(const char *str) { return _xy_str_to_terminal_style (_XY_Str_Cyan, str); }
|
||||
|
||||
|
||||
|
||||
bool
|
||||
xy_streql (const char *str1, const char *str2)
|
||||
{
|
||||
if (NULL==str1 || NULL==str2)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
return strcmp (str1, str2) == 0 ? true : false;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_str_to_quietcmd (const char *cmd)
|
||||
{
|
||||
char *ret = NULL;
|
||||
#ifdef _WIN32
|
||||
ret = xy_2strjoin (cmd, " >nul 2>nul ");
|
||||
#else
|
||||
ret = xy_2strjoin (cmd, " 1>/dev/null 2>&1 ");
|
||||
#endif
|
||||
return ret;
|
||||
}
|
||||
|
||||
bool
|
||||
xy_str_end_with (const char *str, const char *suffix)
|
||||
{
|
||||
size_t len1 = strlen (str);
|
||||
size_t len2 = strlen (suffix);
|
||||
|
||||
if (0 == len2)
|
||||
return true; // 空字符串直接返回
|
||||
if (len1 < len2)
|
||||
return false;
|
||||
|
||||
const char *cur1 = str + len1 - 1;
|
||||
const char *cur2 = suffix + len2 - 1;
|
||||
|
||||
for (int i = 0; i < len2; i++)
|
||||
{
|
||||
if (*cur1 != *cur2)
|
||||
return false;
|
||||
cur1--;
|
||||
cur2--;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool
|
||||
xy_str_start_with (const char *str, const char *prefix)
|
||||
{
|
||||
if (NULL==str || NULL==prefix)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
size_t len1 = strlen (str);
|
||||
size_t len2 = strlen (prefix);
|
||||
|
||||
if (0 == len2)
|
||||
return true; // 空字符串直接返回
|
||||
if (len1 < len2)
|
||||
return false;
|
||||
|
||||
const char *cur1 = str;
|
||||
const char *cur2 = prefix;
|
||||
|
||||
for (int i = 0; i < len2; i++)
|
||||
{
|
||||
if (*cur1 != *cur2)
|
||||
return false;
|
||||
cur1++;
|
||||
cur2++;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_str_delete_prefix (const char *str, const char *prefix)
|
||||
{
|
||||
char *new = xy_strdup (str);
|
||||
bool yes = xy_str_start_with (str, prefix);
|
||||
if (!yes)
|
||||
return new;
|
||||
|
||||
size_t len = strlen (prefix);
|
||||
char *cur = new + len;
|
||||
return cur;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_str_delete_suffix (const char *str, const char *suffix)
|
||||
{
|
||||
char *new = xy_strdup (str);
|
||||
bool yes = xy_str_end_with (str, suffix);
|
||||
if (!yes)
|
||||
return new;
|
||||
|
||||
size_t len1 = strlen (str);
|
||||
size_t len2 = strlen (suffix);
|
||||
char *cur = new + len1 - len2;
|
||||
*cur = '\0';
|
||||
return new;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_str_strip (const char *str)
|
||||
{
|
||||
char *new = xy_strdup (str);
|
||||
|
||||
while (strchr ("\n\r\v\t\f ", new[0]))
|
||||
{
|
||||
new += 1;
|
||||
}
|
||||
|
||||
size_t len = strlen (new);
|
||||
|
||||
char *last = new + len - 1;
|
||||
|
||||
while (strchr ("\n\r\v\t\f ", *last))
|
||||
{
|
||||
*last = '\0';
|
||||
last -= 1;
|
||||
}
|
||||
return new;
|
||||
}
|
||||
|
||||
|
||||
/******************************************************
|
||||
* Logging
|
||||
******************************************************/
|
||||
|
||||
#define _XY_Log_Plain 000000001
|
||||
#define _XY_Log_Success 000000001 << 1
|
||||
#define _XY_Log_Info 000000001 << 2
|
||||
#define _XY_Log_Warn 000000001 << 3
|
||||
#define _XY_Log_Error 000000001 << 4
|
||||
|
||||
static void
|
||||
_xy_log (int level, const char *prompt, const char *content)
|
||||
{
|
||||
char *str = NULL;
|
||||
|
||||
bool to_stderr = false;
|
||||
|
||||
/**
|
||||
* 'app: content'
|
||||
*/
|
||||
if (level & _XY_Log_Plain)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", content);
|
||||
}
|
||||
else if (level & _XY_Log_Success)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", xy_str_to_green (content));
|
||||
}
|
||||
else if (level & _XY_Log_Info)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", xy_str_to_blue (content));
|
||||
}
|
||||
else if (level & _XY_Log_Warn)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", xy_str_to_yellow (content));
|
||||
to_stderr = true;
|
||||
}
|
||||
else if (level & _XY_Log_Error)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", xy_str_to_red (content));
|
||||
to_stderr = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
// xy_assert ("CAN'T REACH!");
|
||||
}
|
||||
|
||||
if (to_stderr)
|
||||
{
|
||||
fprintf (stderr, "%s\n", str);
|
||||
}
|
||||
else
|
||||
{
|
||||
puts (str);
|
||||
}
|
||||
free (str);
|
||||
}
|
||||
|
||||
void xy_log (const char *prompt, const char *str) { _xy_log (_XY_Log_Plain, prompt, str); }
|
||||
void xy_succ(const char *prompt, const char *str) { _xy_log (_XY_Log_Success, prompt, str); }
|
||||
void xy_info(const char *prompt, const char *str) { _xy_log (_XY_Log_Info, prompt, str); }
|
||||
void xy_warn(const char *prompt, const char *str) { _xy_log (_XY_Log_Warn, prompt, str); }
|
||||
void xy_error(const char *prompt, const char *str) { _xy_log (_XY_Log_Error, prompt, str); }
|
||||
|
||||
|
||||
/**
|
||||
* brkt 系列输出受 pip 启发，为了输出方便，使用 xy.h 的程序应该
|
||||
*
|
||||
* 1.若想完全自定义颜色和输出位置：
|
||||
*
|
||||
* 应基于下述 xy_log_brkt_to 定义自己的输出函数
|
||||
*
|
||||
* 2.若想自定义颜色，而保持输出到stdout：
|
||||
*
|
||||
* 应基于下述 xy_log_brkt 定义
|
||||
*
|
||||
* 3.若对log无细致要求，想要快速使用log功能：
|
||||
*
|
||||
* 应基于下述 xy_<level>_brkt 定义自己的 app_<level>_brkt()，或直接使用 xy_<level>_brkt
|
||||
*/
|
||||
|
||||
void
|
||||
xy_log_brkt_to (const char *prompt, const char *content, FILE *stream)
|
||||
{
|
||||
char *str = xy_strjoin (4, "[", prompt, "] ", content);
|
||||
fprintf (stream, "%s\n", str);
|
||||
free (str);
|
||||
}
|
||||
|
||||
|
||||
|
||||
static void
|
||||
_xy_log_brkt (int level, const char *prompt1, const char *prompt2, const char *content)
|
||||
{
|
||||
char *str = NULL;
|
||||
|
||||
bool to_stderr = false;
|
||||
|
||||
if (level & _XY_Log_Plain)
|
||||
{
|
||||
str = xy_strjoin (6, "[", prompt1, " ", prompt2, "] ", content);
|
||||
}
|
||||
else if (level & _XY_Log_Success)
|
||||
{
|
||||
/* [app 成功] [app success] */
|
||||
str = xy_strjoin (6,
|
||||
"[", xy_str_to_green (prompt1), " ", xy_str_to_bold (xy_str_to_green (prompt2)), "] ", xy_str_to_green (content));
|
||||
}
|
||||
else if (level & _XY_Log_Info)
|
||||
{
|
||||
/* [app 信息] [app info]
|
||||
[app 提示] [app notice]
|
||||
*/
|
||||
str = xy_strjoin (6,
|
||||
"[", xy_str_to_blue (prompt1), " ", xy_str_to_bold (xy_str_to_blue (prompt2)), "] ", xy_str_to_blue (content));
|
||||
}
|
||||
else if (level & _XY_Log_Warn)
|
||||
{
|
||||
/* [app 警告] [app warn] */
|
||||
str = xy_strjoin (6,
|
||||
"[", xy_str_to_yellow (prompt1), " ", xy_str_to_bold (xy_str_to_yellow (prompt2)), "] ", xy_str_to_yellow (content));
|
||||
to_stderr = true;
|
||||
}
|
||||
else if (level & _XY_Log_Error)
|
||||
{
|
||||
/* [app 错误] [app error] */
|
||||
str = xy_strjoin (6,
|
||||
"[", xy_str_to_red (prompt1), " ", xy_str_to_bold (xy_str_to_red (prompt2)), "] ", xy_str_to_red (content));
|
||||
to_stderr = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
// xy_assert ("CAN'T REACH!");
|
||||
}
|
||||
|
||||
if (to_stderr)
|
||||
{
|
||||
fprintf (stderr, "%s\n", str);
|
||||
}
|
||||
else
|
||||
{
|
||||
puts (str);
|
||||
}
|
||||
free (str);
|
||||
}
|
||||
|
||||
|
||||
void xy_log_brkt (const char *prompt1, const char *prompt2, const char *content) { _xy_log_brkt(_XY_Log_Plain, prompt1,prompt2,content); }
|
||||
void xy_succ_brkt(const char *prompt1, const char *prompt2, const char *content) { _xy_log_brkt(_XY_Log_Success,prompt1,prompt2,content); }
|
||||
void xy_info_brkt(const char *prompt1, const char *prompt2, const char *content) { _xy_log_brkt(_XY_Log_Info, prompt1,prompt2,content); }
|
||||
void xy_warn_brkt(const char *prompt1, const char *prompt2, const char *content) { _xy_log_brkt(_XY_Log_Warn, prompt1,prompt2,content); }
|
||||
void xy_error_brkt(const char *prompt1,const char *prompt2, const char *content) { _xy_log_brkt(_XY_Log_Error, prompt1,prompt2,content); }
|
||||
|
||||
|
||||
|
||||
/******************************************************
|
||||
* System
|
||||
******************************************************/
|
||||
/**
|
||||
* 执行cmd，返回某行输出结果，并对已经遍历过的行执行iter_func
|
||||
*
|
||||
* @param cmd 要执行的命令
|
||||
* @param n 指定命令执行输出的结果行中的某一行，0 表示最后一行，n (n>0) 表示第n行
|
||||
* 该函数会返回这一行的内容
|
||||
* @param iter_func 对遍历时经过的行的内容，进行函数调用
|
||||
*
|
||||
* @note 返回的字符串最后面一般有换行符号
|
||||
*
|
||||
* 由于目标行会被返回出来，所以 iter_func() 并不执行目标行，只会执行遍历过的行
|
||||
*/
|
||||
char *
|
||||
xy_run_iter (const char *cmd, unsigned long n, void (*iter_func) (const char *))
|
||||
{
|
||||
const int size = 512;
|
||||
char *buf = (char *) malloc (size);
|
||||
|
||||
FILE *stream = popen (cmd, "r");
|
||||
if (stream == NULL)
|
||||
{
|
||||
fprintf (stderr, "xy: 命令执行失败\n");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char *ret = NULL;
|
||||
unsigned long count = 0;
|
||||
|
||||
while (true)
|
||||
{
|
||||
if (NULL == fgets (buf, size, stream))
|
||||
break;
|
||||
/* 存在换行的总是会把换行符读出来，删掉 */
|
||||
ret = xy_str_delete_suffix (buf, "\n");
|
||||
count += 1;
|
||||
if (n == count)
|
||||
break;
|
||||
if (iter_func)
|
||||
{
|
||||
iter_func (buf);
|
||||
}
|
||||
}
|
||||
|
||||
pclose (stream);
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_run (const char *cmd, unsigned long n)
|
||||
{
|
||||
return xy_run_iter (cmd, n, NULL);
|
||||
}
|
||||
|
||||
|
||||
char *
|
||||
xy_os_home ()
|
||||
{
|
||||
char *home = NULL;
|
||||
if (xy_on_windows)
|
||||
home = getenv ("USERPROFILE");
|
||||
else
|
||||
home = getenv ("HOME");
|
||||
return home;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_win_powershell_profile ()
|
||||
{
|
||||
return xy_2strjoin (
|
||||
xy_os_home(), "\\Documents\\PowerShell\\Microsoft.PowerShell_profile.ps1");
|
||||
}
|
||||
|
||||
char *
|
||||
xy_win_powershellv5_profile ()
|
||||
{
|
||||
return xy_2strjoin (
|
||||
xy_os_home(),
|
||||
"\\Documents\\WindowsPowerShell\\Microsoft.PowerShell_profile.ps1");
|
||||
}
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* @note Windows上，`path` 不要夹带变量名，因为最终 access() 不会帮你转换
|
||||
*/
|
||||
bool
|
||||
xy_file_exist (const char *path)
|
||||
{
|
||||
const char *new_path = path;
|
||||
if (xy_str_start_with (path, "~"))
|
||||
{
|
||||
new_path = xy_2strjoin (xy_os_home(), path + 1);
|
||||
}
|
||||
// 0 即 F_OK
|
||||
return (0==access (new_path, 0)) ? true : false;
|
||||
}
|
||||
|
||||
/**
|
||||
* @note xy_file_exist() 和 xy_dir_exist() 两个函数在所有平台默认都支持使用 '~'，
|
||||
* 但实现中都没有调用 xy_normalize_path()，以防万一，调用前可能需要用户手动调用它
|
||||
*/
|
||||
bool
|
||||
xy_dir_exist (const char *path)
|
||||
{
|
||||
const char *dir = path;
|
||||
if (xy_on_windows)
|
||||
{
|
||||
if (xy_str_start_with (path, "~"))
|
||||
{
|
||||
dir = xy_2strjoin (xy_os_home(), path + 1);
|
||||
}
|
||||
}
|
||||
|
||||
if (xy_on_windows)
|
||||
{
|
||||
#ifdef XY_On_Windows
|
||||
// 也可以用 opendir() #include <dirent.h>
|
||||
DWORD attr = GetFileAttributesA (dir);
|
||||
|
||||
if (attr == INVALID_FILE_ATTRIBUTES)
|
||||
{
|
||||
// Q: 我们应该报错吗？
|
||||
return false;
|
||||
}
|
||||
else if (attr & FILE_ATTRIBUTE_DIRECTORY)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
return false;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
else
|
||||
{
|
||||
int status = system (xy_2strjoin ("test -d ", dir));
|
||||
|
||||
if (0==status)
|
||||
return true;
|
||||
else
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 1. 删除路径左右两边多出来的空白符
|
||||
* 2. 将 ~/ 转换为绝对路径
|
||||
*/
|
||||
char *
|
||||
xy_normalize_path (const char *path)
|
||||
{
|
||||
char *new = xy_str_strip (path); // 防止开发者多写了空白符
|
||||
|
||||
if (xy_on_windows)
|
||||
{
|
||||
if (xy_str_start_with (new, "~/"))
|
||||
{
|
||||
// 或 %USERPROFILE%
|
||||
new = xy_strjoin (3, xy_os_home, "\\",
|
||||
xy_str_delete_prefix (new, "~/"));
|
||||
}
|
||||
new = xy_str_gsub (new, "/", "\\");
|
||||
}
|
||||
else
|
||||
{
|
||||
if (xy_str_start_with (new, "~/"))
|
||||
{
|
||||
new = xy_strjoin (3, xy_os_home, "/",
|
||||
xy_str_delete_prefix (new, "~/"));
|
||||
}
|
||||
}
|
||||
|
||||
return new;
|
||||
}
|
||||
|
||||
char *
|
||||
xy_parent_dir (const char *path)
|
||||
{
|
||||
char *dir = xy_normalize_path (path);
|
||||
char *last = NULL;
|
||||
if (xy_on_windows)
|
||||
{
|
||||
last = strrchr (dir, '\\');
|
||||
if (!last)
|
||||
{
|
||||
/* current dir */
|
||||
return ".";
|
||||
}
|
||||
*last = '\0';
|
||||
}
|
||||
else
|
||||
{
|
||||
last = strrchr (dir, '/');
|
||||
if (!last)
|
||||
{
|
||||
/* current dir */
|
||||
return ".";
|
||||
}
|
||||
*last = '\0';
|
||||
}
|
||||
return dir;
|
||||
}
|
||||
786
lib/xy.h
786
lib/xy.h
|
|
@ -2,23 +2,25 @@
|
|||
* Copyright © 2023-2025 Aoran Zeng, Heng Guo
|
||||
* SPDX-License-Identifier: MIT
|
||||
* -------------------------------------------------------------
|
||||
* Lib Name : xy.h
|
||||
* Lib Name : xy
|
||||
* Lib Authors : Aoran Zeng <ccmywish@qq.com>
|
||||
* | Heng Guo <2085471348@qq.com>
|
||||
* Contributors : juzeon <skyjuzheng@gmail.com>
|
||||
* |
|
||||
* Created On : <2023-08-28>
|
||||
* Last Modified : <2024-12-14>
|
||||
* Last Modified : <2025-03-06>
|
||||
*
|
||||
* xy: 襄阳、咸阳
|
||||
* Corss-Platform C utilities for CLI applications in Ruby flavor
|
||||
* Corss-Platform C lib for CLI applications in Ruby & Perl flavor
|
||||
* ------------------------------------------------------------*/
|
||||
|
||||
#ifndef XY_H
|
||||
#define XY_H
|
||||
|
||||
#define _XY_Version "v0.1.4.4-2024/12/14"
|
||||
#define _XY_Maintain_URL "https://gitee.com/RubyMetric/chsrc/blob/main/include/xy.h"
|
||||
|
||||
#define _XY_Version "v0.1.5-2025/03/06"
|
||||
#define _XY_Maintain_URL "https://gitee.com/RubyMetric/chsrc/blob/main/lib/xy.h"
|
||||
|
||||
|
||||
#include <assert.h>
|
||||
#include <stdarg.h>
|
||||
|
|
@ -29,11 +31,14 @@
|
|||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
|
||||
/* Global */
|
||||
bool xy_enable_color = true;
|
||||
|
||||
// #define NDEBUG
|
||||
|
||||
|
||||
/* Global */
|
||||
extern bool xy_enable_color; /* Default value defined in xy.c */
|
||||
|
||||
|
||||
#ifdef _WIN32
|
||||
#define XY_On_Windows 1
|
||||
#define xy_on_windows true
|
||||
|
|
@ -72,733 +77,116 @@ bool xy_enable_color = true;
|
|||
#define xy_useutf8()
|
||||
#endif
|
||||
|
||||
void putf (double n) { printf ("%f\n", n); }
|
||||
void puti (long long n) { printf ("%lld\n", n); }
|
||||
void putb (bool n) { if (n) puts ("true"); else puts ("false"); }
|
||||
void print (const char *s) { printf ("%s", s); }
|
||||
void println (const char *s) { printf ("%s\n", s); }
|
||||
void say (const char *s) { printf ("%s\n", s); }
|
||||
void br () { puts (""); }
|
||||
void p (const char *s) { printf ("%s\n", s); }
|
||||
|
||||
#define assert_str(a, b) assert (xy_streql ((a), (b)))
|
||||
|
||||
#define xy_unsupported() assert(!"Unsuppoted")
|
||||
#define xy_unimplemented() assert(!"Unimplemented temporarily")
|
||||
#define xy_unreached() assert(!"This code shouldn't be reached")
|
||||
/******************************************************
|
||||
* Convenience
|
||||
******************************************************/
|
||||
void putf (double n);
|
||||
void puti (long long n);
|
||||
void putb (bool n);
|
||||
void print (const char *s);
|
||||
void println (const char *s);
|
||||
void say (const char *s);
|
||||
void br ();
|
||||
void p (const char *s);
|
||||
|
||||
#define xy_arylen(x) (sizeof (x) / sizeof (x[0]))
|
||||
|
||||
static inline void *
|
||||
xy_malloc0 (size_t size)
|
||||
{
|
||||
void *ptr = malloc (size);
|
||||
memset (ptr, 0, size);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
|
||||
/******************************************************
|
||||
* Assertion
|
||||
******************************************************/
|
||||
#define assert_str(a, b) assert (xy_streql ((a), (b)))
|
||||
#define xy_unsupported() assert(!"Unsuppoted")
|
||||
#define xy_unimplemented() assert(!"Unimplemented temporarily")
|
||||
#define xy_unreached() assert(!"This code shouldn't be reached")
|
||||
|
||||
|
||||
|
||||
/******************************************************
|
||||
* Memory
|
||||
******************************************************/
|
||||
inline void * xy_malloc0 (size_t size);
|
||||
|
||||
|
||||
|
||||
/******************************************************
|
||||
* String
|
||||
******************************************************/
|
||||
char * xy_str_gsub (const char *str, const char *pat, const char *replace);
|
||||
char * xy_2strjoin (const char *str1, const char *str2);
|
||||
char * xy_strjoin (unsigned int count, ...);
|
||||
char * xy_strdup (const char *str);
|
||||
bool xy_streql (const char *str1, const char *str2);
|
||||
char * xy_str_to_quietcmd (const char *cmd);
|
||||
|
||||
/**
|
||||
* 将str中所有的pat字符串替换成replace，返回一个全新的字符串
|
||||
*/
|
||||
static char *
|
||||
xy_str_gsub (const char *str, const char *pat, const char *replace)
|
||||
{
|
||||
size_t replace_len = strlen (replace);
|
||||
size_t pat_len = strlen (pat);
|
||||
bool xy_str_end_with (const char *str, const char *suffix);
|
||||
bool xy_str_start_with (const char *str, const char *prefix);
|
||||
char * xy_str_delete_prefix (const char *str, const char *prefix);
|
||||
char * xy_str_delete_suffix (const char *str, const char *suffix);
|
||||
char * xy_str_strip (const char *str);
|
||||
|
||||
int unit = replace_len - pat_len;
|
||||
if (unit <= 0)
|
||||
unit = 0;
|
||||
char * xy_str_to_bold(const char *str);
|
||||
char * xy_str_to_faint(const char *str);
|
||||
char * xy_str_to_italic(const char *str);
|
||||
char * xy_str_to_underline(const char *str);
|
||||
char * xy_str_to_blink(const char *str);
|
||||
char * xy_str_to_cross(const char *str);
|
||||
|
||||
size_t len = strlen (str);
|
||||
char * xy_str_to_red(const char *str);
|
||||
char * xy_str_to_green(const char *str);
|
||||
char * xy_str_to_yellow(const char *str);
|
||||
char * xy_str_to_blue(const char *str);
|
||||
char * xy_str_to_magenta(const char *str);
|
||||
char * xy_str_to_purple(const char *str);
|
||||
char * xy_str_to_cyan(const char *str);
|
||||
|
||||
const char *cur = str;
|
||||
int count = 0;
|
||||
|
||||
while (cur < str + len)
|
||||
{
|
||||
char *fnd = strstr (cur, pat);
|
||||
if (fnd)
|
||||
{
|
||||
count++;
|
||||
cur = fnd + pat_len;
|
||||
}
|
||||
else
|
||||
break;
|
||||
}
|
||||
// puti(count); DEBUG 匹配次数
|
||||
|
||||
char *ret = malloc (unit * count + len + 1);
|
||||
char *retcur = ret;
|
||||
|
||||
cur = str;
|
||||
while (cur < str + len)
|
||||
{
|
||||
char *fnd = strstr (cur, pat);
|
||||
if (fnd)
|
||||
{
|
||||
ptrdiff_t diff = fnd - cur;
|
||||
strncpy (retcur, cur, diff);
|
||||
cur = fnd + pat_len;
|
||||
|
||||
retcur += diff;
|
||||
strcpy (retcur, replace);
|
||||
retcur += replace_len;
|
||||
}
|
||||
else
|
||||
break;
|
||||
}
|
||||
strcpy (retcur, cur);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_2strjoin (const char *str1, const char *str2)
|
||||
{
|
||||
size_t len = strlen (str1);
|
||||
size_t size = len + strlen (str2) + 1;
|
||||
char *ret = malloc (size);
|
||||
strcpy (ret, str1);
|
||||
strcpy (ret + len, str2);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_strjoin (unsigned int count, ...)
|
||||
{
|
||||
size_t al_fixed = 128;
|
||||
char *ret = calloc (1, al_fixed);
|
||||
// 已分配次数
|
||||
int al_times = 1;
|
||||
// 当前已分配量
|
||||
size_t al_cur = al_fixed;
|
||||
|
||||
const char *str = NULL;
|
||||
// 需要分配的量
|
||||
size_t al_need = 0;
|
||||
// 用于 strcpy() 到 ret 的哪个位置
|
||||
char *cur = ret + 0;
|
||||
|
||||
va_list args;
|
||||
va_start (args, count);
|
||||
|
||||
for (int i = 0; i < count; i++)
|
||||
{
|
||||
// 是否需要重新分配
|
||||
bool need_realloc = false;
|
||||
|
||||
str = va_arg (args, const char *);
|
||||
al_need += strlen (str);
|
||||
while (al_need > al_cur)
|
||||
{
|
||||
al_times += 1;
|
||||
al_cur = al_times * al_fixed;
|
||||
need_realloc = true;
|
||||
}
|
||||
// printf("al_times %d, al_need %zd, al_cur %zd\n", al_times, al_need,
|
||||
// al_cur);
|
||||
if (need_realloc)
|
||||
{
|
||||
ptrdiff_t diff = cur - ret;
|
||||
ret = realloc (ret, al_cur);
|
||||
cur = ret + diff;
|
||||
}
|
||||
if (NULL == ret)
|
||||
{
|
||||
fprintf (stderr, "xy.h: No availble memory!");
|
||||
return NULL;
|
||||
}
|
||||
strcpy (cur, str);
|
||||
// puts(ret);
|
||||
cur += strlen (str);
|
||||
}
|
||||
va_end (args);
|
||||
|
||||
*cur = '\0';
|
||||
return ret;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_strdup (const char *str)
|
||||
{
|
||||
size_t len = strlen (str);
|
||||
char *new = xy_malloc0 (len + 1);
|
||||
strcpy (new, str);
|
||||
return new;
|
||||
}
|
||||
|
||||
#define _XY_Str_Bold 1
|
||||
#define _XY_Str_Faint 2
|
||||
#define _XY_Str_Italic 3
|
||||
#define _XY_Str_Underline 4
|
||||
#define _XY_Str_Blink 5
|
||||
#define _XY_Str_Cross 9
|
||||
|
||||
#define xy_str_to_bold(str) _xy_str_to_terminal_style (_XY_Str_Bold, str)
|
||||
#define xy_str_to_faint(str) _xy_str_to_terminal_style (_XY_Str_Faint, str)
|
||||
#define xy_str_to_italic(str) _xy_str_to_terminal_style (_XY_Str_Italic, str)
|
||||
#define xy_str_to_underline(str) _xy_str_to_terminal_style (_XY_Str_Underline, str)
|
||||
#define xy_str_to_blink(str) _xy_str_to_terminal_style (_XY_Str_Blink, str)
|
||||
#define xy_str_to_cross(str) _xy_str_to_terminal_style (_XY_Str_Cross, str)
|
||||
|
||||
#define _XY_Str_Red 31
|
||||
#define _XY_Str_Green 32
|
||||
#define _XY_Str_Yellow 33
|
||||
#define _XY_Str_Blue 34
|
||||
#define _XY_Str_Magenta 35
|
||||
#define _XY_Str_Cyan 36
|
||||
|
||||
#define xy_str_to_red(str) _xy_str_to_terminal_style (_XY_Str_Red, str)
|
||||
#define xy_str_to_green(str) _xy_str_to_terminal_style (_XY_Str_Green, str)
|
||||
#define xy_str_to_yellow(str) _xy_str_to_terminal_style (_XY_Str_Yellow, str)
|
||||
#define xy_str_to_blue(str) _xy_str_to_terminal_style (_XY_Str_Blue, str)
|
||||
#define xy_str_to_magenta(str) _xy_str_to_terminal_style (_XY_Str_Magenta, str)
|
||||
#define xy_str_to_purple xy_str_to_magenta
|
||||
#define xy_str_to_cyan(str) _xy_str_to_terminal_style (_XY_Str_Cyan, str)
|
||||
|
||||
static char *
|
||||
_xy_str_to_terminal_style (int style, const char *str)
|
||||
{
|
||||
char *color_fmt_str = NULL;
|
||||
|
||||
if (!xy_enable_color)
|
||||
{
|
||||
color_fmt_str = "%s";
|
||||
goto new_str;
|
||||
}
|
||||
|
||||
switch (style)
|
||||
{
|
||||
case _XY_Str_Red:
|
||||
color_fmt_str = "\e[31m%s\e[0m"; break;
|
||||
case _XY_Str_Green:
|
||||
color_fmt_str = "\e[32m%s\e[0m"; break;
|
||||
case _XY_Str_Yellow:
|
||||
color_fmt_str = "\e[33m%s\e[0m"; break;
|
||||
case _XY_Str_Blue:
|
||||
color_fmt_str = "\e[34m%s\e[0m"; break;
|
||||
case _XY_Str_Magenta:
|
||||
color_fmt_str = "\e[35m%s\e[0m"; break;
|
||||
case _XY_Str_Cyan:
|
||||
color_fmt_str = "\e[36m%s\e[0m"; break;
|
||||
case _XY_Str_Bold:
|
||||
color_fmt_str = "\e[1m%s\e[0m"; break;
|
||||
case _XY_Str_Faint:
|
||||
color_fmt_str = "\e[2m%s\e[0m"; break;
|
||||
case _XY_Str_Italic:
|
||||
color_fmt_str = "\e[3m%s\e[0m"; break;
|
||||
case _XY_Str_Underline:
|
||||
color_fmt_str = "\e[4m%s\e[0m"; break;
|
||||
case _XY_Str_Blink:
|
||||
color_fmt_str = "\e[5m%s\e[0m"; break;
|
||||
case _XY_Str_Cross:
|
||||
color_fmt_str = "\e[9m%s\e[0m"; break;
|
||||
}
|
||||
|
||||
|
||||
size_t len = 0;
|
||||
new_str:
|
||||
// -2 把中间%s减掉
|
||||
len = strlen (color_fmt_str) - 2;
|
||||
char *buf = malloc (strlen (str) + len + 1);
|
||||
sprintf (buf, color_fmt_str, str);
|
||||
return buf;
|
||||
}
|
||||
|
||||
static bool
|
||||
xy_streql (const char *str1, const char *str2)
|
||||
{
|
||||
if (NULL==str1 || NULL==str2)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
return strcmp (str1, str2) == 0 ? true : false;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_str_to_quietcmd (const char *cmd)
|
||||
{
|
||||
char *ret = NULL;
|
||||
#ifdef _WIN32
|
||||
ret = xy_2strjoin (cmd, " >nul 2>nul ");
|
||||
#else
|
||||
ret = xy_2strjoin (cmd, " 1>/dev/null 2>&1 ");
|
||||
#endif
|
||||
return ret;
|
||||
}
|
||||
|
||||
static bool
|
||||
xy_str_end_with (const char *str, const char *suffix)
|
||||
{
|
||||
size_t len1 = strlen (str);
|
||||
size_t len2 = strlen (suffix);
|
||||
|
||||
if (0 == len2)
|
||||
return true; // 空字符串直接返回
|
||||
if (len1 < len2)
|
||||
return false;
|
||||
|
||||
const char *cur1 = str + len1 - 1;
|
||||
const char *cur2 = suffix + len2 - 1;
|
||||
|
||||
for (int i = 0; i < len2; i++)
|
||||
{
|
||||
if (*cur1 != *cur2)
|
||||
return false;
|
||||
cur1--;
|
||||
cur2--;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static bool
|
||||
xy_str_start_with (const char *str, const char *prefix)
|
||||
{
|
||||
if (NULL==str || NULL==prefix)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
size_t len1 = strlen (str);
|
||||
size_t len2 = strlen (prefix);
|
||||
|
||||
if (0 == len2)
|
||||
return true; // 空字符串直接返回
|
||||
if (len1 < len2)
|
||||
return false;
|
||||
|
||||
const char *cur1 = str;
|
||||
const char *cur2 = prefix;
|
||||
|
||||
for (int i = 0; i < len2; i++)
|
||||
{
|
||||
if (*cur1 != *cur2)
|
||||
return false;
|
||||
cur1++;
|
||||
cur2++;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_str_delete_prefix (const char *str, const char *prefix)
|
||||
{
|
||||
char *new = xy_strdup (str);
|
||||
bool yes = xy_str_start_with (str, prefix);
|
||||
if (!yes)
|
||||
return new;
|
||||
|
||||
size_t len = strlen (prefix);
|
||||
char *cur = new + len;
|
||||
return cur;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_str_delete_suffix (const char *str, const char *suffix)
|
||||
{
|
||||
char *new = xy_strdup (str);
|
||||
bool yes = xy_str_end_with (str, suffix);
|
||||
if (!yes)
|
||||
return new;
|
||||
|
||||
size_t len1 = strlen (str);
|
||||
size_t len2 = strlen (suffix);
|
||||
char *cur = new + len1 - len2;
|
||||
*cur = '\0';
|
||||
return new;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_str_strip (const char *str)
|
||||
{
|
||||
char *new = xy_strdup (str);
|
||||
|
||||
while (strchr ("\n\r\v\t\f ", new[0]))
|
||||
{
|
||||
new += 1;
|
||||
}
|
||||
|
||||
size_t len = strlen (new);
|
||||
|
||||
char *last = new + len - 1;
|
||||
|
||||
while (strchr ("\n\r\v\t\f ", *last))
|
||||
{
|
||||
*last = '\0';
|
||||
last -= 1;
|
||||
}
|
||||
return new;
|
||||
}
|
||||
|
||||
|
||||
/******************************************************
|
||||
* Logging
|
||||
******************************************************/
|
||||
void xy_log (const char *prompt, const char *str);
|
||||
void xy_succ(const char *prompt, const char *str);
|
||||
void xy_info(const char *prompt, const char *str);
|
||||
void xy_warn(const char *prompt, const char *str);
|
||||
void xy_error(const char *prompt, const char *str);
|
||||
|
||||
#define _XY_Log_Plain 000000001
|
||||
#define _XY_Log_Success 000000001 << 1
|
||||
#define _XY_Log_Info 000000001 << 2
|
||||
#define _XY_Log_Warn 000000001 << 3
|
||||
#define _XY_Log_Error 000000001 << 4
|
||||
void xy_log_brkt (const char *prompt1, const char *prompt2, const char *content);
|
||||
void xy_succ_brkt(const char *prompt1, const char *prompt2, const char *content);
|
||||
void xy_info_brkt(const char *prompt1, const char *prompt2, const char *content);
|
||||
void xy_warn_brkt(const char *prompt1, const char *prompt2, const char *content);
|
||||
void xy_error_brkt(const char *prompt1,const char *prompt2, const char *content);
|
||||
|
||||
#define xy_log(prompt, str) _xy_log (_XY_Log_Plain, prompt, str)
|
||||
#define xy_succ(prompt,str) _xy_log (_XY_Log_Success, prompt, str)
|
||||
#define xy_info(prompt,str) _xy_log (_XY_Log_Info, prompt, str)
|
||||
#define xy_warn(prompt,str) _xy_log (_XY_Log_Warn, prompt, str)
|
||||
#define xy_error(prompt,str) _xy_log (_XY_Log_Error, prompt, str)
|
||||
|
||||
static void
|
||||
_xy_log (int level, const char *prompt, const char *content)
|
||||
{
|
||||
char *str = NULL;
|
||||
|
||||
bool to_stderr = false;
|
||||
|
||||
/**
|
||||
* 'app: content'
|
||||
*/
|
||||
if (level & _XY_Log_Plain)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", content);
|
||||
}
|
||||
else if (level & _XY_Log_Success)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", xy_str_to_green (content));
|
||||
}
|
||||
else if (level & _XY_Log_Info)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", xy_str_to_blue (content));
|
||||
}
|
||||
else if (level & _XY_Log_Warn)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", xy_str_to_yellow (content));
|
||||
to_stderr = true;
|
||||
}
|
||||
else if (level & _XY_Log_Error)
|
||||
{
|
||||
str = xy_strjoin (3, prompt, ": ", xy_str_to_red (content));
|
||||
to_stderr = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
// xy_assert ("CAN'T REACH!");
|
||||
}
|
||||
|
||||
if (to_stderr)
|
||||
{
|
||||
fprintf (stderr, "%s\n", str);
|
||||
}
|
||||
else
|
||||
{
|
||||
puts (str);
|
||||
}
|
||||
free (str);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* brkt 系列输出受 pip 启发，为了输出方便，使用 xy.h 的程序应该
|
||||
*
|
||||
* 1.若想完全自定义颜色和输出位置：
|
||||
*
|
||||
* 应基于下述 xy_log_brkt_to 定义自己的输出函数
|
||||
*
|
||||
* 2.若想自定义颜色，而保持输出到stdout：
|
||||
*
|
||||
* 应基于下述 xy_log_brkt 定义
|
||||
*
|
||||
* 3.若对log无细致要求，想要快速使用log功能：
|
||||
*
|
||||
* 应基于下述 xy_<level>_brkt 定义自己的 app_<level>_brkt()，或直接使用 xy_<level>_brkt
|
||||
*/
|
||||
|
||||
static void
|
||||
xy_log_brkt_to (const char *prompt, const char *content, FILE *stream)
|
||||
{
|
||||
char *str = xy_strjoin (4, "[", prompt, "] ", content);
|
||||
fprintf (stream, "%s\n", str);
|
||||
free (str);
|
||||
}
|
||||
|
||||
#define xy_log_brkt(prompt1,prompt2,content) _xy_log_brkt(_XY_Log_Plain, prompt1,prompt2,content)
|
||||
#define xy_succ_brkt(prompt1,prompt2,content) _xy_log_brkt(_XY_Log_Success,prompt1,prompt2,content)
|
||||
#define xy_info_brkt(prompt1,prompt2,content) _xy_log_brkt(_XY_Log_Info, prompt1,prompt2,content)
|
||||
#define xy_warn_brkt(prompt1,prompt2,content) _xy_log_brkt(_XY_Log_Warn, prompt1,prompt2,content)
|
||||
#define xy_error_brkt(prompt1,prompt2,content) _xy_log_brkt(_XY_Log_Error, prompt1,prompt2,content)
|
||||
|
||||
static void
|
||||
_xy_log_brkt (int level, const char *prompt1, const char *prompt2, const char *content)
|
||||
{
|
||||
char *str = NULL;
|
||||
|
||||
bool to_stderr = false;
|
||||
|
||||
if (level & _XY_Log_Plain)
|
||||
{
|
||||
str = xy_strjoin (6, "[", prompt1, " ", prompt2, "] ", content);
|
||||
}
|
||||
else if (level & _XY_Log_Success)
|
||||
{
|
||||
/* [app 成功] [app success] */
|
||||
str = xy_strjoin (6,
|
||||
"[", xy_str_to_green (prompt1), " ", xy_str_to_bold (xy_str_to_green (prompt2)), "] ", xy_str_to_green (content));
|
||||
}
|
||||
else if (level & _XY_Log_Info)
|
||||
{
|
||||
/* [app 信息] [app info]
|
||||
[app 提示] [app notice]
|
||||
*/
|
||||
str = xy_strjoin (6,
|
||||
"[", xy_str_to_blue (prompt1), " ", xy_str_to_bold (xy_str_to_blue (prompt2)), "] ", xy_str_to_blue (content));
|
||||
}
|
||||
else if (level & _XY_Log_Warn)
|
||||
{
|
||||
/* [app 警告] [app warn] */
|
||||
str = xy_strjoin (6,
|
||||
"[", xy_str_to_yellow (prompt1), " ", xy_str_to_bold (xy_str_to_yellow (prompt2)), "] ", xy_str_to_yellow (content));
|
||||
to_stderr = true;
|
||||
}
|
||||
else if (level & _XY_Log_Error)
|
||||
{
|
||||
/* [app 错误] [app error] */
|
||||
str = xy_strjoin (6,
|
||||
"[", xy_str_to_red (prompt1), " ", xy_str_to_bold (xy_str_to_red (prompt2)), "] ", xy_str_to_red (content));
|
||||
to_stderr = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
// xy_assert ("CAN'T REACH!");
|
||||
}
|
||||
|
||||
if (to_stderr)
|
||||
{
|
||||
fprintf (stderr, "%s\n", str);
|
||||
}
|
||||
else
|
||||
{
|
||||
puts (str);
|
||||
}
|
||||
free (str);
|
||||
}
|
||||
void xy_log_brkt_to (const char *prompt, const char *content, FILE *stream);
|
||||
|
||||
|
||||
|
||||
/******************************************************
|
||||
* System
|
||||
******************************************************/
|
||||
/**
|
||||
* 执行cmd，返回某行输出结果，并对已经遍历过的行执行iter_func
|
||||
*
|
||||
* @param cmd 要执行的命令
|
||||
* @param n 指定命令执行输出的结果行中的某一行，0 表示最后一行，n (n>0) 表示第n行
|
||||
* 该函数会返回这一行的内容
|
||||
* @param iter_func 对遍历时经过的行的内容，进行函数调用
|
||||
*
|
||||
* @note 返回的字符串最后面一般有换行符号
|
||||
*
|
||||
* 由于目标行会被返回出来，所以 iter_func() 并不执行目标行，只会执行遍历过的行
|
||||
*/
|
||||
static char *
|
||||
xy_run_iter (const char *cmd, unsigned long n, void (*iter_func) (const char *))
|
||||
{
|
||||
const int size = 512;
|
||||
char *buf = (char *) malloc (size);
|
||||
char * xy_run_iter (const char *cmd, unsigned long n, void (*iter_func) (const char *));
|
||||
char * xy_run (const char *cmd, unsigned long n);
|
||||
|
||||
FILE *stream = popen (cmd, "r");
|
||||
if (stream == NULL)
|
||||
{
|
||||
fprintf (stderr, "xy: 命令执行失败\n");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char *ret = NULL;
|
||||
unsigned long count = 0;
|
||||
|
||||
while (true)
|
||||
{
|
||||
if (NULL == fgets (buf, size, stream))
|
||||
break;
|
||||
/* 存在换行的总是会把换行符读出来，删掉 */
|
||||
ret = xy_str_delete_suffix (buf, "\n");
|
||||
count += 1;
|
||||
if (n == count)
|
||||
break;
|
||||
if (iter_func)
|
||||
{
|
||||
iter_func (buf);
|
||||
}
|
||||
}
|
||||
|
||||
pclose (stream);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_run (const char *cmd, unsigned long n)
|
||||
{
|
||||
return xy_run_iter (cmd, n, NULL);
|
||||
}
|
||||
bool xy_file_exist (const char *path);
|
||||
bool xy_dir_exist (const char *path);
|
||||
char * xy_normalize_path (const char *path);
|
||||
char * xy_parent_dir (const char *path);
|
||||
|
||||
|
||||
#define xy_os_home _xy_os_home ()
|
||||
static char *
|
||||
_xy_os_home ()
|
||||
{
|
||||
char *home = NULL;
|
||||
if (xy_on_windows)
|
||||
home = getenv ("USERPROFILE");
|
||||
else
|
||||
home = getenv ("HOME");
|
||||
return home;
|
||||
}
|
||||
|
||||
#define xy_win_powershell_profile _xy_win_powershell_profile ()
|
||||
#define xy_win_powershellv5_profile _xy_win_powershellv5_profile ()
|
||||
static char *
|
||||
_xy_win_powershell_profile ()
|
||||
{
|
||||
return xy_2strjoin (
|
||||
xy_os_home, "\\Documents\\PowerShell\\Microsoft.PowerShell_profile.ps1");
|
||||
}
|
||||
|
||||
char *
|
||||
_xy_win_powershellv5_profile ()
|
||||
{
|
||||
return xy_2strjoin (
|
||||
xy_os_home,
|
||||
"\\Documents\\WindowsPowerShell\\Microsoft.PowerShell_profile.ps1");
|
||||
}
|
||||
|
||||
/******************************************************
|
||||
* OS specific
|
||||
******************************************************/
|
||||
#define xy_zshrc "~/.zshrc"
|
||||
#define xy_bashrc "~/.bashrc"
|
||||
#define xy_fishrc "~/.config/fish/config.fish"
|
||||
|
||||
/**
|
||||
* @note Windows上，`path` 不要夹带变量名，因为最终 access() 不会帮你转换
|
||||
*/
|
||||
static bool
|
||||
xy_file_exist (const char *path)
|
||||
{
|
||||
const char *new_path = path;
|
||||
if (xy_str_start_with (path, "~"))
|
||||
{
|
||||
new_path = xy_2strjoin (xy_os_home, path + 1);
|
||||
}
|
||||
// 0 即 F_OK
|
||||
return (0==access (new_path, 0)) ? true : false;
|
||||
}
|
||||
char * xy_os_home ();
|
||||
char * xy_win_powershell_profile ();
|
||||
char * xy_win_powershellv5_profile ();
|
||||
|
||||
/**
|
||||
* @note xy_file_exist() 和 xy_dir_exist() 两个函数在所有平台默认都支持使用 '~'，
|
||||
* 但实现中都没有调用 xy_normalize_path()，以防万一，调用前可能需要用户手动调用它
|
||||
*/
|
||||
static bool
|
||||
xy_dir_exist (const char *path)
|
||||
{
|
||||
const char *dir = path;
|
||||
if (xy_on_windows)
|
||||
{
|
||||
if (xy_str_start_with (path, "~"))
|
||||
{
|
||||
dir = xy_2strjoin (xy_os_home, path + 1);
|
||||
}
|
||||
}
|
||||
|
||||
if (xy_on_windows)
|
||||
{
|
||||
#ifdef XY_On_Windows
|
||||
// 也可以用 opendir() #include <dirent.h>
|
||||
DWORD attr = GetFileAttributesA (dir);
|
||||
|
||||
if (attr == INVALID_FILE_ATTRIBUTES)
|
||||
{
|
||||
// Q: 我们应该报错吗？
|
||||
return false;
|
||||
}
|
||||
else if (attr & FILE_ATTRIBUTE_DIRECTORY)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
return false;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
else
|
||||
{
|
||||
int status = system (xy_2strjoin ("test -d ", dir));
|
||||
|
||||
if (0==status)
|
||||
return true;
|
||||
else
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 1. 删除路径左右两边多出来的空白符
|
||||
* 2. 将 ~/ 转换为绝对路径
|
||||
*/
|
||||
static char *
|
||||
xy_normalize_path (const char *path)
|
||||
{
|
||||
char *new = xy_str_strip (path); // 防止开发者多写了空白符
|
||||
|
||||
if (xy_on_windows)
|
||||
{
|
||||
if (xy_str_start_with (new, "~/"))
|
||||
{
|
||||
// 或 %USERPROFILE%
|
||||
new = xy_strjoin (3, xy_os_home, "\\",
|
||||
xy_str_delete_prefix (new, "~/"));
|
||||
}
|
||||
new = xy_str_gsub (new, "/", "\\");
|
||||
}
|
||||
else
|
||||
{
|
||||
if (xy_str_start_with (new, "~/"))
|
||||
{
|
||||
new = xy_strjoin (3, xy_os_home, "/",
|
||||
xy_str_delete_prefix (new, "~/"));
|
||||
}
|
||||
}
|
||||
|
||||
return new;
|
||||
}
|
||||
|
||||
static char *
|
||||
xy_parent_dir (const char *path)
|
||||
{
|
||||
char *dir = xy_normalize_path (path);
|
||||
char *last = NULL;
|
||||
if (xy_on_windows)
|
||||
{
|
||||
last = strrchr (dir, '\\');
|
||||
if (!last)
|
||||
{
|
||||
/* current dir */
|
||||
return ".";
|
||||
}
|
||||
*last = '\0';
|
||||
}
|
||||
else
|
||||
{
|
||||
last = strrchr (dir, '/');
|
||||
if (!last)
|
||||
{
|
||||
/* current dir */
|
||||
return ".";
|
||||
}
|
||||
*last = '\0';
|
||||
}
|
||||
return dir;
|
||||
}
|
||||
|
||||
#endif
|
||||
|
|
|
|||
Loading…
Reference in New Issue