118 lines
2.7 KiB
Go
118 lines
2.7 KiB
Go
package controllers
|
|
|
|
import (
|
|
"hr_receiver/config"
|
|
"hr_receiver/models"
|
|
"hr_receiver/util"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type LoginRequest struct {
|
|
Username string `json:"username" binding:"required"`
|
|
Password string `json:"password" binding:"required"`
|
|
}
|
|
|
|
type RegisterRequest struct {
|
|
Username string `json:"username" form:"username"`
|
|
Password string `json:"password" form:"password"`
|
|
}
|
|
|
|
type AuthResponse struct {
|
|
Token string `json:"token"`
|
|
User models.User `json:"user"`
|
|
}
|
|
|
|
// Register 用户注册
|
|
func Register(c *gin.Context) {
|
|
var req RegisterRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// 检查用户名是否已存在
|
|
var existingUser models.User
|
|
if result := config.DB.Where("username = ?", req.Username).First(&existingUser); result.Error == nil {
|
|
c.JSON(http.StatusConflict, gin.H{"error": "Username already exists"})
|
|
return
|
|
}
|
|
|
|
// 创建新用户
|
|
user := models.User{
|
|
Username: req.Username,
|
|
Password: req.Password, // BeforeCreate钩子会自动加密
|
|
}
|
|
|
|
if result := config.DB.Create(&user); result.Error != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create user"})
|
|
return
|
|
}
|
|
|
|
// 生成Token
|
|
token, err := util.GenerateToken(user.ID, user.Username)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to generate token"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusCreated, AuthResponse{
|
|
Token: token,
|
|
User: user,
|
|
})
|
|
}
|
|
|
|
// Login 用户登录
|
|
func Login(c *gin.Context) {
|
|
var req LoginRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// 查找用户
|
|
var user models.User
|
|
result := config.DB.Where("username = ?", req.Username).First(&user)
|
|
|
|
if result.Error != nil {
|
|
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid username or password"})
|
|
return
|
|
}
|
|
|
|
// 验证密码
|
|
if !user.CheckPassword(req.Password) {
|
|
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid username or password"})
|
|
return
|
|
}
|
|
|
|
// 生成JWT Token
|
|
token, err := util.GenerateToken(user.ID, user.Username)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to generate token"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, AuthResponse{
|
|
Token: token,
|
|
User: user,
|
|
})
|
|
}
|
|
|
|
// GetProfile 获取用户信息（需要认证）
|
|
func GetProfile(c *gin.Context) {
|
|
userID, exists := c.Get("userID")
|
|
if !exists {
|
|
c.JSON(http.StatusUnauthorized, gin.H{"error": "Unauthorized"})
|
|
return
|
|
}
|
|
|
|
var user models.User
|
|
if result := config.DB.First(&user, userID); result.Error != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, user)
|
|
}
|