Files
2026-08-19 22:05:49 +08:00

131 lines
2.8 KiB
Go

package site
import (
"net/http"
"server/internal/middleware"
"server/internal/model/common"
"server/internal/model/request"
"server/internal/pkg/errs"
"server/internal/pkg/httputil"
"server/internal/router"
"server/internal/service/web"
"github.com/go-chi/chi/v5"
)
var _ router.Registrar = (*PostHandler)(nil)
type PostHandler struct {
postService *web.PostService
}
func (h *PostHandler) Register(r chi.Router) {
r.Get("/posts", h.List)
r.Get("/posts/search", h.ListSearch)
r.Get("/posts/archives", h.ListArchives)
r.Get("/posts/tags", h.ListPostTags)
r.Get("/posts/{slug}", h.GetPost)
r.Get("/categories/stats", h.ListCategoryStats)
r.Get("/posts/sitemap", h.GetPostsForSitemap)
}
func NewPostHandler(postService *web.PostService) *PostHandler {
return &PostHandler{postService: postService}
}
func (h *PostHandler) List(w http.ResponseWriter, r *http.Request) {
pagination := httputil.Pagination(r)
result, err := h.postService.List(r.Context(), pagination)
if err != nil {
httputil.Fail(w, err)
return
}
resp := common.PageResponse{
Page: pagination.Page,
PageSize: pagination.PageSize,
List: result.List,
Total: result.Total,
}
httputil.OkWithPage(w, &resp)
}
func (h *PostHandler) ListSearch(w http.ResponseWriter, r *http.Request) {
params := request.SearchPublishedPostsParams{
CategoryCode: r.URL.Query().Get("category_code"),
TagCode: r.URL.Query().Get("tag_code"),
}
list, err := h.postService.ListSearch(r.Context(), params)
if err != nil {
httputil.Fail(w, err)
return
}
httputil.Ok(w, list)
}
func (h *PostHandler) GetPost(w http.ResponseWriter, r *http.Request) {
slug := chi.URLParam(r, "slug")
reqCtx := middleware.GetRequestContext(r.Context())
if slug == "" {
httputil.Fail(w, errs.ErrSlugRequired)
return
}
if reqCtx == nil {
httputil.Fail(w, errs.ErrInternalServer)
return
}
post, err := h.postService.GetPost(r.Context(), slug, reqCtx.ClientIp)
if err != nil {
httputil.Fail(w, err)
return
}
httputil.Ok(w, post)
}
func (h *PostHandler) ListCategoryStats(w http.ResponseWriter, r *http.Request) {
list, err := h.postService.ListCategoryStats(r.Context())
if err != nil {
httputil.Fail(w, err)
return
}
httputil.Ok(w, list)
}
func (h *PostHandler) ListArchives(w http.ResponseWriter, r *http.Request) {
list, err := h.postService.ListArchives(r.Context())
if err != nil {
httputil.Fail(w, err)
return
}
httputil.Ok(w, list)
}
func (h *PostHandler) ListPostTags(w http.ResponseWriter, r *http.Request) {
list, err := h.postService.ListPostTags(r.Context())
if err != nil {
httputil.Fail(w, err)
return
}
httputil.Ok(w, list)
}
func (h *PostHandler) GetPostsForSitemap(w http.ResponseWriter, r *http.Request) {
list, err := h.postService.GetPostsForSitemap(r.Context())
if err != nil {
httputil.Fail(w, err)
return
}
httputil.Ok(w, list)
}