Files
healthflow/server/internal/handler/user.go
2025-12-20 23:51:40 +08:00

116 lines
2.8 KiB
Go

package handler
import (
"database/sql"
"net/http"
"healthflow/internal/config"
"healthflow/internal/middleware"
"healthflow/internal/model"
"github.com/gin-gonic/gin"
)
type UserHandler struct {
db *sql.DB
cfg *config.Config
}
func NewUserHandler(db *sql.DB, cfg *config.Config) *UserHandler {
return &UserHandler{db: db, cfg: cfg}
}
func (h *UserHandler) GetProfile(c *gin.Context) {
userID := middleware.GetUserID(c)
var user model.User
var createdAt string
err := h.db.QueryRow(
"SELECT id, username, nickname, avatar_url, bio, is_admin, created_at FROM users WHERE id = ?",
userID,
).Scan(&user.ID, &user.Username, &user.Nickname, &user.AvatarURL, &user.Bio, &user.IsAdmin, &createdAt)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "database error"})
return
}
c.JSON(http.StatusOK, model.ProfileResponse{
User: &user,
CreatedAt: createdAt,
})
}
func (h *UserHandler) UpdateProfile(c *gin.Context) {
userID := middleware.GetUserID(c)
var req model.UpdateProfileRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
_, err := h.db.Exec(
"UPDATE users SET nickname = ?, bio = ? WHERE id = ?",
req.Nickname, req.Bio, userID,
)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to update profile"})
return
}
c.JSON(http.StatusOK, gin.H{"message": "profile updated"})
}
func (h *UserHandler) UpdateAvatar(c *gin.Context) {
userID := middleware.GetUserID(c)
var req struct {
AvatarURL string `json:"avatar_url" binding:"required"`
}
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
_, err := h.db.Exec("UPDATE users SET avatar_url = ? WHERE id = ?", req.AvatarURL, userID)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to update avatar"})
return
}
c.JSON(http.StatusOK, gin.H{"message": "avatar updated"})
}
func (h *UserHandler) GetSettings(c *gin.Context) {
rows, err := h.db.Query("SELECT key, value FROM settings")
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "database error"})
return
}
defer rows.Close()
settings := make(map[string]string)
for rows.Next() {
var key, value string
rows.Scan(&key, &value)
settings[key] = value
}
c.JSON(http.StatusOK, settings)
}
func (h *UserHandler) UpdateSettings(c *gin.Context) {
var settings map[string]string
if err := c.ShouldBindJSON(&settings); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
for key, value := range settings {
h.db.Exec("INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)", key, value)
}
c.JSON(http.StatusOK, gin.H{"message": "settings updated"})
}