package archive

import (
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"path"
	"path/filepath"
	"testing"

	"github.com/docker/docker/pkg/ioutils"
	"github.com/stretchr/testify/assert"
)

func listFiles(dir string) []string {
	items := make([]string, 0)
	filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
		if path == dir {
			return nil
		}
		items = append(items, path)
		return nil
	})

	return items
}

func Test_shouldCreateArhive(t *testing.T) {
	tmpdir, _ := ioutils.TempDir("", "backup")
	defer os.RemoveAll(tmpdir)

	content := []byte("content")
	ioutil.WriteFile(path.Join(tmpdir, "outer"), content, 0600)
	os.MkdirAll(path.Join(tmpdir, "dir"), 0700)
	ioutil.WriteFile(path.Join(tmpdir, "dir", ".dotfile"), content, 0600)
	ioutil.WriteFile(path.Join(tmpdir, "dir", "inner"), content, 0600)

	gzPath, err := TarGzDir(tmpdir)
	assert.Nil(t, err)
	assert.Equal(t, filepath.Join(tmpdir, fmt.Sprintf("%s.tar.gz", filepath.Base(tmpdir))), gzPath)

	extractionDir, _ := ioutils.TempDir("", "extract")
	defer os.RemoveAll(extractionDir)

	cmd := exec.Command("tar", "-xzf", gzPath, "-C", extractionDir)
	err = cmd.Run()
	if err != nil {
		t.Fatal("Failed to extract archive: ", err)
	}
	extractedFiles := listFiles(extractionDir)

	wasExtracted := func(p string) {
		fullpath := path.Join(extractionDir, p)
		assert.Contains(t, extractedFiles, fullpath)
		copyContent, _ := ioutil.ReadFile(fullpath)
		assert.Equal(t, content, copyContent)
	}

	wasExtracted("outer")
	wasExtracted("dir/inner")
	wasExtracted("dir/.dotfile")
}

func Test_shouldCreateArhiveXXXXX(t *testing.T) {
	tmpdir, _ := ioutils.TempDir("", "backup")
	defer os.RemoveAll(tmpdir)

	content := []byte("content")
	ioutil.WriteFile(path.Join(tmpdir, "outer"), content, 0600)
	os.MkdirAll(path.Join(tmpdir, "dir"), 0700)
	ioutil.WriteFile(path.Join(tmpdir, "dir", ".dotfile"), content, 0600)
	ioutil.WriteFile(path.Join(tmpdir, "dir", "inner"), content, 0600)

	gzPath, err := TarGzDir(tmpdir)
	assert.Nil(t, err)
	assert.Equal(t, filepath.Join(tmpdir, fmt.Sprintf("%s.tar.gz", filepath.Base(tmpdir))), gzPath)

	extractionDir, _ := ioutils.TempDir("", "extract")
	defer os.RemoveAll(extractionDir)

	r, _ := os.Open(gzPath)
	ExtractTarGz(r, extractionDir)
	if err != nil {
		t.Fatal("Failed to extract archive: ", err)
	}
	extractedFiles := listFiles(extractionDir)

	wasExtracted := func(p string) {
		fullpath := path.Join(extractionDir, p)
		assert.Contains(t, extractedFiles, fullpath)
		copyContent, _ := ioutil.ReadFile(fullpath)
		assert.Equal(t, content, copyContent)
	}

	wasExtracted("outer")
	wasExtracted("dir/inner")
	wasExtracted("dir/.dotfile")
}