Generator cleanup.

generator/postgresgen/generator.go

@@ -0,0 +1,126 @@
+package postgresgen
+
+import (
+	"database/sql"
+	"fmt"
+	"github.com/go-jet/jet/generator/internal/metadata"
+	"github.com/go-jet/jet/generator/internal/metadata/postgres-metadata"
+	"github.com/go-jet/jet/generator/internal/utils"
+	_ "github.com/lib/pq"
+	"path"
+	"path/filepath"
+)
+
+type DBConnection struct {
+	Host     string
+	Port     string
+	User     string
+	Password string
+	SslMode  string
+	Params   string
+
+	DBName     string
+	SchemaName string
+}
+
+func Generate(destDir string, genData DBConnection) error {
+
+	connectionString := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=%s %s",
+		genData.Host, genData.Port, genData.User, genData.Password, genData.DBName, genData.SslMode, genData.Params)
+
+	fmt.Println("Connecting to postgres database: " + connectionString)
+
+	db, err := sql.Open("postgres", connectionString)
+	if err != nil {
+		return err
+	}
+	defer db.Close()
+
+	err = db.Ping()
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Println("Retrieving schema information...")
+	schemaInfo, err := postgres_metadata.GetSchemaInfo(db, genData.DBName, genData.SchemaName)
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Println("	FOUND", len(schemaInfo.TableInfos), " table(s), ", len(schemaInfo.EnumInfos), " enum(s)")
+
+	fmt.Println("Cleaning up destination directory...")
+	err = utils.CleanUpGeneratedFiles(path.Join(destDir, genData.DBName, genData.SchemaName))
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Println("Generating table sql builder files...")
+	err = generate(schemaInfo, destDir, "table", sqlBuilderTableTemplate, schemaInfo.TableInfos)
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Println("Generating table model files...")
+	err = generate(schemaInfo, destDir, "model", dataModelTemplate, schemaInfo.TableInfos)
+
+	if err != nil {
+		return err
+	}
+
+	if len(schemaInfo.EnumInfos) > 0 {
+		fmt.Println("Generating enum sql builder files...")
+		err = generate(schemaInfo, destDir, "enum", enumTypeTemplate, schemaInfo.EnumInfos)
+
+		if err != nil {
+			return err
+		}
+
+		fmt.Println("Generating enum model files...")
+		err = generate(schemaInfo, destDir, "model", enumModelTemplate, schemaInfo.EnumInfos)
+
+		if err != nil {
+			return err
+		}
+	}
+
+	fmt.Println("Done")
+
+	return nil
+}
+
+func generate(schemaInfo postgres_metadata.SchemaInfo, dirPath, packageName string, template string, metaDataList []metadata.MetaData) error {
+	modelDirPath := filepath.Join(dirPath, schemaInfo.DatabaseName, schemaInfo.Name, packageName)
+
+	err := utils.EnsureDirPath(modelDirPath)
+
+	if err != nil {
+		return err
+	}
+
+	autoGenWarning, err := utils.GenerateTemplate(autoGenWarningTemplate, nil)
+
+	if err != nil {
+		return err
+	}
+
+	for _, metaData := range metaDataList {
+		text, err := utils.GenerateTemplate(template, metaData)
+
+		if err != nil {
+			return err
+		}
+
+		err = utils.SaveGoFile(modelDirPath, metaData.Name(), append(autoGenWarning, text...))
+
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}

generator/postgresgen/templates.go

@@ -0,0 +1,136 @@
+package postgresgen
+
+var autoGenWarningTemplate = `
+//
+// Code generated by go-jet DO NOT EDIT.
+// Generated at {{now}}
+//
+// WARNING: Changes to this file may cause incorrect behavior and will be lost 
+// if the code is regenerated
+//
+// Licence under ...
+//
+
+`
+
+var sqlBuilderTableTemplate = ` 
+{{define "column-list" -}}
+	{{- range $i, $c := . }}
+		{{- if gt $i 0 }}, {{end}}{{camelize $c.Name}}Column
+	{{- end}}
+{{- end}}
+
+package table
+
+import (
+	"github.com/go-jet/jet"
+)
+
+type {{.GoStructName}} struct {
+	jet.Table
+	
+	//Columns
+{{- range .Columns}}
+	{{camelize .Name}} jet.Column{{.SqlBuilderColumnType}}
+{{- end}}
+
+	AllColumns jet.ColumnList
+}
+
+var {{camelize .Name}} = new{{.GoStructName}}()
+
+func new{{.GoStructName}}() *{{.GoStructName}} {
+	var (
+	{{- range .Columns}}
+		{{camelize .Name}}Column = jet.{{.SqlBuilderColumnType}}Column("{{.Name}}")
+	{{- end}}
+	)
+
+	return &{{.GoStructName}}{
+		Table: jet.NewTable("{{.SchemaName}}", "{{.Name}}", {{template "column-list" .Columns}}),
+
+		//Columns
+{{- range .Columns}}
+		{{camelize .Name}}: {{camelize .Name}}Column,
+{{- end}}
+
+		AllColumns: jet.ColumnList{ {{template "column-list" .Columns}} },
+	}
+}
+
+func (a *{{.GoStructName}}) AS(alias string) *{{.GoStructName}} {
+	aliasTable := new{{.GoStructName}}()
+
+	aliasTable.Table.AS(alias)
+
+	return aliasTable
+}
+
+`
+
+var dataModelTemplate = `package model
+
+{{ if .GetImports }}
+import (
+{{- range .GetImports}}
+	"{{.}}"
+{{- end}}
+)
+{{end}}
+
+
+type {{camelize .Name}} struct {
+{{- range .Columns}}
+	{{camelize .Name}} {{.GoModelType}} ` + "{{.GoModelTag ($.IsUnique .Name)}}" + `
+{{- end}}
+}
+`
+
+var enumModelTemplate = `package model
+
+import "errors"
+
+type {{camelize $.Name}} string
+
+const (
+{{- range $index, $element := .Values}}
+	{{camelize $.Name}}_{{camelize $element}} {{camelize $.Name}} = "{{$element}}"
+{{- end}}
+)
+
+func (e *{{camelize $.Name}}) Scan(value interface{}) error {
+	if v, ok := value.(string); !ok {
+		return errors.New("Invalid data for {{camelize $.Name}} enum")
+	} else {
+		switch string(v) {
+{{- range $index, $element := .Values}}
+		case "{{$element}}":
+			*e = {{camelize $.Name}}_{{camelize $element}}
+{{- end}}
+		default:
+			return errors.New("Inavlid data " + string(v) + "for {{camelize $.Name}} enum")
+		}
+
+		return nil
+	}
+}
+
+func (e {{camelize $.Name}}) String() string {
+	return string(e)
+}
+
+`
+var enumTypeTemplate = `package enum
+
+import "github.com/go-jet/jet"
+
+var {{camelize $.Name}} = &struct {
+{{- range $index, $element := .Values}}
+	{{camelize $element}} jet.StringExpression
+{{- end}}
+} {
+{{- range $index, $element := .Values}}
+	{{camelize $element}}: jet.NewEnumValue("{{$element}}"),
+{{- end}}
+}
+`