| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848 | package modelsimport (	"bytes"	"crypto/md5"	"errors"	"fmt"	"io"	"io/ioutil"	"os"	"path/filepath"	"regexp"	"strconv"	"strings"	"time"	"github.com/PuerkitoBio/goquery"	"github.com/astaxie/beego"	"github.com/astaxie/beego/logs"	"github.com/astaxie/beego/orm"	"github.com/lifei6671/mindoc/conf"	"github.com/lifei6671/mindoc/utils/cryptil"	"github.com/lifei6671/mindoc/utils/filetil"	"github.com/lifei6671/mindoc/utils/requests"	"github.com/lifei6671/mindoc/utils/ziptil"	"gopkg.in/russross/blackfriday.v2")// Book struct .type Book struct {	BookId int `orm:"pk;auto;unique;column(book_id)" json:"book_id"`	// BookName 项目名称.	BookName string `orm:"column(book_name);size(500)" json:"book_name"`	// Identify 项目唯一标识.	Identify string `orm:"column(identify);size(100);unique" json:"identify"`	//是否是自动发布 0 否/1 是	AutoRelease int `orm:"column(auto_release);type(int);default(0)" json:"auto_release"`	//是否开启下载功能 0 是/1 否	IsDownload int `orm:"column(is_download);type(int);default(0)" json:"is_download"`	OrderIndex int `orm:"column(order_index);type(int);default(0)" json:"order_index"`	// Description 项目描述.	Description string `orm:"column(description);size(2000)" json:"description"`	//发行公司	Publisher string `orm:"column(publisher);size(500)" json:"publisher"`	Label     string `orm:"column(label);size(500)" json:"label"`	// PrivatelyOwned 项目私有: 0 公开/ 1 私有	PrivatelyOwned int `orm:"column(privately_owned);type(int);default(0)" json:"privately_owned"`	// 当项目是私有时的访问Token.	PrivateToken string `orm:"column(private_token);size(500);null" json:"private_token"`	//状态:0 正常/1 已删除	Status int `orm:"column(status);type(int);default(0)" json:"status"`	//默认的编辑器.	Editor string `orm:"column(editor);size(50)" json:"editor"`	// DocCount 包含文档数量.	DocCount int `orm:"column(doc_count);type(int)" json:"doc_count"`	// CommentStatus 评论设置的状态:open 为允许所有人评论,closed 为不允许评论, group_only 仅允许参与者评论 ,registered_only 仅允许注册者评论.	CommentStatus string `orm:"column(comment_status);size(20);default(open)" json:"comment_status"`	CommentCount  int    `orm:"column(comment_count);type(int)" json:"comment_count"`	//封面地址	Cover string `orm:"column(cover);size(1000)" json:"cover"`	//主题风格	Theme string `orm:"column(theme);size(255);default(default)" json:"theme"`	// CreateTime 创建时间 .	CreateTime time.Time `orm:"type(datetime);column(create_time);auto_now_add" json:"create_time"`	//每个文档保存的历史记录数量,0 为不限制	HistoryCount int `orm:"column(history_count);type(int);default(0)" json:"history_count"`	//是否启用分享,0启用/1不启用	IsEnableShare int       `orm:"column(is_enable_share);type(int);default(0)" json:"is_enable_share"`	MemberId      int       `orm:"column(member_id);size(100)" json:"member_id"`	ModifyTime    time.Time `orm:"type(datetime);column(modify_time);null;auto_now" json:"modify_time"`	Version       int64     `orm:"type(bigint);column(version)" json:"version"`	//是否使用第一篇文章项目为默认首页,0 否/1 是	IsUseFirstDocument int `orm:"column(is_use_first_document);type(int);default(0)" json:"is_use_first_document"`}// TableName 获取对应数据库表名.func (book *Book) TableName() string {	return "books"}// TableEngine 获取数据使用的引擎.func (book *Book) TableEngine() string {	return "INNODB"}func (book *Book) TableNameWithPrefix() string {	return conf.GetDatabasePrefix() + book.TableName()}func NewBook() *Book {	return &Book{}}//添加一个项目func (book *Book) Insert() error {	o := orm.NewOrm()	//	o.Begin()	_, err := o.Insert(book)	if err == nil {		if book.Label != "" {			NewLabel().InsertOrUpdateMulti(book.Label)		}		relationship := NewRelationship()		relationship.BookId = book.BookId		relationship.RoleId = 0		relationship.MemberId = book.MemberId		err = relationship.Insert()		if err != nil {			logs.Error("插入项目与用户关联 => ", err)			//o.Rollback()			return err		}		document := NewDocument()		document.BookId = book.BookId		document.DocumentName = "空白文档"		document.MemberId = book.MemberId		err = document.InsertOrUpdate()		if err != nil {			//o.Rollback()			return err		}		//o.Commit()		return nil	}	//o.Rollback()	return err}func (book *Book) Find(id int) (*Book, error) {	if id <= 0 {		return book, ErrInvalidParameter	}	o := orm.NewOrm()	err := o.QueryTable(book.TableNameWithPrefix()).Filter("book_id", id).One(book)	return book, err}//更新一个项目func (book *Book) Update(cols ...string) error {	o := orm.NewOrm()	temp := NewBook()	temp.BookId = book.BookId	if err := o.Read(temp); err != nil {		return err	}	if book.Label != "" || temp.Label != "" {		go NewLabel().InsertOrUpdateMulti(book.Label + "," + temp.Label)	}	_, err := o.Update(book, cols...)	return err}//复制项目func (book *Book) Copy(identify string) error {	o := orm.NewOrm()	err := o.QueryTable(book.TableNameWithPrefix()).Filter("identify",identify).One(book)	if err != nil {		beego.Error("查询项目时出错 -> ",err)		return err	}	if err := o.Begin();err != nil {		beego.Error("开启事物时出错 -> ",err)		return err	}	bookId := book.BookId	book.BookId = 0	book.Identify = book.Identify + fmt.Sprintf("%s-%s",identify,strconv.FormatInt(time.Now().UnixNano(), 32))	book.BookName = book.BookName + "[副本]"	book.CreateTime = time.Now()	book.CommentCount = 0	book.HistoryCount = 0	if _,err := o.Insert(book);err != nil {		beego.Error("复制项目时出错 -> ",err)		o.Rollback()		return err	}	var rels []*Relationship	if _,err := o.QueryTable(NewRelationship().TableNameWithPrefix()).Filter("book_id",bookId).All(&rels); err != nil {		beego.Error("复制项目关系时出错 -> ",err)		o.Rollback()		return err	}	for _,rel := range rels {		rel.BookId = book.BookId		rel.RelationshipId = 0		if _,err := o.Insert(rel);err != nil {			beego.Error("复制项目关系时出错 -> ",err)			o.Rollback()			return err		}	}	var docs []*Document	if _,err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id",bookId).Filter("parent_id",0).All(&docs);err != nil && err != orm.ErrNoRows {		beego.Error("读取项目文档时出错 -> ",err)		o.Rollback()		return err	}	if len(docs) > 0 {		if err := recursiveInsertDocument(docs,o,book.BookId,0);err != nil {			beego.Error("复制项目时出错 -> ",err)			o.Rollback()			return err		}	}	return o.Commit()}//递归的复制文档func recursiveInsertDocument(docs []*Document,o orm.Ormer,bookId int,parentId int) error {	for _,doc := range docs {		docId := doc.DocumentId		doc.DocumentId = 0		doc.ParentId = parentId		doc.BookId = bookId		doc.Version = time.Now().Unix()		if _,err := o.Insert(doc);err != nil {			beego.Error("插入项目时出错 -> ",err)			return err		}		var attachList []*Attachment		//读取所有附件列表		if _,err := o.QueryTable(NewAttachment().TableNameWithPrefix()).Filter("document_id",docId).All(&attachList); err == nil {			for _,attach := range attachList {				attach.BookId = bookId				attach.DocumentId = doc.DocumentId				attach.AttachmentId = 0				if _,err := o.Insert(attach);err != nil {					return err				}			}		}		var subDocs []*Document		if _,err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("parent_id",docId).All(&subDocs);err != nil && err != orm.ErrNoRows {			beego.Error("读取文档时出错 -> ",err)			return err		}		if len(subDocs) > 0{			if err := recursiveInsertDocument(subDocs,o,bookId,doc.DocumentId);err != nil {				return err			}		}	}	return nil}//根据指定字段查询结果集.func (book *Book) FindByField(field string, value interface{},cols ...string) ([]*Book, error) {	o := orm.NewOrm()	var books []*Book	_, err := o.QueryTable(book.TableNameWithPrefix()).Filter(field, value).All(&books,cols...)	return books, err}//根据指定字段查询一个结果.func (book *Book) FindByFieldFirst(field string, value interface{}) (*Book, error) {	o := orm.NewOrm()	err := o.QueryTable(book.TableNameWithPrefix()).Filter(field, value).One(book)	return book, err}//根据项目标识查询项目func (book *Book) FindByIdentify(identify string) (*Book, error) {	o := orm.NewOrm()	err := o.QueryTable(book.TableNameWithPrefix()).Filter("identify", identify).One(book)	return book, err}//分页查询指定用户的项目func (book *Book) FindToPager(pageIndex, pageSize, memberId int) (books []*BookResult, totalCount int, err error) {	relationship := NewRelationship()	o := orm.NewOrm()	sql1 := "SELECT COUNT(book.book_id) AS total_count FROM " + book.TableNameWithPrefix() + " AS book LEFT JOIN " +		relationship.TableNameWithPrefix() + " AS rel ON book.book_id=rel.book_id AND rel.member_id = ? WHERE rel.relationship_id > 0 "	err = o.Raw(sql1, memberId).QueryRow(&totalCount)	if err != nil {		return	}	offset := (pageIndex - 1) * pageSize	sql2 := "SELECT book.*,rel.member_id,rel.role_id,m.account as create_name FROM " + book.TableNameWithPrefix() + " AS book" +		" LEFT JOIN " + relationship.TableNameWithPrefix() + " AS rel ON book.book_id=rel.book_id AND rel.member_id = ?" +		" LEFT JOIN " + relationship.TableNameWithPrefix() + " AS rel1 ON book.book_id=rel1.book_id  AND rel1.role_id=0" +		" LEFT JOIN " + NewMember().TableNameWithPrefix() + " AS m ON rel1.member_id=m.member_id " +		" WHERE rel.relationship_id > 0 ORDER BY book.order_index DESC,book.book_id DESC LIMIT " + fmt.Sprintf("%d,%d", offset, pageSize)	_, err = o.Raw(sql2, memberId).QueryRows(&books)	if err != nil {		logs.Error("分页查询项目列表 => ", err)		return	}	sql := "SELECT m.account,doc.modify_time FROM md_documents AS doc LEFT JOIN md_members AS m ON doc.modify_at=m.member_id WHERE book_id = ? LIMIT 1 ORDER BY doc.modify_time DESC"	if err == nil && len(books) > 0 {		for index, book := range books {			var text struct {				Account    string				ModifyTime time.Time			}			err1 := o.Raw(sql, book.BookId).QueryRow(&text)			if err1 == nil {				books[index].LastModifyText = text.Account + " 于 " + text.ModifyTime.Format("2006-01-02 15:04:05")			}			if book.RoleId == 0 {				book.RoleName = "创始人"			} else if book.RoleId == 1 {				book.RoleName = "管理员"			} else if book.RoleId == 2 {				book.RoleName = "编辑者"			} else if book.RoleId == 3 {				book.RoleName = "观察者"			}		}	}	return}// 彻底删除项目.func (book *Book) ThoroughDeleteBook(id int) error {	if id <= 0 {		return ErrInvalidParameter	}	o := orm.NewOrm()	book, err := book.Find(id)	if err != nil {		return err	}	o.Begin()	sql2 := "DELETE FROM " + NewDocument().TableNameWithPrefix() + " WHERE book_id = ?"	_, err = o.Raw(sql2, book.BookId).Exec()	if err != nil {		o.Rollback()		return err	}	sql3 := "DELETE FROM " + book.TableNameWithPrefix() + " WHERE book_id = ?"	_, err = o.Raw(sql3, book.BookId).Exec()	if err != nil {		o.Rollback()		return err	}	sql4 := "DELETE FROM " + NewRelationship().TableNameWithPrefix() + " WHERE book_id = ?"	_, err = o.Raw(sql4, book.BookId).Exec()	if err != nil {		o.Rollback()		return err	}	if book.Label != "" {		NewLabel().InsertOrUpdateMulti(book.Label)	}	os.RemoveAll(filepath.Join(conf.WorkingDirectory, "uploads", "books", strconv.Itoa(id)))	return o.Commit()}//分页查找系统首页数据.func (book *Book) FindForHomeToPager(pageIndex, pageSize, member_id int) (books []*BookResult, totalCount int, err error) {	o := orm.NewOrm()	offset := (pageIndex - 1) * pageSize	//如果是登录用户	if member_id > 0 {		sql1 := "SELECT COUNT(*) FROM md_books AS book LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ? WHERE relationship_id > 0 OR book.privately_owned = 0"		err = o.Raw(sql1, member_id).QueryRow(&totalCount)		if err != nil {			return		}		sql2 := `SELECT book.*,rel1.*,member.account AS create_name,member.real_name FROM md_books AS book			LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?			LEFT JOIN md_relationship AS rel1 ON rel1.book_id = book.book_id AND rel1.role_id = 0			LEFT JOIN md_members AS member ON rel1.member_id = member.member_id			WHERE rel.relationship_id > 0 OR book.privately_owned = 0 ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`		_, err = o.Raw(sql2, member_id, offset, pageSize).QueryRows(&books)	} else {		count, err1 := o.QueryTable(book.TableNameWithPrefix()).Filter("privately_owned", 0).Count()		if err1 != nil {			err = err1			return		}		totalCount = int(count)		sql := `SELECT book.*,rel.*,member.account AS create_name,member.real_name FROM md_books AS book			LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.role_id = 0			LEFT JOIN md_members AS member ON rel.member_id = member.member_id			WHERE book.privately_owned = 0 ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`		_, err = o.Raw(sql, offset, pageSize).QueryRows(&books)	}	return}//分页全局搜索.func (book *Book) FindForLabelToPager(keyword string, pageIndex, pageSize, memberId int) (books []*BookResult, totalCount int, err error) {	o := orm.NewOrm()	keyword = "%" + keyword + "%"	offset := (pageIndex - 1) * pageSize	//如果是登录用户	if memberId > 0 {		sql1 := "SELECT COUNT(*) FROM md_books AS book LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ? WHERE (relationship_id > 0 OR book.privately_owned = 0) AND book.label LIKE ?"		err = o.Raw(sql1, memberId, keyword).QueryRow(&totalCount)		if err != nil {			return		}		sql2 := `SELECT book.*,rel1.*,member.account AS create_name FROM md_books AS book			LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?			LEFT JOIN md_relationship AS rel1 ON rel1.book_id = book.book_id AND rel1.role_id = 0			LEFT JOIN md_members AS member ON rel1.member_id = member.member_id			WHERE (rel.relationship_id > 0 OR book.privately_owned = 0) AND book.label LIKE ? ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`		_, err = o.Raw(sql2, memberId, keyword, offset, pageSize).QueryRows(&books)		return	} else {		count, err1 := o.QueryTable(NewBook().TableNameWithPrefix()).Filter("privately_owned", 0).Filter("label__icontains", keyword).Count()		if err1 != nil {			err = err1			return		}		totalCount = int(count)		sql := `SELECT book.*,rel.*,member.account AS create_name FROM md_books AS book			LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.role_id = 0			LEFT JOIN md_members AS member ON rel.member_id = member.member_id			WHERE book.privately_owned = 0 AND book.label LIKE ? ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`		_, err = o.Raw(sql, keyword, offset, pageSize).QueryRows(&books)		return	}}//发布文档func (book *Book) ReleaseContent(bookId int) {	o := orm.NewOrm()	var docs []*Document	_, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", bookId).All(&docs, "document_id", "identify", "content")	if err != nil {		beego.Error("发布失败 =>",bookId, err)		return	}	for _, item := range docs {		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 = 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)		} 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)))			}		}	}}//重置文档数量func (book *Book) ResetDocumentNumber(bookId int) {	o := orm.NewOrm()	totalCount, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", bookId).Count()	if err == nil {		_,err = o.Raw("UPDATE md_books SET doc_count = ? WHERE book_id = ?", int(totalCount), bookId).Exec()		if err != nil {			beego.Error("重置文档数量失败 =>",bookId,err)		}	} else {		beego.Error("获取文档数量失败 =>",bookId,err)	}}//导入项目func (book *Book) ImportBook(zipPath string) error {	if !filetil.FileExists(zipPath) {		return errors.New("文件不存在 => " + zipPath)	}	w := md5.New()	io.WriteString(w, zipPath) //将str写入到w中	io.WriteString(w, time.Now().String())	io.WriteString(w, book.BookName)	md5str := fmt.Sprintf("%x", w.Sum(nil)) //w.Sum(nil)将w的hash转成[]byte格式	tempPath := filepath.Join(os.TempDir(), md5str)	if err := os.MkdirAll(tempPath, 0766); err != nil {		beego.Error("创建导入目录出错 => ",err)	}	//如果加压缩失败	if err := ziptil.Unzip(zipPath, tempPath); err != nil {		return err	}	//当导入结束后,删除临时文件	defer os.RemoveAll(tempPath)	for {		//如果当前目录下只有一个目录,则重置根目录		if entries, err := ioutil.ReadDir(tempPath); err == nil && len(entries) == 1 {			dir := entries[0]			if dir.IsDir() && dir.Name() != "." && dir.Name() != ".." {				tempPath = filepath.Join(tempPath, dir.Name())				break			}		} else {			break		}	}	tempPath = strings.Replace(tempPath, "\\", "/", -1)	docMap := make(map[string]int, 0)	o := orm.NewOrm()	o.Insert(book)	relationship := NewRelationship()	relationship.BookId = book.BookId	relationship.RoleId = 0	relationship.MemberId = book.MemberId	relationship.Insert()	err := filepath.Walk(tempPath, func(path string, info os.FileInfo, err error) error {		path = strings.Replace(path, "\\", "/", -1)		if path == tempPath {			return nil		}		if !info.IsDir() {			ext := filepath.Ext(info.Name())			//如果是Markdown文件			if strings.EqualFold(ext, ".md") || strings.EqualFold(ext, ".markdown") {				beego.Info("正在处理 =>",path,info.Name())				doc := NewDocument()				doc.BookId = book.BookId				doc.MemberId = book.MemberId				docIdentify := strings.Replace(strings.TrimPrefix(path, tempPath+"/"), "/", "-", -1)				if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, docIdentify); !ok || err != nil {					docIdentify = "import-" + docIdentify				}				doc.Identify = docIdentify				//匹配图片,如果图片语法是在代码块中,这里同样会处理				re := regexp.MustCompile(`!\[(.*?)\]\((.*?)\)`)				markdown, err := filetil.ReadFileAndIgnoreUTF8BOM(path)				if err != nil {					return err				}				//处理图片				doc.Markdown = re.ReplaceAllStringFunc(string(markdown), func(image string) string {					images := re.FindAllSubmatch([]byte(image), -1)					if len(images) <= 0 || len(images[0]) < 3 {						return image					}					originalImageUrl := string(images[0][2])					imageUrl := strings.Replace(string(originalImageUrl), "\\", "/", -1)					//如果是本地路径,则需要将图片复制到项目目录					if !strings.HasPrefix(imageUrl, "http://") && !strings.HasPrefix(imageUrl, "https://") {						//如果路径中存在参数						if l := strings.Index(imageUrl,"?"); l > 0 {							imageUrl = imageUrl[:l]						}						if strings.HasPrefix(imageUrl, "/") {							imageUrl = filepath.Join(tempPath, imageUrl)						} else if strings.HasPrefix(imageUrl, "./") {							imageUrl = filepath.Join(filepath.Dir(path), strings.TrimPrefix(imageUrl, "./"))						} else if strings.HasPrefix(imageUrl, "../") {							imageUrl = filepath.Join(filepath.Dir(path), imageUrl)						} else {							imageUrl = filepath.Join(filepath.Dir(path), imageUrl)						}						imageUrl = strings.Replace(imageUrl, "\\", "/", -1)						dstFile := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), strings.TrimPrefix(imageUrl, tempPath))						if filetil.FileExists(imageUrl) {							filetil.CopyFile(imageUrl, dstFile)							imageUrl = strings.TrimPrefix(strings.Replace(dstFile,"\\","/",-1), strings.Replace(conf.WorkingDirectory,"\\","/",-1))							if !strings.HasPrefix(imageUrl, "/") && !strings.HasPrefix(imageUrl, "\\") {								imageUrl = "/" + imageUrl							}						}					} else {						imageExt := cryptil.Md5Crypt(imageUrl) + filepath.Ext(imageUrl)						dstFile := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), imageExt)						if err := requests.DownloadAndSaveFile(imageUrl, dstFile); err == nil {							imageUrl = strings.TrimPrefix(strings.Replace(dstFile, "\\", "/", -1), strings.Replace(conf.WorkingDirectory, "\\", "/", -1))							if !strings.HasPrefix(imageUrl, "/") && !strings.HasPrefix(imageUrl, "\\") {								imageUrl = "/" + imageUrl							}						}					}					imageUrl = strings.Replace(strings.TrimSuffix(image, originalImageUrl+")")+ conf.URLForWithCdnImage(imageUrl) +")", "\\", "/", -1)					return imageUrl				})				linkRegexp := regexp.MustCompile(`\[(.*?)\]\((.*?)\)`)				doc.Markdown = linkRegexp.ReplaceAllStringFunc(doc.Markdown, func(link string) string {					links := linkRegexp.FindAllStringSubmatch(link, -1)					originalLink := links[0][2]					linkPath, err := filepath.Abs(filepath.Join(filepath.Dir(path), originalLink))					if err == nil {						//如果本地存在该链接						if filetil.FileExists(linkPath) {							ext := filepath.Ext(linkPath)							//如果链接是Markdown文件,则生成文档标识,否则,将目标文件复制到项目目录							if strings.EqualFold(ext, ".md") || strings.EqualFold(ext, ".markdown") {								docIdentify := strings.Replace(strings.TrimPrefix(strings.Replace(linkPath, "\\", "/", -1), tempPath+"/"), "/", "-", -1)								//beego.Info(originalLink, "|", linkPath, "|", docIdentify)								if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, docIdentify); !ok || err != nil {									docIdentify = "import-" + docIdentify								}								docIdentify = strings.TrimSuffix(docIdentify, "-README.md")								link = strings.TrimSuffix(link, originalLink+")") + conf.URLFor("DocumentController.Read", ":key", book.Identify, ":id", docIdentify) + ")"							} else {								dstPath := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), originalLink)								filetil.CopyFile(linkPath, dstPath)								tempLink := conf.BaseUrl + strings.TrimPrefix(strings.Replace(dstPath, "\\", "/", -1), strings.Replace(conf.WorkingDirectory, "\\", "/", -1))								link = strings.TrimSuffix(link, originalLink+")") + tempLink + ")"							}						}					}					return link				})				//codeRe := regexp.MustCompile("```\\w+")				//doc.Markdown = codeRe.ReplaceAllStringFunc(doc.Markdown, func(s string) string {				//	//beego.Info(s)				//	return strings.Replace(s,"```","``` ",-1)				//})				doc.Content = string(blackfriday.Run([]byte(doc.Markdown)))				doc.Version = time.Now().Unix()				//解析文档名称,默认使用第一个h标签为标题				docName := strings.TrimSuffix(info.Name(), ext)				for _, line := range strings.Split(doc.Markdown, "\n") {					if strings.HasPrefix(line, "#") {						docName = strings.TrimLeft(line, "#")						break					}				}				doc.DocumentName = strings.TrimSpace(docName)				parentId := 0				parentIdentify := strings.Replace(strings.Trim(strings.TrimSuffix(strings.TrimPrefix(path, tempPath), info.Name()), "/"), "/", "-", -1)				if parentIdentify != "" {					if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, parentIdentify); !ok || err != nil {						parentIdentify = "import-" + parentIdentify					}					if id, ok := docMap[parentIdentify]; ok {						parentId = id					}				}				if strings.EqualFold(info.Name(), "README.md") {					beego.Info(path, "|", info.Name(), "|", parentIdentify, "|", parentId)				}				isInsert := false				//如果当前文件是README.md,则将内容更新到父级				if strings.EqualFold(info.Name(), "README.md") && parentId != 0 {					doc.DocumentId = parentId					//beego.Info(path,"|",parentId)				} else {					//beego.Info(path,"|",parentIdentify)					doc.ParentId = parentId					isInsert = true				}				if err := doc.InsertOrUpdate("document_name", "markdown", "content"); err != nil {					beego.Error(doc.DocumentId, err)				}				if isInsert {					docMap[docIdentify] = doc.DocumentId				}			}		} else {			//如果当前目录下存在Markdown文件,则需要创建此节点			if filetil.HasFileOfExt(path, []string{".md", ".markdown"}) {				beego.Info("正在处理 =>",path,info.Name())				identify := strings.Replace(strings.Trim(strings.TrimPrefix(path, tempPath), "/"), "/", "-", -1)				if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, identify); !ok || err != nil {					identify = "import-" + identify				}				parentDoc := NewDocument()				parentDoc.MemberId = book.MemberId				parentDoc.BookId = book.BookId				parentDoc.Identify = identify				parentDoc.Version = time.Now().Unix()				parentDoc.DocumentName = "空白文档"				parentId := 0				parentIdentify := strings.TrimSuffix(identify, "-"+info.Name())				if id, ok := docMap[parentIdentify]; ok {					parentId = id				}				parentDoc.ParentId = parentId				if err := parentDoc.InsertOrUpdate(); err != nil {					beego.Error(err)				}				docMap[identify] = parentDoc.DocumentId				//beego.Info(path,"|",parentDoc.DocumentId,"|",identify,"|",info.Name(),"|",parentIdentify)			}		}		return nil	})	if err != nil {		beego.Error("导入项目异常 => ", err)		book.Description = "【项目导入存在错误:" + err.Error() + "】"	}	beego.Info("项目导入完毕 => ",book.BookName)	book.ReleaseContent(book.BookId)	return err}
 |