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/helpers.go
2017-07-03 02:03:57 +02:00

80 lignes
2,3 Kio
Go

package users
import (
"errors"
"net/http"
"strconv"
"github.com/NyaaPantsu/nyaa/models"
"github.com/NyaaPantsu/nyaa/utils/log"
"github.com/NyaaPantsu/nyaa/utils/validator/user"
"golang.org/x/crypto/bcrypt"
)
// CheckEmail : check if email is in database
func CheckEmail(email string) bool {
if len(email) == 0 {
return false
}
var count int
models.ORM.Model(models.User{}).Where("email = ?", email).Count(&count)
return count != 0
}
// Exists : check if the users credentials match to a user in db
func Exists(email string, pass string) (user *models.User, status int, err error) {
if email == "" || pass == "" {
return user, http.StatusNotFound, errors.New("no_username_password")
}
var userExist = &models.User{}
// search by email or username
if userValidator.EmailValidation(email) {
if models.ORM.Where("email = ?", email).First(userExist).RecordNotFound() {
status, err = http.StatusNotFound, errors.New("user_not_found")
return
}
} else if models.ORM.Where("username = ?", email).First(userExist).RecordNotFound() {
status, err = http.StatusNotFound, errors.New("user_not_found")
return
}
user = userExist
err = bcrypt.CompareHashAndPassword([]byte(userExist.Password), []byte(pass))
if err != nil {
status, err = http.StatusUnauthorized, errors.New("incorrect_password")
return
}
if userExist.IsBanned() {
status, err = http.StatusUnauthorized, errors.New("account_banned")
return
}
if userExist.IsScraped() {
status, err = http.StatusUnauthorized, errors.New("account_need_activation")
return
}
status, err = http.StatusOK, nil
return
}
// SuggestUsername suggest user's name if user's name already occupied.
func SuggestUsername(username string) string {
var count int
var usernameCandidate string
models.ORM.Model(models.User{}).Where(&models.User{Username: username}).Count(&count)
log.Debugf("count Before : %d", count)
if count == 0 {
return username
}
var postfix int
for {
usernameCandidate = username + strconv.Itoa(postfix)
log.Debugf("usernameCandidate: %s\n", usernameCandidate)
models.ORM.Model(models.User{}).Where(&models.User{Username: usernameCandidate}).Count(&count)
log.Debugf("count after : %d\n", count)
postfix = postfix + 1
if count == 0 {
break
}
}
return usernameCandidate
}