Go: Call go mod vendor to synchronise vendor directory when it exists

mbg · mbg · commit 4d28c0d2a93b · 2024-02-15T16:19:07.000Z
diff --git a/go/extractor/cli/go-autobuilder/go-autobuilder.go b/go/extractor/cli/go-autobuilder/go-autobuilder.go
@@ -480,9 +480,17 @@ func installDependencies(workspace project.GoWorkspace) {
 
 		// get dependencies for all modules
 		for _, module := range workspace.Modules {
+			path := filepath.Dir(module.Path)
+
+			if util.DirExists(filepath.Join(path, "vendor")) {
+				vendor := toolchain.VendorModule(path)
+				log.Printf("Synchronizing vendor file using `go mod vendor` in %s.\n", path)
+				util.RunCmd(vendor)
+			}
+
 			install = exec.Command("go", "get", "-v", "./...")
-			install.Dir = filepath.Dir(module.Path)
-			log.Printf("Installing dependencies using `go get -v ./...` in `%s`.\n", filepath.Dir(module.Path))
+			install.Dir = path
+			log.Printf("Installing dependencies using `go get -v ./...` in `%s`.\n", path)
 			util.RunCmd(install)
 		}
 	}
diff --git a/go/extractor/toolchain/toolchain.go b/go/extractor/toolchain/toolchain.go
@@ -85,3 +85,10 @@ func InitModule(path string) *exec.Cmd {
 	modInit.Dir = path
 	return modInit
 }
+
+// Constructs a command to run `go mod vendor` in the directory given by `path`.
+func VendorModule(path string) *exec.Cmd {
+	modVendor := exec.Command("go", "mod", "vendor")
+	modVendor.Dir = path
+	return modVendor
+}
