diff --git a/internal/handler/notification_handler.go b/internal/handler/notification_handler.go index f67cb7f..5dbd68a 100644 --- a/internal/handler/notification_handler.go +++ b/internal/handler/notification_handler.go @@ -205,11 +205,10 @@ func (h *NotificationHandler) CreateNotification(c *gin.Context) { api.Created(c, notification) } -// RegisterRoutes registers all notification routes to the given router group -func (h *NotificationHandler) RegisterRoutes(rg *gin.RouterGroup) { +// RegisterUserRoutes registers notification routes for regular users +func (h *NotificationHandler) RegisterUserRoutes(rg *gin.RouterGroup) { notifications := rg.Group("/notifications") { - notifications.POST("", h.CreateNotification) notifications.GET("", h.GetNotifications) notifications.GET("/unread-count", h.GetUnreadCount) notifications.PUT("/:id/read", h.MarkAsRead) @@ -217,3 +216,11 @@ func (h *NotificationHandler) RegisterRoutes(rg *gin.RouterGroup) { notifications.DELETE("/:id", h.DeleteNotification) } } + +// RegisterAdminRoutes registers notification routes for admin/system use +func (h *NotificationHandler) RegisterAdminRoutes(rg *gin.RouterGroup) { + notifications := rg.Group("/notifications") + { + notifications.POST("", h.CreateNotification) + } +} diff --git a/internal/router/router.go b/internal/router/router.go index ea485f2..fb5161c 100644 --- a/internal/router/router.go +++ b/internal/router/router.go @@ -226,7 +226,8 @@ func Setup(db *gorm.DB, yunAPIClient *service.YunAPIClient, cfg *config.Config) aiHandler.RegisterRoutes(protected) // Feature: notification-center - Register notification routes - notificationHandler.RegisterRoutes(protected) + notificationHandler.RegisterUserRoutes(protected) + notificationHandler.RegisterAdminRoutes(v1) // Register report routes protected.GET("/reports/summary", reportHandler.GetTransactionSummary) @@ -520,7 +521,8 @@ func SetupWithRedis(db *gorm.DB, yunAPIClient *service.YunAPIClient, redisClient v1.POST("/app-lock/password/change", appLockHandler.ChangePassword) // Register notification routes - notificationHandler.RegisterRoutes(protected) + notificationHandler.RegisterUserRoutes(protected) + notificationHandler.RegisterAdminRoutes(v1) } return r, syncScheduler