netbird/management/server/http/middleware/authenticated.go

package middleware

import (
	"github.com/gorilla/sessions"
	"net/http"
)

type AuthMiddleware struct {
	sessionStore sessions.Store
}

func NewAuth(sessionStore sessions.Store) *AuthMiddleware {
	return &AuthMiddleware{sessionStore: sessionStore}
}

func (am *AuthMiddleware) IsAuthenticated(w http.ResponseWriter, r *http.Request, next http.HandlerFunc) {

	session, err := am.sessionStore.Get(r, "auth-session")
	if err != nil {
		//todo redirect to the error page stating: "error occurred plz try again later and a link to login"
		//http.Error(w, err.Error(), http.StatusInternalServerError)
		http.Redirect(w, r, "/login", http.StatusSeeOther)
		return
	}

	if _, ok := session.Values["profile"]; !ok {
		http.Redirect(w, r, "/login", http.StatusSeeOther)
	} else {
		next(w, r)
	}
}