Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/models/users/create.go

77 lignes
2,3 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)
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(&registrationForm)
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(&registrationForm)
if err != nil {
messages.Error(err)
return user, http.StatusInternalServerError
}
return user, http.StatusOK
}