working commit

pkg/auxutar/utar.go

@@ -0,0 +1,120 @@
+package auxutar
+
+import (
+	"archive/tar"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+// TODO: file and dir modes
+
+func Archive(srcdir, dstpath string) error {
+	var err error
+	srcdir = filepath.Clean(srcdir)
+	dstpath = filepath.Clean(dstpath)
+
+	err = os.MkdirAll(filepath.Dir(dstpath), 0755)
+	if err != nil {
+		return err
+	}
+	tarFile, err := os.OpenFile(dstpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0640)
+	if err != nil {
+		return err
+	}
+	defer tarFile.Close()
+
+	tarWriter := tar.NewWriter(tarFile)
+	defer tarWriter.Close()
+
+	walker := func(filename string, fileInfo os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !fileInfo.Mode().IsRegular() {
+			return nil
+		}
+		header, err := tar.FileInfoHeader(fileInfo, fileInfo.Name())
+		if err != nil {
+			return err
+		}
+		header.Name = strings.TrimPrefix(filename, filepath.Clean(srcdir))
+		header.Name = strings.TrimPrefix(header.Name, string(filepath.Separator))
+
+		err = tarWriter.WriteHeader(header)
+		if err != nil {
+			return err
+		}
+		file, err := os.Open(filename)
+		if err != nil {
+			return err
+		}
+		defer file.Close()
+		_, err = io.Copy(tarWriter, file)
+		if err != nil {
+			return err
+		}
+		return nil
+	}
+	err = filepath.Walk(srcdir, walker)
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func Unarchive(filename, dstdir string) error {
+	var err error
+	err = os.MkdirAll(dstdir, 0755)
+	if err != nil {
+		return err
+	}
+	file, err := os.OpenFile(filename, os.O_RDONLY, 0)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	tarReader := tar.NewReader(file)
+	for {
+		header, err := tarReader.Next()
+		switch {
+		case err == io.EOF:
+			return nil
+		case err != nil:
+			return err
+		case header == nil:
+			continue
+		}
+		target := filepath.Join(dstdir, header.Name)
+		target = filepath.Clean(target)
+		//fileInfo := header.FileInfo()
+		switch header.Typeflag {
+		case tar.TypeDir:
+			_, err := os.Stat(target)
+			if err != nil {
+				err := os.MkdirAll(target, 0755)
+				if err != nil {
+					return err
+				}
+
+			}
+		case tar.TypeReg:
+			err := os.MkdirAll(filepath.Dir(target), 0755)
+			if err != nil {
+				return err
+			}
+			file, err := os.OpenFile(target, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, os.FileMode(header.Mode))
+			if err != nil {
+				return err
+			}
+			_, err = io.Copy(file, tarReader)
+			if err != nil {
+				return err
+			}
+			file.Close()
+		}
+	}
+	return err
+}