75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
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)
|
|
}
|