feature: add ability to assign tasks

api/router/auth.go

@@ -42,7 +42,7 @@ func (h *CitadelHandler) RefreshTokenHandler(w http.ResponseWriter, r *http.Requ
 		w.WriteHeader(http.StatusInternalServerError)
 	}
 
-	accessTokenString, err := NewAccessToken("1")
+	accessTokenString, err := NewAccessToken(token.UserID.String())
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)
 	}
@@ -57,6 +57,25 @@ func (h *CitadelHandler) RefreshTokenHandler(w http.ResponseWriter, r *http.Requ
 	json.NewEncoder(w).Encode(LoginResponseData{AccessToken: accessTokenString})
 }
 
+func (h *CitadelHandler) LogoutHandler(w http.ResponseWriter, r *http.Request) {
+	c, err := r.Cookie("refreshToken")
+	if err != nil {
+		if err == http.ErrNoCookie {
+			w.WriteHeader(http.StatusBadRequest)
+			return
+		}
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+	refreshTokenID := uuid.MustParse(c.Value)
+	err = h.repo.DeleteRefreshTokenByID(r.Context(), refreshTokenID)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+	json.NewEncoder(w).Encode(LogoutResponseData{Status: "success"})
+}
+
 func (h *CitadelHandler) LoginHandler(w http.ResponseWriter, r *http.Request) {
 	var requestData LoginRequestData
 	err := json.NewDecoder(r.Body).Decode(&requestData)
@@ -85,12 +104,11 @@ func (h *CitadelHandler) LoginHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	userID := uuid.MustParse("0183d9ab-d0ed-4c9b-a3df-77a0cdd93dca")
 	refreshCreatedAt := time.Now().UTC()
 	refreshExpiresAt := refreshCreatedAt.AddDate(0, 0, 1)
-	refreshTokenString, err := h.repo.CreateRefreshToken(r.Context(), pg.CreateRefreshTokenParams{userID, refreshCreatedAt, refreshExpiresAt})
+	refreshTokenString, err := h.repo.CreateRefreshToken(r.Context(), pg.CreateRefreshTokenParams{user.UserID, refreshCreatedAt, refreshExpiresAt})
 
-	accessTokenString, err := NewAccessToken("1")
+	accessTokenString, err := NewAccessToken(user.UserID.String())
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)
 	}
@@ -109,5 +127,6 @@ func (rs authResource) Routes(citadelHandler CitadelHandler) chi.Router {
 	r := chi.NewRouter()
 	r.Post("/login", citadelHandler.LoginHandler)
 	r.Post("/refresh_token", citadelHandler.RefreshTokenHandler)
+	r.Post("/logout", citadelHandler.LogoutHandler)
 	return r
 }

api/router/middleware.go

@@ -5,6 +5,7 @@ import (
 	"net/http"
 	"strings"
 
+	"github.com/google/uuid"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -40,7 +41,13 @@ func AuthenticationMiddleware(next http.Handler) http.Handler {
 			return
 		}
 
-		ctx := context.WithValue(r.Context(), "accessClaims", accessClaims)
+		userID, err := uuid.Parse(accessClaims.UserID)
+		if err != nil {
+			log.Error(err)
+			w.WriteHeader(http.StatusBadRequest)
+			return
+		}
+		ctx := context.WithValue(r.Context(), "userID", userID)
 
 		next.ServeHTTP(w, r.WithContext(ctx))
 	})

api/router/models.go

@@ -23,6 +23,10 @@ type LoginResponseData struct {
 	AccessToken string `json:"accessToken"`
 }
 
+type LogoutResponseData struct {
+	Status string `json:"status"`
+}
+
 type RefreshTokenResponseData struct {
 	AccessToken string `json:"accessToken"`
 }