-
Notifications
You must be signed in to change notification settings - Fork 150
/
create.go
77 lines (71 loc) · 2.32 KB
/
create.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
package users
import (
"errors"
"net/http"
"time"
"github.com/NyaaPantsu/nyaa/models"
"github.com/NyaaPantsu/nyaa/utils/crypto"
"github.com/NyaaPantsu/nyaa/utils/log"
msg "github.com/NyaaPantsu/nyaa/utils/messages"
"github.com/NyaaPantsu/nyaa/utils/validator"
"github.com/NyaaPantsu/nyaa/utils/validator/user"
"github.com/gin-gonic/gin"
"golang.org/x/crypto/bcrypt"
)
// CreateUserFromForm creates a user from a registration form.
func CreateUserFromRequest(registrationForm *userValidator.RegistrationForm) (*models.User, error) {
var user = &models.User{}
log.Debugf("registrationForm %+v\n", registrationForm)
validator.Bind(user, registrationForm)
if user.Email == "" {
user.MD5 = ""
} else {
// Despite the email not being verified yet we calculate this for convenience reasons
var err error
user.MD5, err = crypto.GenerateMD5Hash(user.Email)
if err != nil {
return user, err
}
}
user.Email = "" // unset email because it will be verified later
user.CreatedAt = time.Now()
// User settings to default
user.Settings.ToDefault()
user.SaveSettings()
// currently unused but needs to be set:
user.APIToken, _ = crypto.GenerateRandomToken32()
user.APITokenExpiry = time.Unix(0, 0)
if models.ORM.Create(user).Error != nil {
return user, errors.New("user not created")
}
return user, nil
}
// CreateUser creates a user.
func CreateUser(c *gin.Context) (*models.User, int) {
var user = &models.User{}
var registrationForm userValidator.RegistrationForm
var err error
messages := msg.GetMessages(c)
c.Bind(®istrationForm)
usernameCandidate := SuggestUsername(registrationForm.Username)
if usernameCandidate != registrationForm.Username {
messages.AddErrorTf("username", "username_taken", usernameCandidate)
return user, http.StatusInternalServerError
}
if registrationForm.Email != "" && CheckEmail(registrationForm.Email) {
messages.AddErrorT("email", "email_in_db")
return user, http.StatusInternalServerError
}
password, err := bcrypt.GenerateFromPassword([]byte(registrationForm.Password), 10)
if err != nil {
messages.Error(err)
return user, http.StatusInternalServerError
}
registrationForm.Password = string(password)
user, err = CreateUserFromRequest(®istrationForm)
if err != nil {
messages.Error(err)
return user, http.StatusInternalServerError
}
return user, http.StatusOK
}