ctx_middlewares.go

package gobis

import (
	log "github.com/sirupsen/logrus"
	"net/http"
)

const (
	GroupContextKey MiddlewareContextKey = iota
	UsernameContextKey
)

type MiddlewareContextKey int

// SetUsername Set the username to a request context
func SetUsername(req *http.Request, username string) {
	userPtr := usernamePtr(req)
	if userPtr == nil {
		user := username
		AddContextValue(req, UsernameContextKey, &user)
		return
	}
	*userPtr = username
}

// Username Retrieve username from a request context
func Username(req *http.Request) string {
	userPtr := usernamePtr(req)
	if userPtr == nil {
		return ""
	}
	return *userPtr
}
func usernamePtr(req *http.Request) *string {
	var username *string
	if err := InjectContextValue(req, UsernameContextKey, &username); err != nil {
		log.Errorf("got error when injecting context value: %s", err)
	}
	return username
}

// SetGroups add groups to a request context
// this call AddGroups, it's simply for UX
func SetGroups(req *http.Request, groups ...string) {
	AddGroups(req, groups...)
}

// AddGroups add groups to a request context
func AddGroups(req *http.Request, groups ...string) {
	if len(groups) == 0 {
		return
	}
	groupsPtr := groupsPtr(req)
	if groupsPtr == nil {
		m := sliceToMap(groups)
		AddContextValue(req, GroupContextKey, &m)
		return
	}
	origGroups := *groupsPtr
	for _, group := range groups {
		origGroups[group] = true
	}
	*groupsPtr = origGroups
}

// Groups retrieve groups from request context
func Groups(req *http.Request) []string {
	groupsPtr := groupsPtr(req)
	if groupsPtr == nil {
		return make([]string, 0)
	}
	return mapToSlice(*groupsPtr)
}
func groupsPtr(req *http.Request) *map[string]bool {
	var groups *map[string]bool
	if err := InjectContextValue(req, GroupContextKey, &groups); err != nil {
		log.Errorf("got error when injecting context value: %s", err)
	}
	return groups
}
func mapToSlice(m map[string]bool) []string {
	s := make([]string, 0)
	for key := range m {
		s = append(s, key)
	}
	return s
}
func sliceToMap(s []string) map[string]bool {
	m := make(map[string]bool)
	for _, v := range s {
		m[v] = true
	}
	return m
}