package bean_factory

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

var factoryBeanMap *ccmap.ConcurrentMap
var log = new(logrus.Logger)

func init() {
	factoryBeanMap = ccmap.NewConcurrentMap()
}

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

func Add(beanName string, bean interface{}) {
	logging.GetLog(log)
	saveOrUpdateBean(beanName, bean)
	log.WithFields(logrus.Fields{
		"Name": beanName,
		"Type": reflect.TypeOf(bean).String(),
	}).Trace("bean registered")
}

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

func Set(beanName string, bean interface{}) {
	factoryBeanMap.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 {
		factoryBeanMap.Set(beanName, bean)
	}
}

func Autowired(bean interface{}) {
	reflectx.WalkStructField(bean, func(field reflect.StructField, value reflect.Value) {
		if canAutowired(field, value) {
			var beanName = GetFieldBeanName(field)
			beanValue := GetBean(beanName)
			if beanValue != nil {
				value.Set(reflect.ValueOf(beanValue))
			} else {
				log.WithFields(logrus.Fields{
					"beanName": beanName,
					"beanType": field.Type.String(),
				}).Error("bean not found")
			}
		}
	})
}

func canAutowired(field reflect.StructField, value reflect.Value) bool {
	if field.Type.Kind() == reflect.Ptr && value.CanSet() {
		return true
	}
	if _, ok := field.Tag.Lookup("bean"); ok {
		return true
	}
	return false
}

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

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

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