package auth

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

type MeHandler struct {
	GetUseCase       *useCases.GetUserUseCase
	GetAuthorUseCase *useCases.GetMyAuthorUseCase
}

func NewMeHandler(getUser *useCases.GetUserUseCase, getAuthor *useCases.GetMyAuthorUseCase) *MeHandler {
	return &MeHandler{
		GetUseCase:       getUser,
		GetAuthorUseCase: getAuthor,
	}
}

type meAuthor struct {
	author.Byline
	ArticleCount int `json:"article_count"`
}

type meResponse struct {
	publicuser.PublicUser
	Author *meAuthor `json:"author"`
}

func toMeAuthor(found *author.Author) *meAuthor {
	if found == nil {
		return nil
	}
	return &meAuthor{Byline: found.Byline(), ArticleCount: found.ArticleCount}
}

func (h *MeHandler) Me(w http.ResponseWriter, r *http.Request) {
	userId, ok := claims.UserId(r)
	if !ok {
		http.Error(w, apperr.UnauthorizedErrorMessage, http.StatusUnauthorized)
		return
	}

	loadedUser, err := h.GetUseCase.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
	}

	linked, err := h.GetAuthorUseCase.Get(loadedUser.ID)
	if err != nil {
		http.Error(w, "Internal server error", http.StatusInternalServerError)
		return
	}

	body, err := json.Marshal(meResponse{PublicUser: publicuser.New(loadedUser), Author: toMeAuthor(linked)})
	if err != nil {
		http.Error(w, "Internal server error", http.StatusInternalServerError)
		return
	}
	w.Header().Set("Cache-Control", "no-store")
	_, _ = w.Write(body)
}