import { wrapperEnv } from "./utils";
import dayjs, { type Dayjs } from "dayjs";
import gradientString from "gradient-string";
import duration from "dayjs/plugin/duration";
import boxen, { type Options as BoxenOptions } from "boxen";

dayjs.extend(duration);

const boxenOptions: BoxenOptions = {
  padding: 0.94,
  borderColor: "cyan",
  borderStyle: "round",
  textAlignment: "left",
};

/* 输出日志信息 */
const printLogMessage = (VITE_PORT: number) => {
  return gradientString("cyan", "magenta").multiline(
    `保存成功！服务器重新启动
项目访问地址如下：
http://localhost:${VITE_PORT}`
  );
};

export const viteConsoleLog = (mode) => {
  const { VITE_PORT } = wrapperEnv(mode);

  let config: { command: string };
  let startTime: Dayjs;
  let endTime: Dayjs;
  return {
    name: "vite:buildInfo",
    configResolved(resolvedConfig) {
      config = resolvedConfig;
    },
    buildStart() {
      console.log(boxen(printLogMessage(VITE_PORT), boxenOptions));
      if (config.command === "build") {
        startTime = dayjs(new Date());
      }
    },
    closeBundle() {
      if (config.command === "build") {
        endTime = dayjs(new Date());
        const format = dayjs.duration(endTime.diff(startTime)).format("mm分ss秒");
        console.log(
          boxen(
            gradientString("cyan", "magenta").multiline(`🎉 恭喜打包完成（总用时${format}）`),
            boxenOptions
          )
        );
      }
    },
  };
};