package xmux

import (

"net/http"

"regexp"

"github.com/hyahm/xmux/helper"

)

// 初始化临时使用， 最后会合并到 router

type route struct {

// 组里面也包括路由 后面的其实还是patter和handle, 还没到handle， 这里的key是个method

handle http.Handler // handle

module *module // 增加的 modules

postModule *module // 增加的 modules

delmodule map[string]struct{} // 删除的modules

delPostModule map[string]struct{} // 删除的modules

params []string // path正则名

pagekeys map[string]struct{} // 页面权限

delPageKeys map[string]struct{} // 删除的权限

header map[string]string // 请求头

delheader map[string]struct{} // 删除的请求头

methods []string // 删除的请求头

prefixs []string

delprefix map[string]struct{}

responseData interface{} // 接口返回实例

bindResponseData bool

summary string

bindType bindType // 数据绑定格式

dataSource interface{} // 数据源

query []Parameter

denyPrefix bool

middleware onion

regex *regexp.Regexp // 预编译的正则表达式

menuTree *MenuTree

url string

ignoreMenu bool

}

func (rt *route) Use(m ...Middleware) {

rt.middleware.Use(m...)

}

func (rt *route) SetMeta(mt Meta) *route {

rt.menuTree.Meta = mt

return rt

}

func (rt *route) Prefix(prefix string) *route {

rt.prefixs = append(rt.prefixs, prefix)

return rt

}

func (rt *route) DelPrefix(prefixs ...string) *route {

for _, prefix := range prefixs {

rt.delprefix[prefix] = struct{}{}

}

return rt

}

func (rt *route) GetHeader() map[string]string {

return rt.header

}

func (rt *route) DenyPrefix() {

rt.denyPrefix = true

}

// 这个路由的注释, 使用swagger加上这个字段才能显示执行的窗口

func (rt *route) SwaggerSummary(summary string) *route {

rt.summary = summary

return rt

}

func (rt *route) BindResponse(response interface{}) *route {

rt.bindResponseData = true

rt.responseData = response

// if response == nil {

// rt.bindResponseData = false

// return rt

// }

return rt

}

func (rt *route) AddPageKeys(pagekeys ...string) *route {

// 退出文档的组

for _, v := range pagekeys {

if rt.pagekeys == nil {

rt.pagekeys = make(map[string]struct{})

}

rt.pagekeys[v] = struct{}{}

}

return rt

}

func (rt *route) DelPageKeys(pagekeys ...string) *route {

if rt.delPageKeys == nil {

rt.delPageKeys = make(map[string]struct{})

}

for _, key := range pagekeys {

rt.delPageKeys[key] = struct{}{}

}

for _, v := range pagekeys {

delete(rt.pagekeys, v)

}

return rt

}

// 数据绑定

func (rt *route) Bind(s interface{}) *route {

rt.dataSource = s

return rt

}

// json数据绑定

func (rt *route) BindJson(s interface{}) *route {

// 接口补充说明

rt.dataSource = s

rt.bindType = jsonT

return rt

}

func (rt *route) BindByContentType(s interface{}) *route {

// 接口补充说明

rt.dataSource = s

rt.bindType = headT

return rt

}

func (rt *route) BindForm(s interface{}) *route {

// 接口补充说明

rt.dataSource = s

rt.bindType = formT

return rt

}

// yaml数据绑定

func (rt *route) BindYaml(s interface{}) *route {

// 接口补充说明

rt.dataSource = s

rt.bindType = yamlT

return rt

}

// xml数据绑定

func (rt *route) BindXml(s interface{}) *route {

// 接口补充说明

rt.dataSource = s

rt.bindType = xmlT

return rt

}

// 组里面也包括路由 后面的其实还是patter和handle

func (rt *route) SetHeader(k, v string) *route {

if rt.header == nil {

rt.header = map[string]string{}

}

rt.header[k] = v

return rt

}

func (rt *route) DelHeader(dh ...string) *route {

if rt.delheader == nil {

rt.delheader = make(map[string]struct{})

}

for _, v := range dh {

rt.delheader[v] = struct{}{}

}

return rt

}

func (rt *route) AddModule(handles ...func(http.ResponseWriter, *http.Request) bool) *route {

if rt.module == nil {

rt.module = &module{

filter: make(map[string]struct{}),

funcOrder: make([]func(w http.ResponseWriter, r *http.Request) bool, 0),

}

}

rt.module.add(handles...)

return rt

}

func (rt *route) AddPostModule(handles ...func(http.ResponseWriter, *http.Request) bool) *route {

if rt.postModule == nil {

rt.postModule = &module{

filter: make(map[string]struct{}),

funcOrder: make([]func(w http.ResponseWriter, r *http.Request) bool, 0),

}

}

rt.postModule.add(handles...)

return rt

}

func (rt *route) DelModule(handles ...func(http.ResponseWriter, *http.Request) bool) *route {

if rt.delmodule == nil {

rt.delmodule = make(map[string]struct{})

}

for _, handle := range handles {

rt.delmodule[helper.GetFuncName(handle)] = struct{}{}

}

rt.module.delete(rt.delmodule)

return rt

}

func (rt *route) DelPostModule(handles ...func(http.ResponseWriter, *http.Request) bool) *route {

if rt.delPostModule == nil {

rt.delPostModule = make(map[string]struct{})

}

for _, handle := range handles {

rt.delPostModule[helper.GetFuncName(handle)] = struct{}{}

}

rt.postModule.delete(rt.delPostModule)

return rt

}

func (rt *route) SwaggerAddParameter(pt Parameter) *route {

rt.query = append(rt.query, pt)

return rt

}