Merge pull request #87 from tobikris/fix-cli-gen-version

fix cli version in generated file

README.md

@@ -311,7 +311,7 @@ USAGE:
    gogmcli [global options] command [command options] [arguments...]
 
 VERSION:
-   2.0.0
+   2.1.1
 
 COMMANDS:
    generate, g, gen  to generate link and unlink functions for nodes

cmd/gogmcli/gen/gen.go

@@ -39,7 +39,7 @@ import (
 // Generate searches for all go source files, then generates link and unlink functions for all gogm structs
 // takes in root directory and whether to log in debug mode
 // note: Generate is not recursive, it only looks in the target directory
-func Generate(directory string, debug bool) error {
+func Generate(directory string, debug bool, generator string) error {
 	confs := map[string][]*relConf{}
 	imps := map[string][]string{}
 	var edges []string
@@ -203,6 +203,7 @@ func Generate(directory string, debug bool) error {
 
 	buf := new(bytes.Buffer)
 	err = tpl.Execute(buf, templateConfig{
+		Generator:   generator,
 		Imports:     imports,
 		PackageName: packageName,
 		Funcs:       funcs,

cmd/gogmcli/gen/templ.go

@@ -260,7 +260,7 @@ func(l *{{ .StructName }}) UnlinkFrom{{ .OtherStructName }}OnField{{.StructField
 `
 
 var masterTpl = `
-{{ define "linkFile" }}// Code generated by GoGM v1.0.1. DO NOT EDIT
+{{ define "linkFile" }}// Code generated by {{ .Generator }}. DO NOT EDIT
 package {{ .PackageName }}
 
 import (
@@ -279,6 +279,7 @@ import (
 `
 
 type templateConfig struct {
+	Generator   string
 	Imports     []string
 	PackageName string
 	// type: funcs

cmd/gogmcli/gogm.go

@@ -21,10 +21,12 @@ package main
 
 import (
 	"errors"
-	"github.com/mindstand/gogm/v2/cmd/gogmcli/gen"
-	"github.com/urfave/cli/v2"
+	"fmt"
 	"log"
 	"os"
+
+	"github.com/mindstand/gogm/v2/cmd/gogmcli/gen"
+	"github.com/urfave/cli/v2"
 )
 
 //main is the main function
@@ -34,7 +36,7 @@ func main() {
 	app := &cli.App{
 		Name:                 "gogmcli",
 		HelpName:             "gogmcli",
-		Version:              "2.1.0",
+		Version:              "2.1.1",
 		Usage:                "used for neo4j operations from gogm schema",
 		Description:          "cli for generating and executing migrations with gogm",
 		EnableBashCompletion: true,
@@ -58,7 +60,7 @@ func main() {
 						log.Printf("generating link and unlink from directory [%s]", directory)
 					}
 
-					return gen.Generate(directory, debug)
+					return gen.Generate(directory, debug, fmt.Sprintf("GoGM %s", c.App.Version))
 				},
 			},
 		},