コード例 #1
0
ファイル: archive.go プロジェクト: jfindley/automate
func archive(fs testfs.FileSystem) (out []byte, err error) {
	buf := new(bytes.Buffer)
	t := tar.NewWriter(buf)

	origDir, err := fs.Getwd()
	if err != nil {
		return
	}

	// Start at the root of the fs
	fs.Chdir("/")

	err = walk(fs, t)
	if err != nil {
		return
	}

	err = fs.Chdir(origDir)
	if err != nil {
		return
	}

	err = t.Close()

	return buf.Bytes(), err
}
コード例 #2
0
ファイル: archive.go プロジェクト: jfindley/automate
func walk(fs testfs.FileSystem, t *tar.Writer) (err error) {

	cwd, err := fs.Getwd()
	if err != nil {
		return
	}

	f, err := fs.OpenFile(cwd, os.O_RDONLY, 0)
	if err != nil {
		return
	}

	fi, err := f.Readdir(-1)
	if err != nil {
		return
	}

	for _, file := range fi {
		// Add a new file to the archive
		var hdr *tar.Header
		hdr, err = tarHdr(file)
		if err != nil {
			return
		}

		// Make header name fully qualified
		if cwd[len(cwd)-1] == '/' {
			hdr.Name = cwd + hdr.Name
		} else {
			hdr.Name = cwd + "/" + hdr.Name
		}

		// Strip leading slashes
		if hdr.Name[0] == '/' {
			hdr.Name = hdr.Name[1:]
		}

		err = t.WriteHeader(hdr)
		if err != nil {
			return
		}

		// Write the file data to the archive
		if !file.IsDir() && file.Size() > 0 {
			var target testfs.File

			target, err = fs.Open(file.Name())
			if err != nil {
				return
			}

			_, err = io.Copy(t, target)
			if err != nil {
				return
			}
		}

		// Close the file in the archive
		err = t.Flush()
		if err != nil {
			return
		}

		// If a directory, recurse into it
		if file.IsDir() {
			err = fs.Chdir(file.Name())
			if err != nil {
				return
			}

			return walk(fs, t)
		}

	}

	return
}