You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package lql
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
auth "github.com/abbot/go-http-auth"
|
|
"github.com/gin-gonic/gin"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
// Logger is the Global logrus logger
|
|
Logger *log.Logger
|
|
)
|
|
|
|
// CtxKeyLQLClient is they key for the Client in the gin context.
|
|
const CtxKeyLQLClient = "lqlClient"
|
|
|
|
// GinGetLqlClient gets the LQL Client from a gin context
|
|
func GinGetLqlClient(c *gin.Context) (*Client, error) {
|
|
clientIface, ok := c.Get(CtxKeyLQLClient)
|
|
if !ok {
|
|
return nil, errors.New("Failed to get the LQL client from context")
|
|
}
|
|
|
|
return clientIface.(*Client), nil
|
|
}
|
|
|
|
func clientInjectorMiddleware(client *Client) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
c.Set(CtxKeyLQLClient, client)
|
|
}
|
|
}
|
|
|
|
func basicAuthMiddleware(a *auth.BasicAuth) gin.HandlerFunc {
|
|
realmHeader := "Basic realm=" + strconv.Quote(a.Realm)
|
|
|
|
return func(c *gin.Context) {
|
|
user := a.CheckAuth(c.Request)
|
|
|
|
if user == "" {
|
|
c.Header("WWW-Authenticate", realmHeader)
|
|
c.AbortWithStatus(http.StatusUnauthorized)
|
|
return
|
|
}
|
|
|
|
c.Set("user", user)
|
|
}
|
|
}
|