refactor: core

src/common/utils/helper.ts

@@ -18,3 +18,20 @@ export function isNull(value: any) {
 export function isNumeric(str: string) {
   return /^\d+$/.test(str);
 }
+
+export function truncateString(obj: any, maxLength = 500) {
+  if (obj !== null && typeof obj === 'object') {
+    Object.keys(obj).forEach(key => {
+      if (typeof obj[key] === 'string') {
+        // 如果是字符串且超过指定长度，则截断
+        if (obj[key].length > maxLength) {
+          obj[key] = obj[key].substring(0, maxLength) + '...';
+        }
+      } else if (typeof obj[key] === 'object') {
+        // 如果是对象或数组，则递归调用
+        truncateString(obj[key], maxLength);
+      }
+    });
+  }
+  return obj;
+}

src/common/utils/log.ts

@@ -1,4 +1,85 @@
+import log4js from 'log4js';
+import { truncateString } from '@/common/utils/helper';
+import path from 'node:path';
+
+export enum LogLevel {
+  DEBUG = 'debug',
+  INFO = 'info',
+  WARN = 'warn',
+  ERROR = 'error',
+  FATAL = 'fatal',
+}
+
+const logPath = path.join(path.resolve(__dirname), 'logs', 'napcat.log');
+
+function genLogConfig(fileLogLevel: LogLevel, consoleLogLevel: LogLevel) {
+  return {
+    appenders: {
+      fileAppender: { // 输出到文件的appender
+        type: 'file',
+        filename: logPath, // 指定日志文件的位置和文件名
+        maxLogSize: 10485760, // 日志文件的最大大小（单位：字节），这里设置为10MB
+      },
+      consoleAppender: { // 输出到控制台的appender
+        type: 'console'
+      }
+    },
+    categories: {
+      default: { appenders: ['fileAppender', 'consoleAppender'], level: 'info' }, // 默认情况下同时输出到文件和控制台
+      file: { appenders: ['fileAppender'], level: fileLogLevel },
+      console: { appenders: ['consoleAppender'], level: fileLogLevel }
+    }
+  };
+}
+
+
+export function setLogLevel(fileLogLevel: LogLevel, consoleLogLevel: LogLevel) {
+  log4js.configure(genLogConfig(fileLogLevel, consoleLogLevel));
+}
+
+setLogLevel(LogLevel.DEBUG, LogLevel.INFO);
+
+let fileLogEnabled = true;
+let consoleLogEnabled = true;
+export function enableFileLog(enable: boolean) {
+  fileLogEnabled = enable;
+}
+export function enableConsoleLog(enable: boolean) {
+  consoleLogEnabled = enable;
+}
+
+function formatMsg(msg: any[]){
+  let logMsg = '';
+  for (const msgItem of msg) {
+    // 判断是否是对象
+    if (typeof msgItem === 'object') {
+      const obj = JSON.parse(JSON.stringify(msgItem, null, 2));
+      logMsg += JSON.stringify(truncateString(obj)) + ' ';
+      continue;
+    }
+    logMsg += msgItem + ' ';
+  }
+  return logMsg;
+}
+
+function _log(level: LogLevel, ...args: any[]){
+  if (consoleLogEnabled){
+    log4js.getLogger('console')[level](formatMsg(args));
+  }
+  if (fileLogEnabled){
+    log4js.getLogger('file')[level](formatMsg(args));
+  }
+}
 
 export function log(...args: any[]) {
-  console.log(...args);
-}
+  // info 等级
+  _log(LogLevel.INFO, ...args);
+}
+
+export function logDebug(...args: any[]) {
+  _log(LogLevel.DEBUG, ...args);
+}
+
+export function logError(...args: any[]) {
+  _log(LogLevel.ERROR, ...args);
+}