package ziputil

import (
	"archive/zip"
	"go-admin/app/brush/utils"
	"sync"
	"io"
	"os"
	"path/filepath"
	log "github.com/go-admin-team/go-admin-core/logger"
)

// Zip 將指定的文件夾或文件壓縮為 .zip 文件
func Zip(source, zipFile string) error {
	// 創(chuàng)建一個(gè)新的 zip 文件
	zipFileWriter, err := os.Create(zipFile)
	if err != nil {
		return err
	}
	defer func(zipFileWriter *os.File) {
		err := zipFileWriter.Close()
		if err != nil {
			log.Errorf("關(guān)閉 zip 文件失敗: %s", err)
		}
	}(zipFileWriter)

	// 創(chuàng)建 zip 寫入器
	zipWriter := zip.NewWriter(zipFileWriter)
	defer func(zipWriter *zip.Writer) {
		err := zipWriter.Close()
		if err != nil {
			log.Errorf("關(guān)閉 zip 寫入器失敗: %s", err)
		}
	}(zipWriter)

	// 獲取源文件的絕對(duì)路徑
	absSource, err := filepath.Abs(source)
	if err != nil {
		return err
	}

	// 遍歷文件夾并添加到 zip 文件中
	return filepath.Walk(absSource, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}

		// 計(jì)算文件相對(duì)路徑
		relPath, err := filepath.Rel(absSource, path)
		if err != nil {
			return err
		}

		// 如果是目錄，則在 zip 文件中創(chuàng)建一個(gè)目錄項(xiàng)
		if info.IsDir() {
			if relPath != "." {
				_, err := zipWriter.Create(relPath + "/")
				if err != nil {
					return err
				}
			}
			return nil
		}

		// 否則將文件添加到 zip 文件
		return addFileToZip(zipWriter, path, relPath)
	})
}

// addFileToZip 將單個(gè)文件添加到 zip 寫入器
func addFileToZip(zipWriter *zip.Writer, file string, relPath string) error {
	f, err := os.Open(file)
	if err != nil {
		return err
	}
	defer func(f *os.File) {
		err := f.Close()
		if err != nil {
			log.Errorf("關(guān)閉文件失敗: %s", err)
		}
	}(f)

	// 在 zip 文件中創(chuàng)建該文件
	writer, err := zipWriter.Create(relPath)
	if err != nil {
		return err
	}

	// 將文件內(nèi)容寫入 zip
	_, err = io.Copy(writer, f)
	if err != nil {
		return err
	}

	return nil
}

// UnZip 解壓 zip 文件到目標(biāo)目錄
func UnZip(zipFile, destDir string) error {
	log.Debugf("解壓文件: %s 到 %s", zipFile, destDir)
	r, err := zip.OpenReader(zipFile)
	if err != nil {
		return err
	}

	defer func(r *zip.ReadCloser) {
		err := r.Close()
		if err != nil {
			log.Errorf("關(guān)閉 zip 文件失敗: %s", err)
		}
	}(r)

	log.Debugf("總共 %d 個(gè)文件", len(r.File))

	// 并發(fā)解壓每個(gè)文件
	wg := sync.WaitGroup{}
	for _, f := range r.File {
		wg.Add(1)
		go func(rf *zip.File, w *sync.WaitGroup) {
			defer w.Done()
			if err := unzipFile(rf, destDir); err != nil {
				log.Errorf("解壓文件 [%s] 失敗: %v", rf.Name, err)
			}
		}(f, &wg)
	}

	wg.Wait()

	return nil
}

// unzipFile 解壓?jiǎn)蝹€(gè)文件到目標(biāo)目錄
func unzipFile(f *zip.File, destDir string) error {
	// 將文件名轉(zhuǎn)換為 UTF-8
	filename := utils.ConvertToUTF8([]byte(f.Name))
	filePath := filepath.Join(destDir, filename)

	// 創(chuàng)建文件夾
	if f.FileInfo().IsDir() {
		return os.MkdirAll(filePath, os.ModePerm)
	}

	// 創(chuàng)建文件的父目錄
	if err := os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err != nil {
		log.Errorf("創(chuàng)建目錄 [%s] 失敗: %v", filepath.Dir(filePath), err)
		return err
	}

	// 打開文件
	file, err := f.Open()
	if err != nil {
		log.Errorf("打開文件 [%s] 失敗: %v", filePath, err)
		return err
	}

	defer func(file io.ReadCloser) {
		err := file.Close()
		if err != nil {
			log.Errorf("關(guān)閉文件 [%s] 失敗: %v", filePath, err)
		}
	}(file)

	// 創(chuàng)建文件
	outFile, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
	if err != nil {
		log.Errorf("創(chuàng)建文件 [%s] 失敗: %v", filePath, err)
		return err
	}

	defer func(outFile *os.File) {
		err := outFile.Close()
		if err != nil {
			log.Errorf("關(guān)閉文件 [%s] 失敗: %v", filePath, err)
		}
	}(outFile)

	// 將文件內(nèi)容寫入
	_, err = io.Copy(outFile, file)
	if err != nil {
		log.Errorf("復(fù)制文件 [%s] 失敗: %v", filePath, err)
		return err
	}

	return nil
}