-
Notifications
You must be signed in to change notification settings - Fork 1
/
admin_user.go
106 lines (87 loc) · 2.35 KB
/
admin_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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package logic
import (
"errors"
"github.com/ic3network/mccs-alpha-api/internal/app/repository/mongo"
"github.com/ic3network/mccs-alpha-api/internal/app/repository/redis"
"github.com/ic3network/mccs-alpha-api/internal/app/types"
"github.com/ic3network/mccs-alpha-api/util/bcrypt"
"github.com/spf13/viper"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type adminUser struct{}
var AdminUser = &adminUser{}
func (a *adminUser) FindByID(id primitive.ObjectID) (*types.AdminUser, error) {
adminUser, err := mongo.AdminUser.FindByID(id)
if err != nil {
return nil, err
}
return adminUser, nil
}
func (a *adminUser) FindByIDString(id string) (*types.AdminUser, error) {
objectID, err := primitive.ObjectIDFromHex(id)
if err != nil {
return nil, err
}
adminUser, err := mongo.AdminUser.FindByID(objectID)
if err != nil {
return nil, err
}
return adminUser, nil
}
func (a *adminUser) FindByEmail(email string) (*types.AdminUser, error) {
adminUser, err := mongo.AdminUser.FindByEmail(email)
if err != nil {
return nil, err
}
return adminUser, nil
}
func (a *adminUser) Login(email string, password string) (*types.AdminUser, error) {
user, err := mongo.AdminUser.FindByEmail(email)
if err != nil {
return &types.AdminUser{}, err
}
attempts := redis.GetLoginAttempts(email)
if attempts >= viper.GetInt("login_attempts.limit") {
return nil, ErrLoginLocked
}
err = bcrypt.CompareHash(user.Password, password)
if err != nil {
if attempts+1 >= viper.GetInt("login_attempts.limit") {
return nil, ErrLoginLocked
}
return nil, errors.New("Invalid password.")
}
redis.ResetLoginAttempts(email)
return user, nil
}
func (u *adminUser) IncLoginAttempts(email string) error {
err := redis.IncLoginAttempts(email)
if err != nil {
return err
}
return nil
}
// POST /admin/login
func (a *adminUser) UpdateLoginInfo(id primitive.ObjectID, ip string) (*types.LoginInfo, error) {
info, err := mongo.AdminUser.UpdateLoginInfo(id, ip)
if err != nil {
return nil, err
}
return info, nil
}
func (a *adminUser) ResetPassword(email string, newPassword string) error {
user, err := mongo.AdminUser.FindByEmail(email)
if err != nil {
return err
}
hashedPassword, err := bcrypt.Hash(newPassword)
if err != nil {
return err
}
user.Password = hashedPassword
err = mongo.AdminUser.UpdatePassword(user)
if err != nil {
return err
}
return nil
}