package middleware

import (
	"RenewCMS/api/pkg/apperr"
	"RenewCMS/api/pkg/claims"
	"RenewCMS/internal/domain/user"
	"RenewCMS/internal/infrastructure/useCases"
	"errors"
	"net/http"
)

func JsonContentTypeMiddleware(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		next.ServeHTTP(w, r)
	})
}

func RequireVerifiedUser(getUser *useCases.GetUserUseCase) func(http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			userId, ok := claims.UserId(r)
			if !ok {
				http.Error(w, apperr.UnauthorizedErrorMessage, http.StatusUnauthorized)
				return
			}

			currentUser, err := getUser.GetUser(userId)
			if err != nil {
				if errors.Is(err, user.ErrUserNotFound) {
					http.Error(w, apperr.UnauthorizedErrorMessage, http.StatusUnauthorized)
					return
				}
				http.Error(w, "Internal server error", http.StatusInternalServerError)
				return
			}

			if !currentUser.IsVerified {
				http.Error(w, apperr.VerificationRequiredErrorMessage, http.StatusForbidden)
				return
			}

			next.ServeHTTP(w, r)
		})
	}
}