gosora/query_gen/querygen.go

/* WIP Under Construction */
package qgen // import "github.com/Azareal/Gosora/query_gen"

import (
	"database/sql"
	"errors"
	"strconv"
	"strings"
	"sync"
)

var Registry []Adapter
var ErrNoAdapter = errors.New("This adapter doesn't exist")
var queryStrPool = sync.Pool{}

type DBTableColumn struct {
	Name          string
	Type          string
	Size          int
	Null          bool
	AutoIncrement bool
	Default       string
}

type DBTableKey struct {
	Columns string
	Type    string

	// Foreign keys only
	FTable  string
	Cascade bool
}

type DBSelect struct {
	Table   string
	Columns string
	Where   string
	Orderby string
	Limit   string
}

type DBJoin struct {
	Table1  string
	Table2  string
	Columns string
	Joiners string
	Where   string
	Orderby string
	Limit   string
}

type DBInsert struct {
	Table   string
	Columns string
	Fields  string
}

type DBColumn struct {
	Table string
	Left  string // Could be a function or a column, so I'm naming this Left
	Alias string // aka AS Blah, if it's present
	//Type  string // function or column
	Type int
}

const (
	IdenFunc = iota
	IdenColumn
	IdenString
	IdenLiteral
)

type DBField struct {
	Name string
	//Type string
	Type int
}

type DBWhere struct {
	Expr []DBToken // Simple expressions, the innards of functions are opaque for now.
}

type DBJoiner struct {
	LeftTable   string
	LeftColumn  string
	RightTable  string
	RightColumn string
	Operator    string
}

type DBOrder struct {
	Column string
	Order  string
}

const (
	TokenFunc = iota
	TokenOp
	TokenColumn
	TokenNumber
	TokenString
	TokenSub
	TokenOr
	TokenNot
	TokenLike
)

type DBToken struct {
	Contents string
	//Type     string // function, op, column, number, string, sub, not, like
	Type int
}

type DBSetter struct {
	Column string
	Expr   []DBToken // Simple expressions, the innards of functions are opaque for now.
}

type DBLimit struct {
	Offset   string // ? or int
	MaxCount string // ? or int
}

type DBStmt struct {
	Contents string
	Type     string // create-table, insert, update, delete
}

// TODO: Add the DropTable, TableExists, AddColumn, ColumnExists, and RemoveColumn methods
type Adapter interface {
	GetName() string
	BuildConn(config map[string]string) (*sql.DB, error)
	DbVersion() string

	DropTable(name, table string) (string, error)
	CreateTable(name, table, charset, collation string, cols []DBTableColumn, keys []DBTableKey) (string, error)
	// TODO: Some way to add indices and keys
	// TODO: Test this
	AddColumn(name, table string, col DBTableColumn, key *DBTableKey) (string, error)
	DropColumn(name, table, colName string) (string, error)
	RenameColumn(name, table, oldName, newName string) (string, error)
	ChangeColumn(name, table, colName string, col DBTableColumn) (string, error)
	SetDefaultColumn(name, table, colName, colType, defaultStr string) (string, error)
	AddIndex(name, table, iname, colname string) (string, error)
	AddKey(name, table, col string, key DBTableKey) (string, error)
	RemoveIndex(name, table, col string) (string, error)
	AddForeignKey(name, table, col, ftable, fcol string, cascade bool) (out string, e error)
	SimpleInsert(name, table, cols, fields string) (string, error)
	SimpleBulkInsert(name, table, cols string, fieldSet []string) (string, error)
	SimpleUpdate(b *updatePrebuilder) (string, error)
	SimpleUpdateSelect(b *updatePrebuilder) (string, error) // ! Experimental
	SimpleDelete(name, table, where string) (string, error)
	Purge(name, table string) (string, error)
	SimpleSelect(name, table, cols, where, orderby, limit string) (string, error)
	ComplexDelete(b *deletePrebuilder) (string, error)
	SimpleLeftJoin(name, table1, table2, cols, joiners, where, orderby, limit string) (string, error)
	SimpleInnerJoin(string, string, string, string, string, string, string, string) (string, error)
	SimpleInsertSelect(string, DBInsert, DBSelect) (string, error)
	SimpleInsertLeftJoin(string, DBInsert, DBJoin) (string, error)
	SimpleInsertInnerJoin(string, DBInsert, DBJoin) (string, error)
	SimpleCount(string, string, string, string) (string, error)

	ComplexSelect(*selectPrebuilder) (string, error)

	Builder() *prebuilder
	Write() error
}

func GetAdapter(name string) (adap Adapter, err error) {
	for _, adapter := range Registry {
		if adapter.GetName() == name {
			return adapter, nil
		}
	}
	return adap, ErrNoAdapter
}

type QueryPlugin interface {
	Hook(name string, args ...interface{}) error
	Write() error
}

type MySQLUpsertCallback struct {
	stmt *sql.Stmt
}

func (double *MySQLUpsertCallback) Exec(args ...interface{}) (res sql.Result, err error) {
	if len(args) < 2 {
		return res, errors.New("Need two or more arguments")
	}
	args = args[:len(args)-1]
	return double.stmt.Exec(append(args, args...)...)
}

func PrepareMySQLUpsertCallback(db *sql.DB, query string) (*MySQLUpsertCallback, error) {
	stmt, err := db.Prepare(query)
	return &MySQLUpsertCallback{stmt}, err
}

type LitStr string

// TODO: Test this
func InterfaceMapToInsertStrings(data map[string]interface{}, order string) (cols, values string) {
	done := make(map[string]bool)
	addValue := func(value interface{}) {
		switch value := value.(type) {
		case string:
			values += "'" + strings.Replace(value, "'", "\\'", -1) + "',"
		case int:
			values += strconv.Itoa(value) + ","
		case LitStr:
			values += string(value) + ","
		case bool:
			if value {
				values += "1,"
			} else {
				values += "0,"
			}
		}
	}

	// Add the ordered items
	for _, col := range strings.Split(order, ",") {
		col = strings.TrimSpace(col)
		value, ok := data[col]
		if ok {
			cols += col + ","
			addValue(value)
			done[col] = true
		}
	}

	// Go over any unordered items and add them at the end
	if len(data) > len(done) {
		for col, value := range data {
			_, ok := done[col]
			if ok {
				continue
			}
			cols += col + ","
			addValue(value)
		}
	}

	if cols != "" {
		cols = cols[:len(cols)-1]
		values = values[:len(values)-1]
	}
	return cols, values
}