Du kan inte välja fler än 25 ämnen
Ämnen måste starta med en bokstav eller siffra, kan innehålla bindestreck ('-') och vara max 35 tecken långa.
107 rader
2.7 KiB
107 rader
2.7 KiB
package controllers
|
|
|
|
import (
|
|
"github.com/astaxie/beego"
|
|
"beeblog/service"
|
|
"beeblog/models"
|
|
"strings"
|
|
"crypto/md5"
|
|
"time"
|
|
"strconv"
|
|
"encoding/hex"
|
|
)
|
|
|
|
type UserController struct {
|
|
beego.Controller
|
|
}
|
|
|
|
func (u *UserController) LoginPage() {
|
|
|
|
//
|
|
//for i:=0; i<len(tests) ; i++ {
|
|
// fmt.Println(tests[i].Title,tests[i].Id)
|
|
//}
|
|
//fmt.Println(len(tests))
|
|
//num, err = o.Delete(&u)
|
|
//fmt.Printf("NUM: %d, ERR: %v\n", num, err)
|
|
|
|
//u.Ctx.WriteString("login page")
|
|
u.TplName = "login.html"
|
|
}
|
|
|
|
func (this *UserController) Login() {
|
|
username := this.GetString("username")
|
|
userpwd := this.GetString("userpwd")
|
|
if username == "" {
|
|
this.Data["json"] = models.ReurnError("用户名为空")
|
|
this.ServeJSON()
|
|
}
|
|
if len(username) < 4 {
|
|
this.Data["json"] = models.ReurnError("用户名最低4位")
|
|
this.ServeJSON()
|
|
}
|
|
if userpwd == "" {
|
|
this.Data["json"] = models.ReurnError("密码为空")
|
|
this.ServeJSON()
|
|
}
|
|
if len(userpwd) < 6 {
|
|
this.Data["json"] = models.ReurnError("密码最低6位")
|
|
this.ServeJSON()
|
|
}
|
|
user, error := service.FindByUserName(username)
|
|
if error == nil && user != nil {
|
|
h := md5.New()
|
|
h.Write([]byte(userpwd + user.Salt))
|
|
userpwd = hex.EncodeToString(h.Sum(nil))
|
|
if userpwd == user.UserPwd {
|
|
this.Data["json"] = models.ReurnSuccess("")
|
|
} else {
|
|
this.Data["json"] = models.ReurnError("用户名或密码错误")
|
|
}
|
|
} else {
|
|
this.Data["json"] = models.ReurnError("用户名不存在")
|
|
}
|
|
this.ServeJSON()
|
|
}
|
|
|
|
func (this *UserController) Regist() {
|
|
username := this.GetString("username")
|
|
userpwd := this.GetString("userpwd")
|
|
username = strings.Replace(username, " ", "", -1)
|
|
userpwd = strings.Replace(userpwd, " ", "", -1)
|
|
if username == "" {
|
|
this.Data["json"] = models.ReurnError("用户名为空")
|
|
this.ServeJSON()
|
|
}
|
|
if len(username) < 4 {
|
|
this.Data["json"] = models.ReurnError("用户名最低4位")
|
|
this.ServeJSON()
|
|
}
|
|
if userpwd == "" {
|
|
this.Data["json"] = models.ReurnError("密码为空")
|
|
this.ServeJSON()
|
|
}
|
|
if len(userpwd) < 6 {
|
|
this.Data["json"] = models.ReurnError("密码最低6位")
|
|
this.ServeJSON()
|
|
}
|
|
user, _ := service.FindByUserName(username)
|
|
if user != nil {
|
|
this.Data["json"] = models.ReurnError("用户已经存在")
|
|
this.ServeJSON()
|
|
}
|
|
h := md5.New()
|
|
h.Write([]byte(strconv.FormatInt(time.Now().Unix(), 10) + beego.AppConfig.String("host")))
|
|
salt := hex.EncodeToString(h.Sum(nil))
|
|
h = md5.New()
|
|
h.Write([]byte(userpwd + salt))
|
|
userpwd = hex.EncodeToString(h.Sum(nil))
|
|
user = &models.User{UserName: username, UserPwd: userpwd, Salt: salt}
|
|
err := service.SaveUser(user)
|
|
if err == nil {
|
|
this.Data["json"] = user
|
|
} else {
|
|
this.Data["json"] = models.ReurnError("注册失败")
|
|
}
|
|
this.ServeJSON()
|
|
}
|
|
|