Blame view

Yi.Vben5.Vue3/internal/vite-config/src/plugins/archiver.ts 1.85 KB
515fceeb   “wangming”   框架初始化
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
  import type { PluginOption } from 'vite';
  
  import type { ArchiverPluginOptions } from '../typing';
  
  import fs from 'node:fs';
  import fsp from 'node:fs/promises';
  import { join } from 'node:path';
  
  import archiver from 'archiver';
  
  export const viteArchiverPlugin = (
    options: ArchiverPluginOptions = {},
  ): PluginOption => {
    return {
      apply: 'build',
      closeBundle: {
        handler() {
          const { name = 'dist', outputDir = '.' } = options;
  
          setTimeout(async () => {
            const folderToZip = 'dist';
  
            const zipOutputDir = join(process.cwd(), outputDir);
            const zipOutputPath = join(zipOutputDir, `${name}.zip`);
            try {
              await fsp.mkdir(zipOutputDir, { recursive: true });
            } catch {
              // ignore
            }
  
            try {
              await zipFolder(folderToZip, zipOutputPath);
              console.log(`Folder has been zipped to: ${zipOutputPath}`);
            } catch (error) {
              console.error('Error zipping folder:', error);
            }
          }, 0);
        },
        order: 'post',
      },
      enforce: 'post',
      name: 'vite:archiver',
    };
  };
  
  async function zipFolder(
    folderPath: string,
    outputPath: string,
  ): Promise<void> {
    return new Promise((resolve, reject) => {
      const output = fs.createWriteStream(outputPath);
      const archive = archiver('zip', {
        zlib: { level: 9 }, // 设置压缩级别为 9 以实现最高压缩率
      });
  
      output.on('close', () => {
        console.log(
          `ZIP file created: ${outputPath} (${archive.pointer()} total bytes)`,
        );
        resolve();
      });
  
      archive.on('error', (err) => {
        reject(err);
      });
  
      archive.pipe(output);
  
      // 使用 directory 方法以流的方式压缩文件夹,减少内存消耗
      archive.directory(folderPath, false);
  
      // 流式处理完成
      archive.finalize();
    });
  }