package bean_factory

import (
	"gitee.com/kristas/booting-go/framework/common/util/ccmap"
	"gitee.com/kristas/booting-go/framework/common/util/reflectx"
	. "gitee.com/kristas/booting-go/framework/core/bean"
	"gitee.com/kristas/booting-go/framework/logging"
	"reflect"
)

var (
	beanCache *ccmap.ConcurrentMap
	log       logging.Logger
	injectors []injectProcessor
)

func init() {
	beanCache = ccmap.NewConcurrentMap()
	injectors = []injectProcessor{
		new(beanTagInjector),
		new(emptyTagPtrInjector),
		new(InterfaceInjector),
		new(InterfaceSliceInjector),
	}
}

func AddBean(bean Bean) {
	beanName := GetBeanName(bean)
	Add(beanName, bean)
}

func Add(beanName string, bean interface{}) {
	log = logging.GetLog()
	saveOrUpdateBean(beanName, bean)
	log.Tracef("bean registered: name=%s, type=%s", beanName, reflect.TypeOf(bean).String())
}

func SetBean(bean Bean) {
	beanName := GetBeanName(bean)
	Set(beanName, bean)
}

func Set(beanName string, bean interface{}) {
	beanCache.Set(beanName, bean)
}

func saveOrUpdateBean(beanName string, bean interface{}) {
	if ExistsBean(beanName) {
		log.Tracef("bean [%s] exists, try update", beanName)
		oldBean := GetBean(beanName)
		oldVal := reflect.ValueOf(oldBean).Elem()
		newVal := reflect.ValueOf(bean).Elem()
		oldVal.Set(newVal)
		log.Tracef("bean [%s] updated", beanName)
	} else {
		beanCache.Set(beanName, bean)
	}
}

func Autowired(bean interface{}) {
	reflectx.WalkStructField(bean, func(field reflect.StructField, value reflect.Value) {
		err := runInjectProcessor(injectors, field, value)
		if err != nil {
			log.Errorf("Inject value in %s failed: field[name=%s, type=%s]\ncaused by: %s",
				reflect.TypeOf(bean).String(),
				GetFieldBeanName(field), field.Type.String(),
				err.Error())
		}
	}, true)
}

func GetFieldBeanName(field reflect.StructField) string {
	var beanName string
	if beanName = field.Tag.Get("bean"); beanName == "" {
		if field.Type.Kind() == reflect.Ptr {
			beanName = field.Type.Elem().String()
		} else {
			beanName = field.Type.String()
		}
	}
	return beanName
}

func GetBean(beanName string) interface{} {
	return beanCache.Get(beanName)
}

func ExistsBean(beanName string) bool {
	return beanCache.Exists(beanName)
}