mindoc/models/Template.go

package models

import (
	"errors"
	"time"

	"github.com/beego/beego/v2/adapter/logs"
	"github.com/beego/beego/v2/client/orm"
	"github.com/mindoc-org/mindoc/conf"
)

type Template struct {
	TemplateId   int    `orm:"column(template_id);pk;auto;unique;" json:"template_id"`
	TemplateName string `orm:"column(template_name);size(500);" json:"template_name"`
	MemberId     int    `orm:"column(member_id);index" json:"member_id"`
	BookId       int    `orm:"column(book_id);index" json:"book_id"`
	BookName     string `orm:"-" json:"book_name"`
	//是否是全局模板：0 否/1 是; 全局模板在所有项目中都可以使用；否则只能在创建模板的项目中使用
	IsGlobal        int       `orm:"column(is_global);default(0)" json:"is_global"`
	TemplateContent string    `orm:"column(template_content);type(text);null" json:"template_content"`
	CreateTime      time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
	CreateName      string    `orm:"-" json:"create_name"`
	ModifyTime      time.Time `orm:"column(modify_time);type(datetime);auto_now" json:"modify_time"`
	ModifyAt        int       `orm:"column(modify_at);type(int)" json:"-"`
	ModifyName      string    `orm:"-" json:"modify_name"`
	Version         int64     `orm:"type(bigint);column(version)" json:"version"`
}

// TableName 获取对应数据库表名.
func (m *Template) TableName() string {
	return "templates"
}

// TableEngine 获取数据使用的引擎.
func (m *Template) TableEngine() string {
	return "INNODB"
}

func (m *Template) TableNameWithPrefix() string {
	return conf.GetDatabasePrefix() + m.TableName()
}

func NewTemplate() *Template {
	return &Template{}
}

//查询指定ID的模板
func (t *Template) Find(templateId int) (*Template, error) {
	if templateId <= 0 {
		return t, ErrInvalidParameter
	}

	o := orm.NewOrm()

	err := o.QueryTable(t.TableNameWithPrefix()).Filter("template_id", templateId).One(t)

	if err != nil {
		logs.Error("查询模板时失败 ->%s", err)
	}
	return t, err
}

//查询属于指定项目的模板.
func (t *Template) FindByBookId(bookId int) ([]*Template, error) {
	if bookId <= 0 {
		return nil, ErrInvalidParameter
	}
	o := orm.NewOrm()

	var templateList []*Template

	_, err := o.QueryTable(t.TableNameWithPrefix()).Filter("book_id", bookId).OrderBy("-template_id").All(&templateList)

	if err != nil {
		logs.Error("查询模板列表失败 ->", err)
	}
	return templateList, err
}

//查询指定项目所有可用模板列表.
func (t *Template) FindAllByBookId(bookId int) ([]*Template, error) {
	if bookId <= 0 {
		return nil, ErrInvalidParameter
	}
	o := orm.NewOrm()

	cond := orm.NewCondition()

	cond1 := cond.And("book_id", bookId).Or("is_global", 1)

	qs := o.QueryTable(t.TableNameWithPrefix())

	var templateList []*Template

	_, err := qs.SetCond(cond1).OrderBy("-template_id").All(&templateList)

	if err != nil {
		logs.Error("查询模板列表失败 ->", err)
	}
	return templateList, err
}

//删除一个模板
func (t *Template) Delete(templateId int, memberId int) error {
	if templateId <= 0 {
		return ErrInvalidParameter
	}

	o := orm.NewOrm()

	qs := o.QueryTable(t.TableNameWithPrefix()).Filter("template_id", templateId)

	if memberId > 0 {
		qs = qs.Filter("member_id", memberId)
	}
	_, err := qs.Delete()

	if err != nil {
		logs.Error("删除模板失败 ->", err)
	}
	return err
}

//添加或更新模板
func (t *Template) Save(cols ...string) (err error) {

	if t.BookId <= 0 {
		return ErrInvalidParameter
	}
	o := orm.NewOrm()

	if !o.QueryTable(NewBook().TableNameWithPrefix()).Filter("book_id", t.BookId).Exist() {
		return errors.New("项目不存在")
	}
	if !o.QueryTable(NewMember().TableNameWithPrefix()).Filter("member_id", t.MemberId).Filter("status", 0).Exist() {
		return errors.New("用户已被禁用")
	}
	t.Version = time.Now().Unix()

	if t.TemplateId > 0 {
		t.ModifyTime = time.Now()
		_, err = o.Update(t, cols...)
	} else {
		t.CreateTime = time.Now()
		_, err = o.Insert(t)
	}

	return
}

//预加载一些数据
func (t *Template) Preload() *Template {
	if t != nil {
		if t.MemberId > 0 {
			m, err := NewMember().Find(t.MemberId, "account", "real_name")
			if err == nil {
				if m.RealName != "" {
					t.CreateName = m.RealName
				} else {
					t.CreateName = m.Account
				}
			} else {
				logs.Error("加载模板所有者失败 ->", err)
			}
		}
		if t.ModifyAt > 0 {
			if m, err := NewMember().Find(t.ModifyAt, "account", "real_name"); err == nil {
				if m.RealName != "" {
					t.ModifyName = m.RealName
				} else {
					t.ModifyName = m.Account
				}
			}
		}
		if t.BookId > 0 {
			if b, err := NewBook().Find(t.BookId, "book_name"); err == nil {
				t.BookName = b.BookName
			}
		}
	}
	return t
}