bob/drop.go

package bob

import (
	"bytes"
	"errors"

	"github.com/lann/builder"
)

type DropBuilder builder.Builder

type dropData struct {
	TableName string
	IfExists bool
}

func init() {
	builder.Register(DropBuilder{}, dropData{})
}

// DropTable sets which table to be dropped
func (b DropBuilder) DropTable(name string) DropBuilder {
	return builder.Set(b, "TableName", name).(DropBuilder)
}

func (b DropBuilder) IfExists() DropBuilder {
	return builder.Set(b, "IfExists", true).(DropBuilder)
}

// ToSql returns 3 variables filled out with the correct values based on bindings, etc.
func (b DropBuilder) ToSql() (string, []interface{}, error) {
	data := builder.GetStruct(b).(dropData)
	return data.ToSql()
}

// ToSql returns 3 variables filled out with the correct values based on bindings, etc.
func (d *dropData) ToSql() (sqlStr string, args []interface{}, err error) {
	if len(d.TableName) == 0 || d.TableName == "" {
		err = errors.New("drop statement must specify a table")
	}
	sql := &bytes.Buffer{}

	sql.WriteString("DROP TABLE ")

	if d.IfExists {
		sql.WriteString("IF EXISTS ")
	}
	
	sql.WriteString("\""+d.TableName+"\";")

	sqlStr = sql.String()
	return
}