This commit is contained in:
Aoran Zeng 2024-04-18 22:12:22 +08:00
parent d7bc099788
commit 0ef468f524

269
xy.h
View File

@ -3,7 +3,7 @@
* License : MIT * License : MIT
* Authors : Aoran Zeng <ccmywish@qq.com> * Authors : Aoran Zeng <ccmywish@qq.com>
* Created on : <2023-08-28> * Created on : <2023-08-28>
* Last modified : <2023-09-29> * Last modified : <2024-04-18>
* *
* xy: * xy:
* *
@ -17,14 +17,14 @@
#ifndef XY_H #ifndef XY_H
#define XY_H #define XY_H
#include <stdio.h>
#include <stdarg.h>
#include <string.h>
#include <stdlib.h>
#include <stdbool.h>
#include <assert.h> #include <assert.h>
#include <stdarg.h>
#include <stdbool.h>
#include <stddef.h> #include <stddef.h>
#include <unistd.h> // For access() #include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
// #define NDEBUG // #define NDEBUG
@ -62,18 +62,39 @@
#define xy_useutf8() #define xy_useutf8()
#endif #endif
void
void putf(double n) {printf("%f\n", n);} putf (double n)
void puti(long long n) {printf("%lld\n", n);} {
void putb(bool n) {if(n) puts("true"); else puts("false");} printf ("%f\n", n);
void print(char* s) {printf("%s", s);} }
void println(char* s) {printf("%s\n", s);} void
puti (long long n)
{
printf ("%lld\n", n);
}
void
putb (bool n)
{
if (n)
puts ("true");
else
puts ("false");
}
void
print (char *s)
{
printf ("%s", s);
}
void
println (char *s)
{
printf ("%s\n", s);
}
#define xy_arylen(x) (sizeof (x) / sizeof (x[0])) #define xy_arylen(x) (sizeof (x) / sizeof (x[0]))
#define assert_str(a, b) assert (xy_streql ((a), (b))) #define assert_str(a, b) assert (xy_streql ((a), (b)))
static inline void * static inline void *
xy_malloc0 (size_t size) xy_malloc0 (size_t size)
{ {
@ -82,7 +103,6 @@ xy_malloc0 (size_t size)
return ptr; return ptr;
} }
#define XY_Log_Info 00001 #define XY_Log_Info 00001
#define XY_Log_Success 00001 << 1 #define XY_Log_Success 00001 << 1
#define XY_Log_Warn 00001 << 2 #define XY_Log_Warn 00001 << 2
@ -100,21 +120,26 @@ _xy_log (int level, const char* str)
bool to_stderr = false; bool to_stderr = false;
if (level & XY_Log_Info) { if (level & XY_Log_Info)
{
color_fmt_str = "\033[34m%s\033[0m"; // 蓝色 color_fmt_str = "\033[34m%s\033[0m"; // 蓝色
} }
else if (level & XY_Log_Success) { else if (level & XY_Log_Success)
{
color_fmt_str = "\033[32m%s\033[0m"; // 绿色 color_fmt_str = "\033[32m%s\033[0m"; // 绿色
} }
else if (level & XY_Log_Warn) { else if (level & XY_Log_Warn)
{
color_fmt_str = "\033[33m%s\033[0m\n"; // 黄色 color_fmt_str = "\033[33m%s\033[0m\n"; // 黄色
to_stderr = true; to_stderr = true;
} }
else if (level & XY_Log_Error) { else if (level & XY_Log_Error)
{
color_fmt_str = "\033[31m%s\033[0m\n"; // 红色 color_fmt_str = "\033[31m%s\033[0m\n"; // 红色
to_stderr = true; to_stderr = true;
} }
else { else
{
// xy_assert ("CAN'T REACH!"); // xy_assert ("CAN'T REACH!");
} }
@ -123,16 +148,17 @@ _xy_log (int level, const char* str)
char *buf = malloc (strlen (str) + len + 1); char *buf = malloc (strlen (str) + len + 1);
sprintf (buf, color_fmt_str, str); sprintf (buf, color_fmt_str, str);
if (to_stderr) { if (to_stderr)
{
fprintf (stderr, "%s", buf); fprintf (stderr, "%s", buf);
} else { }
else
{
puts (buf); puts (buf);
} }
free (buf); free (buf);
} }
/** /**
* str中所有的pat字符串替换成replace * str中所有的pat字符串替换成replace
*/ */
@ -143,19 +169,24 @@ xy_str_gsub (const char* str, const char* pat, const char* replace)
size_t pat_len = strlen (pat); size_t pat_len = strlen (pat);
int unit = replace_len - pat_len; int unit = replace_len - pat_len;
if (unit<=0) unit = 0; if (unit <= 0)
unit = 0;
size_t len = strlen (str); size_t len = strlen (str);
const char *cur = str; const char *cur = str;
int count = 0; int count = 0;
while (cur<str+len) { while (cur < str + len)
{
char *fnd = strstr (cur, pat); char *fnd = strstr (cur, pat);
if (fnd) { if (fnd)
{
count++; count++;
cur = fnd + pat_len; cur = fnd + pat_len;
} else break; }
else
break;
} }
// puti(count); DEBUG 匹配次数 // puti(count); DEBUG 匹配次数
@ -163,9 +194,11 @@ xy_str_gsub (const char* str, const char* pat, const char* replace)
char *retcur = ret; char *retcur = ret;
cur = str; cur = str;
while(cur<str+len) { while (cur < str + len)
{
char *fnd = strstr (cur, pat); char *fnd = strstr (cur, pat);
if (fnd) { if (fnd)
{
ptrdiff_t diff = fnd - cur; ptrdiff_t diff = fnd - cur;
strncpy (retcur, cur, diff); strncpy (retcur, cur, diff);
cur = fnd + pat_len; cur = fnd + pat_len;
@ -173,14 +206,15 @@ xy_str_gsub (const char* str, const char* pat, const char* replace)
retcur += diff; retcur += diff;
strcpy (retcur, replace); strcpy (retcur, replace);
retcur += replace_len; retcur += replace_len;
} else break; }
else
break;
} }
strcpy (retcur, cur); strcpy (retcur, cur);
return ret; return ret;
} }
static char * static char *
xy_2strjoin (const char *str1, const char *str2) xy_2strjoin (const char *str1, const char *str2)
{ {
@ -192,7 +226,6 @@ xy_2strjoin (const char* str1, const char* str2)
return ret; return ret;
} }
static char * static char *
xy_strjoin (unsigned int count, ...) xy_strjoin (unsigned int count, ...)
{ {
@ -219,18 +252,24 @@ xy_strjoin (unsigned int count, ...)
str = va_arg (args, const char *); str = va_arg (args, const char *);
al_need += strlen (str); al_need += strlen (str);
while (al_need > al_cur) { while (al_need > al_cur)
al_times += 1; al_cur = al_times * al_fixed; {
al_times += 1;
al_cur = al_times * al_fixed;
need_realloc = true; need_realloc = true;
} }
// printf("al_times %d, al_need %zd, al_cur %zd\n", al_times, al_need, al_cur); // printf("al_times %d, al_need %zd, al_cur %zd\n", al_times, al_need,
if (need_realloc) { // al_cur);
if (need_realloc)
{
ptrdiff_t diff = cur - ret; ptrdiff_t diff = cur - ret;
ret = realloc (ret, al_cur); ret = realloc (ret, al_cur);
cur = ret + diff; cur = ret + diff;
} }
if (NULL==ret) { if (NULL == ret)
xy_error ("xy: No availble memory!"); return NULL; {
xy_error ("xy: No availble memory!");
return NULL;
} }
strcpy (cur, str); strcpy (cur, str);
// puts(ret); // puts(ret);
@ -242,7 +281,6 @@ xy_strjoin (unsigned int count, ...)
return ret; return ret;
} }
static char * static char *
xy_strdup (const char *str) xy_strdup (const char *str)
{ {
@ -252,7 +290,6 @@ xy_strdup(const char* str)
return new; return new;
} }
#define XY_Str_Bold 1 #define XY_Str_Bold 1
#define XY_Str_Faint 2 #define XY_Str_Faint 2
#define XY_Str_Italic 3 #define XY_Str_Italic 3
@ -263,7 +300,8 @@ xy_strdup(const char* str)
#define xy_str_to_bold(str) _xy_str_to_terminal_style (XY_Str_Bold, str) #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_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_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_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_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_to_cross(str) _xy_str_to_terminal_style (XY_Str_Cross, str)
@ -286,54 +324,32 @@ static char*
_xy_str_to_terminal_style (int style, const char *str) _xy_str_to_terminal_style (int style, const char *str)
{ {
char *color_fmt_str = NULL; char *color_fmt_str = NULL;
if (XY_Str_Red==style)
{
color_fmt_str = "\e[31m%s\e[0m"; // 红色
}
else if (XY_Str_Green==style)
{
color_fmt_str = "\e[32m%s\e[0m"; // 绿色
}
else if (XY_Str_Yellow==style)
{
color_fmt_str = "\e[33m%s\e[0m"; // 黄色
}
else if (XY_Str_Blue==style)
{
color_fmt_str = "\e[34m%s\e[0m"; // 蓝色
}
else if (XY_Str_Magenta==style)
{
color_fmt_str = "\e[35m%s\e[0m"; // 蓝色
}
else if (XY_Str_Cyan==style)
{
color_fmt_str = "\e[36m%s\e[0m"; // 蓝色
}
switch (style)
else if (XY_Str_Bold==style)
{ {
case XY_Str_Red:
color_fmt_str = "\e[31m%s\e[0m";
case XY_Str_Green:
color_fmt_str = "\e[32m%s\e[0m";
case XY_Str_Yellow:
color_fmt_str = "\e[33m%s\e[0m";
case XY_Str_Blue:
color_fmt_str = "\e[34m%s\e[0m";
case XY_Str_Magenta:
color_fmt_str = "\e[35m%s\e[0m";
case XY_Str_Cyan:
color_fmt_str = "\e[36m%s\e[0m";
case XY_Str_Bold:
color_fmt_str = "\e[1m%s\e[0m"; color_fmt_str = "\e[1m%s\e[0m";
} case XY_Str_Faint:
else if (XY_Str_Faint==style)
{
color_fmt_str = "\e[2m%s\e[0m"; color_fmt_str = "\e[2m%s\e[0m";
} case XY_Str_Italic:
else if (XY_Str_Italic==style)
{
color_fmt_str = "\e[3m%s\e[0m"; color_fmt_str = "\e[3m%s\e[0m";
} case XY_Str_Underline:
else if (XY_Str_Underline==style)
{
color_fmt_str = "\e[4m%s\e[0m"; color_fmt_str = "\e[4m%s\e[0m";
} case XY_Str_Blink:
else if (XY_Str_Blink==style)
{
color_fmt_str = "\e[5m%s\e[0m"; color_fmt_str = "\e[5m%s\e[0m";
} case XY_Str_Cross:
else if (XY_Str_Cross==style)
{
color_fmt_str = "\e[9m%s\e[0m"; color_fmt_str = "\e[9m%s\e[0m";
} }
@ -344,13 +360,12 @@ _xy_str_to_terminal_style(int style, const char* str)
return buf; return buf;
} }
static bool static bool
xy_streql(const char* str1, const char* str2) { xy_streql (const char *str1, const char *str2)
{
return strcmp (str1, str2) == 0 ? true : false; return strcmp (str1, str2) == 0 ? true : false;
} }
static char * static char *
xy_str_to_quietcmd (const char *cmd) xy_str_to_quietcmd (const char *cmd)
{ {
@ -363,23 +378,26 @@ xy_str_to_quietcmd (const char* cmd)
return ret; return ret;
} }
static bool static bool
xy_str_end_with (const char *str, const char *suffix) xy_str_end_with (const char *str, const char *suffix)
{ {
size_t len1 = strlen (str); size_t len1 = strlen (str);
size_t len2 = strlen (suffix); size_t len2 = strlen (suffix);
if (0==len2) return true; // 空字符串直接返回 if (0 == len2)
if (len1 < len2) return false; return true; // 空字符串直接返回
if (len1 < len2)
return false;
const char *cur1 = str + len1 - 1; const char *cur1 = str + len1 - 1;
const char *cur2 = suffix + len2 - 1; const char *cur2 = suffix + len2 - 1;
for (int i = 0; i < len2; i++) for (int i = 0; i < len2; i++)
{ {
if (*cur1 != *cur2) return false; if (*cur1 != *cur2)
cur1--; cur2--; return false;
cur1--;
cur2--;
} }
return true; return true;
} }
@ -390,16 +408,20 @@ xy_str_start_with (const char* str, const char* prefix)
size_t len1 = strlen (str); size_t len1 = strlen (str);
size_t len2 = strlen (prefix); size_t len2 = strlen (prefix);
if (0==len2) return true; // 空字符串直接返回 if (0 == len2)
if (len1 < len2) return false; return true; // 空字符串直接返回
if (len1 < len2)
return false;
const char *cur1 = str; const char *cur1 = str;
const char *cur2 = prefix; const char *cur2 = prefix;
for (int i = 0; i < len2; i++) for (int i = 0; i < len2; i++)
{ {
if (*cur1 != *cur2) return false; if (*cur1 != *cur2)
cur1++; cur2++; return false;
cur1++;
cur2++;
} }
return true; return true;
} }
@ -409,20 +431,21 @@ xy_str_delete_prefix (const char* str, const char* prefix)
{ {
char *new = xy_strdup (str); char *new = xy_strdup (str);
bool yes = xy_str_start_with (str, prefix); bool yes = xy_str_start_with (str, prefix);
if (!yes) return new; if (!yes)
return new;
size_t len = strlen (prefix); size_t len = strlen (prefix);
char *cur = new + len; char *cur = new + len;
return cur; return cur;
} }
static char * static char *
xy_str_delete_suffix (const char *str, const char *suffix) xy_str_delete_suffix (const char *str, const char *suffix)
{ {
char *new = xy_strdup (str); char *new = xy_strdup (str);
bool yes = xy_str_end_with (str, suffix); bool yes = xy_str_end_with (str, suffix);
if (!yes) return new; if (!yes)
return new;
size_t len1 = strlen (str); size_t len1 = strlen (str);
size_t len2 = strlen (suffix); size_t len2 = strlen (suffix);
@ -431,7 +454,6 @@ xy_str_delete_suffix (const char* str, const char* suffix)
return new; return new;
} }
static char * static char *
xy_str_strip (const char *str) xy_str_strip (const char *str)
{ {
@ -454,7 +476,6 @@ xy_str_strip (const char* str)
return new; return new;
} }
/** /**
* cmd * cmd
* *
@ -471,7 +492,8 @@ xy_getcmd (const char* cmd, unsigned long n, void (*func)(const char*))
char *buf = (char *)malloc (size); char *buf = (char *)malloc (size);
FILE *stream = popen (cmd, "r"); FILE *stream = popen (cmd, "r");
if (stream == NULL) { if (stream == NULL)
{
fprintf (stderr, "xy: 命令执行失败\n"); fprintf (stderr, "xy: 命令执行失败\n");
return NULL; return NULL;
} }
@ -479,19 +501,24 @@ xy_getcmd (const char* cmd, unsigned long n, void (*func)(const char*))
char *ret = NULL; char *ret = NULL;
unsigned long count = 0; unsigned long count = 0;
while (true) { while (true)
if (NULL==fgets(buf, size, stream)) break; {
if (NULL == fgets (buf, size, stream))
break;
ret = buf; ret = buf;
count += 1; count += 1;
if (n==count) break; if (n == count)
if (func) { func(buf); } break;
if (func)
{
func (buf);
}
} }
pclose (stream); pclose (stream);
return ret; return ret;
} }
#define xy_os_home _xy_os_home () #define xy_os_home _xy_os_home ()
static char * static char *
_xy_os_home () _xy_os_home ()
@ -504,22 +531,23 @@ _xy_os_home ()
return home; return home;
} }
#define xy_win_powershell_profile _xy_win_powershell_profile () #define xy_win_powershell_profile _xy_win_powershell_profile ()
#define xy_win_powershellv5_profile _xy_win_powershellv5_profile () #define xy_win_powershellv5_profile _xy_win_powershellv5_profile ()
static char * static char *
_xy_win_powershell_profile () _xy_win_powershell_profile ()
{ {
return xy_2strjoin(xy_os_home, "\\Documents\\PowerShell\\Microsoft.PowerShell_profile.ps1"); return xy_2strjoin (
xy_os_home, "\\Documents\\PowerShell\\Microsoft.PowerShell_profile.ps1");
} }
char * char *
_xy_win_powershellv5_profile () _xy_win_powershellv5_profile ()
{ {
return xy_2strjoin(xy_os_home, "\\Documents\\WindowsPowerShell\\Microsoft.PowerShell_profile.ps1"); return xy_2strjoin (
xy_os_home,
"\\Documents\\WindowsPowerShell\\Microsoft.PowerShell_profile.ps1");
} }
/** /**
* @note Windows上`path` access() * @note Windows上`path` access()
*/ */
@ -529,14 +557,14 @@ xy_file_exist (const char* path)
const char *newpath = path; const char *newpath = path;
if (xy_on_windows) if (xy_on_windows)
{ {
if (xy_str_start_with(path, "~")) { if (xy_str_start_with (path, "~"))
{
newpath = xy_2strjoin (xy_os_home, path + 1); newpath = xy_2strjoin (xy_os_home, path + 1);
} }
} }
return access (newpath, 0) ? false : true; return access (newpath, 0) ? false : true;
} }
/** /**
* 使Windows下也可调用 * 使Windows下也可调用
*/ */
@ -546,10 +574,13 @@ xy_uniform_path (const char* path)
char *new = xy_str_strip (path); // 防止开发者多写了空白符 char *new = xy_str_strip (path); // 防止开发者多写了空白符
// 这个函数仅在Windows上才进行替换 // 这个函数仅在Windows上才进行替换
if (xy_on_windows) { if (xy_on_windows)
if (xy_str_start_with(new, "~/")){ {
if (xy_str_start_with (new, "~/"))
{
// 或 %USERPROFILE% // 或 %USERPROFILE%
new = xy_strjoin(3, xy_os_home, "\\", xy_str_delete_prefix(new, "~/")); new = xy_strjoin (3, xy_os_home, "\\",
xy_str_delete_prefix (new, "~/"));
} }
new = xy_str_gsub (new, "/", "\\"); new = xy_str_gsub (new, "/", "\\");
} }
@ -557,16 +588,18 @@ xy_uniform_path (const char* path)
return new; return new;
} }
static char * static char *
xy_parent_dir (const char *path) xy_parent_dir (const char *path)
{ {
char *dir = xy_uniform_path (path); char *dir = xy_uniform_path (path);
char *last = NULL; char *last = NULL;
if (xy_on_windows) { if (xy_on_windows)
{
last = strrchr (dir, '\\'); last = strrchr (dir, '\\');
*last = '\0'; *last = '\0';
} else { }
else
{
last = strrchr (dir, '/'); last = strrchr (dir, '/');
*last = '\0'; *last = '\0';
} }