Initial commit

backend/pkgs/automapper/automapper.go

@@ -0,0 +1,92 @@
+package automapper
+
+import (
+	"bytes"
+	"fmt"
+	"go/format"
+	"os"
+	"reflect"
+	"strings"
+	"text/template"
+)
+
+type FieldAssignment struct {
+	ModelField  string
+	SchemaField string
+}
+
+type Model struct {
+	Type      interface{}
+	Prefix    string
+	Fields    []reflect.StructField
+	Reference string
+}
+
+type Schema struct {
+	Name      string
+	Type      interface{}
+	Prefix    string
+	Fields    []reflect.StructField
+	Reference string
+}
+
+type AutoMapper struct {
+	Name             string
+	Package          string
+	Prefix           string
+	Schema           Schema
+	Model            Model
+	Imports          []string
+	FieldAssignments []FieldAssignment
+}
+
+func (mapper *AutoMapper) ExecuteTemplates(conf *AutoMapperConf) {
+	t := template.New("automapper")
+	t, err := t.Parse(automapperTemplate)
+	if err != nil {
+		fmt.Println(err)
+	}
+
+	// Ensure the output directory exists
+	os.MkdirAll(conf.OutDir, 0755)
+
+	var path = fmt.Sprintf("%s/%s", conf.OutDir, mapper.GetFileName())
+
+	f, err := os.Create(path)
+	if err != nil {
+		panic(err)
+	}
+	defer f.Close()
+
+	var buf bytes.Buffer
+
+	err = t.Execute(&buf, mapper)
+	if err != nil {
+		fmt.Println(err)
+	}
+
+	text, err := format.Source(buf.Bytes())
+
+	if err != nil {
+		fmt.Println(err)
+	}
+
+	f.Write(text)
+
+}
+
+// GetFileName returns the computed file name based off user preference.
+// If the Prefix has been specified on the AutoMapper it will be used
+// in place of the Struct name. If the Prefix is not specified, the
+// Struct name will be used.
+//
+// Examples:
+// prefix_automapper.go
+// mystructname_automapper.go
+func (mapper *AutoMapper) GetFileName() string {
+	if mapper.Prefix == "" {
+		return strings.ToLower(mapper.Schema.Reference) + "_" + "automapper.go"
+	}
+	return strings.ToLower(mapper.Prefix) + "_" + "automapper.go"
+
+}