码农笔录博客源码
Ви не можете вибрати більше 25 тем Теми мають розпочинатися з літери або цифри, можуть містити дефіси (-) і не повинні перевищувати 35 символів.

309 рядки
7.6 KiB

6 роки тому
package controllers
6 роки тому
import (
"github.com/astaxie/beego"
"beeblog/service"
"beeblog/models"
"strings"
"crypto/md5"
"time"
"strconv"
"encoding/hex"
)
6 роки тому
type UserController struct {
beego.Controller
}
func (u *UserController) LoginPage() {
6 роки тому
u.TplName = "login.html"
6 роки тому
}
6 роки тому
func (u *UserController) RegistPage() {
u.TplName = "regist.html"
}
6 роки тому
6 роки тому
func (this *UserController) UserInfo() {
idStr := this.Ctx.Input.Param(":id")
id, _ := strconv.ParseInt(idStr, 10, 64)
user, err := service.GetUser(id)
if err != nil {
this.Redirect("/500", 302)
return
}
size := 15
num, _ := this.GetInt("num")
if num <= 0 {
num = 1
}
flag, _ := this.GetInt("flag")
if page, err := service.MeBlogs(num, size, flag, id); err == nil {
this.Data["Page"] = page
}
this.Data["User"] = user
6 роки тому
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
6 роки тому
this.Data["IsDDesc"] = true
6 роки тому
this.TplName = "user.html"
return
}
6 роки тому
func (this *UserController) PersonBlog() {
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 := service.MeBlogs(num, size, flag, uid.(int64))
if err != nil {
6 роки тому
this.Redirect("/500", 302)
return
6 роки тому
}
user, uerr := service.GetUser(uid.(int64))
if uerr != nil {
6 роки тому
this.Redirect("/500", 302)
return
6 роки тому
}
6 роки тому
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
6 роки тому
this.Data["Page"] = page
6 роки тому
this.Data["IsMeBlog"] = true
this.Data["Flag"] = 0
this.Data["User"] = user
this.TplName = "ublogs.html"
}
func (this *UserController) PersonNote() {
uid := this.GetSession("userid")
if uid == nil {
this.Redirect("/login", 302)
return
}
notColl, err := service.GetNoteColl(uid.(int64))
if err == nil {
if len(notColl) > 0 {
for i := 0; i < len(notColl); i++ {
count, _ := service.CountNote(notColl[i].Id)
notColl[i].Count = count
}
}
} else {
notColl = make([]*models.NoteColl, 0)
}
if err != nil {
6 роки тому
this.Redirect("/500", 302)
return
6 роки тому
}
user, uerr := service.GetUser(uid.(int64))
if uerr != nil {
if uid == nil {
this.Redirect("/500", 302)
return
}
}
6 роки тому
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
6 роки тому
this.Data["Note"] = notColl
this.Data["IsMeNote"] = true
this.Data["User"] = user
this.TplName = "unote.html"
}
func (this *UserController) PersonLike() {
uid := this.GetSession("userid")
if uid == nil {
this.Redirect("/login", 302)
return
}
size := 15
num, _ := this.GetInt("num")
if num <= 0 {
num = 1
}
6 роки тому
page, err := service.MeLikes(num, size, uid.(int64))
6 роки тому
if err != nil {
6 роки тому
this.Redirect("/500", 302)
return
6 роки тому
}
user, uerr := service.GetUser(uid.(int64))
if uerr != nil {
if uid == nil {
this.Redirect("/500", 302)
return
}
}
6 роки тому
this.Data["UserId"] = this.GetSession("userid")
this.Data["HeadImg"] = this.GetSession("headimg")
6 роки тому
this.Data["NickName"] = this.GetSession("nickname")
this.Data["IsLogin"] = this.GetSession("nickname") != nil
6 роки тому
this.Data["Page"] = page
6 роки тому
this.Data["IsMeLike"] = true
6 роки тому
this.Data["User"] = user
6 роки тому
this.TplName = "ulike.html"
}
func (this *UserController) PersonInfo() {
uid := this.GetSession("userid")
if uid == nil {
this.Redirect("/login", 302)
return
}
user, err := service.GetUser(uid.(int64))
if err != nil {
this.Redirect("/500", 302)
return
}
this.Data["IsLogin"] = this.GetSession("nickname") != nil
6 роки тому
this.Data["UserId"] = this.GetSession("userid")
this.Data["HeadImg"] = this.GetSession("headimg")
this.Data["NickName"] = this.GetSession("nickname")
6 роки тому
this.Data["IsMeInfo"] = true
this.Data["User"] = user
this.TplName = "uinfo.html"
}
func (this *UserController) Edit() {
uid := this.GetSession("userid")
if uid == nil {
models.ReurnError(401, "")
this.ServeJSON()
return
}
user, err := service.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")
6 роки тому
if _, err := service.EditUser(user); err != nil {
6 роки тому
this.Data["json"] = models.ReurnError(500, "")
6 роки тому
} else {
6 роки тому
this.Data["json"] = models.ReurnSuccess("")
}
this.ServeJSON()
return
6 роки тому
}
6 роки тому
func (this *UserController) Login() {
username := this.GetString("username")
userpwd := this.GetString("userpwd")
if username == "" {
6 роки тому
this.Data["json"] = models.ReurnError(1, "用户名为空")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
if len(username) < 4 {
6 роки тому
this.Data["json"] = models.ReurnError(1, "用户名最低4位")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
if userpwd == "" {
6 роки тому
this.Data["json"] = models.ReurnError(1, "密码为空")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
if len(userpwd) < 6 {
6 роки тому
this.Data["json"] = models.ReurnError(1, "密码最低6位")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
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("")
6 роки тому
this.SetSession("userid", user.Id)
6 роки тому
this.SetSession("nickname", user.NickName)
6 роки тому
this.SetSession("headimg", user.Headimg)
6 роки тому
} else {
6 роки тому
this.Data["json"] = models.ReurnError(1, "用户名或密码错误")
6 роки тому
}
} else {
6 роки тому
this.Data["json"] = models.ReurnError(1, "用户名不存在")
6 роки тому
}
this.ServeJSON()
6 роки тому
return
6 роки тому
}
func (this *UserController) Regist() {
username := this.GetString("username")
userpwd := this.GetString("userpwd")
username = strings.Replace(username, " ", "", -1)
userpwd = strings.Replace(userpwd, " ", "", -1)
if username == "" {
6 роки тому
this.Data["json"] = models.ReurnError(1, "用户名为空")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
if len(username) < 4 {
6 роки тому
this.Data["json"] = models.ReurnError(1, "用户名最低4位")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
if userpwd == "" {
6 роки тому
this.Data["json"] = models.ReurnError(1, "密码为空")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
if len(userpwd) < 6 {
6 роки тому
this.Data["json"] = models.ReurnError(1, "密码最低6位")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
user, _ := service.FindByUserName(username)
if user != nil {
6 роки тому
this.Data["json"] = models.ReurnError(1, "用户已经存在")
6 роки тому
this.ServeJSON()
6 роки тому
return
6 роки тому
}
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))
6 роки тому
user = &models.User{UserName: username, NickName: username, UserPwd: userpwd, Salt: salt}
6 роки тому
user.Headimg = "https://www.aiprose.com/foss/timg.jpg"
6 роки тому
err := service.SaveUser(user)
if err == nil {
6 роки тому
this.Data["json"] = models.ReurnSuccess("")
6 роки тому
} else {
6 роки тому
this.Data["json"] = models.ReurnError(1, "注册失败")
6 роки тому
}
this.ServeJSON()
6 роки тому
return
6 роки тому
}
6 роки тому
func (this *UserController) Logout() {
this.DelSession("userid")
this.DelSession("nickname")
6 роки тому
this.Redirect("/", 302)
6 роки тому
return
6 роки тому
}