package controller

import (
	"gitee.com/kristas/booting-go/framework/common/util/lang"
	"gitee.com/kristas/booting-go/framework/common/util/reflectx"
	"gitee.com/kristas/booting-go/framework/core/component"
	"gitee.com/kristas/booting-go/framework/web"
	"reflect"
)

func HandleRestApi(controller component.Controller, handler func(apiInfo web.ApiInfo, svm web.ServiceMethod)) {
	typ := reflect.TypeOf(controller)
	val := reflect.ValueOf(controller)
	reflectx.WalkStructField(controller, func(field reflect.StructField, value reflect.Value) error {
		if isRestAPI(field) {
			if method, ok := typ.MethodByName(methodName(field)); ok {
				httpMethod, url := findHttpMethodAndUrl(field)
				url = controller.Controller() + url
				methodVal := val.MethodByName(methodName(field))
				paramMap := buildParam(field.Tag.Get("param"), method.Type)
				description := field.Tag.Get("desc")

				apiInfo := &ApiInfo{
					HttpMethod:  httpMethod,
					URL:         url,
					Description: description,
					Params:      paramMap,
				}
				svm := NewServiceMethod(&method, &methodVal)
				NewApiInfo(controller.Controller(), field, svm)
				handler(apiInfo, svm)
			}
		}
		return nil
	}, false)
}

func HandleFsApi(controller component.Controller, handler func(prefix, dir string)) {
	typ := reflect.TypeOf(controller)
	val := reflect.ValueOf(controller)
	reflectx.WalkStructField(controller, func(field reflect.StructField, value reflect.Value) error {
		if isFsAPI(field) {
			if _, ok := typ.MethodByName(methodName(field)); ok {
				url := field.Tag.Get("url")
				url = controller.Controller() + url
				methodVal := val.MethodByName(methodName(field))
				handler(url, methodVal.Call(nil)[0].String())
			}
		}
		return nil
	}, false)
}

func isRestAPI(field reflect.StructField) bool {
	return field.Type.Implements(reflect.TypeOf(new(web.Rest)).Elem())
}

func isFsAPI(field reflect.StructField) bool {
	return field.Type == reflect.TypeOf(web.FS{})
}

func findHttpMethodAndUrl(field reflect.StructField) (method string, url string) {
	m := reflect.New(field.Type).Interface().(web.Rest)
	method = m.GetType()
	url = field.Tag.Get("url")
	return
}

func methodName(field reflect.StructField) string {
	method := lang.NewString(field.Name).FirstLetterUpper()
	return method.String()
}