gosora/router_gen/route_group.go

86 lines
1.8 KiB
Go
Raw Normal View History

package main
import "strings"
type RouteGroup struct {
Path string
RouteList []*RouteImpl
RunBefore []Runnable
NoHead bool
}
func newRouteGroup(path string, routes ...*RouteImpl) *RouteGroup {
g := &RouteGroup{Path: path}
for _, route := range routes {
route.Parent = g
g.RouteList = append(g.RouteList, route)
}
return g
}
func (g *RouteGroup) Not(path ...string) *RouteSubset {
routes := make([]*RouteImpl, len(g.RouteList))
copy(routes, g.RouteList)
for i, route := range routes {
if inStringList(route.Path, path) {
routes = append(routes[:i], routes[i+1:]...)
}
}
return &RouteSubset{routes}
}
func inStringList(needle string, list []string) bool {
for _, item := range list {
if item == needle {
return true
}
}
return false
}
func (g *RouteGroup) NoHeader() *RouteGroup {
g.NoHead = true
return g
}
func (g *RouteGroup) Before(lines ...string) *RouteGroup {
for _, line := range lines {
g.RunBefore = append(g.RunBefore, Runnable{line, false})
}
return g
}
func (g *RouteGroup) LitBefore(lines ...string) *RouteGroup {
for _, line := range lines {
g.RunBefore = append(g.RunBefore, Runnable{line, true})
}
return g
}
/*func (g *RouteGroup) Routes(routes ...*RouteImpl) *RouteGroup {
for _, route := range routes {
route.Parent = g
g.RouteList = append(g.RouteList, route)
}
return g
}*/
func (g *RouteGroup) Routes(routes ...interface{}) *RouteGroup {
for _, route := range routes {
switch r := route.(type) {
case *RouteImpl:
r.Parent = g
g.RouteList = append(g.RouteList, r)
case RouteSet:
for _, rr := range r.Items {
rr.Name = r.Name + rr.Name
rr.Path = strings.TrimSuffix(r.Path, "/") + "/" + strings.TrimPrefix(rr.Path, "/")
rr.Parent = g
g.RouteList = append(g.RouteList, rr)
}
}
}
return g
}