127 lines
3.2 KiB
Go
127 lines
3.2 KiB
Go
package handler
|
|
|
|
import (
|
|
"database/sql"
|
|
"net/http"
|
|
|
|
"memory/internal/config"
|
|
"memory/internal/middleware"
|
|
"memory/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
|
|
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, &user.CreatedAt)
|
|
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "database error"})
|
|
return
|
|
}
|
|
|
|
// 获取统计数据
|
|
var postCount, likeCount int
|
|
h.db.QueryRow("SELECT COUNT(*) FROM posts WHERE user_id = ?", userID).Scan(&postCount)
|
|
h.db.QueryRow("SELECT COUNT(*) FROM likes l JOIN posts p ON l.post_id = p.id WHERE p.user_id = ?", userID).Scan(&likeCount)
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"user": user,
|
|
"post_count": postCount,
|
|
"like_count": likeCount,
|
|
})
|
|
}
|
|
|
|
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 = COALESCE(NULLIF(?, ''), 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 req map[string]string
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
for key, value := range req {
|
|
_, err := h.db.Exec("INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)", key, value)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to update settings"})
|
|
return
|
|
}
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "settings updated"})
|
|
}
|