From 5c12764c5fdde32dfcbf615cbd3d0960dca9b7b7 Mon Sep 17 00:00:00 2001
From: Sergey Lanzman <Sergeylanz@gettaxi.com>
Date: Thu, 7 Jul 2016 15:21:47 +0300
Subject: [PATCH] add Postgresql support generate(migrate,Scaffold..)

---
 g_scaffold.go | 56 +++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 52 insertions(+), 4 deletions(-)

diff --git a/g_scaffold.go b/g_scaffold.go
index a9df9d8..db57767 100644
--- a/g_scaffold.go
+++ b/g_scaffold.go
@@ -51,25 +51,46 @@ func generateSQLFromFields(fields string) string {
 			ColorLog("[ERRO] the fields format is wrong. should key:type,key:type " + v + "\n")
 			return ""
 		}
-		typ, tag := getSqlType(kv[1])
+		typ, tag := "", ""
+		switch driver {
+		case "mysql":
+			typ, tag = getSqlTypeMysql(kv[1])
+		case "postgres":
+			typ, tag = getSqlTypePostgresql(kv[1])
+		default:
+			typ, tag = getSqlTypeMysql(kv[1])
+		}
 		if typ == "" {
 			ColorLog("[ERRO] the fields format is wrong. should key:type,key:type " + v + "\n")
 			return ""
 		}
 		if i == 0 && strings.ToLower(kv[0]) != "id" {
-			sql = sql + "`id` int(11) NOT NULL AUTO_INCREMENT,"
-			tags = tags + "PRIMARY KEY (`id`),"
+			switch driver {
+			case "mysql":
+				sql = sql + "`id` int(11) NOT NULL AUTO_INCREMENT,"
+				tags = tags + "PRIMARY KEY (`id`),"
+			case "postgres":
+				sql = sql + "id interger serial primary key,"
+			default:
+				sql = sql + "`id` int(11) NOT NULL AUTO_INCREMENT,"
+				tags = tags + "PRIMARY KEY (`id`),"
+			}
 		}
+
 		sql = sql + "`" + snakeString(kv[0]) + "` " + typ + ","
 		if tag != "" {
 			tags = tags + fmt.Sprintf(tag, "`"+snakeString(kv[0])+"`") + ","
 		}
 	}
+	if driver == "postgres" {
+		sql = strings.Replace(sql, "`", "", -1)
+		tags = strings.Replace(tags, "`", "", -1)
+	}
 	sql = strings.TrimRight(sql+tags, ",")
 	return sql
 }
 
-func getSqlType(ktype string) (tp, tag string) {
+func getSqlTypeMysql(ktype string) (tp, tag string) {
 	kv := strings.SplitN(ktype, ":", 2)
 	switch kv[0] {
 	case "string":
@@ -99,3 +120,30 @@ func getSqlType(ktype string) (tp, tag string) {
 	}
 	return "", ""
 }
+
+func getSqlTypePostgresql(ktype string) (tp, tag string) {
+	kv := strings.SplitN(ktype, ":", 2)
+	switch kv[0] {
+	case "string":
+		if len(kv) == 2 {
+			return "char(" + kv[1] + ") NOT NULL", ""
+		} else {
+			return "TEXT NOT NULL", ""
+		}
+	case "text":
+		return "TEXT NOT NULL", ""
+	case "auto", "pk":
+		return "serial primary key", ""
+	case "datetime":
+		return "TIMESTAMP WITHOUT TIME ZONE NOT NULL", ""
+	case "int", "int8", "int16", "int32", "int64":
+		fallthrough
+	case "uint", "uint8", "uint16", "uint32", "uint64":
+		return "integer DEFAULT NULL", ""
+	case "bool":
+		return "boolean NOT NULL", ""
+	case "float32", "float64", "float":
+		return "numeric NOT NULL", ""
+	}
+	return "", ""
+}