選択できるのは25トピックまでです。
トピックは、先頭が英数字で、英数字とダッシュ('-')を使用した35文字以内のものにしてください。
320 行
8.2 KiB
320 行
8.2 KiB
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 := 15
|
|
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
|
|
}
|
|
birthday := this.GetString("birthday")
|
|
if birthday != "" {
|
|
birthday += " 00:00:00"
|
|
if localTime, errt := time.ParseInLocation("2006-01-02 15:04:05", birthday, time.Local); errt == nil {
|
|
user.Birthday = localTime
|
|
}
|
|
}
|
|
user.NickName = this.GetString("nickName")
|
|
user.Email = this.GetString("email")
|
|
user.Mobile = this.GetString("mobile")
|
|
user.QQ = this.GetString("qqnum")
|
|
user.Sex, _ = this.GetInt("catory")
|
|
user.DescInfo = this.GetString("mdesc")
|
|
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
|
|
}
|
|
|