jet/qrm/utill.go

package qrm

import (
	"database/sql"
	"fmt"
	"github.com/go-jet/jet/v2/internal/utils/must"
	"github.com/go-jet/jet/v2/internal/utils/strslice"
	"github.com/go-jet/jet/v2/qrm/internal"
	"github.com/google/uuid"
	"reflect"
	"strings"
	"time"
)

var scannerInterfaceType = reflect.TypeOf((*sql.Scanner)(nil)).Elem()

func implementsScannerType(fieldType reflect.Type) bool {
	if fieldType.Implements(scannerInterfaceType) {
		return true
	}

	fieldTypePtr := reflect.New(fieldType).Type()

	return fieldTypePtr.Implements(scannerInterfaceType)
}

func getScanner(value reflect.Value) sql.Scanner {
	if scanner, ok := value.Interface().(sql.Scanner); ok {
		return scanner
	}

	return value.Addr().Interface().(sql.Scanner)
}

func getSliceElemType(slicePtrValue reflect.Value) reflect.Type {
	sliceTypePtr := slicePtrValue.Type()
	elemType := indirectType(sliceTypePtr).Elem()

	return indirectType(elemType)
}

func getSliceElemPtrAt(slicePtrValue reflect.Value, index int) reflect.Value {
	sliceValue := slicePtrValue.Elem()
	elem := sliceValue.Index(index)

	if elem.Kind() == reflect.Ptr {
		return elem
	}

	return elem.Addr()
}

func appendElemToSlice(slicePtrValue reflect.Value, objPtrValue reflect.Value) error {
	must.BeTrue(!slicePtrValue.IsNil(), "jet: internal, slice is nil")

	sliceValue := slicePtrValue.Elem()
	sliceElemType := sliceValue.Type().Elem()

	var newSliceElemValue reflect.Value

	if objPtrValue.Type().AssignableTo(sliceElemType) {
		newSliceElemValue = objPtrValue
	} else if objPtrValue.Elem().Type().AssignableTo(sliceElemType) {
		newSliceElemValue = objPtrValue.Elem()
	} else {
		newSliceElemValue = reflect.New(sliceElemType).Elem()

		var err error

		if newSliceElemValue.Kind() == reflect.Ptr {
			newSliceElemValue.Set(reflect.New(newSliceElemValue.Type().Elem()))
			err = assign(objPtrValue.Elem(), newSliceElemValue.Elem())
		} else {
			err = assign(objPtrValue.Elem(), newSliceElemValue)
		}

		if err != nil {
			return fmt.Errorf("can't append %T to %T slice: %w", objPtrValue.Elem().Interface(), sliceValue.Interface(), err)
		}
	}

	sliceValue.Set(reflect.Append(sliceValue, newSliceElemValue))

	return nil
}

func newElemPtrValueForSlice(slicePtrValue reflect.Value) reflect.Value {
	destinationSliceType := slicePtrValue.Type().Elem()
	elemType := indirectType(destinationSliceType.Elem())

	return reflect.New(elemType)
}

func getTypeName(structType reflect.Type, parentField *reflect.StructField) string {
	if parentField == nil {
		return structType.Name()
	}

	aliasTag := parentField.Tag.Get("alias")

	if aliasTag == "" {
		return structType.Name()
	}

	aliasParts := strings.Split(aliasTag, ".")

	return toCommonIdentifier(aliasParts[0])
}

func getTypeAndFieldName(structType string, field reflect.StructField) (string, string) {
	aliasTag := field.Tag.Get("alias")

	if aliasTag == "" {
		return structType, field.Name
	}

	aliasParts := strings.Split(aliasTag, ".")

	if len(aliasParts) == 1 {
		return structType, toCommonIdentifier(aliasParts[0])
	}

	return toCommonIdentifier(aliasParts[0]), toCommonIdentifier(aliasParts[1])
}

var replacer = strings.NewReplacer(" ", "", "-", "", "_", "")

func toCommonIdentifier(name string) string {
	return strings.ToLower(replacer.Replace(name))
}

func initializeValueIfNilPtr(value reflect.Value) {
	if !value.IsValid() || !value.CanSet() {
		return
	}

	if value.Kind() == reflect.Ptr && value.IsNil() {
		value.Set(reflect.New(value.Type().Elem()))
	}
}

var timeType = reflect.TypeOf(time.Now())
var uuidType = reflect.TypeOf(uuid.New())
var byteArrayType = reflect.TypeOf([]byte(""))

func isSimpleModelType(objType reflect.Type) bool {
	objType = indirectType(objType)

	switch objType.Kind() {
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64,
		reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64,
		reflect.Float32, reflect.Float64,
		reflect.String,
		reflect.Bool:
		return true
	}

	return objType == timeType || objType == uuidType || objType == byteArrayType
}

// source can't be pointer
// destination can be pointer
func assign(source, destination reflect.Value) error {
	if destination.Kind() == reflect.Ptr {
		if destination.IsNil() {
			initializeValueIfNilPtr(destination)
		}

		destination = destination.Elem()
	}

	err := tryAssign(source, destination)

	if err != nil {
		// needs for the type conversions are rare, so we leave conversion as a last assign step if everything else fails
		if tryConvert(source, destination) {
			return nil
		}

		return err
	}

	return nil
}

func assignIfAssignable(source, destination reflect.Value) bool {
	sourceType := source.Type()
	if sourceType.AssignableTo(destination.Type()) {
		switch sourceType {
		case byteArrayType:
			destination.SetBytes(cloneBytes(source.Interface().([]byte)))
		default:
			destination.Set(source)
		}
		return true
	}

	return false
}

// source and destination are non-ptr values
func tryAssign(source, destination reflect.Value) error {

	if assignIfAssignable(source, destination) {
		return nil
	}

	sourceInterface := source.Interface()

	switch destination.Type().Kind() {
	case reflect.Bool:
		var nullBool internal.NullBool

		err := nullBool.Scan(sourceInterface)

		if err != nil {
			return err
		}

		destination.SetBool(nullBool.Bool)

	case reflect.Float32, reflect.Float64:
		var nullFloat sql.NullFloat64

		err := nullFloat.Scan(sourceInterface)
		if err != nil {
			return err
		}

		if nullFloat.Valid {
			destination.SetFloat(nullFloat.Float64)
		}
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		var integer sql.NullInt64

		err := integer.Scan(sourceInterface)
		if err != nil {
			return err
		}

		if integer.Valid {
			destination.SetInt(integer.Int64)
		}

	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
		var uInt internal.NullUInt64

		err := uInt.Scan(sourceInterface)

		if err != nil {
			return err
		}

		if uInt.Valid {
			destination.SetUint(uInt.UInt64)
		}

	case reflect.String:
		var str sql.NullString

		err := str.Scan(sourceInterface)
		if err != nil {
			return err
		}

		if str.Valid {
			destination.SetString(str.String)
		}

	default:
		switch destination.Interface().(type) {
		case time.Time:
			var nullTime internal.NullTime

			err := nullTime.Scan(sourceInterface)
			if err != nil {
				return err
			}

			if nullTime.Valid {
				destination.Set(reflect.ValueOf(nullTime.Time))
			}
		default:
			return fmt.Errorf("can't assign %T to %T", sourceInterface, destination.Interface())
		}
	}

	return nil
}

func tryConvert(source, destination reflect.Value) bool {
	destinationType := destination.Type()

	if source.Type().ConvertibleTo(destinationType) {
		source = source.Convert(destinationType)
		return assignIfAssignable(source, destination)
	}

	return false
}

func setZeroValue(value reflect.Value) {
	if !value.IsZero() {
		value.Set(reflect.Zero(value.Type()))
	}
}

func isPrimaryKey(field reflect.StructField, primaryKeyOverwrites []string) bool {
	if len(primaryKeyOverwrites) > 0 {
		return strslice.Contains(primaryKeyOverwrites, field.Name)
	}

	sqlTag := field.Tag.Get("sql")

	return sqlTag == "primary_key"
}

func parentFieldPrimaryKeyOverwrite(parentField *reflect.StructField) []string {
	if parentField == nil {
		return nil
	}

	sqlTag := parentField.Tag.Get("sql")

	if !strings.HasPrefix(sqlTag, "primary_key") {
		return nil
	}

	parts := strings.Split(sqlTag, "=")

	if len(parts) < 2 {
		return nil
	}

	return strings.Split(parts[1], ",")
}

func indirectType(reflectType reflect.Type) reflect.Type {
	if reflectType.Kind() != reflect.Ptr {
		return reflectType
	}
	return reflectType.Elem()
}

func fieldToString(field *reflect.StructField) string {
	if field == nil {
		return ""
	}

	return " at '" + field.Name + " " + field.Type.String() + "'"
}

func cloneBytes(b []byte) []byte {
	if b == nil {
		return nil
	}
	c := make([]byte, len(b))
	copy(c, b)
	return c
}

func concat(stringList ...string) string {
	var b strings.Builder

	var length int
	for i := 0; i < len(stringList); i++ {
		length += len(stringList[i])
	}

	b.Grow(length)

	for _, str := range stringList {
		b.WriteString(str)
	}
	return b.String()
}