feat: 增加推送通知配置

2026-06-19 16:28:49 +08:00
parent a4cdc3e806
commit 5ff1ea40b0
15 changed files with 1376 additions and 51 deletions
@@ -0,0 +1,169 @@
+package adminpush
+
+import (
+	"errors"
+	"strconv"
+
+	"hfb_sys/backend/internal/middleware"
+	"hfb_sys/backend/pkg/response"
+
+	"github.com/gin-gonic/gin"
+)
+
+type Handler struct {
+	service *Service
+}
+
+func NewHandler(service *Service) *Handler {
+	return &Handler{service: service}
+}
+
+// ── 渠道 ──
+
+func (h *Handler) ListChannels(c *gin.Context) {
+	items, err := h.service.ListChannels(c.Request.Context())
+	if err != nil {
+		writePushError(c, err)
+		return
+	}
+	response.OK(c, gin.H{"items": items})
+}
+
+func (h *Handler) CreateChannel(c *gin.Context) {
+	if _, ok := currentAdminID(c); !ok {
+		response.Unauthorized(c, "缺少管理员上下文")
+		return
+	}
+	var req CreateChannelRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.BadRequest(c, "名称、类型和配置不能为空")
+		return
+	}
+	item, err := h.service.CreateChannel(c.Request.Context(), req)
+	if err != nil {
+		writePushError(c, err)
+		return
+	}
+	response.Created(c, item)
+}
+
+func (h *Handler) TestChannel(c *gin.Context) {
+	if _, ok := currentAdminID(c); !ok {
+		response.Unauthorized(c, "缺少管理员上下文")
+		return
+	}
+	id, err := parseID(c)
+	if err != nil {
+		response.BadRequest(c, "ID 不正确")
+		return
+	}
+	if err := h.service.TestChannel(c.Request.Context(), id); err != nil {
+		writePushError(c, err)
+		return
+	}
+	response.OK(c, gin.H{"sent": true})
+}
+
+func (h *Handler) UpdateChannel(c *gin.Context) {
+	if _, ok := currentAdminID(c); !ok {
+		response.Unauthorized(c, "缺少管理员上下文")
+		return
+	}
+	id, err := parseID(c)
+	if err != nil {
+		response.BadRequest(c, "ID 不正确")
+		return
+	}
+	var req UpdateChannelRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.BadRequest(c, "请求格式不正确")
+		return
+	}
+	item, err := h.service.UpdateChannel(c.Request.Context(), id, req)
+	if err != nil {
+		writePushError(c, err)
+		return
+	}
+	response.OK(c, item)
+}
+
+func (h *Handler) DeleteChannel(c *gin.Context) {
+	if _, ok := currentAdminID(c); !ok {
+		response.Unauthorized(c, "缺少管理员上下文")
+		return
+	}
+	id, err := parseID(c)
+	if err != nil {
+		response.BadRequest(c, "ID 不正确")
+		return
+	}
+	if err := h.service.DeleteChannel(c.Request.Context(), id); err != nil {
+		writePushError(c, err)
+		return
+	}
+	response.OK(c, gin.H{"deleted": true})
+}
+
+// ── 规则 ──
+
+func (h *Handler) ListRules(c *gin.Context) {
+	items, err := h.service.ListRules(c.Request.Context())
+	if err != nil {
+		writePushError(c, err)
+		return
+	}
+	response.OK(c, gin.H{"items": items})
+}
+
+func (h *Handler) UpdateRule(c *gin.Context) {
+	if _, ok := currentAdminID(c); !ok {
+		response.Unauthorized(c, "缺少管理员上下文")
+		return
+	}
+	id, err := parseID(c)
+	if err != nil {
+		response.BadRequest(c, "ID 不正确")
+		return
+	}
+	var req UpdateRuleRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.BadRequest(c, "请求格式不正确")
+		return
+	}
+	item, err := h.service.UpdateRule(c.Request.Context(), id, req)
+	if err != nil {
+		writePushError(c, err)
+		return
+	}
+	response.OK(c, item)
+}
+
+// ── 辅助 ──
+
+func parseID(c *gin.Context) (uint64, error) {
+	return strconv.ParseUint(c.Param("id"), 10, 64)
+}
+
+func currentAdminID(c *gin.Context) (uint64, bool) {
+	value, ok := c.Get(middleware.ContextAdminID)
+	if !ok {
+		return 0, false
+	}
+	adminID, ok := value.(uint64)
+	return adminID, ok
+}
+
+func writePushError(c *gin.Context, err error) {
+	switch {
+	case errors.Is(err, ErrDependencyUnavailable):
+		response.ServiceUnavailable(c, "数据库未连接")
+	case errors.Is(err, ErrChannelNotFound):
+		response.NotFound(c, "推送渠道不存在")
+	case errors.Is(err, ErrRuleNotFound):
+		response.NotFound(c, "推送规则不存在")
+	case errors.Is(err, ErrInvalidChannel):
+		response.BadRequest(c, "渠道类型或配置不正确")
+	default:
+		response.InternalServerError(c, "推送服务暂时不可用")
+	}
+}