package ginAutoRoute

import (
	"github.com/gin-gonic/gin"
	"net/http"
	"reflect"
	"strings"
)

var methods = make(map[string]map[string]reflect.Value)

func AutoRoute(s interface{}) gin.HandlerFunc {
	return func(c *gin.Context) {
		action := c.Param("action")
		realAction := strings.ToLower(action)
		rt := reflect.TypeOf(s)
		pkgName := rt.String()
		v, ok := methods[pkgName][realAction]
		if !ok && len(methods[pkgName]) == 0 {
			rv := reflect.ValueOf(s)
			methods[pkgName] = make(map[string]reflect.Value)
			for i := 0; i < rv.NumMethod(); i++ {
				fn := rv.Method(i)
				ft := rt.Method(i)
				methods[pkgName][strings.ToLower(ft.Name)] = fn
			}
			v, ok = methods[pkgName][realAction]
			if !ok && len(methods[pkgName]) > 0 {
				http.NotFound(c.Writer, c.Request)
				c.Abort()
				return
			}
		} else if !ok && len(methods[pkgName]) > 0 {
			http.NotFound(c.Writer, c.Request)
			c.Abort()
			return
		}

		arguments := make([]reflect.Value, 1)
		arguments[0] = reflect.ValueOf(c) // *gin.Context
		v.Call(arguments)
	}
}