init

2026-07-10 18:55:55 +08:00
commit fce40c7d6c
317 changed files with 170079 additions and 0 deletions
@@ -0,0 +1,207 @@
+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))}）`,
+//               ),
+//             ),
+//           )
+//         })
+//       }
+//     },
+//   }
+// }