mindoc/models/Blog.go

package models

import (
	"time"
	"github.com/lifei6671/mindoc/conf"
	"github.com/astaxie/beego/orm"
	"github.com/astaxie/beego"
	"github.com/lifei6671/mindoc/cache"
	"fmt"
	"github.com/PuerkitoBio/goquery"
	"strings"
	"bytes"
	"github.com/lifei6671/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 (m *Blog) TableUnique() [][]string {
	return [][]string{
		{"blog_id", "blog_identify"},
	}
}

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

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

func (m *Blog) TableNameWithPrefix() string {
	return conf.GetDatabasePrefix() + m.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 {
		beego.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()
		beego.Debug("从缓存读取文章成功 ->", key)
		return b,nil
	}else {
		beego.Error("读取缓存失败 ->",err)
	}

	blog,err = b.Find(blogId)
	if err == nil {
		//默认一个小时
		if err := cache.Put(key,blog,time.Hour * 1); err != nil {
			beego.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 {
		beego.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 {
		beego.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 {
			beego.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 {
				beego.Error("查询关联文档的项目时出错 ->",err)
			}else{
				b.BookIdentify = book.Identify
				b.BookId = doc.BookId
			}
		}
	}

	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://") {
					//beego.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 := beego.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
		}
		beego.Error("获取文章列表时出错 ->",err)
		return
	}
	count,err := query.Count()

	if err != nil {
		beego.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 {
		beego.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 {
		beego.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{
		beego.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 {
		beego.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{
		beego.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{
			beego.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{
			beego.Error("查询文章附件时出错 ->", err)
		}
	}
	b.AttachList = attachList
	return
}