/
user.go
90 lines (74 loc) · 2.54 KB
/
user.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package services
import (
"errors"
"strings"
"github.com/wuttinanhi/code-judge-system/entities"
"github.com/wuttinanhi/code-judge-system/repositories"
"golang.org/x/crypto/bcrypt"
)
type UserService interface {
Register(email, password, displayname string) (user *entities.User, err error)
Login(email, password string) (user *entities.User, err error)
UpdateRole(user *entities.User, role string) (err error)
FindUserByID(id uint) (user *entities.User, err error)
Pagination(options *entities.PaginationOptions) (result *entities.PaginationResult[*entities.User], err error)
}
type userService struct {
userRepo repositories.UserRepository
}
// Pagination implements UserService.
func (s *userService) Pagination(options *entities.PaginationOptions) (result *entities.PaginationResult[*entities.User], err error) {
result, err = s.userRepo.Pagination(options)
return
}
// FindUserByID implements UserService.
func (s *userService) FindUserByID(id uint) (user *entities.User, err error) {
user, err = s.userRepo.FindUserByID(id)
return user, err
}
// UpdateRole implements UserService.
func (s *userService) UpdateRole(user *entities.User, role string) (err error) {
user.Role = role
err = s.userRepo.UpdateUser(user)
return err
}
// Login implements services.UserService.
func (s *userService) Login(email string, password string) (user *entities.User, err error) {
user, err = s.userRepo.FindUserByEmail(email)
if err != nil {
return nil, errors.New("failed to get user")
}
err = bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password))
if err != nil {
return nil, errors.New("invalid password")
}
return user, nil
}
// Register implements services.UserService.
func (s *userService) Register(email string, password string, displayname string) (user *entities.User, err error) {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
if err != nil {
return nil, errors.New("failed to hash password")
}
user = &entities.User{
Email: email,
Password: string(hashedPassword),
DisplayName: displayname,
Role: "user",
}
// Save the user to the repository
user, err = s.userRepo.CreateUser(user)
if err != nil {
// if error contains "UNIQUE constraint failed"
if strings.Contains(err.Error(), "UNIQUE constraint failed") {
return nil, errors.New("user already exists")
}
return nil, errors.New("failed to register user")
}
return user, nil
}
func NewUserService(userRepo repositories.UserRepository) UserService {
return &userService{
userRepo: userRepo,
}
}