From be7a64e03e426247e7ccd1220d139cf786990259 Mon Sep 17 00:00:00 2001
From: Minho <longfei6671@163.com>
Date: Tue, 27 Mar 2018 01:45:34 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=AF=BC=E5=87=BAMarkdown?=
 =?UTF-8?q?=E5=86=85=E9=93=BE=E8=B7=AF=E5=BE=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/book_result.go | 67 +++++++++++++++++++++++++++++++++++++++----
 1 file changed, 61 insertions(+), 6 deletions(-)

diff --git a/models/book_result.go b/models/book_result.go
index ef42c508..2b25f618 100644
--- a/models/book_result.go
+++ b/models/book_result.go
@@ -400,12 +400,15 @@ func (m *BookResult) ExportMarkdown(sessionId string) (string, error) {
 
 	defer os.RemoveAll(tempOutputPath)
 
-	err := exportMarkdown(tempOutputPath, 0, m.BookId,tempOutputPath)
+	bookUrl := conf.URLFor("DocumentController.Index",":key" , m.Identify) + "/"
+
+	err := exportMarkdown(tempOutputPath, 0, m.BookId,tempOutputPath,bookUrl)
 
 	if err != nil {
 		return "", err
 	}
 
+
 	if err := ziptil.Compress(outputPath, tempOutputPath); err != nil {
 		beego.Error("导出Markdown失败=>", err)
 		return "", err
@@ -413,7 +416,8 @@ func (m *BookResult) ExportMarkdown(sessionId string) (string, error) {
 	return outputPath, nil
 }
 
-func exportMarkdown(p string, parentId int, bookId int,baseDir string) error {
+//递归导出Markdown文档
+func exportMarkdown(p string, parentId int, bookId int,baseDir string,bookUrl string) error {
 	o := orm.NewOrm()
 
 	var docs []*Document
@@ -443,9 +447,13 @@ func exportMarkdown(p string, parentId int, bookId int,baseDir string) error {
 			}
 		} else {
 			if doc.Identify != "" {
-				docPath = filepath.Join(p, doc.Identify+".md")
+				if strings.HasSuffix(doc.Identify,".md") || strings.HasSuffix(doc.Identify,".markdown") {
+					docPath = filepath.Join(p, doc.Identify)
+				}else {
+					docPath = filepath.Join(p, doc.Identify+".md")
+				}
 			} else {
-				docPath = filepath.Join(p, doc.DocumentName+".md")
+				docPath = filepath.Join(p, strings.TrimSpace(doc.DocumentName)+".md")
 			}
 		}
 		dirPath := filepath.Dir(docPath)
@@ -493,7 +501,31 @@ func exportMarkdown(p string, parentId int, bookId int,baseDir string) error {
 				if len(links) > 0 && len(links[0]) >= 3 {
 					originalLink := links[0][2]
 
-					link = strings.TrimSuffix(link, originalLink+")") + strings.TrimPrefix(originalLink, conf.BaseUrl) + ")"
+					//如果当前链接位于当前项目内
+					if strings.HasPrefix(originalLink,bookUrl) {
+						docIdentify := strings.TrimSpace(strings.TrimPrefix(originalLink, bookUrl))
+						tempDoc := NewDocument()
+						if id,err := strconv.Atoi(docIdentify);err == nil && id > 0 {
+							err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("document_id",id).One(tempDoc,"identify","parent_id","document_id")
+							if err != nil {
+								beego.Error(err)
+								return link
+							}
+						}else{
+							err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("identify",docIdentify).One(tempDoc,"identify","parent_id","document_id")
+							if err != nil {
+								beego.Error(err)
+								return link
+							}
+						}
+						tempLink := recursiveJoinDocumentIdentify(tempDoc.ParentId,"") + strings.TrimPrefix(originalLink, bookUrl)
+
+						if !strings.HasSuffix(tempLink,".md") && !strings.HasSuffix(doc.Identify,".markdown") {
+							tempLink = tempLink + ".md"
+						}
+
+						link = strings.TrimSuffix(link, originalLink+")") + tempLink + ")"
+					}
 				}
 
 				return link
@@ -508,7 +540,7 @@ func exportMarkdown(p string, parentId int, bookId int,baseDir string) error {
 		}
 
 		if subDocCount > 0 {
-			if err = exportMarkdown(dirPath, doc.DocumentId, bookId,baseDir); err != nil {
+			if err = exportMarkdown(dirPath, doc.DocumentId, bookId,baseDir,bookUrl); err != nil {
 				return err
 			}
 		}
@@ -516,6 +548,29 @@ func exportMarkdown(p string, parentId int, bookId int,baseDir string) error {
 	return nil
 }
 
+func recursiveJoinDocumentIdentify(parentDocId int,identify string) string {
+	o := orm.NewOrm()
+
+	doc := NewDocument()
+
+	err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("document_id",parentDocId).One(doc,"identify","parent_id","document_id")
+
+	if err != nil {
+		beego.Error(err)
+		return identify
+	}
+
+	if doc.Identify == "" {
+		identify = strconv.Itoa(doc.DocumentId) + "/" + identify
+	}else{
+		identify = doc.Identify + "/" + identify
+	}
+	if doc.ParentId > 0 {
+		identify = recursiveJoinDocumentIdentify(doc.ParentId,identify)
+	}
+	return identify
+}
+
 //查询项目的第一篇文档
 func (m *BookResult) FindFirstDocumentByBookId(bookId int) (*Document, error) {