vendor dependencies with dep

vendor/github.com/go-pg/pg/orm/create_table.go

@@ -0,0 +1,122 @@
+package orm
+
+import (
+	"errors"
+	"strconv"
+)
+
+type CreateTableOptions struct {
+	Temp          bool
+	IfNotExists   bool
+	Varchar       int  // replaces PostgreSQL data type `text` with `varchar(n)`
+	FKConstraints bool // whether to create foreign key constraints
+}
+
+func CreateTable(db DB, model interface{}, opt *CreateTableOptions) (Result, error) {
+	return NewQuery(db, model).CreateTable(opt)
+}
+
+type createTableQuery struct {
+	q   *Query
+	opt *CreateTableOptions
+}
+
+func (q createTableQuery) Copy() QueryAppender {
+	return q
+}
+
+func (q createTableQuery) Query() *Query {
+	return q.q
+}
+
+func (q createTableQuery) AppendQuery(b []byte) ([]byte, error) {
+	if q.q.stickyErr != nil {
+		return nil, q.q.stickyErr
+	}
+	if q.q.model == nil {
+		return nil, errors.New("pg: Model is nil")
+	}
+
+	table := q.q.model.Table()
+
+	b = append(b, "CREATE "...)
+	if q.opt != nil && q.opt.Temp {
+		b = append(b, "TEMP "...)
+	}
+	b = append(b, "TABLE "...)
+	if q.opt != nil && q.opt.IfNotExists {
+		b = append(b, "IF NOT EXISTS "...)
+	}
+	b = q.q.appendTableName(b)
+	b = append(b, " ("...)
+
+	for i, field := range table.Fields {
+		b = append(b, field.Column...)
+		b = append(b, " "...)
+		if q.opt != nil && q.opt.Varchar > 0 &&
+			field.SQLType == "text" && !field.HasFlag(customTypeFlag) {
+			b = append(b, "varchar("...)
+			b = strconv.AppendInt(b, int64(q.opt.Varchar), 10)
+			b = append(b, ")"...)
+		} else {
+			b = append(b, field.SQLType...)
+		}
+		if field.HasFlag(NotNullFlag) {
+			b = append(b, " NOT NULL"...)
+		}
+		if field.HasFlag(UniqueFlag) {
+			b = append(b, " UNIQUE"...)
+		}
+		if field.Default != "" {
+			b = append(b, " DEFAULT "...)
+			b = append(b, field.Default...)
+		}
+
+		if i != len(table.Fields)-1 {
+			b = append(b, ", "...)
+		}
+	}
+
+	b = appendPKConstraint(b, table.PKs)
+
+	if q.opt != nil && q.opt.FKConstraints {
+		for _, rel := range table.Relations {
+			b = q.appendFKConstraint(b, table, rel)
+		}
+	}
+
+	b = append(b, ")"...)
+
+	return b, nil
+}
+
+func appendPKConstraint(b []byte, pks []*Field) []byte {
+	if len(pks) == 0 {
+		return b
+	}
+
+	b = append(b, ", PRIMARY KEY ("...)
+	b = appendColumns(b, pks)
+	b = append(b, ")"...)
+	return b
+}
+
+func (q createTableQuery) appendFKConstraint(b []byte, table *Table, rel *Relation) []byte {
+	if rel.Type != HasOneRelation {
+		return b
+	}
+
+	b = append(b, ", FOREIGN KEY ("...)
+	b = appendColumns(b, rel.FKs)
+	b = append(b, ")"...)
+
+	b = append(b, " REFERENCES "...)
+	b = q.q.FormatQuery(b, string(rel.JoinTable.Name))
+	b = append(b, " ("...)
+	b = appendColumns(b, rel.JoinTable.PKs)
+	b = append(b, ")"...)
+
+	b = append(b, " ON DELETE CASCADE"...)
+
+	return b
+}