mindoc/models/Blog.go

package models

import (
	"bytes"
	"fmt"
	"strings"
	"time"

	"github.com/PuerkitoBio/goquery"
	"github.com/beego/beego/v2/core/logs"
	"github.com/beego/beego/v2/adapter"
	"github.com/beego/beego/v2/adapter/orm"
	"github.com/mindoc-org/mindoc/cache"
	"github.com/mindoc-org/mindoc/conf"
	"github.com/mindoc-org/mindoc/utils"
)

//博文表
type Blog struct {
	BlogId int `orm:"pk;auto;unique;column(blog_id)" json:"blog_id"`
	//文章标题
	BlogTitle string `orm:"column(blog_title);size(500)" json:"blog_title"`
	//文章标识
	BlogIdentify string `orm:"column(blog_identify);size(100);unique" json:"blog_identify"`
	//排序序号
	OrderIndex int `orm:"column(order_index);type(int);default(0)" json:"order_index"`
	//所属用户
	MemberId int `orm:"column(member_id);type(int);default(0);index" json:"member_id"`
	//用户头像
	MemberAvatar string `orm:"-" json:"member_avatar"`
	//文章类型:0 普通文章/1 链接文章
	BlogType int `orm:"column(blog_type);type(int);default(0)" json:"blog_type"`
	//链接到的项目中的文档ID
	DocumentId int `orm:"column(document_id);type(int);default(0)" json:"document_id"`
	//文章的标识
	DocumentIdentify string `orm:"-" json:"document_identify"`
	//关联文档的项目标识
	BookIdentify string `orm:"-" json:"book_identify"`
	//关联文档的项目ID
	BookId int `orm:"-" json:"book_id"`
	//文章摘要
	BlogExcerpt string `orm:"column(blog_excerpt);size(1500)" json:"blog_excerpt"`
	//文章内容
	BlogContent string `orm:"column(blog_content);type(text);null" json:"blog_content"`
	//发布后的文章内容
	BlogRelease string `orm:"column(blog_release);type(text);null" json:"blog_release"`
	//文章当前的状态，枚举enum(’publish’,’draft’,’password’)值，publish为已 发表，draft为草稿，password 为私人内容(不会被公开) 。默认为publish。
	BlogStatus string `orm:"column(blog_status);size(100);default(publish)" json:"blog_status"`
	//文章密码，varchar(100)值。文章编辑才可为文章设定一个密码，凭这个密码才能对文章进行重新强加或修改。
	Password string `orm:"column(password);size(100)" json:"-"`
	//最后修改时间
	Modified time.Time `orm:"column(modify_time);type(datetime);auto_now" json:"modify_time"`
	//修改人id
	ModifyAt       int    `orm:"column(modify_at);type(int)" json:"-"`
	ModifyRealName string `orm:"-" json:"modify_real_name"`
	//创建时间
	Created    time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
	CreateName string    `orm:"-" json:"create_name"`
	//版本号
	Version int64 `orm:"type(bigint);column(version)" json:"version"`
	//附件列表
	AttachList []*Attachment `orm:"-" json:"attach_list"`
}

// 多字段唯一键
func (b *Blog) TableUnique() [][]string {
	return [][]string{
		{"blog_id", "blog_identify"},
	}
}

// TableName 获取对应数据库表名.
func (b *Blog) TableName() string {
	return "blogs"
}

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

func (b *Blog) TableNameWithPrefix() string {
	return conf.GetDatabasePrefix() + b.TableName()
}

func NewBlog() *Blog {
	return &Blog{
		BlogStatus: "public",
	}
}

//根据文章ID查询文章
func (b *Blog) Find(blogId int) (*Blog, error) {
	o := orm.NewOrm()

	err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).One(b)
	if err != nil {
		logs.Error("查询文章时失败 -> ", err)
		return nil, err
	}

	return b.Link()
}

//从缓存中读取文章
func (b *Blog) FindFromCache(blogId int) (blog *Blog, err error) {
	key := fmt.Sprintf("blog-id-%d", blogId)
	var temp Blog
	err = cache.Get(key, &temp)
	if err == nil {
		b = &temp
		b.Link()
		logs.Debug("从缓存读取文章成功 ->", key)
		return b, nil
	} else {
		logs.Error("读取缓存失败 ->", err)
	}

	blog, err = b.Find(blogId)
	if err == nil {
		//默认一个小时
		if err := cache.Put(key, blog, time.Hour*1); err != nil {
			logs.Error("将文章存入缓存失败 ->", err)
		}
	}
	return
}

//查找指定用户的指定文章
func (b *Blog) FindByIdAndMemberId(blogId, memberId int) (*Blog, error) {
	o := orm.NewOrm()

	err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).Filter("member_id", memberId).One(b)
	if err != nil {
		logs.Error("查询文章时失败 -> ", err)
		return nil, err
	}

	return b.Link()
}

//根据文章标识查询文章
func (b *Blog) FindByIdentify(identify string) (*Blog, error) {
	o := orm.NewOrm()

	err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_identify", identify).One(b)
	if err != nil {
		logs.Error("查询文章时失败 -> ", err)
		return nil, err
	}
	return b, nil
}

//获取指定文章的链接内容
func (b *Blog) Link() (*Blog, error) {
	o := orm.NewOrm()
	//如果是链接文章，则需要从链接的项目中查找文章内容
	if b.BlogType == 1 && b.DocumentId > 0 {
		doc := NewDocument()
		if err := o.QueryTable(doc.TableNameWithPrefix()).Filter("document_id", b.DocumentId).One(doc, "release", "markdown", "identify", "book_id"); err != nil {
			logs.Error("查询文章链接对象时出错 -> ", err)
		} else {
			b.DocumentIdentify = doc.Identify
			b.BlogRelease = doc.Release

			//目前仅支持markdown文档进行链接
			b.BlogContent = doc.Markdown
			book := NewBook()
			if err := o.QueryTable(book.TableNameWithPrefix()).Filter("book_id", doc.BookId).One(book, "identify"); err != nil {
				logs.Error("查询关联文档的项目时出错 ->", err)
			} else {
				b.BookIdentify = book.Identify
				b.BookId = doc.BookId
			}
			//处理链接文档存在源文档修改时间的问题
			if content, err := goquery.NewDocumentFromReader(bytes.NewBufferString(b.BlogRelease)); err == nil {
				content.Find(".wiki-bottom").Remove()
				if html, err := content.Html(); err == nil {
					b.BlogRelease = html
				} else {
					logs.Error("处理文章失败 ->", err)
				}
			} else {
				logs.Error("处理文章失败 ->", err)
			}
		}
	}

	if b.ModifyAt > 0 {
		member := NewMember()
		if err := o.QueryTable(member.TableNameWithPrefix()).Filter("member_id", b.ModifyAt).One(member, "real_name", "account"); err == nil {
			if member.RealName != "" {
				b.ModifyRealName = member.RealName
			} else {
				b.ModifyRealName = member.Account
			}
		}
	}
	if b.MemberId > 0 {
		member := NewMember()
		if err := o.QueryTable(member.TableNameWithPrefix()).Filter("member_id", b.MemberId).One(member, "real_name", "account", "avatar"); err == nil {
			if member.RealName != "" {
				b.CreateName = member.RealName
			} else {
				b.CreateName = member.Account
			}
			b.MemberAvatar = member.Avatar
		}
	}

	return b, nil
}

//判断指定的文章标识是否存在
func (b *Blog) IsExist(identify string) bool {
	o := orm.NewOrm()

	return o.QueryTable(b.TableNameWithPrefix()).Filter("blog_identify", identify).Exist()
}

//保存文章
func (b *Blog) Save(cols ...string) error {
	o := orm.NewOrm()

	if b.OrderIndex <= 0 {
		blog := NewBlog()
		if err := o.QueryTable(blog.TableNameWithPrefix()).OrderBy("-blog_id").Limit(1).One(blog, "blog_id"); err == nil {
			b.OrderIndex = blog.BlogId + 1
		} else {
			c, _ := o.QueryTable(b.TableNameWithPrefix()).Count()
			b.OrderIndex = int(c) + 1
		}
	}
	var err error

	b.Processor().Version = time.Now().Unix()

	if b.BlogId > 0 {
		b.Modified = time.Now()
		_, err = o.Update(b, cols...)
		key := fmt.Sprintf("blog-id-%d", b.BlogId)
		cache.Delete(key)

	} else {

		b.Created = time.Now()
		_, err = o.Insert(b)
	}

	return err
}

//过滤文章的危险标签，处理文章外链以及图片.
func (b *Blog) Processor() *Blog {

	b.BlogRelease = utils.SafetyProcessor(b.BlogRelease)

	//解析文档中非本站的链接，并设置为新窗口打开
	if content, err := goquery.NewDocumentFromReader(bytes.NewBufferString(b.BlogRelease)); err == nil {

		content.Find("a").Each(func(i int, contentSelection *goquery.Selection) {
			if src, ok := contentSelection.Attr("href"); ok {
				if strings.HasPrefix(src, "http://") || strings.HasPrefix(src, "https://") {
					//logs.Info(src,conf.BaseUrl,strings.HasPrefix(src,conf.BaseUrl))
					if conf.BaseUrl != "" && !strings.HasPrefix(src, conf.BaseUrl) {
						contentSelection.SetAttr("target", "_blank")
						if html, err := content.Html(); err == nil {
							b.BlogRelease = html
						}
					}
				}

			}
		})
		//设置图片为CDN地址
		if cdnimg := adapter.AppConfig.String("cdnimg"); cdnimg != "" {
			content.Find("img").Each(func(i int, contentSelection *goquery.Selection) {
				if src, ok := contentSelection.Attr("src"); ok && strings.HasPrefix(src, "/uploads/") {
					contentSelection.SetAttr("src", utils.JoinURI(cdnimg, src))
				}
			})
		}
	}

	return b
}

//分页查询文章列表
func (b *Blog) FindToPager(pageIndex, pageSize int, memberId int, status string) (blogList []*Blog, totalCount int, err error) {

	o := orm.NewOrm()

	offset := (pageIndex - 1) * pageSize

	query := o.QueryTable(b.TableNameWithPrefix())

	if memberId > 0 {
		query = query.Filter("member_id", memberId)
	}
	if status != "" {
		query = query.Filter("blog_status", status)
	}

	_, err = query.OrderBy("-order_index", "-blog_id").Offset(offset).Limit(pageSize).All(&blogList)

	if err != nil {
		if err == orm.ErrNoRows {
			err = nil
		}
		logs.Error("获取文章列表时出错 ->", err)
		return
	}
	count, err := query.Count()

	if err != nil {
		logs.Error("获取文章数量时出错 ->", err)
		return nil, 0, err
	}
	totalCount = int(count)
	for _, blog := range blogList {
		if blog.BlogType == 1 {
			blog.Link()
		}
	}

	return
}

//删除文章
func (b *Blog) Delete(blogId int) error {
	o := orm.NewOrm()

	_, err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).Delete()
	if err != nil {
		logs.Error("删除文章失败 ->", err)
	}
	return err
}

//查询下一篇文章
func (b *Blog) QueryNext(blogId int) (*Blog, error) {
	o := orm.NewOrm()
	blog := NewBlog()

	if err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).One(blog, "order_index"); err != nil {
		logs.Error("查询文章时出错 ->", err)
		return b, err
	}

	err := o.QueryTable(b.TableNameWithPrefix()).Filter("order_index__gte", blog.OrderIndex).Filter("blog_id__gt", blogId).OrderBy("order_index", "blog_id").One(blog)

	if err != nil && err != orm.ErrNoRows {
		logs.Error("查询文章时出错 ->", err)
	}
	return blog, err
}

//查询下一篇文章
func (b *Blog) QueryPrevious(blogId int) (*Blog, error) {
	o := orm.NewOrm()
	blog := NewBlog()

	if err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).One(blog, "order_index"); err != nil {
		logs.Error("查询文章时出错 ->", err)
		return b, err
	}

	err := o.QueryTable(b.TableNameWithPrefix()).Filter("order_index__lte", blog.OrderIndex).Filter("blog_id__lt", blogId).OrderBy("-order_index", "-blog_id").One(blog)

	if err != nil && err != orm.ErrNoRows {
		logs.Error("查询文章时出错 ->", err)
	}
	return blog, err
}

//关联文章附件
func (b *Blog) LinkAttach() (err error) {

	o := orm.NewOrm()

	var attachList []*Attachment
	//当不是关联文章时，用文章ID去查询附件
	if b.BlogType != 1 || b.DocumentId <= 0 {
		_, err = o.QueryTable(NewAttachment().TableNameWithPrefix()).Filter("document_id", b.BlogId).Filter("book_id", 0).All(&attachList)
		if err != nil && err != orm.ErrNoRows {
			logs.Error("查询文章附件时出错 ->", err)
		}
	} else {
		_, err = o.QueryTable(NewAttachment().TableNameWithPrefix()).Filter("document_id", b.DocumentId).Filter("book_id", b.BookId).All(&attachList)

		if err != nil && err != orm.ErrNoRows {
			logs.Error("查询文章附件时出错 ->", err)
		}
	}
	b.AttachList = attachList
	return
}