3 files changed, 7 insertions, 41 deletions
diff --git a/compflag/flags.go b/compflag/flags.go
index 91b413a..10fd32a 100644
--- a/compflag/flags.go
+++ b/compflag/flags.go
@@ -1,3 +1,5 @@
+// Autogenerated by go run compflag/gen/main.go. DO NOT EDIT.
+
 package compflag
 
 import (
@@ -8,10 +10,6 @@ import (
 	"github.com/posener/complete/v2/predict"
 )
 
-// Code auto generated with `go run ./gen`. DO NOT EDIT
-
-//go:generate go run ./gen
-
 // String if a flag function for a flag of type string.
 func String(name string, value string, usage string, options ...predict.Option) *string {
 	return CommandLine.String(name, value, usage, options...)
diff --git a/compflag/gen/flags.go.gotmpl b/compflag/gen/flags.go.gotmpl
index 9b65dd5..07dddc7 100644
--- a/compflag/gen/flags.go.gotmpl
+++ b/compflag/gen/flags.go.gotmpl
@@ -1,10 +1,5 @@
 package compflag
 
-// Code auto generated with `go run ./gen`. DO NOT EDIT
-
-//go:generate go run ./gen
-
-
 {{ range . }}
 
 // {{ .Name }} if a flag function for a flag of type {{ .Type }}.
diff --git a/compflag/gen/main.go b/compflag/gen/main.go
index cbea96f..d5601ca 100644
--- a/compflag/gen/main.go
+++ b/compflag/gen/main.go
@@ -3,15 +3,12 @@ package main
 
 import (
 	"log"
-	"os"
-	"path/filepath"
 	"strings"
-	"text/template"
 
-	"github.com/posener/script"
+	"github.com/posener/autogen"
 )
 
-const tmplGlob = "gen/*.go.gotmpl"
+//go:generate go run .
 
 type flag struct {
 	Name          string
@@ -35,33 +32,9 @@ var flags = []flag{
 	{Name: "Duration", Type: "time.Duration"},
 }
 
-var tmpl = template.Must(template.ParseGlob(tmplGlob))
-
 func main() {
-	for _, t := range tmpl.Templates() {
-		fileName := outFileName(t.Name())
-		f, err := os.Create(fileName)
-		if err != nil {
-			panic(err)
-		}
-		defer f.Close()
-
-		log.Printf("Writing %s", fileName)
-		err = t.Execute(f, flags)
-		if err != nil {
-			panic(err)
-		}
-
-		// Format the file.
-		err = script.ExecHandleStderr(os.Stderr, "goimports", "-w", fileName).ToStdout()
-		if err != nil {
-			panic(err)
-		}
+	err := autogen.Execute(flags)
+	if err != nil {
+		log.Fatal(err)
 	}
 }
-
-func outFileName(templateName string) string {
-	name := filepath.Base(templateName)
-	// Remove .gotmpl suffix.
-	return name[:strings.LastIndex(name, ".")]
-}