80 lignes
2,4 Kio
Go
80 lignes
2,4 Kio
Go
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)
|
|
// set default number of PP
|
|
user.Pantsu = 1
|
|
|
|
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
|
|
}
|