diff --git a/cmd/pgen/main.go b/cmd/pgen/main.go index 1485719..876e316 100644 --- a/cmd/pgen/main.go +++ b/cmd/pgen/main.go @@ -11,6 +11,8 @@ import ( "path/filepath" "regexp" "strings" + + "gitlab.wow.st/gmp/persist/generate" ) type visitor struct{} @@ -133,7 +135,7 @@ func main() { if len(ts) == 0 { continue } - g := NewGenerator(pkg.Name) + g := generate.NewGenerator(pkg.Name) for _,v := range ts { g.Add(v.fun, v.name, v.typ) } diff --git a/cmd/pgen/gen.go b/generate/generate.go similarity index 99% rename from cmd/pgen/gen.go rename to generate/generate.go index c13cb4c..537fd2f 100644 --- a/cmd/pgen/gen.go +++ b/generate/generate.go @@ -1,4 +1,4 @@ -package main +package generate import ( "io" diff --git a/cmd/pgen/template.go b/generate/template.go similarity index 97% rename from cmd/pgen/template.go rename to generate/template.go index 34d499a..5cb991d 100644 --- a/cmd/pgen/template.go +++ b/generate/template.go @@ -1,4 +1,4 @@ -package main +package generate const template string = `//Generated by gitlab.wow.st/persist/cmd/pgen. //Do not edit package main