From 82837cf6d59af75de87d8115a6ab8304da05bd86 Mon Sep 17 00:00:00 2001
From: "Dr. Stefan Schimanski" <stefan.schimanski@gmail.com>
Date: Wed, 24 May 2017 17:08:17 +0200
Subject: [PATCH] PoC: support staging directory

---
 internal/gps/pkgtree/pkgtree.go | 12 +++--
 txn_writer.go                   | 80 +++++++++++++++++++++++++++++++++
 2 files changed, 89 insertions(+), 3 deletions(-)

diff --git a/internal/gps/pkgtree/pkgtree.go b/internal/gps/pkgtree/pkgtree.go
index f14598c286..9663cb991b 100644
--- a/internal/gps/pkgtree/pkgtree.go
+++ b/internal/gps/pkgtree/pkgtree.go
@@ -122,6 +122,12 @@ func ListPackages(fileRoot, importRoot string) (PackageTree, error) {
 		// import paths.
 		ip := filepath.ToSlash(filepath.Join(importRoot, strings.TrimPrefix(wp, fileRoot)))
 
+		// translate staging repo import path by cutting off the prefix
+		stagingPattern := "/staging/src/"
+		if p := strings.LastIndex(ip, stagingPattern); p != -1 {
+			ip = ip[p+len(stagingPattern):]
+		}
+
 		// Find all the imports, across all os/arch combos
 		//p, err := fullPackageInDir(wp)
 		p := &build.Package{
@@ -488,10 +494,10 @@ func (t PackageTree) ToReachMap(main, tests, backprop bool, ignore map[string]bo
 				continue
 			}
 
-			if !eqOrSlashedPrefix(imp, t.ImportRoot) {
-				w.ex[imp] = true
-			} else {
+			if _, internal := t.Packages[imp]; internal {
 				w.in[imp] = true
+			} else {
+				w.ex[imp] = true
 			}
 		}
 
diff --git a/txn_writer.go b/txn_writer.go
index 74adaf7465..7e093e1bd6 100644
--- a/txn_writer.go
+++ b/txn_writer.go
@@ -273,6 +273,72 @@ func (sw SafeWriter) validate(root string, sm gps.SourceManager) error {
 	return nil
 }
 
+// createStagingLinksForPackage looks into <pkgPath>/staging/src and symlinks those staging
+// repositories into vendor/. If packages only contain subpackages, a deeper level is linked.
+// E.g. if <pkgPath>/staging/src/github.com has no real files and only directories the algorithm
+// looks at a deeper level, e.g. at pkgPath/staging/src/github.com/foo. If foo has files (for
+// example *.go files), a relative symlink is created:
+//
+//   tempVendor/github.com/foo -> .../../../staging/src/github.com/foo
+//
+// The tempVendor directory corresponds logically to root/vendor. But, because th caller of
+// this func creates vendor/ dirs as temporary directories first, we have to support this
+// distinction.
+func createStagingLinksForPackage(root, pkgPath string, tempVendor string) error {
+	stagingRoot := filepath.Join(pkgPath, "staging", "src")
+	if _, err := os.Lstat(stagingRoot); os.IsNotExist(err) {
+		return nil
+	}
+
+	logicalVendor := filepath.Join(root, "vendor")
+	return filepath.Walk(stagingRoot, func(wp string, fi os.FileInfo, err error) error {
+		if err != nil && err != filepath.SkipDir {
+			return err
+		}
+		if !fi.IsDir() || strings.HasPrefix(fi.Name(), ".") {
+			return filepath.SkipDir
+		}
+
+		// find out whether the directory has directories only
+		f, err := os.Open(wp)
+		if err != nil {
+			return err
+		}
+		ffis, err := f.Readdir(0)
+		f.Close()
+		if err != nil {
+			return err
+		}
+		filesFound := false
+		for _, ffi := range ffis {
+			if !ffi.IsDir() && !strings.HasPrefix(ffi.Name(), ".") {
+				filesFound = true
+				break
+			}
+		}
+
+		// dive deeper if no files where found, only more subdirectories
+		if !filesFound {
+			return nil
+		}
+
+		// files were found. Let's create a symlink in the vendor dir
+		ip := strings.TrimPrefix(wp, stagingRoot)
+		if err := os.MkdirAll(filepath.Join(tempVendor, filepath.Dir(ip)), 0755); err != nil {
+			return err
+		}
+		rel, err := filepath.Rel(filepath.Join(logicalVendor, filepath.Dir(ip)), wp)
+		if err != nil {
+			return err
+		}
+		if err := os.Symlink(rel, filepath.Join(tempVendor, ip)); err != nil {
+			return err
+		}
+
+		return filepath.SkipDir
+	})
+}
+
 // Write saves some combination of config yaml, lock, and a vendor tree.
 // root is the absolute path of root dir in which to write.
 // sm is only required if vendor is being written.
@@ -337,6 +403,20 @@ func (sw *SafeWriter) Write(root string, sm gps.SourceManager, noExamples bool)
 		if err != nil {
 			return errors.Wrap(err, "error while writing out vendor tree")
 		}
+
+		// symlink staging repos of the root package
+		if err := createStagingLinksForPackage(root, root, filepath.Join(td, "vendor")); err != nil {
+			return errors.Wrap(err, "error creating staging symlinks")
+		}
+
+		// symlink staging repos of the vendored packages. These are exported into the temporary
+		// directory td. So we can use that as the root.
+		for _, p := range sw.Lock.Projects() {
+			pkgPath := filepath.Join(td, "vendor", string(p.Ident().ProjectRoot))
+			if err := createStagingLinksForPackage(td, pkgPath, filepath.Join(td, "vendor")); err != nil {
+				return errors.Wrap(err, fmt.Sprintf("error creating staging symlinks for vendored package %q", p.Ident().ProjectRoot))
+			}
+		}
 	}
 
 	// Ensure vendor/.git is preserved if present