Incidents-Data-Portal/frontend/plugins/vite-build-info.ts

import type { Dayjs } from 'dayjs'
import type { Plugin, ResolvedConfig } from 'vite'
import { readdir, stat } from 'node:fs'
import dayjs from 'dayjs'
import duration from 'dayjs/plugin/duration'
import pkg from 'picocolors'

const { green, blue, bold } = pkg
dayjs.extend(duration)

const fileListTotal: number[] = []

// eslint-disable-next-line ts/no-unsafe-function-type
function recursiveDirectory(folder: string, callback: Function): void {
  readdir(folder, (err, files: string[]) => {
    if (err)
      throw err
    let count = 0
    const checkEnd = () => {
      ++count === files.length && callback()
    }
    files.forEach((item: string) => {
      stat(`${folder}/${item}`, async (err, stats) => {
        if (err)
          throw err
        if (stats.isFile()) {
          fileListTotal.push(stats.size)
          checkEnd()
        }
        else if (stats.isDirectory()) {
          recursiveDirectory(`${folder}/${item}/`, checkEnd)
        }
      })
    })
    files.length === 0 && callback()
  })
}

function sum(arr: number[]) {
  return arr.reduce((t: number, c: number) => {
    return t + c
  }, 0)
}
function formatBytes(a: number, b?: number): string {
  if (a === 0)
    return '0 Bytes'
  const c = 1024
  const d = b || 2
  const e = ['Bytes', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB']
  const f = Math.floor(Math.log(a) / Math.log(c))
  return `${Number.parseFloat((a / c ** f).toFixed(d))} ${e[f]}`
}

export function viteBuildInfo(name: string): Plugin {
  let config: ResolvedConfig
  let startTime: Dayjs
  let endTime: Dayjs
  return {
    name: 'vite:buildInfo',
    configResolved(resolvedConfig) {
      config = resolvedConfig
    },
    buildStart() {
      console.log(
        bold(
          green(
            `👏欢迎使用${blue(`[${name}]`)}，现在正全力为您${config.command === 'build' ? '打包' : '编译'
            }`,
          ),
        ),
      )
      if (config.command === 'build')
        startTime = dayjs(new Date())
    },
    closeBundle() {
      if (config.command === 'build') {
        endTime = dayjs(new Date())
        recursiveDirectory(config.build.outDir, () => {
          console.log(
            bold(
              green(
                `恭喜打包完成🎉（总用时${dayjs
                  .duration(endTime.diff(startTime))
                  .format('mm分ss秒')}，打包后的大小为${formatBytes(
                  sum(fileListTotal),
                )}）`,
              ),
            ),
          )
        })
      }
    },
  }
}


// import type { Dayjs } from 'dayjs'
// import type { Plugin, ResolvedConfig } from 'vite'
// import { readdir, stat, existsSync } from 'node:fs' // 👈 新增 existsSync
// import { resolve } from 'node:path' // 👈 新增 resolve
// import dayjs from 'dayjs'
// import duration from 'dayjs/plugin/duration'
// import pkg from 'picocolors'

// const { green, blue, bold } = pkg
// dayjs.extend(duration)

// const fileListTotal: number[] = []

// function recursiveDirectory(folder: string, callback: () => void): void {
//   // 👇 先检查目录是否存在
//   if (!existsSync(folder)) {
//     console.warn(`[vite:buildInfo] 目录不存在: ${folder}`)
//     callback()
//     return
//   }

//   readdir(folder, (err, files: string[]) => {
//     if (err) {
//       console.error('[vite:buildInfo] 读取目录失败:', err)
//       callback()
//       return
//     }

//     let count = 0
//     const checkEnd = () => {
//       if (++count === files.length) callback()
//     }

//     if (files.length === 0) {
//       callback()
//       return
//     }

//     files.forEach((item: string) => {
//       const fullPath = resolve(folder, item) // 👈 使用 resolve 拼接路径
//       stat(fullPath, (err, stats) => {
//         if (err) {
//           checkEnd()
//           return
//         }
//         if (stats.isFile()) {
//           fileListTotal.push(stats.size)
//           checkEnd()
//         } else if (stats.isDirectory()) {
//           recursiveDirectory(fullPath, checkEnd)
//         }
//       })
//     })
//   })
// }

// function sum(arr: number[]) {
//   return arr.reduce((t, c) => t + c, 0)
// }

// function formatBytes(a: number, b = 2): string {
//   if (a === 0) return '0 Bytes'
//   const c = 1024
//   const e = ['Bytes', 'KB', 'MB', 'GB', 'TB']
//   const f = Math.floor(Math.log(a) / Math.log(c))
//   return `${parseFloat((a / c ** f).toFixed(b))} ${e[f]}`
// }

// export function viteBuildInfo(name: string): Plugin {
//   let config: ResolvedConfig
//   let startTime: Dayjs
//   let endTime: Dayjs

//   return {
//     name: 'vite:buildInfo',
//     configResolved(resolvedConfig) {
//       config = resolvedConfig
//     },
//     buildStart() {
//       console.log(
//         bold(
//           green(
//             `👏欢迎使用${blue(`[${name}]`)}，现在正全力为您${config.command === 'build' ? '打包' : '编译'}`,
//           ),
//         ),
//       )
//       if (config.command === 'build') {
//         startTime = dayjs()
//       }
//     },
//     closeBundle() {
//       if (config.command === 'build') {
//         endTime = dayjs()
//         // 👇 关键：将 outDir 转为绝对路径！
//         const outDir = resolve(config.root || process.cwd(), config.build.outDir)
        
//         recursiveDirectory(outDir, () => {
//           console.log(
//             bold(
//               green(
//                 `恭喜打包完成🎉（总用时${dayjs
//                   .duration(endTime.diff(startTime))
//                   .format('mm分ss秒')}，打包后的大小为${formatBytes(sum(fileListTotal))}）`,
//               ),
//             ),
//           )
//         })
//       }
//     },
//   }
// }