RenewCMS/api/handlers/article/articleHandler.go

package article

import (
	"RenewCMS/api/pkg/apperr"
	"RenewCMS/api/pkg/claims"
	validator "RenewCMS/api/pkg/validator"
	domain "RenewCMS/internal/domain/article"
	"RenewCMS/internal/infrastructure/useCases"
	"encoding/json"
	"errors"
	"fmt"
	"log"
	"net/http"
	"strconv"

	"github.com/go-chi/chi/v5"
)

type Handler struct {
	CreateUseCase  *useCases.CreateArticleUseCase
	GetUseCase     *useCases.GetArticleUseCase
	ListUseCase    *useCases.ListArticlesUseCase
	DeleteUseCase  *useCases.DeleteArticleUseCase
	UpdateUseCase  *useCases.UpdateArticleUseCase
	GetUserUseCase *useCases.GetUserUseCase
}

type PostArticle struct {
	Title string `json:"title" validate:"required,min=3,max=200"`
	Body  string `json:"body" validate:"required,max=10000"`
}

func NewHandler(create *useCases.CreateArticleUseCase, get *useCases.GetArticleUseCase, list *useCases.ListArticlesUseCase, delete *useCases.DeleteArticleUseCase, update *useCases.UpdateArticleUseCase, getUser *useCases.GetUserUseCase) *Handler {
	return &Handler{
		CreateUseCase:  create,
		GetUseCase:     get,
		ListUseCase:    list,
		DeleteUseCase:  delete,
		UpdateUseCase:  update,
		GetUserUseCase: getUser,
	}
}

func parseArticleId(r *http.Request) (uint32, error) {
	id, err := strconv.ParseUint(chi.URLParam(r, "id"), 10, 32)
	return uint32(id), err
}

func (h *Handler) GetArticle(w http.ResponseWriter, r *http.Request) {
	id, err := parseArticleId(r)
	if err != nil {
		http.Error(w, apperr.IdUint32ErrorMessage, http.StatusBadRequest)
		return
	}

	var article domain.Article
	if h.callerIsVerified(r) {
		article, err = h.GetUseCase.GetArticle(id)
	} else {
		article, err = h.GetUseCase.GetOnlineArticle(id)
	}
	if err != nil {
		if errors.Is(err, domain.ErrArticleNotFound) {
			http.Error(w, apperr.ArticleNotFoundErrorMessage, http.StatusNotFound)
			return
		}
		http.Error(w, "Internal server error", http.StatusInternalServerError)
		return
	}

	articleJson, err := json.Marshal(article)
	if err != nil {
		http.Error(w, "Failed to encode response", http.StatusInternalServerError)
		return
	}

	if _, err := w.Write(articleJson); err != nil {
		fmt.Printf("Failed to write response: %v", err)
	}
}

func (h *Handler) PostArticle(w http.ResponseWriter, r *http.Request) {
	var localArticle PostArticle
	if err := json.NewDecoder(r.Body).Decode(&localArticle); err != nil {
		http.Error(w, apperr.BodyErrorMessage, http.StatusBadRequest)
		return
	}

	if err := validator.Validate.Struct(localArticle); err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	createdArticle, err := h.CreateUseCase.CreateArticle(useCases.CreateArticleCommand{
		Title: localArticle.Title,
		Body:  localArticle.Body,
	})
	if err != nil {
		if errors.Is(err, domain.ErrInvalidTitle) {
			http.Error(w, apperr.ArticleTitleErrorMessage, http.StatusBadRequest)
			return
		}
		http.Error(w, "Failed to create article", http.StatusInternalServerError)
		return
	}

	articleJson, err := json.Marshal(createdArticle)
	if err != nil {
		http.Error(w, "Failed to encode response", http.StatusInternalServerError)
		return
	}

	w.WriteHeader(http.StatusCreated)
	if _, err := w.Write(articleJson); err != nil {
		log.Printf("Failed to write response: %v", err)
	}
}

func writeArticleErrorResponse(w http.ResponseWriter, err error) {
	if errors.Is(err, domain.ErrArticleNotFound) {
		http.Error(w, apperr.ArticleNotFoundErrorMessage, http.StatusNotFound)
		return
	}
	http.Error(w, "Internal server error", http.StatusInternalServerError)
}

func writeArticleJsonResponse(w http.ResponseWriter, article domain.Article) {
	articleJson, err := json.Marshal(article)
	if err != nil {
		http.Error(w, "Failed to encode response", http.StatusInternalServerError)
		return
	}

	if _, err := w.Write(articleJson); err != nil {
		log.Printf("Failed to write response: %v", err)
	}
}

func (h *Handler) setIsOnline(w http.ResponseWriter, r *http.Request, isOnline bool) {
	id, err := parseArticleId(r)
	if err != nil {
		http.Error(w, apperr.IdUint32ErrorMessage, http.StatusBadRequest)
		return
	}

	updatedArticle, err := h.UpdateUseCase.UpdateIsOnline(id, isOnline)
	if err != nil {
		writeArticleErrorResponse(w, err)
		return
	}

	writeArticleJsonResponse(w, updatedArticle)
}

type UpdateArticleBody struct {
	Body string `json:"body" validate:"required,max=10000"`
}

func (h *Handler) UpdateArticleBody(w http.ResponseWriter, r *http.Request) {
	id, err := parseArticleId(r)
	if err != nil {
		http.Error(w, apperr.IdUint32ErrorMessage, http.StatusBadRequest)
		return
	}

	var localArticle UpdateArticleBody
	if err := json.NewDecoder(r.Body).Decode(&localArticle); err != nil {
		http.Error(w, apperr.BodyErrorMessage, http.StatusBadRequest)
		return
	}

	if err := validator.Validate.Struct(localArticle); err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	updatedArticle, err := h.UpdateUseCase.UpdateBody(id, localArticle.Body)
	if err != nil {
		writeArticleErrorResponse(w, err)
		return
	}

	writeArticleJsonResponse(w, updatedArticle)
}

func (h *Handler) PublishArticle(w http.ResponseWriter, r *http.Request) {
	h.setIsOnline(w, r, true)
}

func (h *Handler) UnpublishArticle(w http.ResponseWriter, r *http.Request) {
	h.setIsOnline(w, r, false)
}

func (h *Handler) callerIsVerified(r *http.Request) bool {
	userId, ok := claims.UserId(r)
	if !ok {
		return false
	}

	currentUser, err := h.GetUserUseCase.GetUser(userId)
	if err != nil {
		return false
	}

	return currentUser.IsVerified
}

func (h *Handler) ListArticles(w http.ResponseWriter, r *http.Request) {
	var articles []domain.Article
	var err error
	if h.callerIsVerified(r) {
		articles, err = h.ListUseCase.ListAllArticles()
	} else {
		articles, err = h.ListUseCase.ListOnlineArticles()
	}
	if err != nil {
		http.Error(w, "Failed to retrieve articles", http.StatusInternalServerError)
		return
	}

	articlesJson, err := json.Marshal(articles)
	if err != nil {
		http.Error(w, "Failed to encode response", http.StatusInternalServerError)
		return
	}

	if _, err := w.Write(articlesJson); err != nil {
		log.Printf("Failed to write response: %v", err)
	}
}

func (h *Handler) DeleteArticle(w http.ResponseWriter, r *http.Request) {
	id, err := parseArticleId(r)
	if err != nil {
		http.Error(w, apperr.IdUint32ErrorMessage, http.StatusBadRequest)
		return
	}

	if err := h.DeleteUseCase.DeleteArticle(id); err != nil {
		if errors.Is(err, domain.ErrArticleNotFound) {
			http.Error(w, apperr.ArticleNotFoundErrorMessage, http.StatusNotFound)
			return
		}
		http.Error(w, "Failed to delete article", http.StatusInternalServerError)
		return
	}

	w.WriteHeader(http.StatusNoContent)
}