Add Dependencies info to README.md.

internal/util/utils.go

@@ -1,25 +0,0 @@
-package util
-
-import (
-	"github.com/go-jet/jet/internal/3rdparty/snaker"
-	"strings"
-)
-
-func ToGoIdentifier(databaseIdentifier string) string {
-	if len(databaseIdentifier) == 0 {
-		return databaseIdentifier
-	}
-
-	return snaker.SnakeToCamel(replaceInvalidChars(databaseIdentifier))
-}
-
-func ToGoFileName(databaseIdentifier string) string {
-	return strings.ToLower(replaceInvalidChars(databaseIdentifier))
-}
-
-func replaceInvalidChars(str string) string {
-	str = strings.Replace(str, " ", "_", -1)
-	str = strings.Replace(str, "-", "_", -1)
-
-	return str
-}

internal/utils/utils.go

@@ -0,0 +1,154 @@
+package utils
+
+import (
+	"bytes"
+	"github.com/go-jet/jet/internal/3rdparty/snaker"
+	"go/format"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"text/template"
+	"time"
+)
+
+func ToGoIdentifier(databaseIdentifier string) string {
+	if len(databaseIdentifier) == 0 {
+		return databaseIdentifier
+	}
+
+	return snaker.SnakeToCamel(replaceInvalidChars(databaseIdentifier))
+}
+
+func ToGoFileName(databaseIdentifier string) string {
+	return strings.ToLower(replaceInvalidChars(databaseIdentifier))
+}
+
+func SaveGoFile(dirPath, fileName string, text []byte) error {
+	newGoFilePath := filepath.Join(dirPath, fileName) + ".go"
+
+	file, err := os.Create(newGoFilePath)
+
+	if err != nil {
+		return err
+	}
+
+	defer file.Close()
+
+	p, err := format.Source(text)
+	if err != nil {
+		return err
+	}
+
+	_, err = file.Write(p)
+
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func EnsureDirPath(dirPath string) error {
+	if _, err := os.Stat(dirPath); os.IsNotExist(err) {
+		err := os.MkdirAll(dirPath, os.ModePerm)
+
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func GenerateTemplate(templateText string, templateData interface{}) ([]byte, error) {
+
+	t, err := template.New("sqlBuilderTableTemplate").Funcs(template.FuncMap{
+		"ToGoIdentifier": ToGoIdentifier,
+		"now": func() string {
+			return time.Now().Format(time.RFC850)
+		},
+	}).Parse(templateText)
+
+	if err != nil {
+		return nil, err
+	}
+
+	var buf bytes.Buffer
+	if err := t.Execute(&buf, templateData); err != nil {
+		return nil, err
+	}
+
+	return buf.Bytes(), nil
+}
+
+func CleanUpGeneratedFiles(dir string) error {
+	exist, err := DirExists(dir)
+
+	if err != nil {
+		return err
+	}
+
+	if exist {
+		err := os.RemoveAll(dir)
+
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func DirExists(path string) (bool, error) {
+	_, err := os.Stat(path)
+	if err == nil {
+		return true, nil
+	}
+	if os.IsNotExist(err) {
+		return false, nil
+	}
+	return true, err
+}
+
+func replaceInvalidChars(str string) string {
+	str = strings.Replace(str, " ", "_", -1)
+	str = strings.Replace(str, "-", "_", -1)
+
+	return str
+}
+
+// github.com/lib/pq
+// FormatTimestamp formats t into Postgres' text format for timestamps.
+func FormatTimestamp(t time.Time) []byte {
+	// Need to send dates before 0001 A.D. with " BC" suffix, instead of the
+	// minus sign preferred by Go.
+	// Beware, "0000" in ISO is "1 BC", "-0001" is "2 BC" and so on
+	bc := false
+	if t.Year() <= 0 {
+		// flip year sign, and add 1, e.g: "0" will be "1", and "-10" will be "11"
+		t = t.AddDate((-t.Year())*2+1, 0, 0)
+		bc = true
+	}
+	b := []byte(t.Format("2006-01-02 15:04:05.999999999Z07:00"))
+
+	_, offset := t.Zone()
+	offset = offset % 60
+	if offset != 0 {
+		// RFC3339Nano already printed the minus sign
+		if offset < 0 {
+			offset = -offset
+		}
+
+		b = append(b, ':')
+		if offset < 10 {
+			b = append(b, '0')
+		}
+		b = strconv.AppendInt(b, int64(offset), 10)
+	}
+
+	if bc {
+		b = append(b, " BC"...)
+	}
+	return b
+}

internal/utils/utils_test.go

@@ -1,4 +1,4 @@
-package util
+package utils
 
 import (
 	"github.com/stretchr/testify/assert"