func CreateUser(u *models.User, salt *models.Salts, body *user.Account) {
u.Mail = body.Mail
u.Name = body.Name
u.Phone = body.Phone
u.CreatedAt = time.Now().Unix()
u.UpdatedAt = time.Now().Unix()
salt.Salt = utils.EncryptRandSequence(5)
DecodeSalt, _ := utils.TokenDecode(salt.Salt)
u.Password, _ = utils.PsdHandler(body.Password, []byte(DecodeSalt))
}
func (this *Account) Register(body *user.Account) datatype.Response {
var res datatype.Response
//check mail
reg := regexp.MustCompile(`^(\w)+(\.\w+)*@(\w)+((\.\w+)+)$`)
sel := [...]bool{true, true, true}
u := models.User{}
sheet := models.Sheet{}
salt := models.Salts{}
flag := 0
// begin transaction
tx := this.DB.Begin()
if body.Password == "" {
goto BadRequest
}
if body.Name == "" || body.Mail == "" || body.Phone == "" {
goto BadRequest
}
if reg.FindAllString(body.Mail, -1) == nil {
goto BadRequest
}
if body.Name != "" {
tx.Where("name=?", body.Name).First(&u)
sel[0] = checkUser(u)
}
if body.Mail != "" {
tx.Where("mail=?", body.Mail).First(&u)
sel[1] = checkUser(u)
}
if body.Phone != "" {
tx.Where("phone=?", body.Phone).First(&u)
sel[2] = checkUser(u)
}
for i, v := range sel {
if v {
flag = i
goto Forbidden
}
}
CreateUser(&u, &salt, body)
tx.Create(&u)
//Create default sheet for user
tx.Where("mail=?", body.Mail).First(&u)
sheet.UserID = u.ID
sheet.Name = "default#" + strconv.Itoa(u.ID)
sheet.Privilege = "privacy"
sheet.CreatedAt = time.Now().Unix()
sheet.UpdatedAt = time.Now().Unix()
tx.Create(&sheet)
// transaction commit
salt.UserID = u.ID
tx.Create(&salt)
tx.Commit()
res = datatype.Response{
Status: http.StatusOK,
Body: u,
}
return res
BadRequest:
res = datatype.Response{
Status: http.StatusBadRequest,
}
return res
Forbidden:
var resText string
switch flag {
case 0:
resText = "Name existed"
case 1:
resText = "Mail existed"
case 2:
resText = "Phone existed"
}
res = datatype.Response{
Status: http.StatusForbidden,
ResponseText: resText,
}
return res
}