condition.go

package dbr

import (
	"fmt"
	"strings"
)

type condition struct {
	operator operator
	query    interface{}
	values   []interface{}

	db *db
}

func newCondition(db *db, operator operator, query interface{}, values ...interface{}) *condition {
	return &condition{operator: OperatorAnd, query: query, values: values, db: db}
}

func (c *condition) Build(db ...*db) (query string, err error) {

	if len(db) > 0 {
		c.db = db[0]
	}

	switch stmt := c.query.(type) {
	case *StmtSelect:
		query, err = stmt.Build()
		if err != nil {
			return "", err
		}
		query = fmt.Sprintf("(%s)", query)
	default:
		if impl, ok := stmt.(iFunction); ok {
			query, err = impl.Build(c.db)
			if err != nil {
				return "", err
			}
		} else {
			query = fmt.Sprintf("%+v", stmt)
		}
	}

	if strings.Count(query, c.db.Dialect.Placeholder()) != len(c.values) {
		return "", ErrorNumberOfConditionValues
	}

	var value interface{}

	for _, v := range c.values {

		switch stmt := v.(type) {
		case *StmtSelect:
			value, err = stmt.Build()
			if err != nil {
				return "", err
			}
			value = fmt.Sprintf("(%s)", value)
		default:
			if impl, ok := stmt.(iFunction); ok {
				value, err = impl.Build(c.db)
				if err != nil {
					return "", err
				}
			} else {
				value = stmt
			}
		}

		query = strings.Replace(query, c.db.Dialect.Placeholder(), c.db.Dialect.Encode(value), 1)
	}

	return query, nil
}