fix: use directory instead of project name when directory is set

mukezhz · Jan 9, 2024 · d91c09f · d91c09f
1 parent d570b31
commit d91c09f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmd/add_infrastructure.go b/cmd/add_infrastructure.go
@@ -99,7 +99,11 @@ func addInfrastructure(
 		templatePath := filepath.Join(".", "templates", "wesionary", "infrastructure", infrasTmpl[i])
 		var targetRoot string
 		if isNewProject {
-			targetRoot = filepath.Join(data.PackageName, "pkg", "infrastructure", strings.Replace(infrasTmpl[i], ".tmpl", ".go", 1))
+			if data.Directory == "" {
+				targetRoot = filepath.Join(data.PackageName, "pkg", "infrastructure", strings.Replace(infrasTmpl[i], ".tmpl", ".go", 1))
+			} else {
+				targetRoot = filepath.Join(data.Directory, "pkg", "infrastructure", strings.Replace(infrasTmpl[i], ".tmpl", ".go", 1))
+			}
 		} else {
 			targetRoot = filepath.Join(".", "pkg", "infrastructure", strings.Replace(infrasTmpl[i], ".tmpl", ".go", 1))
 		}