codegen: add helpers

pkg/schema/table.go

@@ -1,6 +1,10 @@
 package schema
 
-import "strings"
+import (
+	"strings"
+
+	"git.offline-twitter.com/offline-labs/gas-stack/pkg/textutils"
+)
 
 // Column represents a single column in a table.
 type Column struct {
@@ -26,6 +30,39 @@ func (c Column) IsNonCodeTableForeignKey() bool {
 	return c.IsForeignKey && strings.HasSuffix(c.Name, "_id")
 }
 
+func (c Column) GoFieldName() string {
+	if c.Name == "rowid" {
+		return "ID"
+	}
+	if c.IsNonCodeTableForeignKey() {
+		return textutils.SnakeToCamel(strings.TrimSuffix(c.Name, "_id")) + "ID"
+	}
+	return textutils.SnakeToCamel(c.Name)
+}
+
+func (c Column) GoType() string {
+	if c.IsNonCodeTableForeignKey() {
+		return TypenameFromTablename(c.ForeignKeyTargetTable) + "ID"
+	}
+	switch c.Type {
+	case "integer", "int":
+		if strings.HasPrefix(c.Name, "is_") || strings.HasPrefix(c.Name, "has_") {
+			return "bool"
+		} else if strings.HasSuffix(c.Name, "_at") {
+			return "Timestamp"
+		}
+		return "int"
+	case "text":
+		return "string"
+	case "real":
+		return "float32"
+	case "blob":
+		return "[]byte"
+	default:
+		panic("Unrecognized sqlite column type: " + c.Type)
+	}
+}
+
 // Table is a single SQLite table.
 type Table struct {
 	TableName string `db:"name"`