134 lines
2.8 KiB
Go
134 lines
2.8 KiB
Go
package controllers
|
|
|
|
// Пример генератора токена
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/base64"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-playground/validator/v10"
|
|
"golang-test/database"
|
|
"golang-test/libs"
|
|
"golang-test/message"
|
|
"golang-test/validators"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
//func tokenGenerator() string {
|
|
// b := make([]byte, 4)
|
|
// rand.Read(b)
|
|
// return fmt.Sprintf("%x", b)
|
|
//}
|
|
|
|
// func main() {
|
|
// a := tokenGenerator()
|
|
// fmt.Println(a)
|
|
// }
|
|
|
|
type TokenStruct struct {
|
|
UserID uint
|
|
token string
|
|
}
|
|
|
|
func RegisterUser(c *gin.Context, user database.User) {
|
|
db := database.Connector()
|
|
validate := validators.Validate
|
|
response := message.Response{Status: 200}
|
|
if err := validate.Struct(user); err != nil {
|
|
response.Error = libs.GetValidationErrors(err.(validator.ValidationErrors))
|
|
response.Status = 400
|
|
message.SendResponse(c, response)
|
|
return
|
|
}
|
|
hash, err := bcrypt.GenerateFromPassword([]byte(user.Password), 12)
|
|
if err != nil {
|
|
return
|
|
}
|
|
user.Password = string(hash)
|
|
if err := db.Create(user).Error; err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func LoginUser(c *gin.Context, email, password string) *Token {
|
|
var user database.User
|
|
db := database.Connector()
|
|
validate := validators.Validate
|
|
if err := db.Where("email = ?", email).First(&user).Error; err != nil {
|
|
return nil
|
|
}
|
|
if err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password)); err != nil {
|
|
return nil
|
|
}
|
|
token := &Token{
|
|
UserID: user.ID,
|
|
Token: generateToken(),
|
|
}
|
|
if err := db.Create(token).Error; err != nil {
|
|
return nil
|
|
}
|
|
return token
|
|
}
|
|
|
|
func generateToken() string {
|
|
b := make([]byte, 4)
|
|
if _, err := rand.Read(b); err != nil {
|
|
return ""
|
|
}
|
|
return base64.StdEncoding.EncodeToString(b)
|
|
}
|
|
|
|
func GetUserByToken(c *gin.Context, token string) (database.User, error) {
|
|
var tokenObject TokenStruct
|
|
var err error
|
|
var user database.User
|
|
|
|
db := database.Connector()
|
|
if err := db.Where("token = ?", token).First(&tokenObject).Error; err != nil {
|
|
return user, err
|
|
}
|
|
if err := db.First(&user, tokenObject.UserID).Error; err != nil {
|
|
return user, err
|
|
}
|
|
return user, err
|
|
}
|
|
|
|
// asdasdasd
|
|
func EditUser(c *gin.Context, token string, user database.User) {
|
|
db := database.Connector()
|
|
validate := validators.Validate
|
|
u := GetUserByToken(c, token)
|
|
if err := validate.Struct(user); err != nil {
|
|
return
|
|
}
|
|
if user.Name != "" {
|
|
u.Name = user.Name
|
|
}
|
|
if user.Email != "" {
|
|
u.Email = user.Email
|
|
}
|
|
if user.Password != "" {
|
|
hash, err := bcrypt.GenerateFromPassword([]byte(user.Password), 12)
|
|
if err != nil {
|
|
return
|
|
}
|
|
u.Password = string(hash)
|
|
}
|
|
if err := db.Save(u).Error; err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func DeleteUser(c *gin.Context, token string) {
|
|
u, err := GetUserByToken(db, token)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if err := db.Delete(u).Error; err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|