Add the ability to fully customize jet generated files.

generator/metadata/column_meta_data.go

@@ -0,0 +1,27 @@
+package metadata
+
+// Column struct
+type Column struct {
+	Name         string
+	IsPrimaryKey bool
+	IsNullable   bool
+	DataType     DataType
+}
+
+// DataTypeKind is database type kind(base, enum, user-defined, array)
+type DataTypeKind string
+
+// DataTypeKind possible values
+const (
+	BaseType        DataTypeKind = "base"
+	EnumType        DataTypeKind = "enum"
+	UserDefinedType DataTypeKind = "user-defined"
+	ArrayType       DataTypeKind = "array"
+)
+
+// DataType contains information about column data type
+type DataType struct {
+	Name       string
+	Kind       DataTypeKind
+	IsUnsigned bool
+}

generator/metadata/dialect_query_set.go

@@ -0,0 +1,35 @@
+package metadata
+
+import (
+	"database/sql"
+	"fmt"
+)
+
+// TableType is type of database table(view or base)
+type TableType string
+
+const (
+	baseTable TableType = "BASE TABLE"
+	viewTable TableType = "VIEW"
+)
+
+// DialectQuerySet is set of methods necessary to retrieve dialect meta data information
+type DialectQuerySet interface {
+	GetTablesMetaData(db *sql.DB, schemaName string, tableType TableType) []Table
+	GetEnumsMetaData(db *sql.DB, schemaName string) []Enum
+}
+
+// GetSchema retrieves Schema information from database
+func GetSchema(db *sql.DB, querySet DialectQuerySet, schemaName string) Schema {
+	ret := Schema{
+		Name:           schemaName,
+		TablesMetaData: querySet.GetTablesMetaData(db, schemaName, baseTable),
+		ViewsMetaData:  querySet.GetTablesMetaData(db, schemaName, viewTable),
+		EnumsMetaData:  querySet.GetEnumsMetaData(db, schemaName),
+	}
+
+	fmt.Println("	FOUND", len(ret.TablesMetaData), "table(s),", len(ret.ViewsMetaData), "view(s),",
+		len(ret.EnumsMetaData), "enum(s)")
+
+	return ret
+}

generator/metadata/enum_meta_data.go

@@ -0,0 +1,7 @@
+package metadata
+
+// Enum metadata struct
+type Enum struct {
+	Name   string `sql:"primary_key"`
+	Values []string
+}

generator/metadata/schema_meta_data.go

@@ -0,0 +1,14 @@
+package metadata
+
+// Schema struct
+type Schema struct {
+	Name           string
+	TablesMetaData []Table
+	ViewsMetaData  []Table
+	EnumsMetaData  []Enum
+}
+
+// IsEmpty returns true if schema info does not contain any table, views or enums metadata
+func (s Schema) IsEmpty() bool {
+	return len(s.TablesMetaData) == 0 && len(s.ViewsMetaData) == 0 && len(s.EnumsMetaData) == 0
+}

generator/metadata/table_meta_data.go

@@ -0,0 +1,22 @@
+package metadata
+
+// Table metadata struct
+type Table struct {
+	Name    string
+	Columns []Column
+}
+
+// MutableColumns returns list of mutable columns for table
+func (t Table) MutableColumns() []Column {
+	var ret []Column
+
+	for _, column := range t.Columns {
+		if column.IsPrimaryKey {
+			continue
+		}
+
+		ret = append(ret, column)
+	}
+
+	return ret
+}