gosora/router_gen/route_impl.go

package main

import "strings"

type RouteImpl struct {
	Name      string
	Path      string
	Action    bool
	NoHead    bool
	Vars      []string
	RunBefore []Runnable

	Parent *RouteGroup
}

type Runnable struct {
	Contents string
	Literal  bool
}

func (r *RouteImpl) Before(items ...string) *RouteImpl {
	for _, item := range items {
		r.RunBefore = append(r.RunBefore, Runnable{item, false})
	}
	return r
}

func (r *RouteImpl) LitBefore(items ...string) *RouteImpl {
	for _, item := range items {
		r.RunBefore = append(r.RunBefore, Runnable{item, true})
	}
	return r
}

func (r *RouteImpl) LitBeforeMultiline(items ...string) *RouteImpl {
	for _, item := range items {
		for _, line := range strings.Split(item, "\n") {
			r.LitBefore(strings.TrimSpace(line))
		}
	}
	return r
}

func (r *RouteImpl) hasBefore(items ...string) bool {
	for _, item := range items {
		if r.hasBeforeItem(item) {
			return true
		}
	}
	return false
}

func (r *RouteImpl) hasBeforeItem(item string) bool {
	for _, before := range r.RunBefore {
		if before.Contents == item {
			return true
		}
	}
	return false
}

func (r *RouteImpl) NoGzip() *RouteImpl {
	return r.LitBeforeMultiline(`gzw, ok := w.(c.GzipResponseWriter)
	if ok {
		w = gzw.ResponseWriter
		h := w.Header()
		h.Del("Content-Type")
		h.Del("Content-Encoding")
	}`)
}

func (r *RouteImpl) NoHeader() *RouteImpl {
	r.NoHead = true
	return r
}

func blankRoute() *RouteImpl {
	return &RouteImpl{"", "", false, false, []string{}, []Runnable{}, nil}
}

func route(fname, path string, action, special bool, args ...string) *RouteImpl {
	return &RouteImpl{fname, path, action, special, args, []Runnable{}, nil}
}

func View(fname, path string, args ...string) *RouteImpl {
	return route(fname, path, false, false, args...)
}

func MView(fname, path string, args ...string) *RouteImpl {
	route := route(fname, path, false, false, args...)
	if !route.hasBefore("SuperModOnly", "AdminOnly") {
		route.Before("MemberOnly")
	}
	return route
}

func MemberView(fname, path string, args ...string) *RouteImpl {
	route := route(fname, path, false, false, args...)
	if !route.hasBefore("SuperModOnly", "AdminOnly") {
		route.Before("MemberOnly")
	}
	return route
}

func ModView(fname, path string, args ...string) *RouteImpl {
	route := route(fname, path, false, false, args...)
	if !route.hasBefore("AdminOnly") {
		route.Before("SuperModOnly")
	}
	return route
}

func Action(fname, path string, args ...string) *RouteImpl {
	route := route(fname, path, true, false, args...)
	route.Before("NoSessionMismatch")
	if !route.hasBefore("SuperModOnly", "AdminOnly") {
		route.Before("MemberOnly")
	}
	return route
}

func AnonAction(fname, path string, args ...string) *RouteImpl {
	return route(fname, path, true, false, args...).Before("ParseForm")
}

func Special(fname, path string, args ...string) *RouteImpl {
	return route(fname, path, false, true, args...).LitBefore("req.URL.Path += extraData")
}

// Make this it's own type to force the user to manipulate methods on it to set parameters
type uploadAction struct {
	Route *RouteImpl
}

func UploadAction(fname, path string, args ...string) *uploadAction {
	route := route(fname, path, true, false, args...)
	if !route.hasBefore("SuperModOnly", "AdminOnly") {
		route.Before("MemberOnly")
	}
	return &uploadAction{route}
}

func (a *uploadAction) MaxSizeVar(varName string) *RouteImpl {
	a.Route.LitBeforeMultiline(`err = c.HandleUploadRoute(w,req,user,` + varName + `)
			if err != nil {
				return err
			}`)
	a.Route.Before("NoUploadSessionMismatch")
	return a.Route
}

type RouteSet struct {
	Name  string
	Path  string
	Items []*RouteImpl
}

func Set(name, path string, routes ...*RouteImpl) RouteSet {
	return RouteSet{name, path, routes}
}