mindoc/models/DocumentModel.go

package models

import (
	"time"

	"fmt"
	"strconv"

	"github.com/astaxie/beego"
	"github.com/astaxie/beego/orm"
	"github.com/lifei6671/mindoc/cache"
	"github.com/lifei6671/mindoc/conf"
	"github.com/PuerkitoBio/goquery"
	"strings"
	"bytes"
	"os"
	"path/filepath"
)

// Document struct.
type Document struct {
	DocumentId   int    `orm:"pk;auto;unique;column(document_id)" json:"doc_id"`
	DocumentName string `orm:"column(document_name);size(500)" json:"doc_name"`
	// Identify 文档唯一标识
	Identify  string `orm:"column(identify);size(100);index;null;default(null)" json:"identify"`
	BookId    int    `orm:"column(book_id);type(int);index" json:"book_id"`
	ParentId  int    `orm:"column(parent_id);type(int);index;default(0)" json:"parent_id"`
	OrderSort int    `orm:"column(order_sort);default(0);type(int);index" json:"order_sort"`
	// Markdown markdown格式文档.
	Markdown string `orm:"column(markdown);type(text);null" json:"markdown"`
	// Release 发布后的Html格式内容.
	Release string `orm:"column(release);type(text);null" json:"release"`
	// Content 未发布的 Html 格式内容.
	Content    string        `orm:"column(content);type(text);null" json:"content"`
	CreateTime time.Time     `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
	MemberId   int           `orm:"column(member_id);type(int)" json:"member_id"`
	ModifyTime time.Time     `orm:"column(modify_time);type(datetime);auto_now" json:"modify_time"`
	ModifyAt   int           `orm:"column(modify_at);type(int)" json:"-"`
	Version    int64         `orm:"column(version);type(bigint);" json:"version"`
	//是否展开子目录：0 否/1 是
	IsOpen	   int 			 `orm:"column(is_open);type(int);default(0)" json:"is_open"`
	AttachList []*Attachment `orm:"-" json:"attach"`
}

// 多字段唯一键
func (m *Document) TableUnique() [][]string {
	return [][]string{
		[]string{"book_id", "identify"},
	}
}

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

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

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

func NewDocument() *Document {
	return &Document{
		Version: time.Now().Unix(),
	}
}

//根据文档ID查询指定文档.
func (m *Document) Find(id int) (*Document, error) {
	if id <= 0 {
		return m, ErrInvalidParameter
	}

	o := orm.NewOrm()

	err := o.QueryTable(m.TableNameWithPrefix()).Filter("document_id", id).One(m)

	if err == orm.ErrNoRows {
		return m, ErrDataNotExist
	}

	return m, nil
}

//插入和更新文档.
func (m *Document) InsertOrUpdate(cols ...string) error {
	o := orm.NewOrm()
	var err error
	if m.DocumentId > 0 {
		_, err = o.Update(m, cols...)
	} else {
		if m.Identify == "" {
			book := NewBook()
			identify := "docs"
			if err := o.QueryTable(book.TableNameWithPrefix()).Filter("book_id", m.BookId).One(book, "identify"); err == nil {
				identify = book.Identify
			}

			m.Identify = fmt.Sprintf("%s-%s", identify, strconv.FormatInt(time.Now().UnixNano(), 32))
		}

		if m.OrderSort == 0 {
			sort, _ := o.QueryTable(m.TableNameWithPrefix()).Filter("book_id", m.BookId).Filter("parent_id", m.ParentId).Count()
			m.OrderSort = int(sort) + 1
		}
		_, err = o.Insert(m)
		NewBook().ResetDocumentNumber(m.BookId)
	}
	if err != nil {
		return err
	}

	return nil
}

//根据文档识别编号和项目id获取一篇文档
func (m *Document) FindByIdentityFirst(identify string, bookId int) (*Document, error) {
	o := orm.NewOrm()

	err := o.QueryTable(m.TableNameWithPrefix()).Filter("book_id", bookId).Filter("identify", identify).One(m)

	return m, err
}

//递归删除一个文档.
func (m *Document) RecursiveDocument(docId int) error {

	o := orm.NewOrm()

	if doc, err := m.Find(docId); err == nil {
		o.Delete(doc)
		NewDocumentHistory().Clear(doc.DocumentId)
	}
	var maps []orm.Params

	_, err := o.Raw("SELECT document_id FROM " + m.TableNameWithPrefix() + " WHERE parent_id=" + strconv.Itoa(docId)).Values(&maps)
	if err != nil {
		beego.Error("RecursiveDocument => ", err)
		return err
	}

	for _, item := range maps {
		if docId, ok := item["document_id"].(string); ok {
			id, _ := strconv.Atoi(docId)
			o.QueryTable(m.TableNameWithPrefix()).Filter("document_id", id).Delete()
			m.RecursiveDocument(id)
		}
	}

	return nil
}

//将文档写入缓存
func (m *Document) PutToCache() {
	go func(m Document) {

		if m.Identify == "" {

			if err := cache.Put("Document.Id."+strconv.Itoa(m.DocumentId), m, time.Second*3600); err != nil {
				beego.Info("文档缓存失败:", m.DocumentId)
			}
		} else {
			if err := cache.Put(fmt.Sprintf("Document.BookId.%d.Identify.%s", m.BookId, m.Identify), m, time.Second*3600); err != nil {
				beego.Info("文档缓存失败:", m.DocumentId)
			}
		}

	}(*m)
}

//清除缓存
func (m *Document) RemoveCache() {
	go func(m Document) {
		cache.Put("Document.Id."+strconv.Itoa(m.DocumentId), m, time.Second*3600)

		if m.Identify != "" {
			cache.Put(fmt.Sprintf("Document.BookId.%d.Identify.%s", m.BookId, m.Identify), m, time.Second*3600)
		}
	}(*m)
}

//从缓存获取
func (m *Document) FromCacheById(id int) (*Document, error) {

	if err := cache.Get("Document.Id."+strconv.Itoa(id), &m); err == nil && m.DocumentId > 0 {
		beego.Info("从缓存中获取文档信息成功 ->", m.DocumentId)
		return m, nil
	}

	if m.DocumentId > 0 {
		m.PutToCache()
	}
	m, err := m.Find(id)

	if err == nil {
		m.PutToCache()
	}
	return m, err
}

//根据文档标识从缓存中查询文档
func (m *Document) FromCacheByIdentify(identify string, bookId int) (*Document, error) {

	key := fmt.Sprintf("Document.BookId.%d.Identify.%s", bookId, identify)

	if err := cache.Get(key, m); err == nil && m.DocumentId > 0 {
		beego.Info("从缓存中获取文档信息成功 ->", key)
		return m, nil
	}

	defer func() {
		if m.DocumentId > 0 {
			m.PutToCache()
		}
	}()
	return m.FindByIdentityFirst(identify, bookId)
}

//根据项目ID查询文档列表.
func (m *Document) FindListByBookId(bookId int) (docs []*Document, err error) {
	o := orm.NewOrm()

	_, err = o.QueryTable(m.TableNameWithPrefix()).Filter("book_id", bookId).OrderBy("order_sort").All(&docs)

	return
}

//判断文章是否存在
func (m *Document) IsExist(documentId int) bool {
	o := orm.NewOrm()

	return o.QueryTable(m.TableNameWithPrefix()).Filter("document_id", documentId).Exist()
}

//发布单篇文档
func (item *Document) ReleaseContent() error {

	o := orm.NewOrm()

	bookId := item.BookId

	if item.Content != "" {
		item.Release = item.Content
		bufio := bytes.NewReader([]byte(item.Content))
		//解析文档中非本站的链接，并设置为新窗口打开
		if content, err := goquery.NewDocumentFromReader(bufio); 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 {
								item.Release = strings.TrimSuffix(strings.TrimPrefix(strings.TrimSpace(html),"<html><head></head><body>"),"</body></html>")
							}
						}
					}

				}
			})
		}
	}

	attachList, err := NewAttachment().FindListByDocumentId(item.DocumentId)
	if err == nil && len(attachList) > 0 {
		content := bytes.NewBufferString("<div class=\"attach-list\"><strong>附件</strong><ul>")
		for _, attach := range attachList {
			if strings.HasPrefix(attach.HttpPath, "/") {
				attach.HttpPath = strings.TrimSuffix(conf.BaseUrl, "/") + attach.HttpPath
			}
			li := fmt.Sprintf("<li><a href=\"%s\" target=\"_blank\" title=\"%s\">%s</a></li>", attach.HttpPath, attach.FileName, attach.FileName)

			content.WriteString(li)
		}
		content.WriteString("</ul></div>")
		item.Release += content.String()
	}
	_, err = o.Update(item, "release")
	if err != nil {
		beego.Error(fmt.Sprintf("发布失败 => %+v", item), err)
		return err
	} else {
		//当文档发布后，需要清除已缓存的转换文档和文档缓存
		if doc, err := NewDocument().Find(item.DocumentId); err == nil {
			doc.PutToCache()
		} else {
			doc.RemoveCache()
		}

		if err := os.RemoveAll(filepath.Join(conf.WorkingDirectory, "uploads", "books", strconv.Itoa(bookId))); err != nil {
			beego.Error("删除已缓存的文档目录失败 => ",filepath.Join(conf.WorkingDirectory, "uploads", "books", strconv.Itoa(bookId)))
			return err
		}
	}
	return nil
}

//追加一些文档信息.
func (doc *Document) AppendInfo() *Document {

	docCreator, err := NewMember().Find(doc.MemberId,"real_name","account")

	doc.Release = strings.TrimSuffix(strings.TrimPrefix(strings.TrimSpace(doc.Release),"<html><head></head><body>"),"</body></html>")
	suffix := ""
	if doc.Release != "" {
		beego.Info(doc.Release)
		if strings.HasPrefix(doc.Release,"<div class=\"markdown-toc") {

			doc.Release = strings.TrimSuffix(doc.Release,"</div>")
			suffix = "</div>"
		}
		doc.Release += "<div class=\"wiki-bottom\">文档更新时间: " + doc.ModifyTime.Local().Format("2006-01-02 15:04") + " &nbsp;&nbsp;作者：";
		if err == nil && docCreator != nil {
			if docCreator.RealName != "" {
				doc.Release += docCreator.RealName
			} else {
				doc.Release += docCreator.Account
			}
		}
		doc.Release += "</div>" + suffix
	}

	return doc
}