33 lines
716 B
Go
33 lines
716 B
Go
package middlewares
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/uptrace/bunrouter"
|
|
)
|
|
|
|
func (self *Handlers) CheckPresharedKey(
|
|
next bunrouter.HandlerFunc,
|
|
) bunrouter.HandlerFunc {
|
|
return func(w http.ResponseWriter, req bunrouter.Request) error {
|
|
authHeader := strings.Split(req.Header.Get("Authorization"), " ")
|
|
if len(authHeader) != 2 || authHeader[0] != "Bearer" {
|
|
return HTTPError{
|
|
StatusCode: http.StatusUnauthorized,
|
|
Message: "missing preshared key",
|
|
}
|
|
}
|
|
|
|
if authHeader[1] != viper.GetString("preshared-key") {
|
|
return HTTPError{
|
|
StatusCode: http.StatusUnauthorized,
|
|
Message: "preshared key mismatched",
|
|
}
|
|
}
|
|
|
|
return next(w, req)
|
|
}
|
|
}
|