mirror of
https://github.com/RubyMetric/chsrc.git
synced 2024-11-27 11:43:40 +08:00
129 lines
2.6 KiB
C
129 lines
2.6 KiB
C
/* --------------------------------------------------------------
|
||
* File : helper.h
|
||
* Authors : Aoran Zeng <ccmywish@qq.com>
|
||
* Created on : <2023-08-28>
|
||
* Last modified : <2023-08-30>
|
||
*
|
||
* helper:
|
||
*
|
||
* helper functions and macros
|
||
* -------------------------------------------------------------*/
|
||
|
||
#ifndef XY_H
|
||
#define XY_H
|
||
|
||
#include <string.h>
|
||
#include <stdlib.h>
|
||
#include <stdbool.h>
|
||
#include <assert.h>
|
||
|
||
// #define NDEBUG
|
||
|
||
#ifdef _WIN32
|
||
#include <windows.h>
|
||
#define xy_useutf8() SetConsoleOutputCP(65001)
|
||
#else
|
||
#define xy_useutf8()
|
||
#endif
|
||
|
||
#define Array_Size(x) (sizeof(x) / sizeof(x[0]))
|
||
|
||
|
||
static inline void*
|
||
xy_malloc0 (size_t size)
|
||
{
|
||
void* ptr = malloc(size);
|
||
memset(ptr, 0, size);
|
||
return ptr;
|
||
}
|
||
|
||
|
||
static char*
|
||
xy_strjoin (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;
|
||
}
|
||
|
||
|
||
#define XY_INFO 1
|
||
#define XY_SUCCESS 1<<1
|
||
#define XY_WARN 1<<2
|
||
#define XY_ERROR 1<<3
|
||
|
||
static void
|
||
xy_log (int level, const char* str)
|
||
{
|
||
char* color_fmt_str = NULL;
|
||
|
||
bool to_stderr = false;
|
||
|
||
if (level & XY_INFO) {
|
||
color_fmt_str = "\033[34m%s\033[0m"; // 蓝色
|
||
}
|
||
else if (level & XY_SUCCESS) {
|
||
color_fmt_str = "\033[32m%s\033[0m"; // 绿色
|
||
}
|
||
else if (level & XY_WARN) {
|
||
color_fmt_str = "\033[33m%s\033[0m\n"; // 黄色
|
||
to_stderr = true;
|
||
}
|
||
else if (level & XY_ERROR) {
|
||
color_fmt_str = "\033[31m%s\033[0m\n"; // 红色
|
||
to_stderr = true;
|
||
}
|
||
else {
|
||
//xy_assert ("CAN'T REACH!");
|
||
}
|
||
|
||
// -2 把中间%s减掉，-1 把末尾nul减掉
|
||
size_t len = sizeof(color_fmt_str) -2 -1;
|
||
char* buf = malloc(strlen(str) + len + 1);
|
||
|
||
sprintf (buf, color_fmt_str, str);
|
||
if (to_stderr) {
|
||
fprintf(stderr, buf);
|
||
} else {
|
||
puts(buf);
|
||
}
|
||
free(buf);
|
||
}
|
||
|
||
#define xy_info(str) xy_log (XY_INFO, str)
|
||
#define xy_error(str) xy_log (XY_ERROR, str)
|
||
|
||
|
||
static void
|
||
xy_success (const char* str1)
|
||
{
|
||
char color_fmt_str[] = "\033[32m%s\033[0m";
|
||
// -2 把中间%s减掉，-1 把末尾nul减掉
|
||
size_t len = sizeof(color_fmt_str) -2 -1;
|
||
char* buf = malloc(strlen(str1) + len + 1);
|
||
|
||
sprintf (buf, color_fmt_str, str1);
|
||
puts(buf);
|
||
free(buf);
|
||
}
|
||
|
||
|
||
static void
|
||
xy_warn (const char* str1)
|
||
{
|
||
// 注意，我们这里相比于xy_suceess()多了一个换行符
|
||
char color_fmt_str[] = "\033[33m%s\033[0m\n";
|
||
// -2 把中间%s减掉，-1 把末尾nul减掉
|
||
size_t len = sizeof(color_fmt_str) -2 -1;
|
||
char* buf = malloc(strlen(str1) + len + 1);
|
||
|
||
sprintf (buf, color_fmt_str, str1);
|
||
fprintf(stderr, buf);
|
||
free(buf);
|
||
}
|
||
|
||
#endif
|