beeblog/controllers/UserController.go

package controllers

import (
	"beeblog/models"
	"beeblog/service"
	"beeblog/utils"
	"crypto/md5"
	"encoding/hex"
	"fmt"
	beego "github.com/beego/beego/v2/server/web"
	"github.com/mojocn/base64Captcha"
	"strconv"
	"strings"
	"time"
)

type UserController struct {
	beego.Controller
}

func (u *UserController) LoginPage() {
	u.TplName = "login.html"
}
func (u *UserController) RegistPage() {
	u.TplName = "regist.html"
}

func (this *UserController) UserInfo() {
	userService := service.UserService{}
	blogService := service.BlogService{}
	idStr := this.Ctx.Input.Param(":id")
	id, _ := strconv.ParseInt(idStr, 10, 64)
	user, err := userService.GetUser(id)
	if err != nil {
		this.Redirect("/404", 302)
		return
	}
	size := 15
	num, _ := this.GetInt("num")
	if num <= 0 {
		num = 1
	}
	flag, _ := this.GetInt("flag")
	if page, err := blogService.MeBlogs(num, size, flag, id); err == nil {
		this.Data["Page"] = page
	}
	this.Data["User"] = user
	this.Data["UserId"] = this.GetSession("userid")
	this.Data["HeadImg"] = this.GetSession("headimg")
	this.Data["NickName"] = this.GetSession("nickname")
	this.Data["IsLogin"] = this.GetSession("nickname") != nil
	this.Data["IsDDesc"] = true
	this.TplName = "user.html"
	return
}

func (this *UserController) PersonBlog() {
	userService := service.UserService{}
	blogService := service.BlogService{}
	uid := this.GetSession("userid")
	if uid == nil {
		this.Redirect("/login", 302)
		return
	}
	size := 10
	num, _ := this.GetInt("num")
	if num <= 0 {
		num = 1
	}
	flag, _ := this.GetInt("flag")
	page, err := blogService.MeBlogs(num, size, flag, uid.(int64))
	if err != nil {
		this.Redirect("/404", 302)
		return
	}
	user, uerr := userService.GetUser(uid.(int64))
	if uerr != nil {
		this.Redirect("/404", 302)
		return
	}
	this.Data["UserId"] = this.GetSession("userid")
	this.Data["HeadImg"] = this.GetSession("headimg")
	this.Data["NickName"] = this.GetSession("nickname")
	this.Data["IsLogin"] = this.GetSession("nickname") != nil
	this.Data["Page"] = page
	this.Data["IsMeBlog"] = true
	this.Data["Flag"] = 0
	this.Data["User"] = user
	this.TplName = "ublogs.html"
}
func (this *UserController) PersonInfo() {
	userService := service.UserService{}
	uid := this.GetSession("userid")
	if uid == nil {
		this.Redirect("/login", 302)
		return
	}
	user, err := userService.GetUser(uid.(int64))
	if err != nil {
		this.Redirect("/404", 302)
		return
	}
	this.Data["IsLogin"] = this.GetSession("nickname") != nil
	this.Data["UserId"] = this.GetSession("userid")
	this.Data["HeadImg"] = this.GetSession("headimg")
	this.Data["NickName"] = this.GetSession("nickname")
	this.Data["IsMeInfo"] = true
	this.Data["User"] = user
	this.TplName = "uinfo.html"
}

func (this *UserController) Edit() {
	userService := service.UserService{}
	uid := this.GetSession("userid")
	if uid == nil {
		models.ReurnError(401, "")
		this.ServeJSON()
		return
	}
	user, err := userService.GetUser(uid.(int64))
	if err != nil {
		this.Data["json"] = models.ReurnError(500, "")
		this.ServeJSON()
		return
	}
	user.NickName = this.GetString("nickName")
	user.Email = this.GetString("email")
	user.Mobile = this.GetString("mobile")
	if _, err := userService.EditUser(user); err != nil {
		this.Data["json"] = models.ReurnError(500, "")
	} else {
		this.Data["json"] = models.ReurnSuccess("")
	}
	this.ServeJSON()
	return
}

func (this *UserController) Login() {
	userService := service.UserService{}
	verifycode := this.GetString("captcha")
	if len(verifycode) == 0 {
		this.Data["json"] = models.ReurnError(1, "验证码错误")
		this.ServeJSON()
		return
	}
	captchaid := this.GetString("captchaId")
	if len(captchaid) == 0 {
		this.Data["json"] = models.ReurnError(1, "验证码错误")
		this.ServeJSON()
		return
	}
	has := store.Has(captchaid)
	if has == 0 {
		this.Data["json"] = models.ReurnError(1, "验证码过期")
		this.ServeJSON()
		return
	}
	verify := store.Verify(captchaid, verifycode, true)
	if !verify {
		this.Data["json"] = models.ReurnError(1, "验证码错误")
		this.ServeJSON()
		return
	}
	username := this.GetString("username")
	userpwd := this.GetString("userpwd")
	if username == "" {
		this.Data["json"] = models.ReurnError(1, "用户名为空")
		this.ServeJSON()
		return
	}
	if len(username) < 4 {
		this.Data["json"] = models.ReurnError(1, "用户名最低4位")
		this.ServeJSON()
		return
	}
	if userpwd == "" {
		this.Data["json"] = models.ReurnError(1, "密码为空")
		this.ServeJSON()
		return
	}
	if len(userpwd) < 6 {
		this.Data["json"] = models.ReurnError(1, "密码最低6位")
		this.ServeJSON()
		return
	}
	user, error := userService.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("")
			this.SetSession("userid", user.Id)
			this.SetSession("nickname", user.NickName)
			this.SetSession("headimg", user.Headimg)
		} else {
			this.Data["json"] = models.ReurnError(1, "用户名或密码错误")
		}
	} else {
		this.Data["json"] = models.ReurnError(1, "用户名不存在")
	}
	this.ServeJSON()
	return
}

func (this *UserController) Regist() {
	userService := service.UserService{}
	username := this.GetString("username")
	userpwd := this.GetString("userpwd")
	verifycode := this.GetString("captcha")
	if len(verifycode) == 0 {
		this.Data["json"] = models.ReurnError(1, "验证码错误")
		this.ServeJSON()
		return
	}
	captchaid := this.GetString("captchaId")
	if len(captchaid) == 0 {
		this.Data["json"] = models.ReurnError(1, "验证码错误")
		this.ServeJSON()
		return
	}
	has := store.Has(captchaid)
	if has == 0 {
		this.Data["json"] = models.ReurnError(1, "验证码过期")
		this.ServeJSON()
		return
	}
	verify := store.Verify(captchaid, verifycode, true)
	if !verify {
		this.Data["json"] = models.ReurnError(1, "验证码错误")
		this.ServeJSON()
		return
	}
	username = strings.Replace(username, " ", "", -1)
	userpwd = strings.Replace(userpwd, " ", "", -1)
	if username == "" {
		this.Data["json"] = models.ReurnError(1, "用户名为空")
		this.ServeJSON()
		return
	}
	if len(username) < 4 {
		this.Data["json"] = models.ReurnError(1, "用户名最低4位")
		this.ServeJSON()
		return
	}
	if userpwd == "" {
		this.Data["json"] = models.ReurnError(1, "密码为空")
		this.ServeJSON()
		return
	}
	if len(userpwd) < 6 {
		this.Data["json"] = models.ReurnError(1, "密码最低6位")
		this.ServeJSON()
		return
	}
	user, _ := userService.FindByUserName(username)
	if user != nil {
		this.Data["json"] = models.ReurnError(1, "用户已经存在")
		this.ServeJSON()
		return
	}
	h := md5.New()
	h.Write([]byte(strconv.FormatInt(time.Now().Unix(), 10) + "aiprose:nelson"))
	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, NickName: username, UserPwd: userpwd, Salt: salt}
	user.Headimg = "https://oss.aiprose.com/aiprose/timg.jpg"
	err := userService.SaveUser(user)
	if err == nil {
		this.Data["json"] = models.ReurnSuccess("")
	} else {
		this.Data["json"] = models.ReurnError(1, "注册失败")
	}
	this.ServeJSON()
	return
}

func (this *UserController) Logout() {
	this.DelSession("userid")
	this.DelSession("nickname")
	this.Redirect("/", 302)
	return
}

var store = utils.RedisStore{}

func (u *UserController) Captcha() {
	type CaptchaResult struct {
		Id         string `json:"captchaId"`
		Base64Blob string `json:"img"`
	}
	driver := base64Captcha.NewDriverMath(40, 150, 0, 0, nil, nil, nil)
	//driver := base64Captcha.DefaultDriverDigit
	c := base64Captcha.NewCaptcha(driver, store)
	id, b64s, err := c.Generate()
	if err != nil {
		fmt.Println(err)
		u.Data["json"] = models.ReurnError(500, "验证码获取异常")
		u.ServeJSON()
		return
	}
	detail := make(map[string]string, 2)
	detail["captchaId"] = id
	detail["img"] = b64s
	u.Data["json"] = models.ReurnData("ok", detail)
	u.ServeJSON()
	return
}