scriptcat-mv3/src/app/logger/logger.ts

import dayjs from "dayjs";
import LoggerCore, { LogLabel, LogLevel } from "./core";

const levelNumber = {
  trace: 10,
  debug: 100,
  info: 1000,
  warn: 10000,
  error: 100000,
};

function buildLabel(...label: LogLabel[][]): LogLabel {
  const ret: LogLabel = {};
  label.forEach((item) => {
    item.forEach((item2) => {
      Object.keys(item2).forEach((key) => {
        ret[key] = item2[key];
      });
    });
  });
  return ret;
}

export default class Logger {
  core: LoggerCore;

  label: LogLabel[];

  constructor(core: LoggerCore, ...label: LogLabel[]) {
    this.core = core;
    this.label = label;
  }

  log(level: LogLevel, message: string, ...label: LogLabel[]) {
    if (levelNumber[level] >= levelNumber[this.core.level]) {
      this.core.writer.write(level, message, buildLabel(this.label, label));
    }
    if (this.core.debug !== "none" && levelNumber[level] >= levelNumber[this.core.debug]) {
      if (typeof message === "object") {
        message = JSON.stringify(message);
      }
      const msg = `${dayjs(new Date()).format("YYYY-MM-DD HH:mm:ss")} [${level}] msg=${message} label=${JSON.stringify(
        buildLabel(this.label, label)
      )}`;
      switch (level) {
        case "error":
          console.error(msg);
          break;
        case "warn":
          console.warn(msg);
          break;
        case "trace":
          console.info(msg);
          break;
        default:
          console.info(msg);
          break;
      }
    }
  }

  with(...label: LogLabel[]) {
    return new Logger(this.core, ...this.label, ...label);
  }

  trace(message: string, ...label: LogLabel[]) {
    this.log("trace", message, ...label);
  }

  debug(message: string, ...label: LogLabel[]) {
    this.log("debug", message, ...label);
  }

  info(message: string, ...label: LogLabel[]) {
    this.log("info", message, ...label);
  }

  warn(message: string, ...label: LogLabel[]) {
    this.log("warn", message, ...label);
  }

  error(message: string, ...label: LogLabel[]) {
    this.log("error", message, ...label);
  }

  static E(e: unknown): LogLabel {
    if (typeof e === "string") {
      return { error: e };
    }
    if (e instanceof Error) {
      return { error: e.message };
    }
    if (typeof e === "object") {
      return e as never;
    }
    return {};
  }
}