nepiedg
9200600b1c
- 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.
112 lines
3.4 KiB
Go
112 lines
3.4 KiB
Go
package handler
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"wx_service/internal/middleware"
|
|
"wx_service/internal/model"
|
|
smokeservice "wx_service/internal/smoke/service"
|
|
)
|
|
|
|
type upsertSmokeProfileRequest struct {
|
|
BaselineCigsPerDay *int `json:"baseline_cigs_per_day"`
|
|
SmokingYears *float64 `json:"smoking_years"`
|
|
PackPriceCent *int `json:"pack_price_cent"`
|
|
|
|
SmokeMotivations *[]string `json:"smoke_motivations"`
|
|
QuitMotivations *[]string `json:"quit_motivations"`
|
|
|
|
WakeUpTime *string `json:"wake_up_time"`
|
|
SleepTime *string `json:"sleep_time"`
|
|
|
|
QuitDate *string `json:"quit_date"`
|
|
}
|
|
|
|
func (h *SmokeHandler) GetProfile(c *gin.Context) {
|
|
user := middleware.MustCurrentUser(c)
|
|
|
|
view, err := h.smokeProfileService.GetView(c.Request.Context(), int(user.ID))
|
|
if err != nil {
|
|
if errors.Is(err, smokeservice.ErrSmokeProfileInvalidTime) {
|
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "作息时间格式错误，应为 HH:MM"))
|
|
return
|
|
}
|
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取基础信息失败，请稍后重试"))
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.Success(view))
|
|
}
|
|
|
|
func (h *SmokeHandler) UpsertProfile(c *gin.Context) {
|
|
user := middleware.MustCurrentUser(c)
|
|
|
|
var req upsertSmokeProfileRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil && !errors.Is(err, io.EOF) {
|
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请求参数错误"))
|
|
return
|
|
}
|
|
|
|
if req.BaselineCigsPerDay != nil {
|
|
if *req.BaselineCigsPerDay < 0 || *req.BaselineCigsPerDay > 300 {
|
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "baseline_cigs_per_day 应在 0~300"))
|
|
return
|
|
}
|
|
}
|
|
if req.SmokingYears != nil {
|
|
if *req.SmokingYears < 0 || *req.SmokingYears > 80 {
|
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "smoking_years 应在 0~80"))
|
|
return
|
|
}
|
|
}
|
|
if req.PackPriceCent != nil {
|
|
if *req.PackPriceCent < 0 || *req.PackPriceCent > 1000000 {
|
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "pack_price_cent 应在 0~1000000"))
|
|
return
|
|
}
|
|
}
|
|
|
|
quitDateProvided := false
|
|
var quitDate *time.Time
|
|
if req.QuitDate != nil {
|
|
quitDateProvided = true
|
|
value := strings.TrimSpace(*req.QuitDate)
|
|
if value != "" {
|
|
parsed, err := time.ParseInLocation(dateLayout, value, time.Local)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "quit_date 格式错误，应为 YYYY-MM-DD"))
|
|
return
|
|
}
|
|
quitDate = &parsed
|
|
}
|
|
}
|
|
|
|
view, err := h.smokeProfileService.Upsert(c.Request.Context(), int(user.ID), smokeservice.UpsertSmokeProfileRequest{
|
|
BaselineCigsPerDay: req.BaselineCigsPerDay,
|
|
SmokingYears: req.SmokingYears,
|
|
PackPriceCent: req.PackPriceCent,
|
|
SmokeMotivations: req.SmokeMotivations,
|
|
QuitMotivations: req.QuitMotivations,
|
|
WakeUpTime: req.WakeUpTime,
|
|
SleepTime: req.SleepTime,
|
|
QuitDateProvided: quitDateProvided,
|
|
QuitDate: quitDate,
|
|
})
|
|
if err != nil {
|
|
if errors.Is(err, smokeservice.ErrSmokeProfileInvalidTime) {
|
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "作息时间格式错误，应为 HH:MM"))
|
|
return
|
|
}
|
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "保存基础信息失败，请稍后重试"))
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.Success(view))
|
|
}
|