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 } 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) }