router.go

package td

import (
	"net/http"

	"github.com/swxctx/malatd/httprouter"
)

// Router
type Router struct {
	Plugins  Plugins
	basePath string
	server   *Server
	root     bool
}

// 注册组路由
func (r *Router) Group(relativePath string, plugins ...Plugin) *Router {
	gPlugins := append(r.Plugins, plugins...)
	return &Router{
		Plugins:  gPlugins,
		basePath: getReqPath(r.basePath, relativePath),
		server:   r.server,
		root:     false,
	}
}

// AddPlugin
func (r *Router) AddPlugin(plugins ...Plugin) {
	r.Plugins = append(r.Plugins, plugins...)
}

// Get
func (r *Router) Get(relativePath string, plugins ...Plugin) {
	path := getReqPath(r.basePath, relativePath)
	plugin := append(r.Plugins, plugins...)
	Infof("[ROUTE]: GET %s", path)
	r.handle("GET", path, plugin)
}

// Post
func (r *Router) Post(relativePath string, plugins ...Plugin) {
	path := getReqPath(r.basePath, relativePath)
	plugin := append(r.Plugins, plugins...)
	Infof("[ROUTE]: POST %s", path)
	r.handle("POST", path, plugin)
}

// Options
func (r *Router) Options(relativePath string, plugins ...Plugin) {
	path := getReqPath(r.basePath, relativePath)
	plugin := append(r.Plugins, plugins...)
	Infof("[ROUTE]: OPTIONS %s", path)
	r.handle("OPTIONS", path, plugin)
}

// handle
func (r *Router) handle(httpMethod, relativePath string, plugins Plugins) {
	ctx := Context{
		index:   0,
		server:  r.server,
		plugins: plugins,
	}

	var (
		err error
	)

	switch httpMethod {
	case "GET":
		r.server.router.GET(relativePath, func(response http.ResponseWriter, request *http.Request, params httprouter.Params) {
			defer func() {
				if re := recover(); re != nil {
					Errorf("[GET] err: %v", re)
					ctx.Render(RerrInternalServer)
				}
			}()
			ctx.Request = request
			ctx.ResponseWriter = response
			ctx.Next()
		})
	case "POST":
		r.server.router.POST(relativePath, func(response http.ResponseWriter, request *http.Request, params httprouter.Params) {
			defer func() {
				if re := recover(); re != nil {
					Errorf("[POST] err: %v", re)
					ctx.Render(RerrInternalServer)
				}
			}()
			ctx.Request = request
			ctx.ResponseWriter = response
			ctx.Next()
		})
	case "OPTIONS":
		r.server.router.OPTIONS(relativePath, func(response http.ResponseWriter, request *http.Request, params httprouter.Params) {
			defer func() {
				if re := recover(); re != nil {
					Errorf("[POST] err: %v", re)
					ctx.Render(RerrInternalServer)
				}
			}()
			ctx.Request = request
			ctx.ResponseWriter = response
			ctx.Next()
		})
	}
	if err != nil {
		Errorf("[PLUGIN] handle err: %v", err)
	}
}