type.go

package doob

import (
	"log"
	"regexp"

	"github.com/fudali113/doob/utils"
	"github.com/fudali113/doob/config"
	"github.com/fudali113/doob/http/router"

	. "github.com/fudali113/doob/http/const"
)

// 封装node，对外提供简单方法
type Router struct {
	node *router.Node
}

func (r Router) AddHandlerFunc(allUrl string, handler interface{}, methods ...HttpMethod) {
	urls := utils.Split(allUrl, config.UrlSplitSymbol)
	for _, url := range urls {
		for _, method := range methods {
			methodStr := string(method)
			if checkMethodStr(methodStr) {
				r.node.InsertChild(url, router.GetSimpleRestHandler(methodStr, handler))
				if config.AutoAddHead && methodStr == string(GET) {
					r.node.InsertChild(url, router.GetSimpleRestHandler(string(HEAD), handler))
				}
			} else {
				log.Printf("%s method is unsupport", methodStr)
			}
		}
	}
}

func (r Router) Get(allUrl string, handler interface{}) {
	r.AddHandlerFunc(allUrl, handler, GET)
}

func (r Router) Post(allUrl string, handler interface{}) {
	r.AddHandlerFunc(allUrl, handler, POST)
}

func (r Router) Put(allUrl string, handler interface{}) {
	r.AddHandlerFunc(allUrl, handler, PUT)
}

func (r Router) Delete(allUrl string, handler interface{}) {
	r.AddHandlerFunc(allUrl, handler, DELETE)
}

func (r Router) Options(allUrl string, handler interface{}) {
	r.AddHandlerFunc(allUrl, handler, OPTIONS)
}

func (r Router) Head(allUrl string, handler interface{}) {
	r.AddHandlerFunc(allUrl, handler, HEAD)
}

type HttpMethod string

func checkHttpMethod(httpMethod HttpMethod) bool {
	return checkMethodStr(string(httpMethod))
}

func checkMethodStr(httpMethod string) bool {
	match, _ := regexp.MatchString("get|post|put|delete|options|head", httpMethod)
	return match
}