Ejemplo n.º 1
0
package cf_ignore_test

import (
	"errors"
	"strings"

	"github.com/cloudfoundry-incubator/lattice/ltc/droplet_runner/command_factory/cf_ignore"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("CF Ignore", func() {
	var cfIgnore cf_ignore.CFIgnore

	BeforeEach(func() {
		cfIgnore = cf_ignore.New()
	})

	It("excludes files based on exact path matches", func() {
		Expect(cfIgnore.Parse(strings.NewReader("the-dir/the-path"))).To(Succeed())
		Expect(cfIgnore.ShouldIgnore("the-dir/the-path")).To(BeTrue())
	})

	It("excludes the contents of directories based on exact path matches", func() {
		Expect(cfIgnore.Parse(strings.NewReader("dir1/dir2"))).To(Succeed())
		Expect(cfIgnore.ShouldIgnore("dir1/dir2/the-file")).To(BeTrue())
		Expect(cfIgnore.ShouldIgnore("dir1/dir2/dir3/the-file")).To(BeTrue())
	})

	It("excludes files based on star patterns", func() {
		Expect(cfIgnore.Parse(strings.NewReader("dir1/*.so"))).To(Succeed())
Ejemplo n.º 2
0
func (zipper *DropletArtifactZipper) Zip(srcDir string, cfIgnore cf_ignore.CFIgnore) (string, error) {
	tmpPath, err := ioutil.TempDir(os.TempDir(), "droplet-artifact-zipper")
	if err != nil {
		return "", err
	}

	fileWriter, err := os.OpenFile(filepath.Join(tmpPath, "artifact.zip"), os.O_CREATE|os.O_WRONLY, 0600)
	if err != nil {
		return "", err
	}

	zipWriter := zip.NewWriter(fileWriter)
	defer zipWriter.Close()

	contentsFileInfo, err := os.Stat(srcDir)
	if err != nil {
		return "", err
	}

	if !contentsFileInfo.IsDir() {
		return "", fmt.Errorf("%s must be a directory", srcDir)
	}

	if ignoreFile, err := os.Open(filepath.Join(srcDir, ".cfignore")); err == nil {
		if err := cfIgnore.Parse(ignoreFile); err != nil {
			return "", err
		}
	}

	err = filepath.Walk(srcDir, func(fullPath string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}

		relativePath, err := filepath.Rel(srcDir, fullPath)
		if err != nil {
			return err
		}

		if cfIgnore.ShouldIgnore(relativePath) {
			return nil
		}

		if relativePath == fileWriter.Name() || relativePath == "." || relativePath == ".." {
			return nil
		}

		if h, err := zip.FileInfoHeader(info); err == nil {
			h.Name = relativePath

			if info.IsDir() {
				h.Name = h.Name + "/"
			}

			h.SetMode(info.Mode())

			writer, err := zipWriter.CreateHeader(h)
			if err != nil {
				return err
			}

			if info.IsDir() {
				return nil
			}

			li, err := os.Lstat(fullPath)
			if err != nil {
				return err
			}

			if li.Mode()&os.ModeSymlink == os.ModeSymlink {
				dest, err := os.Readlink(fullPath)
				if err != nil {
					return err
				}
				if _, err := io.Copy(writer, strings.NewReader(dest)); err != nil {
					return err
				}
			} else {
				fr, err := os.Open(fullPath)
				if err != nil {
					return err
				}
				defer fr.Close()
				if _, err := io.Copy(writer, fr); err != nil {
					return err
				}
			}
		}

		return nil
	})

	return fileWriter.Name(), err
}