mindoc/controllers/manager.go

package controllers

import (

	"encoding/json"
	"html/template"
	"strings"
	"regexp"

	"github.com/lifei6671/godoc/conf"
	"github.com/astaxie/beego/logs"
	"github.com/lifei6671/godoc/utils"
	"github.com/lifei6671/godoc/models"
	"github.com/astaxie/beego/orm"
)

type ManagerController struct {
	BaseController
}

func (p *ManagerController) Index() {
	p.TplName = "manager/index.tpl"
}

// 用户列表.
func (c *ManagerController) Users()  {
	c.Prepare()
	c.TplName = "manager/users.tpl"

	if !c.Member.IsAdministrator() {
		c.Abort("403")
	}

	pageIndex,_ := c.GetInt("page",0)

	members,totalCount,err := models.NewMember().FindToPager(pageIndex,15)

	if err != nil {
		c.Data["ErrorMessage"] = err.Error()
		return
	}

	if totalCount > 0 {
		html := utils.GetPagerHtml(c.Ctx.Request.RequestURI, pageIndex, 10, int(totalCount))

		c.Data["PageHtml"] = html
	}else{
		c.Data["PageHtml"] = ""
	}

	b,err := json.Marshal(members)

	if err != nil {
		c.Data["Result"] = template.JS("[]")
	}else{
		c.Data["Result"] = template.JS(string(b))
	}
}

// 添加用户
func (c *ManagerController) CreateMember() {
	c.Prepare()
	if !c.Member.IsAdministrator(){
		c.Abort("403")
	}

	account := strings.TrimSpace(c.GetString("account"))
	password1 := strings.TrimSpace(c.GetString("password1"))
	password2 := strings.TrimSpace(c.GetString("password2"))
	email := strings.TrimSpace(c.GetString("email"))
	phone := strings.TrimSpace(c.GetString("phone"))
	role,_ := c.GetInt("role",1)
	status,_ := c.GetInt("status",0)

	if ok,err := regexp.MatchString(conf.RegexpAccount,account); account == "" || !ok || err != nil {
		c.JsonResult(6001,"账号只能由英文字母数字组成，且在3-50个字符")
	}
	if  l := strings.Count(password1,"") ; password1 == "" || l > 50 || l < 6{
		c.JsonResult(6002,"密码必须在6-50个字符之间")
	}
	if password1 != password2 {
		c.JsonResult(6003,"确认密码不正确")
	}
	if  ok,err := regexp.MatchString(conf.RegexpEmail,email); !ok || err != nil || email == "" {
		c.JsonResult(6004,"邮箱格式不正确")
	}
	if role != 0 && role != 1 {
		role = 1
	}
	if status != 0 && status != 1 {
		status = 0
	}

	member := models.NewMember()

	if err := member.FindByAccount(account); err == nil && member.MemberId > 0 {
		c.JsonResult(6005,"账号已存在")
	}

	member.Account = account
	member.Password = password1
	member.Role = role
	member.Avatar = conf.GetDefaultAvatar()
	member.CreateAt = c.Member.MemberId
	member.Email = email
	if phone != "" {
		member.Phone = phone
	}

	if err := member.Add(); err != nil {
		c.JsonResult(6006,err.Error())
	}

	c.JsonResult(0,"ok",member)
}

//更新用户状态.
func (c *ManagerController) UpdateMemberStatus()  {
	c.Prepare()

	if !c.Member.IsAdministrator() {
		c.Abort("403")
	}

	member_id,_ := c.GetInt("member_id",0)
	status ,_ := c.GetInt("status",0)

	if member_id <= 0 {
		c.JsonResult(6001,"参数错误")
	}
	if status != 0 && status != 1 {
		status = 0
	}
	member := models.NewMember()

	if err := member.Find(member_id); err != nil {
		c.JsonResult(6002,"用户不存在")
	}
	member.Status = status

	if err := member.Update();err != nil {
		logs.Error("",err)
		c.JsonResult(6003,"用户状态设置失败")
	}
	c.JsonResult(0,"ok",member)
}

func (c *ManagerController) Books()  {
	c.Prepare()
	c.TplName = "manager/books.tpl"

	pageIndex, _ := c.GetInt("page", 1)

	books,totalCount,err := models.NewBookResult().FindToPager(pageIndex,conf.PageSize)

	if err != nil {
		c.Abort("500")
	}

	if totalCount > 0 {
		html := utils.GetPagerHtml(c.Ctx.Request.RequestURI, pageIndex, conf.PageSize, totalCount)

		c.Data["PageHtml"] = html
	}else {
		c.Data["PageHtml"] = ""
	}

	c.Data["Lists"] = books
}

func (c *ManagerController) EditBook()  {
	c.TplName = "manager/edit_book.tpl"
	identify := c.GetString(":key")

	if identify == "" {
		c.Abort("404")
	}
	book,err := models.NewBook().FindByFieldFirst("identify",identify)
	if err != nil {
		c.Abort("500")
	}
	c.Data["Model"] = book
}

// 删除项目.
func (c *ManagerController) DeleteBook()  {
	c.Prepare()
	if !c.Member.IsAdministrator() {
		c.Abort("403")
	}

	book_id,_ := c.GetInt("book_id",0)

	if book_id <= 0{
		c.JsonResult(6001,"参数错误")
	}
	book := models.NewBook()

	err := book.ThoroughDeleteBook(book_id)

	if err == orm.ErrNoRows {
		c.JsonResult(6002,"项目不存在")
	}
	if err != nil {
		logs.Error("",err)
		c.JsonResult(6003,"删除失败")
	}
	c.JsonResult(0,"ok")
}

func (c *ManagerController) Comments()  {
	c.Prepare()
	if !c.Member.IsAdministrator() {
		c.Abort("403")
	}
}

//DeleteComment 标记评论为已删除
func (c *ManagerController) DeleteComment()  {
	c.Prepare()
	if !c.Member.IsAdministrator() {
		c.Abort("403")
	}
	comment_id,_ := c.GetInt("comment_id",0)

	if comment_id <= 0 {
		c.JsonResult(6001,"参数错误")
	}

	comment := models.NewComment()

	if err := comment.Find(comment_id); err != nil {
		c.JsonResult(6002,"评论不存在")
	}

	comment.Approved = 3

	if err := comment.Update("approved");err != nil {
		c.JsonResult(6003,"删除评论失败")
	}
	c.JsonResult(0,"ok",comment)
}