package zip

import (
	"github.com/itnotebooks/zip"
	"io"
	"log"
	"os"
	"path/filepath"
	"strings"
)

// ZipLib 遞歸壓縮，默認(rèn)采用AES256加密方式加密
// 支持以下加密方式
// Standard         ZIP標(biāo)準(zhǔn)，安全性最低
// AES128           AES128位，安全性高
// AES192           AES192位，安全性高
// AES256           AES256位，安全性最高，本程序默認(rèn)采用此加密方式
func ZipLib(dst, src string, encrypt bool, password, enc string) (err error) {
	var dstFileBaseName = ""

	// 創(chuàng)建壓縮文件對象
	zfile, err := os.Create(dst)
	defer zfile.Close()

	if err != nil {
		return err
	}

	// 通過文件對象生成寫入對象
	zFileWriter := zip.NewWriter(zfile)
	defer func() {
		// 檢測一下是否成功關(guān)閉
		if err := zFileWriter.Close(); err != nil {
			log.Fatalln(err)
		}
	}()

	info, err := os.Stat(src)
	if err != nil {
		return err
	}

	var baseDir string
	if info.IsDir() {
		baseDir = filepath.Base(src)

		if !strings.HasSuffix(src, "/") {
			dstName := filepath.Base(dst)
			dstFileBaseName = strings.TrimSuffix(dstName, filepath.Ext(dstName))
		}
	}

	// 將文件寫入 zFileWriter 對象 ，可能會有很多個(gè)目錄及文件，遞歸處理
	return filepath.Walk(src, func(path string, fi os.FileInfo, errBack error) (err error) {
		if errBack != nil {
			return errBack
		}

		if strings.HasSuffix(path, ".zip") {
			return
		}
		//創(chuàng)建文件頭
		header, err := zip.FileInfoHeader(fi)
		if err != nil {
			return err
		}

		if baseDir != "" {
			// 如果原目錄是以"/"結(jié)尾，表示打包指定目錄時(shí)不包含該目錄
			if strings.HasSuffix(src, "/") {
				header.Name = strings.TrimPrefix(path, src)
			} else {
				header.Name = filepath.Join(dstFileBaseName, baseDir, strings.TrimPrefix(path, src))
			}
		}

		if fi.IsDir() {
			header.Name += "/"
		} else {
			header.Method = zip.Deflate

		}

		var fh io.Writer
		if encrypt {
			// 加密方式
			var encryption = zip.AES256Encryption

			switch enc {
			case "Standard":
				encryption = zip.StandardEncryption
			case "AES128":
				encryption = zip.AES128Encryption
			case "AES192":
				encryption = zip.AES192Encryption

			}

			// 寫入文件頭信息，并配置密碼
			fh, err = zFileWriter.Encrypt(header, password, encryption)
		} else {
			// 寫入文件頭信息
			fh, err = zFileWriter.CreateHeader(header)
		}

		if err != nil {
			return err
		}

		// 判斷是否是標(biāo)準(zhǔn)文件
		if !header.Mode().IsRegular() {
			return nil
		}

		// 打開要壓縮的文件
		file, err := os.Open(path)
		defer file.Close()
		if err != nil {
			return
		}

		// 將文件對象拷貝到 writer 結(jié)構(gòu)中
		ret, err := io.Copy(fh, file)
		if err != nil {
			return
		}

		log.Printf("added： %s, total: %d\n", path, ret)

		return nil
	})
}

調(diào)用文章來源地址http://www.zghlxwxcb.cn/news/detail-704955.html

func TestZipLib(t *testing.T) {
	enc := "AES256"
	encrypt := true
	pass := tools.RandomString(12)
	baseDir := tools.GetBaseDir()
	ZipLib(baseDir+"/test/202309.zip", baseDir+"/target/2023-09-07/xxx.csv", encrypt, pass, enc)
	log.Println("Zip file password:", pass)
}

到了這里，關(guān)于Golang zip壓縮及加密的文章就介紹完了。如果您還想了解更多內(nèi)容，請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！