56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type Middleware func(http.Handler) http.HandlerFunc
|
|
|
|
func multipleMiddleware(h http.HandlerFunc, m ...Middleware) http.HandlerFunc {
|
|
if len(m) < 1 {
|
|
return h
|
|
}
|
|
wrapped := h
|
|
for i := len(m) - 1; i >= 0; i-- {
|
|
wrapped = m[i](wrapped)
|
|
}
|
|
return wrapped
|
|
}
|
|
|
|
func (api *API) authMiddleware(next http.Handler) http.HandlerFunc {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
cookie, err := r.Cookie("Token")
|
|
if err != nil {
|
|
log.Warn("[middleware] Cookie not found")
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
return
|
|
}
|
|
|
|
// Validate cookie.Value JWT with
|
|
api.Auth.ValidateJWTToken(cookie.Value)
|
|
|
|
|
|
log.Info("[middleware] Cookie Name: ", cookie.Name)
|
|
log.Info("[middleware] Cookie Value: ", cookie.Value)
|
|
|
|
next.ServeHTTP(w, r)
|
|
|
|
// if true {
|
|
// next.ServeHTTP(w, r)
|
|
// } else {
|
|
// w.WriteHeader(http.StatusUnauthorized)
|
|
// }
|
|
})
|
|
}
|
|
|
|
func (api *API) logMiddleware(h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
log.SetOutput(os.Stdout)
|
|
log.Println(r.Method, r.URL)
|
|
h.ServeHTTP(w, r)
|
|
})
|
|
}
|