gosora/router_gen/route_group.go

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
}