Pass build information to templates

src/generator.go

@@ -69,6 +69,7 @@ func generateDbContext(routines []Routine, hashMap *map[string]string, config *C
 	data := &DbContextData{
 		Config:    config,
 		Functions: routines,
+		BuildInfo: GetBuildInfo(),
 	}
 
 	filename := changeCase("DbContext"+config.GeneratedFileExtension, config.GeneratedFileCase)
@@ -107,8 +108,9 @@ func generateModels(routines []Routine, hashMap *map[string]string, config *Conf
 		filePath := filepath.Join(config.OutputFolder, relPath)
 
 		data := &ModelTemplateData{
-			Config:  config,
-			Routine: routine,
+			Config:    config,
+			Routine:   routine,
+			BuildInfo: GetBuildInfo(),
 		}
 
 		changed, err := generateFile(data, moduleTemplate, filePath, hashMap)
@@ -149,8 +151,9 @@ func generateProcessors(routines []Routine, hashMap *map[string]string, config *
 		filePath := filepath.Join(config.OutputFolder, relPath)
 
 		data := &ProcessorTemplateData{
-			Config:  config,
-			Routine: routine,
+			Config:    config,
+			Routine:   routine,
+			BuildInfo: GetBuildInfo(),
 		}
 
 		changed, err := generateFile(data, processorTemplate, filePath, hashMap)

src/types.go

@@ -27,14 +27,17 @@ type Routine struct {
 type DbContextData struct {
 	Config    *Config
 	Functions []Routine
+	BuildInfo *BuildInformation
 }
 
 type ProcessorTemplateData struct {
-	Config  *Config
-	Routine Routine
+	Config    *Config
+	Routine   Routine
+	BuildInfo *BuildInformation
 }
 
 type ModelTemplateData struct {
-	Config  *Config
-	Routine Routine
+	Config    *Config
+	Routine   Routine
+	BuildInfo *BuildInformation
 }

src/version.go

@@ -9,48 +9,55 @@ import (
 )
 
 type BuildInformation struct {
-	builder    string
-	version    string
-	commitHash string
+	Builder    string
+	Version    string
+	CommitHash string
 }
 
-var info *BuildInformation = nil
+const localVersion = "LOCAL"
+const localBuilder = "LOCAL"
+
+var info BuildInformation = BuildInformation{
+	Builder:    localBuilder,
+	Version:    localVersion,
+	CommitHash: "",
+}
 
 func ParseBuildInformation(versionFileText string) error {
 	if strings.HasPrefix(versionFileText, "LOCAL") {
-		common.LogWarn("Running locally build version, be careful")
+		common.LogWarn("Running locally build Version, be careful")
 		return nil
 	}
 
 	splitted := strings.Split(versionFileText, " ")
 
 	if len(splitted) < 2 {
-		return fmt.Errorf("error with build, version file has incorrect format ")
+		return fmt.Errorf("error with build, Version file has incorrect format ")
 	}
 
-	info = &BuildInformation{
-		builder:    splitted[0],
-		version:    splitted[1],
-		commitHash: splitted[2],
+	info = BuildInformation{
+		Builder:    splitted[0],
+		Version:    splitted[1],
+		CommitHash: splitted[2],
 	}
 
 	return nil
 }
 
 func PrintVersion() {
-	if info == nil {
-		log.Printf("Locally build version")
+	if info.Builder == localBuilder {
+		log.Printf("Locally build Version")
 		return
 	}
-	log.Printf("db-gen build by %s ", info.builder)
-	log.Printf("version %s ", info.version)
-	log.Printf("last commit hash %s ", info.commitHash)
+	log.Printf("db-gen build by %s ", info.Builder)
+	log.Printf("Version %s ", info.Version)
+	log.Printf("last commit hash %s ", info.CommitHash)
 }
 
 func GetVersion() string {
-	if info == nil {
-		return "LOCAL"
-	} else {
-		return info.version
-	}
+	return info.Version
+}
+
+func GetBuildInfo() *BuildInformation {
+	return &info
 }

testing/templates/dbcontext.gotmpl

@@ -1,4 +1,4 @@
-// Autogenerated using db-gen
+// Autogenerated using db-gen version: {{.BuildInfo.Version}}
 
 using Database.Common;
 using Database.Generated;

testing/templates/model.gotmpl

@@ -1,4 +1,4 @@
-// Autogenerated using db-gen
+// Autogenerated using db-gen version: {{.BuildInfo.Version}}
 
 using Database.Common;
 

testing/templates/processor.gotmpl

@@ -1,4 +1,4 @@
-// Autogenerated using db-gen
+// Autogenerated using db-gen version: {{.BuildInfo.Version}}
 
 using Database.Common;
 using Npgsql;