Enhance smoking tracking API with new features and improvements
- Added a new API endpoint `GET /api/v1/smoke/home` to consolidate core modules for the home dashboard, reducing the need for multiple requests. - Updated the `smoke` routes to include the new home endpoint and improved user profile management with the addition of a `quit_date` field. - Enhanced the algorithm for calculating daily targets and next smoke suggestions, ensuring accurate future time handling and user-specific recommendations. - Improved API documentation to reflect new endpoints, response formats, and detailed field descriptions for better clarity and usability. - Refactored user authentication handling in various handlers to streamline the process and ensure consistent error responses.
This commit is contained in:
@@ -1,6 +1,8 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"wx_service/internal/model"
|
||||
@@ -16,3 +18,34 @@ func CurrentUser(c *gin.Context) (*model.User, bool) {
|
||||
user, ok := userVal.(*model.User)
|
||||
return user, ok
|
||||
}
|
||||
|
||||
// MustCurrentUser 仅用于已通过鉴权中间件的路由,避免每个 handler 重复判断。
|
||||
func MustCurrentUser(c *gin.Context) *model.User {
|
||||
userVal := c.MustGet(ContextCurrentUserKey)
|
||||
user, ok := userVal.(*model.User)
|
||||
if !ok || user == nil {
|
||||
panic("current user missing in context")
|
||||
}
|
||||
return user
|
||||
}
|
||||
|
||||
// RequireUser 是对 CurrentUser 的封装,复用统一的未登录响应。
|
||||
func RequireUser(c *gin.Context) (*model.User, bool) {
|
||||
user, ok := CurrentUser(c)
|
||||
if ok {
|
||||
return user, true
|
||||
}
|
||||
c.AbortWithStatusJSON(http.StatusUnauthorized, model.Error(http.StatusUnauthorized, "未登录或登录已过期"))
|
||||
return nil, false
|
||||
}
|
||||
|
||||
// RequireUserMiddleware 将登录校验统一下沉到路由层。
|
||||
func RequireUserMiddleware() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
if _, ok := CurrentUser(c); !ok {
|
||||
c.AbortWithStatusJSON(http.StatusUnauthorized, model.Error(http.StatusUnauthorized, "未登录或登录已过期"))
|
||||
return
|
||||
}
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user