package bean_factory

import (
	. "gitee.com/kristas/booting-go/framework/bean"
	"gitee.com/kristas/booting-go/framework/logging"
	"github.com/sirupsen/logrus"
	"reflect"
	"sync"
)

var factoryBeanMap ConcurrentBeanMap
var log = new(logrus.Logger)

func init() {
	factoryBeanMap = ConcurrentBeanMap{
		beans: make(map[string]interface{}),
	}
	logging.GetLog(log)
}

type ConcurrentBeanMap struct {
	beans  map[string]interface{}
	locker sync.RWMutex
}

func (m *ConcurrentBeanMap) set(k string, v interface{}) {
	m.locker.Lock()
	defer m.locker.Unlock()
	m.beans[k] = v
}

func (m *ConcurrentBeanMap) get(k string) interface{} {
	m.locker.RLock()
	defer m.locker.RUnlock()
	return m.beans[k]
}

func (m *ConcurrentBeanMap) exists(k string) bool {
	m.locker.RLock()
	defer m.locker.RUnlock()
	_, ok := m.beans[k]
	return ok
}

func Register(bean interface{}) {
	beanName := DefaultBeanName(bean)
	factoryBeanMap.set(beanName, bean)
	log.WithFields(logrus.Fields{
		"beanName": beanName,
		"beanType": reflect.TypeOf(bean).String(),
	}).Debug("bean registered")
}

func RegisterWithName(beanName string, bean interface{}) {
	factoryBeanMap.set(beanName, bean)
	log.WithFields(logrus.Fields{
		"beanName": beanName,
		"beanType": reflect.TypeOf(bean).String(),
	}).Debug("bean registered")
}

func Autowired(bean interface{}) interface{} {
	WalkStructField(bean, func(field reflect.StructField, value reflect.Value) {
		if value.CanSet() {
			var beanName = GetBeanName(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")
			}
		}
	})
	return bean
}

func WalkStructField(st interface{}, f func(field reflect.StructField, value reflect.Value)) {
	t := reflect.TypeOf(st).Elem()
	v := reflect.ValueOf(st).Elem()
	for i := 0; i < v.NumField(); i++ {
		if !v.Field(i).CanSet() {
			continue
		}
		f(t.Field(i), v.Field(i))
	}
}

func GetBeanName(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)
}