// Gulu - Golang common utilities for everyone.
// Copyright (c) 2019-present, b3log.org
//
// Gulu is licensed under Mulan PSL v2.
// You can use this software according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of Mulan PSL v2 at:
//         http://license.coscl.org.cn/MulanPSL2
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PSL v2 for more details.

package gulu

import (
	"os"
	"path/filepath"
	"testing"
)

var testdataDir = "testdata"
var zipDirPath = filepath.Join(testdataDir, "test_zip")

func TestCreate(t *testing.T) {
	zipFile, err := Zip.Create(zipDirPath + ".zip")
	if nil != err {
		t.Error(err)
		return
	}

	zipFile.AddDirectory(".", testdataDir, func(filename string) {
		t.Logf("zipped %s", filename)
	})
	if nil != err {
		t.Error(err)
		return
	}

	err = zipFile.Close()
	if nil != err {
		t.Error(err)
		return
	}

	err = Zip.Unzip(zipDirPath+".zip", zipDirPath, func(filename string) {
		t.Logf("unzipped %s", filename)
	})
	if nil != err {
		t.Error(err)
		return
	}

	f1, _ := os.Stat(filepath.Join(testdataDir, "README.md"))
	f1ModTime := f1.ModTime()
	f2, _ := os.Stat(filepath.Join(zipDirPath, "README.md"))
	f2ModTime := f2.ModTime()
	if f1ModTime.Unix() != f2ModTime.Unix() {
		t.Error("ModTime error")
		return
	}
}

func TestUnzip(t *testing.T) {
	err := Zip.Unzip(zipDirPath+".zip", zipDirPath)
	if nil != err {
		t.Error(err)
		return
	}
}

func _TestEmptyDir(t *testing.T) {
	dir1 := "/dir/subDir1"
	dir2 := "/dir/subDir2"

	err := os.MkdirAll(zipDirPath+dir1, os.ModeDir)
	if nil != err {
		t.Error(err)

		return
	}

	err = os.MkdirAll(zipDirPath+dir2, os.ModeDir)
	if nil != err {
		t.Error(err)
		return
	}

	f, err := os.Create(zipDirPath + dir2 + "/file")
	if nil != err {
		t.Error(err)
		return
	}
	f.Close()

	zipFile, err := Zip.Create(zipDirPath + "/dir.zip")
	if nil != err {
		t.Error(err)
		return
	}

	zipFile.AddDirectory("dir", zipDirPath+"/dir")
	if nil != err {
		t.Error(err)
		return
	}

	err = zipFile.Close()
	if nil != err {
		t.Error(err)
		return
	}

	err = Zip.Unzip(zipDirPath+"/dir.zip", zipDirPath+"/unzipDir")
	if nil != err {
		t.Error(err)
		return
	}

	if !File.IsExist(zipDirPath+"/unzipDir") || !File.IsDir(zipDirPath+"/unzipDir") {
		t.Error("Unzip failed")
		return
	}

	if !File.IsExist(zipDirPath+"/unzipDir"+dir1) || !File.IsDir(zipDirPath+"/unzipDir"+dir1) {
		t.Error("Unzip failed")
		return
	}

	if !File.IsExist(zipDirPath+"/unzipDir"+dir2) || !File.IsDir(zipDirPath+"/unzipDir"+dir2) {
		t.Error("Unzip failed")
		return
	}

	if !File.IsExist(zipDirPath+"/unzipDir"+dir2+"/file") || File.IsDir(zipDirPath+"/unzipDir"+dir2+"/file") {
		t.Error("Unzip failed")
		return
	}
}

func TestMain(m *testing.M) {
	retCode := m.Run()

	// clean test data
	os.RemoveAll(zipDirPath + ".zip")
	os.RemoveAll(zipDirPath)

	os.RemoveAll(tarDirPath + ".tar")
	os.RemoveAll(tarDirPath)
	os.RemoveAll(untarDirPath)

	os.Exit(retCode)
}