feat:实现自定义模板功能

controllers/TemplateController.go

@@ -0,0 +1,125 @@
+package controllers
+
+import (
+	"github.com/lifei6671/mindoc/models"
+	"github.com/astaxie/beego/orm"
+	"github.com/qiniu/x/errors.v7"
+	"strings"
+)
+
+type TemplateController struct {
+	BaseController
+	BookId int
+}
+
+func (c *TemplateController) isPermission() (error) {
+	c.Prepare()
+
+	if c.IsAjax() {
+		bookIdentify := c.GetString("identify", "")
+
+		if bookIdentify == "" {
+			return  errors.New("参数错误")
+		}
+
+		if !c.Member.IsAdministrator() {
+			book, err := models.NewBookResult().FindByIdentify(bookIdentify, c.Member.MemberId,"book_id")
+			if err != nil {
+				if err == orm.ErrNoRows {
+					return errors.New("项目不存在或没有权限")
+				}
+				return errors.New("查询项目模板失败")
+			}
+			c.BookId = book.BookId
+		}else{
+			book,err := models.NewBook().FindByIdentify(bookIdentify,"book_id")
+			if err != nil {
+				if err == orm.ErrNoRows {
+					return errors.New("项目不存在或没有权限")
+				}
+				return errors.New("查询项目模板失败")
+			}
+			c.BookId = book.BookId
+		}
+		return nil
+	}
+	return errors.New("请求方法不支持")
+}
+
+//获取模板列表
+func (c *TemplateController) List() {
+	if err := c.isPermission() ; err != nil {
+		c.JsonResult(500,err.Error())
+	}
+
+	templateList,err := models.NewTemplate().FindAllByBookId(c.BookId)
+
+	if err != nil {
+		if err == orm.ErrNoRows {
+			c.JsonResult(404,"没有模板")
+		}
+		c.JsonResult(500,"查询项目模板失败")
+	}
+	c.JsonResult(0,"OK",templateList)
+}
+
+func (c *TemplateController) Add() {
+	if err := c.isPermission() ; err != nil {
+		c.JsonResult(500,err.Error())
+	}
+
+	templateId, _ := c.GetInt("template_id",0)
+	content := c.GetString("content")
+	isGlobal,_ := c.GetInt("is_global",0)
+	templateName := c.GetString("template_name","")
+
+	if templateName == "" || strings.Count(templateName,"") > 300 {
+		c.JsonResult(500,"模板名称不能为空且必须小于300字")
+	}
+	template := models.NewTemplate()
+	template.TemplateId = templateId
+	template.BookId = c.BookId
+	template.TemplateContent = content
+	template.MemberId = c.Member.MemberId
+
+	if templateId > 0 {
+		template.ModifyAt = c.Member.MemberId
+	}
+	//只有管理员才能设置全局模板
+	if c.Member.IsAdministrator() {
+		template.IsGlobal = isGlobal
+	}else{
+		template.IsGlobal = 0
+	}
+
+	var cols []string
+
+	if templateId > 0 {
+		cols = []string{ "template_content", "modify_time","modify_at","version" }
+	}
+
+	if err := template.Save(cols...); err != nil {
+		c.JsonResult(500,"报错模板失败")
+	}
+	c.JsonResult(0,"OK",template)
+}
+
+func (c *TemplateController) Delete() {
+	if err := c.isPermission() ; err != nil {
+		c.JsonResult(500,err.Error())
+	}
+	templateId, _ := c.GetInt("template_id",0)
+
+	if c.Member.IsAdministrator() {
+		err := models.NewTemplate().Delete(templateId,0)
+		if err != nil {
+			c.JsonResult(500,"删除失败")
+		}
+	}else{
+		err := models.NewTemplate().Delete(templateId,c.Member.MemberId)
+		if err != nil {
+			c.JsonResult(500,"删除失败")
+		}
+	}
+	c.JsonResult(0,"OK")
+}