package squarehandler import ( "strconv" "github.com/gin-gonic/gin" "spark-server/internal/module/square/entity" squarelogic "spark-server/internal/module/square/logic" "spark-server/internal/pkg/errs" "spark-server/internal/pkg/reqctx" "spark-server/internal/transport/http/respond" ) // Handler is the square HTTP controller for one plaza channel. type Handler struct { square *squarelogic.Service channel entity.Channel } func New(square *squarelogic.Service, channel entity.Channel) *Handler { if channel == "" { channel = entity.ChannelSquare } return &Handler{square: square, channel: channel} } func currentUID(c *gin.Context) (int64, bool) { rc := reqctx.From(c.Request.Context()) if rc == nil || rc.UserID <= 0 { respond.JSONError(c, errs.ErrUnauthorized) return 0, false } return rc.UserID, true } func pathInt(c *gin.Context, name string) (int64, bool) { id, err := strconv.ParseInt(c.Param(name), 10, 64) if err != nil || id <= 0 { respond.JSONError(c, errs.ErrInvalidArgument) return 0, false } return id, true } func queryInt(c *gin.Context, name string) int64 { n, _ := strconv.ParseInt(c.Query(name), 10, 64) return n } func queryFloat(c *gin.Context, name string) float64 { n, _ := strconv.ParseFloat(c.Query(name), 64) return n } func (h *Handler) Categories(c *gin.Context) { respond.JSON(c, gin.H{"items": h.square.Categories()}) } func (h *Handler) Feed(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } cat, ok := entity.ParseCategory(c.Query("category")) if !ok { respond.JSONError(c, errs.ErrInvalidArgument) return } if cat == "" { cat = entity.CategoryRecommend } out, err := h.square.Feed(c.Request.Context(), uid, h.channel, squarelogic.FeedQuery{ Cursor: queryInt(c, "cursor"), Limit: int(queryInt(c, "limit")), Category: cat, Lat: queryFloat(c, "lat"), Lng: queryFloat(c, "lng"), }) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) Publish(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } var in squarelogic.PublishInput if err := respond.BindJSON(c, &in); err != nil { respond.JSONError(c, err) return } out, err := h.square.Publish(c.Request.Context(), uid, h.channel, &in) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) UserPosts(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } owner, ok := pathInt(c, "uid") if !ok { return } out, err := h.square.UserPosts(c.Request.Context(), uid, owner, h.channel, queryInt(c, "cursor"), int(queryInt(c, "limit"))) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) Get(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } id, ok := pathInt(c, "id") if !ok { return } out, err := h.square.Get(c.Request.Context(), uid, id, h.channel) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) Delete(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } id, ok := pathInt(c, "id") if !ok { return } if err := h.square.Delete(c.Request.Context(), uid, id, h.channel); err != nil { respond.JSONError(c, err) return } respond.JSON(c, gin.H{}) } func (h *Handler) Like(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } id, ok := pathInt(c, "id") if !ok { return } out, err := h.square.Like(c.Request.Context(), uid, id, h.channel) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) Unlike(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } id, ok := pathInt(c, "id") if !ok { return } out, err := h.square.Unlike(c.Request.Context(), uid, id, h.channel) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) CreateComment(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } id, ok := pathInt(c, "id") if !ok { return } var in squarelogic.CommentInput if err := respond.BindJSON(c, &in); err != nil { respond.JSONError(c, err) return } out, err := h.square.Comment(c.Request.Context(), uid, id, h.channel, &in) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) DeleteComment(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } postID, ok := pathInt(c, "id") if !ok { return } cid, ok := pathInt(c, "cid") if !ok { return } out, err := h.square.DeleteComment(c.Request.Context(), uid, postID, cid, h.channel) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) ListNotices(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } out, err := h.square.ListNotices(c.Request.Context(), uid, h.channel, queryInt(c, "cursor"), int(queryInt(c, "limit"))) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) NoticeUnread(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } out, err := h.square.NoticeUnread(c.Request.Context(), uid, h.channel) if err != nil { respond.JSONError(c, err) return } respond.JSON(c, out) } func (h *Handler) ReadNotices(c *gin.Context) { uid, ok := currentUID(c) if !ok { return } var in squarelogic.NoticeReadInput if err := respond.BindJSON(c, &in); err != nil { respond.JSONError(c, err) return } if err := h.square.ReadNotices(c.Request.Context(), uid, h.channel, &in); err != nil { respond.JSONError(c, err) return } respond.JSON(c, gin.H{}) }