di-go/registry/definition.go

package registry

import (
	"fmt"
	"reflect"

	cda "git.loafle.net/commons/di-go/annotation"
	cur "git.loafle.net/commons/util-go/reflect"
)

type TypeDefinition struct {
	FullName string
	PkgName  string
	TypeName string
	Type     reflect.Type
	RealType reflect.Type

	TypeAnnotations   map[reflect.Type]cda.Annotation
	MethodAnnotations map[string]map[reflect.Type]cda.Annotation
	Fields            []*FieldDefinition
}

func (td *TypeDefinition) GetTypeAnnotationByType(at reflect.Type, includeEmbedding bool) cda.Annotation {
	if nil == td.TypeAnnotations {
		return nil
	}

	if !includeEmbedding {
		return td.TypeAnnotations[at]
	}

	for _, v := range td.TypeAnnotations {
		if at == reflect.TypeOf(v) {
			return v
		}

		if includeEmbedding {
			if checkAnnotation(reflect.TypeOf(v), at) {
				return v
			}
		}
	}

	return nil
}

func (td *TypeDefinition) GetMethodAnnotationByType(at reflect.Type, methodName string) cda.Annotation {
	if nil == td.MethodAnnotations {
		return nil
	}

	ms, ok := td.MethodAnnotations[methodName]
	if !ok {
		return nil
	}

	return ms[at]
}

func (td *TypeDefinition) GetMethodAnnotationsByType(at reflect.Type) map[string]cda.Annotation {
	if nil == td.MethodAnnotations {
		return nil
	}

	mas := make(map[string]cda.Annotation)
	for k, v := range td.MethodAnnotations {
		a, ok := v[at]
		if !ok {
			continue
		}
		mas[k] = a
	}

	return mas
}

func checkAnnotation(t reflect.Type, st reflect.Type) bool {
	rt, _, _ := cur.GetTypeInfo(t)
	if reflect.Struct != rt.Kind() {
		return false
	}

	for i := 0; i < rt.NumField(); i++ {
		f := rt.Field(i)

		if f.Anonymous {
			if f.Type == st {
				return true
			}
			if checkAnnotation(f.Type, st) {
				return true
			}
		}
	}

	return false
}

type FieldDefinition struct {
	FieldName string
	PkgName   string
	TypeName  string
	Type      reflect.Type
	RealType  reflect.Type

	Annotations map[reflect.Type]cda.Annotation
}

func (fd *FieldDefinition) GetAnnotationByType(at reflect.Type, includeEmbedding bool) cda.Annotation {
	if nil == fd.Annotations {
		return nil
	}

	if !includeEmbedding {
		return fd.Annotations[at]
	}

	for _, v := range fd.Annotations {
		if at == reflect.TypeOf(v) {
			return v
		}

		if includeEmbedding {
			if checkAnnotation(reflect.TypeOf(v), at) {
				return v
			}
		}
	}

	return nil
}

func FullName(pkgName, typeName string) string {
	return fmt.Sprintf("%s/%s", pkgName, typeName)
}