package main

import (
	"archive/zip"
	"fmt"
	"io"
	"log"
	"os"
)

func main() {
	Zip()
	Unzip()
}

func Zip() {
	// 创建 ab.tar 文件
	target, err := os.OpenFile("archive/files/ab.zip", os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		log.Fatalln(err)
	}
	defer target.Close()
	zw := zip.NewWriter(target)
	defer zw.Close()
	files := []string{"archive/files/a.txt", "archive/files/b.txt"}
	for _, fileName := range files {
		file, err := os.Open(fileName)
		if err != nil {
			log.Fatalln(err)
		}
		fileInfo, err := file.Stat()
		if err != nil {
			log.Fatalln(err)
		}
		f, err := zw.Create(fileInfo.Name())
		if err != nil {
			log.Fatal(err)
		}
		// 读取文件内容并写入 zip 文件
		if _, err := io.Copy(f, file); err != nil {
			log.Fatalln(err)
		}
		file.Close()
	}
}

func Unzip() {
	// 解压 archive/files/ab.tar 文件
	zipFile, err := zip.OpenReader("archive/files/ab.zip")
	if err != nil {
		log.Fatal(err)
	}
	defer zipFile.Close()
	// Iterate through the files in the archive,
	// printing some of their contents.
	for _, f := range zipFile.File {
		fmt.Printf("Contents of %s:\n", f.Name)
		rc, err := f.Open()
		if err != nil {
			log.Fatal(err)
		}
		defer rc.Close()
		for {
			_, err = io.CopyN(os.Stdout, rc, 68)
			if err == io.EOF {
				break
			}
			if err != nil {
				log.Fatalln(err)
			}
		}
		fmt.Println()
	}
}