123 lines
3.7 KiB
Go
123 lines
3.7 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/golang-jwt/jwt/v5"
|
|
"golang.org/x/crypto/bcrypt"
|
|
|
|
"mrp/database"
|
|
"mrp/models"
|
|
)
|
|
|
|
func Login(c *gin.Context) {
|
|
jwtKey := []byte(os.Getenv("JWT_SECRET"))
|
|
var input struct {
|
|
Username string `json:"username"`
|
|
Password string `json:"password"`
|
|
}
|
|
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Неверный ввод"})
|
|
return
|
|
}
|
|
|
|
var user models.User
|
|
|
|
err := database.DB.Where("username = ?", input.Username).First(&user).Error
|
|
|
|
passErr := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(input.Password))
|
|
|
|
if err != nil || passErr != nil {
|
|
c.JSON(http.StatusUnauthorized, gin.H{"error": "Неверный логин или пароль"})
|
|
return
|
|
}
|
|
|
|
token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
|
|
"user_id": user.ID,
|
|
"role": user.Role,
|
|
"exp": time.Now().Add(time.Hour * 24).Unix(),
|
|
})
|
|
|
|
tokenString, _ := token.SignedString(jwtKey)
|
|
|
|
c.JSON(http.StatusOK, gin.H{"token": tokenString, "user": gin.H{
|
|
"id": user.ID,
|
|
"first_name": user.FirstName,
|
|
"last_name": user.LastName,
|
|
"phone": user.Phone,
|
|
"department": user.Department,
|
|
"role": user.Role,
|
|
}})
|
|
}
|
|
|
|
func Register(c *gin.Context) {
|
|
var input struct {
|
|
Username string `json:"username" binding:"required"`
|
|
Password string `json:"password" binding:"required,min=8"`
|
|
FirstName string `json:"first_name" binding:"required"`
|
|
LastName string `json:"last_name" binding:"required"`
|
|
Phone string `json:"phone"`
|
|
Department string `json:"department" binding:"required"`
|
|
Role string `json:"role" binding:"required"`
|
|
}
|
|
|
|
// 1. Проверяем входящие данные
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Ошибка валидации ", "details": err.Error()})
|
|
return
|
|
}
|
|
|
|
// 2. Хешируем пароль (чтобы не хранить его в открытом виде)
|
|
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(input.Password), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Ошибка при шифровании пароля"})
|
|
return
|
|
}
|
|
|
|
// 3. Создаем объект пользователя
|
|
user := models.User{
|
|
Username: input.Username,
|
|
Password: string(hashedPassword),
|
|
FirstName: input.FirstName,
|
|
LastName: input.LastName,
|
|
Phone: input.Phone,
|
|
Department: input.Department,
|
|
Role: input.Role, // по умолчанию
|
|
}
|
|
|
|
// 4. Сохраняем в базу через GORM
|
|
if err := database.DB.Create(&user).Error; err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Ошибка при регистрации пользователя", "details": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"message": "Пользователь успешно зарегистрирован"})
|
|
}
|
|
|
|
func UpdateProfile(c *gin.Context) {
|
|
var input models.UpdateUserInput
|
|
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
userID := c.MustGet("user_id")
|
|
|
|
if err := database.DB.Model(&models.User{}).Where("id = ?", userID).Updates(models.User{
|
|
FirstName: input.FirstName,
|
|
LastName: input.LastName,
|
|
Phone: input.Phone,
|
|
Department: input.Department,
|
|
Role: input.Role,
|
|
}).Error; err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Не удалось обновить данные", "details": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "Данные успешно обновлены"})
|
|
}
|