imagini/internal/api/middlewares.go

package api

import (
    "os"
    "context"
    "net/http"
    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) {
        // Acquire Token
        accessCookie, err := r.Cookie("AccessToken")
        if err != nil {
            log.Warn("[middleware] AccessToken not found")
            w.WriteHeader(http.StatusUnauthorized)
            return
        }

        // Validate JWT Tokens
        accessToken, accessOK := api.Auth.ValidateJWTAccessToken(accessCookie.Value)

        if accessOK {
            // Acquire UserID and DeviceID
            reqInfo := make(map[string]string)
            uid, _ := accessToken.Get("sub")
            did, _ := accessToken.Get("did")
            reqInfo["uid"] = uid.(string)
            reqInfo["did"] = did.(string)

            // Add context
            ctx := context.WithValue(r.Context(), "uuids", reqInfo)
            sr := r.WithContext(ctx)

            next.ServeHTTP(w, sr)
        } 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)
   })
}