Add custom flags

anilsenay · Sep 28, 2023 · af19906 · af19906
1 parent 1f58f8d
commit af19906
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+test/
diff --git a/cmd/gonew.go b/cmd/gonew.go
@@ -63,6 +63,10 @@ func main() {
 	log.SetPrefix("gonew: ")
 	log.SetFlags(0)
 	flag.Usage = usage
+
+	srcModuleName := flag.String("srcmod", "", "source module name in go.mod file")
+	dstModuleName := flag.String("dstmod", "", "destination module name in go.mod file")
+
 	flag.Parse()
 	args := flag.Args()
 
@@ -123,6 +127,14 @@ func main() {
 		}
 	}
 
+	// Set custom module name
+	if *srcModuleName != "" {
+		srcMod = *srcModuleName
+	}
+	if *dstModuleName != "" {
+		dstMod = *dstModuleName
+	}
+
 	// Copy from module cache into new directory, making edits as needed.
 	filepath.WalkDir(info.Dir, func(src string, d fs.DirEntry, err error) error {
 		if err != nil {