func (l *LoginController) Recovery() {
email := l.GetString("email")
valid := validation.Validation{}
valid.Email(email, "email")
if valid.HasErrors() {
var msg string
for _, err := range valid.Errors {
msg += fmt.Sprintf("%s: %s\r", err.Key, err.Message)
}
l.Ctx.ResponseWriter.WriteHeader(400)
l.Data["json"] = &map[string]interface{}{"status": "error", "message": msg}
l.ServeJson()
return
}
user, _ := rbacModels.UserGetByEmail(email)
if user.Id == 0 {
l.Data["json"] = &map[string]interface{}{"status": "error", "message": "Пользователь не найден"}
} else {
site_url := beego.AppConfig.String("site_url")
reset_token, _ := user.GenResetPassToken()
params := map[string]interface{}{}
params["user:one-time-login-url"] = fmt.Sprintf("%s/admin/reset?token=%s", site_url, reset_token)
msg := emailSystem.New()
msg.Template = "email_password_reset"
msg.Params = ¶ms
msg.User = user
if err := msg.SendMail(); err == nil {
l.Data["json"] = &map[string]interface{}{"status": "info", "message": "Дальнейшие инструкции высланы на указанный почтовый ящик"}
}
}
l.ServeJson()
}
/**
* 登录
*/
func (this *LoginController) Login() {
val := validation.Validation{}
username := this.GetString("username")
password := this.GetString("password")
val.Required(username, "username")
val.Required(password, "password")
if val.HasErrors() {
for _, v := range val.Errors {
this.Ctx.WriteString("<script>alert('" + v.Key + " " + v.Message + "');</script>") //遇到一个错误就可以StopRun了其实，这里只是练习验证模块
}
this.StopRun()
}
toolbox.Display("u", username)
toolbox.Display("pa", password)
password = utils.LoginPassword(password)
toolbox.Display("paaa", password)
if user.CheckLogin(username, password) == true {
this.SetSession("username", username)
this.Ctx.SetCookie("username", username)
this.Redirect("/admin", 302)
} else {
this.Redirect("/login", 302)
}
}
/* 测试战斗提交表单 */
func (this *ApiController) TestPost() {
//测试提交的表单
valid := validation.Validation{}
valid.Required(this.GetString("my"), "1")
valid.Match(this.GetString("my"), regexp.MustCompile("^[,0-9]+$"), "2")
valid.Required(this.GetString("enemie"), "3")
valid.Match(this.GetString("enemie"), regexp.MustCompile("^[,0-9]+$"), "4")
if valid.HasErrors() { //没有通过验证则退出
return
}
//解析提交的敌我方数据
myArray := strings.Split(this.GetString("my"), ",")
enemieArray := strings.Split(this.GetString("enemie"), ",")
mySlice := make([]string, len(myArray))
enemieSlice := make([]string, len(enemieArray))
for k, v := range myArray {
mySlice[k] = v + ",0,0,0,0,0"
}
for k, v := range enemieArray {
enemieSlice[k] = v + ",0,0,0,0,0"
}
//模拟战斗
index := IndexController{}
result := index.SimulateFight(mySlice, "0;0", enemieSlice, "0;0")
this.Data["json"] = result
this.ServeJson()
}
//用户注册
func (this *User) Register() {
req := this.ReqJson()
if req != nil {
email := req.Get("email").MustString()
nickName := req.Get("nickName").MustString()
password := req.Get("password").MustString()
valid := validation.Validation{}
valid.Email(email, "email")
valid.MinSize(nickName, 6, "nickNameMin")
valid.MaxSize(nickName, 12, "nickNameMax")
valid.MinSize(password, 6, "passwordMin")
valid.MaxSize(password, 12, "passwordMax")
if valid.HasErrors() {
this.CustomAbort(enum.BadRequest.Code(), enum.BadRequest.Str())
}
err := user.AddUser(email, nickName, password)
if err == nil {
this.RespJson(enum.OK, nil)
} else {
if strings.Contains(err.Error(), "email") {
this.RespJson(enum.EmailAlreadyExist, nil)
} else if strings.Contains(err.Error(), "nickname") {
this.RespJson(enum.NickNameAlreadyExist, nil)
} else {
beego.Error(err)
}
}
}
}
func (form *LoginForm) Valid(v *validation.Validation) {
valid := validation.Validation{}
valid.Email(form.Email, "email")
if valid.HasErrors() {
v.SetError("Email", form.Locale.Tr("valid.invalid_email"))
}
}
func (conf *Config) ValidateCommonSpecs() bool {
valid := validation.Validation{}
//Validate mandatory fields.
valid.Required(conf.configLinux.Version, "Version")
//Version must complient with SemVer v2.0.0
valid.Match(conf.configLinux.Version, regexp.MustCompile("^(\\d+\\.)?(\\d+\\.)?(\\*|\\d+)$"), "Version")
valid.Required(conf.configLinux.Platform.OS, "OS")
valid.Required(conf.configLinux.Platform.Arch, "Platform.Arch")
for _, env := range conf.configLinux.Process.Env {
//If Process defined, env cannot be empty
valid.Required(env, "Process.Env")
}
valid.Required(conf.configLinux.Process.User.UID, "User.UID")
valid.Required(conf.configLinux.Process.User.GID, "User.GID")
valid.Required(conf.configLinux.Root.Path, "Root.Path")
//Iterate over Mount array
for _, mount := range conf.configLinux.Mounts {
//If Mount points defined, it must define these three.
valid.Required(mount.Type, "Mount.Type")
valid.Required(mount.Source, "Mount.Source")
valid.Required(mount.Destination, "Mount.Destination")
}
if valid.HasErrors() {
// validation does not pass
for i, err := range valid.Errors {
fmt.Println(i, err.Key, err.Message)
}
return false
}
return true
}
/**
* 登录接收
*/
func (this *UserController) Signup() {
//errmsg := make(map[string]string)
requestEmail := this.GetString("email")
requestPassword := this.GetString("password")
svalid := validation.Validation{}
svalid.Required(requestEmail, "email")
svalid.MaxSize(requestEmail, 40, "email")
svalid.Required(requestPassword, "password")
if svalid.HasErrors() {
for _, err := range svalid.Errors {
//fmt.Println(err.Key, err.Message)
this.showmsg("Error Message", "["+err.Key+"]"+err.Message)
}
}
var user models.User
user.Email = requestEmail
user.Password = models.Md5([]byte(requestPassword))
var Ru models.ReturnUser
Ru = user.Select()
if Ru.Id > 0 {
this.SetSession("nickname", Ru.Nickname)
this.SetSession("email", Ru.Email)
this.SetSession("isadmin", Ru.Isadmin)
this.Redirect("/", 302)
} else {
this.showmsg("Error", "Email OR Password IS WRONG")
}
}
//用户登录
func (this *User) Login() {
json := this.ReqJson()
if json != nil {
email := json.Get("email").MustString()
password := json.Get("password").MustString()
valid := validation.Validation{}
valid.Email(email, "email")
valid.MinSize(password, 6, "passwordMin")
valid.MaxSize(password, 12, "passwordMax")
if valid.HasErrors() {
this.CustomAbort(enum.BadRequest.Code(), enum.BadRequest.Str())
}
u := user.GetUserByEmail(email)
if u == nil {
//用户不存在
this.RespJson(enum.UserNotExist, nil)
} else if util.Md5(u.Salt+password) != u.Password {
//密码错误
this.RespJson(enum.PasswordIncorrect, nil)
} else {
this.SetSession("uId", u.Id.Hex())
user.SetToken(u.Id, this.StartSession().SessionID()) //using cookie as token
this.RespJson(enum.OK, map[string]interface{}{"url": "/"})
}
}
}
func (this *UserController) Login() {
if this.Ctx.Input.Method() == "GET" {
if this.GetSession("user") != nil {
this.Redirect("/", 302)
}
this.TplNames = "console/login.html"
return
}
json := this.ReqJson()
email := json.Get("email").MustString()
password := json.Get("password").MustString()
valid := validation.Validation{}
valid.Email(email, "email")
valid.MinSize(password, 6, "passwordMin")
valid.MaxSize(password, 12, "passwordMax")
if valid.HasErrors() {
this.CustomAbort(enum.RespCode.BadRequest.Code(), enum.RespCode.BadRequest.Str())
}
user := user.GetUserByEmail(email)
if user == nil {
//用户不存在
this.RespJson(enum.RespCode.UserNotExist, nil)
} else if util.Md5(user.Salt+password) != user.Password {
//密码错误
this.RespJson(enum.RespCode.PasswordIncorrect, nil)
} else {
this.SetSession("user", user)
this.RespJson(enum.RespCode.OK, map[string]interface{}{"url": "/"})
}
}
//注册 todo 错误处理的优雅封装
func (this *User) Register() {
reqBody := this.Ctx.Input.RequestBody
reqJson, err := simplejson.NewJson(reqBody)
//非法请求
if err != nil {
this.Abort("400")
}
email := reqJson.Get("email").MustString()
nickname := reqJson.Get("nickname").MustString()
password := reqJson.Get("password").MustString()
valid := validation.Validation{}
valid.Email(email, "email") //邮箱
valid.MinSize(nickname, 5, "nickname") //昵称至少5位
valid.MinSize(password, 6, "password") //密码至少6位
//非法请求
if valid.HasErrors() {
this.Abort("400")
}
enumResult := user.AddUser(email, nickname, password)
fmt.Println(enumResult)
this.Data["json"] = map[string]interface{}{"code": enumResult, "msg": enumResult.String()}
this.ServeJson()
}
// 报错
func (this *Base) HasErrMsgs(valid validation.Validation) (str string, yes bool) {
if valid.HasErrors() {
yes = true
msg := []string{}
for _, err := range valid.Errors {
msg = append(msg, err.Message)
}
str = strings.Join(msg, "; ")
}
return
}
//表单验证
func ValidCompany(m *BCompany) string {
v := validation.Validation{}
v.Required(m.Name, "name").Message("单位名称不能为空")
v.Required(m.Phone, "phone").Message("电话不能为空")
v.Required(m.Address, "address").Message("地址不能为空")
if v.HasErrors() {
// 如果有错误信息，证明验证没通过
return utils.WrapErrors(v.Errors)
}
return ""
}
//表单验证
func ValidUser(m *BUser) string {
v := validation.Validation{}
v.Required(m.Username, "username").Message("邮箱不能为空")
v.Required(m.Password, "password").Message("密码不能为空")
v.Email(m.Username, "username").Message("邮箱格式不正确")
if v.HasErrors() {
// 如果有错误信息，证明验证没通过
return utils.WrapErrors(v.Errors)
}
return ""
}
func (this *MainController) Reset() {
this.activeContent("user/reset")
flash := beego.NewFlash()
u := this.Ctx.Input.Param(":uuid")
o := orm.NewOrm()
o.Using("default")
user := models.AuthUser{Reset_key: u}
err := o.Read(&user, "Reset_key")
if err == nil {
if this.Ctx.Input.Method() == "POST" {
password := this.GetString("password")
password2 := this.GetString("password2")
valid := validation.Validation{}
valid.MinSize(password, 6, "password")
valid.Required(password2, "password2")
if valid.HasErrors() {
errormap := make(map[string]string)
for _, err := range valid.Errors {
errormap[err.Key] = err.Message
}
this.Data["Errors"] = errormap
return
}
if password != password2 {
flash.Error("Passwords don't match")
flash.Store(&this.Controller)
return
}
h := pk.HashPassword(password)
// Convert password hash to string
user.Password = hex.EncodeToString(h.Hash) + hex.EncodeToString(h.Salt)
user.Reset_key = ""
if _, err := o.Update(&user); err != nil {
flash.Error("Internal error")
flash.Store(&this.Controller)
return
}
flash.Notice("Password updated.")
flash.Store(&this.Controller)
this.Redirect("/notice", 302)
}
} else {
flash.Notice("Invalid key.")
flash.Store(&this.Controller)
this.Redirect("/notice", 302)
}
}
func (c *UserController) Login() {
c.setupView("auth/login")
if c.Ctx.Input.Method() == "POST" {
flash := beego.NewFlash()
username := c.GetString("username")
password := c.GetString("password")
valid := validation.Validation{}
valid.Required(username, "username")
valid.Required(password, "password")
if valid.HasErrors() {
errormap := []string{}
for _, err := range valid.Errors {
errormap = append(errormap, "Validation failed on "+err.Key+": "+err.Message+"\n")
}
flash.Error("Invalid data!")
flash.Store(&c.Controller)
c.Data["Errors"] = errormap
return
}
o := orm.NewOrm()
o.Using("default")
user := &models.AuthUser{
Username: username,
}
err := o.Read(user, "Username")
if err == orm.ErrNoRows {
flash.Error("User not found!")
flash.Store(&c.Controller)
return
}
bcryptErr := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password))
if bcryptErr != nil {
flash.Error("Wrong password!")
flash.Store(&c.Controller)
return
}
m := make(map[string]interface{})
m["email"] = user.Email
m["username"] = user.Username
m["fullname"] = user.Firstname + " " + user.Lastname
m["timestamp"] = time.Now()
c.SetSession("user", m)
c.Redirect("/nladmin"+c.Ctx.Input.Param(":id"), 302) // go to previous page after login
}
}
func Create(post Post) (p Post, err error) {
o := orm.NewOrm()
valid := validation.Validation{}
valid.Required(post.Title, "title")
if valid.HasErrors() {
// validation does not pass
// print invalid message
return p, errors.New("hshshshs")
}
id, err := o.Insert(&post)
if err == nil {
fmt.Println(id)
}
return post, nil
}
func AddCategory(c Category) (Category, error) {
o := orm.NewOrm()
valid := validation.Validation{}
valid.Required(c.Name, "name")
if valid.HasErrors() {
return c, errors.New("Error")
}
id, err := o.Insert(&c)
if err == nil {
fmt.Println(id)
} else {
fmt.Println(err)
}
return c, nil
}
// @router /register [post]
func (c *RegisterController) Register() {
var usr models.User
c.ParseForm(&usr)
beego.Notice(usr)
valid := validation.Validation{}
usr.Valid(&valid)
if valid.HasErrors() {
c.Abort("401")
}
n := models.RegisterUser(&usr)
if n <= 0 {
c.Abort("401")
}
c.MainController.LoginSetSession(n)
c.MainController.Get()
}
// @router /register [post]
func (c *RegistController) Register() {
var usr models.User
c.ParseForm(&usr)
beego.Notice(usr)
valid := validation.Validation{}
usr.Valid(&valid)
if valid.HasErrors() {
c.Redirect("/", 302)
}
n := models.RegisterUser(&usr)
if n <= 0 {
c.Redirect("/", 302)
}
c.LoginSetSession(n)
c.Redirect("/", 302)
}
func (c *TUserController) ValidEmail() {
email := c.GetString("email")
valid := validation.Validation{}
valid.Email(email, "Email")
valid.MaxSize(email, 50, "Email")
if valid.HasErrors() {
c.Data["json"] = map[string]string{"error": "邮箱不正确"}
} else {
err := models.ValidEmail(email)
if err == nil {
c.Data["json"] = map[string]string{"error": "邮箱已经存在"}
} else {
c.Data["json"] = map[string]string{"message": "这个邮箱未被注册"}
}
}
c.ServeJson()
}
// We will check if the user data are valid
func (u User) Valid() bool {
v := validation.Validation{}
v.Required(u.Name, "name")
v.MaxSize(u.Name, 20, "nameMax")
v.Required(u.Pass, "pass")
v.MaxSize(u.Pass, 30, "passMax")
v.Email(u.Email, "email")
if v.HasErrors() {
for _, e := range v.Errors {
log.Println("Check valid user data:", e)
}
return false
}
return true
}
func (c *TUserController) ValidUserName() {
username := c.GetString("username")
valid := validation.Validation{}
re, _ := regexp.Compile(`[\p{Han},\w]+`)
valid.Match(username, re, "用户名")
valid.MinSize(username, 4, "用户名")
valid.MaxSize(username, 16, "用户名")
if valid.HasErrors() {
c.Data["json"] = map[string]string{"error": "用户名不正确"}
} else {
err := models.ValidUserName(username)
if err == nil {
c.Data["json"] = map[string]string{"error": "用户名已经存在"}
} else {
c.Data["json"] = map[string]string{"message": "你可以使用此用户名"}
}
}
c.ServeJson()
}
// @router /login [post]
func (c *FileServerController) PostLogin() {
sess, sess_err := models.GlobalSessions.SessionStart(c.Ctx.ResponseWriter, c.Ctx.Request)
var usr models.User
form_err := c.ParseForm(&usr)
if nil != sess_err || nil != form_err {
c.Abort("400")
}
valid := validation.Validation{}
usr.Check(&valid)
if valid.HasErrors() {
c.Data["content"] = "user no auth."
c.TplNames = "error.html"
return
}
sess.Set("gosessionid", "beego1234")
defer sess.SessionRelease(c.Ctx.ResponseWriter)
beego.Debug(sess)
c.Redirect("/", 302)
}
func (conf *Config) ValidateLinuxSpecs() bool {
valid := validation.Validation{}
for _, namespace := range conf.configLinux.Linux.Namespaces {
valid.Required(namespace.Type, "Namespace.Type")
}
if valid.HasErrors() {
// validation does not pass
fmt.Println("\nLinux Specific config errors\n")
for i, err := range valid.Errors {
fmt.Println(i, err.Key, err.Message)
}
return false
}
return true
}
func (this *MainController) Forgot() {
this.activeContent("user/forgot")
if this.Ctx.Input.Method() == "POST" {
email := this.GetString("email")
valid := validation.Validation{}
valid.Email(email, "email")
if valid.HasErrors() {
errormap := make(map[string]string)
for _, err := range valid.Errors {
errormap[err.Key] = err.Message
}
this.Data["Errors"] = errormap
return
}
flash := beego.NewFlash()
o := orm.NewOrm()
o.Using("default")
user := models.AuthUser{Email: email}
err := o.Read(&user, "Email")
if err != nil {
flash.Error("No such user/email in our records")
flash.Store(&this.Controller)
return
}
u := uuid.NewV4()
user.Reset_key = u.String()
_, err = o.Update(&user)
if err != nil {
flash.Error("Internal error")
flash.Store(&this.Controller)
return
}
domainname := this.Data["domainname"]
this.sendRequestReset(email, u.String(), domainname.(string))
flash.Notice("You've been sent a reset password link. You must check your email.")
flash.Store(&this.Controller)
this.Redirect("/notice", 302)
}
}
// @router /home [get]
func (this *HomeController) Get() {
// this.Ctx.WriteString("HomeController")
this.Abort("404")
valid := validation.Validation{}
u := user{20}
valid.MaxSize(u.Age, 15, "ageMax")
if valid.HasErrors() {
for _, err := range valid.Errors {
log.Println(err.Key, err.Message)
}
}
if v := valid.Max(u.Age, 12, "age"); !v.Ok {
log.Println(v.Error.Key, v.Error.Message)
}
this.TplNames = "addUser.html"
}
//登录
func (this *User) Login() {
fmt.Println(this)
reqBody := this.Ctx.Input.RequestBody
fmt.Println(string(reqBody))
reqJson, err := simplejson.NewJson(reqBody)
//非法请求
if err != nil {
this.Abort("400")
}
fmt.Println(reqJson)
email := reqJson.Get("email").MustString()
password := reqJson.Get("password").MustString()
valid := validation.Validation{}
valid.Email(email, "email")
valid.MinSize(password, 6, "password") //密码至少6位
//非法请求
if valid.HasErrors() {
this.Abort("400")
}
fmt.Println(reqJson)
if currentUser := user.FindUser(email); currentUser == nil {
this.Data["json"] = map[string]interface{}{"code": enum.UserNotExist, "msg": enum.UserNotExist.String()}
} else {
if currentUser.Password != common.Md5(password+currentUser.Salt) {
this.Data["json"] = map[string]interface{}{"code": enum.PasswordError, "msg": enum.PasswordError.String()}
} else {
this.Data["json"] = map[string]interface{}{"code": enum.OK, "msg": enum.OK.String()}
//讲token写入cookie
token := user.UpdateCookieToken(*currentUser)
this.SetSession(this.Ctx.GetCookie("token"), token)
}
}
this.ServeJson()
}
/* 文章界面 */
func (this *ArticleController) Article() {
//是否用手机访问
isMobile := IsMobile(this.Ctx.Input.UserAgent())
//验证
valid := validation.Validation{}
valid.Match(this.Ctx.Input.Param(":id"), regexp.MustCompile("^[a-z0-9]+$"), "category")
if valid.HasErrors() { //没有通过验证则返回首页
this.Ctx.Redirect(302, this.UrlFor("IndexController.Index"))
this.StopRun()
}
//寻找文章
article := &models.Article{}
ok := article.FindOne(this.Ctx.Input.Param(":id"))
if !ok { //文章不存在
this.Ctx.Redirect(302, this.UrlFor("IndexController.Index"))
this.StopRun()
}
this.Data["article"] = article
//根据文章存储的分类id查询分类信息
category := &models.Category{}
category.FindOne(article.Category)
this.Data["category"] = category
//根据分类信息查询父分类信息以及自己分类的文章（必定存在父级分类）
parent, articles, hots := category.FindParentAndArticles()
this.Data["parent"] = parent
this.Data["articles"] = articles
this.Data["hots"] = hots
if isMobile { //如果是手机访问
this.TplNames = "mobile/article/article.html"
} else {
this.TplNames = "article/article.html"
}
this.Render()
}
/**
* 注册逻辑处理
* @param {[type]} this *UserController) Join( [description]
* @return {[type]} [description]
*/
func (this *UserController) Join() {
beego.AutoRender = false
errmsg := make(map[string]string)
requestEmail := this.GetString("email")
requestNickname := this.GetString("nickname")
requestPassword := this.GetString("password")
jvalid := validation.Validation{}
jvalid.Required(requestEmail, "email")
jvalid.MaxSize(requestEmail, 40, "email")
jvalid.Required(requestNickname, "nickname")
jvalid.MaxSize(requestNickname, 20, "nickname")
jvalid.Required(requestPassword, "password")
fmt.Println(jvalid.HasErrors())
if jvalid.HasErrors() {
for _, err := range jvalid.Errors {
//fmt.Println(err.Key, err.Message)
this.showmsg("Error Message", "["+err.Key+"]"+err.Message)
}
}
if len(errmsg) == 0 {
var user models.User
user.Email = requestEmail
user.Nickname = requestNickname
user.Password = models.Md5([]byte(requestPassword))
user.Lastlogin = time.Now().Unix()
if err := user.Insert(); err != nil {
this.Ctx.WriteString(err.Error())
}
}
this.SetSession("nickname", requestNickname)
this.SetSession("email", requestEmail)
this.SetSession("isadmin", 0)
this.Redirect("/", 302)
}
func DecodeAndValidate(w http.ResponseWriter, r *http.Request, obj interface{}) (err error) {
decoder := json.NewDecoder(r.Body)
if err = decoder.Decode(obj); err != nil {
log.Println("Error in decoding request body. Error is ", err)
ErrorResponse(w, errors.New("Invalid json details in request body"), http.StatusBadRequest)
return
}
log.Println("decoded object type and value:", fmt.Sprintf("%T", obj), obj)
valid := validation.Validation{}
var b bool
if b, err = valid.RecursiveValid(obj); err != nil || !b {
var buffer bytes.Buffer
if valid.HasErrors() {
for _, validationErr := range valid.Errors {
buffer.WriteString(validationErr.Field + " " + validationErr.Message + ".")
}
}
err = errors.New(buffer.String())
ErrorResponse(w, err, http.StatusBadRequest)
}
return
}
