chore(internal/postprocessor): ensure new modules are added to worksp…

…ace (#8013)
googleapis · May 30, 2023 · 23cf597 · 23cf597
1 parent 71b1326
commit 23cf597
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/internal/postprocessor/execv/gocmd/gocmd.go b/internal/postprocessor/execv/gocmd/gocmd.go
@@ -138,3 +138,10 @@ func EditReplace(dir, mod, modPath string) error {
 	}
 	return c.Run()
 }
+
+// WorkUse updates the go.work file for added modules.
+func WorkUse(dir string) error {
+	c := execv.Command("go", "work", "use", "-r", ".")
+	c.Dir = dir
+	return c.Run()
+}
diff --git a/internal/postprocessor/main.go b/internal/postprocessor/main.go
@@ -250,7 +250,11 @@ func (p *postProcessor) generateModule(modPath, importPath string) error {
 		return err
 	}
 	log.Printf("Creating %s/go.mod", modPath)
-	return gocmd.ModInit(modPath, importPath)
+	if err := gocmd.ModInit(modPath, importPath); err != nil {
+		return err
+	}
+	log.Print("Updating workspace")
+	return gocmd.WorkUse(p.googleCloudDir)
 }
 
 func (p *postProcessor) generateVersionFile(moduleName, path string) error {