v2: Adapt PR 244 for v2

kubernetes · Jan 31, 2024 · e37a30e · e37a30e
1 parent 4f44cb3
commit e37a30e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/v2/execute.go b/v2/execute.go
@@ -63,6 +63,8 @@ func GoBoilerplate(headerFile, buildTag, generatedBy string) ([]byte, error) {
 
 	if generatedBy != "" {
 		generatorName := filepath.Base(os.Args[0])
+		// Strip the extension from the name to normalize output between *nix and Windows.
+		generatorName = generatorName[:len(generatorName)-len(filepath.Ext(generatorName))]
 		generatedByComment := strings.ReplaceAll(generatedBy, "GENERATOR_NAME", generatorName)
 		buf.WriteString(fmt.Sprintf("%s\n\n", generatedByComment))
 	}