76 lines
2.2 KiB
JavaScript
76 lines
2.2 KiB
JavaScript
|
/*
|
|||
|
|
* @FilePath : /public/js/zip.js
|
|||
|
|
* @Description :
|
|||
|
|
*/
|
|||
|
|
/*
|
|||
|
|
* @Author: suhang suhang_max@163.com
|
|||
|
|
* @Date: 2023-06-01 00:14:29
|
|||
|
|
* @LastEditors: suhang suhang_max@163.com
|
|||
|
|
* @LastEditTime: 2023-06-01 10:56:19
|
|||
|
|
* @FilePath: \train-assessd:\Code\basic-frontend\zip.js
|
|||
|
|
* @Description: 将打包后的文件夹自动压缩成zip
|
|||
|
|
*/
|
|||
|
|
// console.log(import.meta, 'import.meta.env.VITE_APP_GISURL')
|
|||
|
|
import path from 'path'
|
|||
|
|
import fs from 'fs'
|
|||
|
|
import JSZip from 'jszip'
|
|||
|
|
const plugin = (fileName = 'pms-front', output) => {
|
|||
|
|
// if (!output) output = path.resolve(__dirname, `../..${import.meta.env.VITE_BASE_URL}`)
|
|||
|
|
output = path.resolve(__dirname, `../..${output}`)
|
|||
|
|
fileName += '.zip'
|
|||
|
|
const makeZip = () => {
|
|||
|
|
const zip = new JSZip()
|
|||
|
|
const distPath = path.resolve(output)
|
|||
|
|
const readDir = (zip, dirPath) => {
|
|||
|
|
// 读取dist下的根文件目录
|
|||
|
|
const files = fs.readdirSync(dirPath)
|
|||
|
|
files.forEach(fileName => {
|
|||
|
|
const fillPath = path.join(dirPath, './', fileName)
|
|||
|
|
const file = fs.statSync(fillPath)
|
|||
|
|
// 如果是文件夹的话需要递归遍历下面的子文件
|
|||
|
|
if (file.isDirectory()) {
|
|||
|
|
const dirZip = zip.folder(fileName)
|
|||
|
|
readDir(dirZip, fillPath)
|
|||
|
|
} else {
|
|||
|
|
// 读取每个文件为buffer存到zip中
|
|||
|
|
zip.file(fileName, fs.readFileSync(fillPath))
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
const removeExistedZip = () => {
|
|||
|
|
const dest = path.join(distPath, './' + fileName)
|
|||
|
|
if (fs.existsSync(dest)) {
|
|||
|
|
fs.unlinkSync(dest)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
const zipDir = () => {
|
|||
|
|
readDir(zip, distPath)
|
|||
|
|
zip
|
|||
|
|
.generateAsync({
|
|||
|
|
type: 'nodebuffer', // 压缩类型
|
|||
|
|
compression: 'DEFLATE', // 压缩算法
|
|||
|
|
compressionOptions: {
|
|||
|
|
// 压缩级别
|
|||
|
|
level: 9,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
.then(content => {
|
|||
|
|
const dest = path.join(distPath, '../' + fileName)
|
|||
|
|
removeExistedZip()
|
|||
|
|
// 把zip包写到硬盘中，这个content现在是一段buffer
|
|||
|
|
fs.writeFileSync(dest, content)
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
removeExistedZip()
|
|||
|
|
zipDir(distPath)
|
|||
|
|
}
|
|||
|
|
return {
|
|||
|
|
name: 'vite-plugin-auto-zip',
|
|||
|
|
apply: 'build',
|
|||
|
|
closeBundle() {
|
|||
|
|
makeZip()
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
export default plugin
|