package wrapper

import (
	"reflect"
	"strconv"
	"strings"
)

func AddTable(builder *strings.Builder, table string) {
	if table == "" {
		return
	}
	AddQuoted(builder, table)
	builder.WriteByte('.')
}

func IsQuoted(v string) bool {
	if v == "" {
		return false
	}
	first := v[0]
	end := v[len(v)-1]
	return first == '`' && end == '`'
}

func AddQuoted(builder *strings.Builder, v string) bool {
	if v == "" {
		return false
	}
	first := v[0]
	end := v[len(v)-1]
	if first != '`' { //引号
		builder.WriteByte('`')
	}
	builder.WriteString(v)
	if end != '`' {
		builder.WriteByte('`')
	}
	return true
}

func AddAlias(builder *strings.Builder, v string) bool {
	if v == "" {
		return false
	}
	builder.WriteString(" AS ")
	builder.WriteString(v)
	return true
}

func AddVar(builder *strings.Builder, val interface{}) {
	ref := reflect.Indirect(reflect.ValueOf(val))

	switch ref.Kind() {

	case reflect.String:
		s := ref.String()
		s = strings.ReplaceAll(s, "\"", "\\\"")
		s = strings.ReplaceAll(s, "'", "\\'")
		builder.WriteByte('\'')
		builder.WriteString(s)
		builder.WriteByte('\'')

	case reflect.Int:
		fallthrough
	case reflect.Int8:
		fallthrough
	case reflect.Int16:
		fallthrough
	case reflect.Int32:
		fallthrough
	case reflect.Int64:
		builder.WriteString(strconv.FormatInt(ref.Int(), 10))

	case reflect.Uint:
		fallthrough
	case reflect.Uint8:
		fallthrough
	case reflect.Uint16:
		fallthrough
	case reflect.Uint32:
		fallthrough
	case reflect.Uint64:
		builder.WriteString(strconv.FormatUint(ref.Uint(), 10))

	case reflect.Float32:
		fallthrough
	case reflect.Float64:
		builder.WriteString(strconv.FormatFloat(ref.Float(), 'f', 5, 64))

	case reflect.Bool:
		if ref.Bool() {
			builder.WriteString("true")
		} else {
			builder.WriteString("false")
		}

	case reflect.Struct:
		if sql, ok := val.(*SQL); ok {
			builder.WriteString(sql.Build())
			return
		}
		if sql, ok := val.(SQL); ok {
			builder.WriteString(sql.Build())
		}
	default:
	}
}