BookResult.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668
  1. package models
  2. import (
  3. "bytes"
  4. "io/ioutil"
  5. "os"
  6. "path/filepath"
  7. "strconv"
  8. "strings"
  9. "time"
  10. "github.com/PuerkitoBio/goquery"
  11. "github.com/astaxie/beego"
  12. "github.com/astaxie/beego/logs"
  13. "github.com/astaxie/beego/orm"
  14. "github.com/lifei6671/mindoc/conf"
  15. "github.com/lifei6671/mindoc/converter"
  16. "github.com/lifei6671/mindoc/utils/filetil"
  17. "github.com/lifei6671/mindoc/utils/ziptil"
  18. "gopkg.in/russross/blackfriday.v2"
  19. "regexp"
  20. "github.com/lifei6671/mindoc/utils/cryptil"
  21. "github.com/lifei6671/mindoc/utils/requests"
  22. "github.com/lifei6671/mindoc/utils/gopool"
  23. "encoding/base64"
  24. "net/http"
  25. )
  26. var(
  27. exportLimitWorkerChannel = gopool.NewChannelPool(conf.GetExportProcessNum(),conf.GetExportQueueLimitNum())
  28. )
  29. type BookResult struct {
  30. BookId int `json:"book_id"`
  31. BookName string `json:"book_name"`
  32. Identify string `json:"identify"`
  33. OrderIndex int `json:"order_index"`
  34. Description string `json:"description"`
  35. Publisher string `json:"publisher"`
  36. PrivatelyOwned int `json:"privately_owned"`
  37. PrivateToken string `json:"private_token"`
  38. DocCount int `json:"doc_count"`
  39. CommentStatus string `json:"comment_status"`
  40. CommentCount int `json:"comment_count"`
  41. CreateTime time.Time `json:"create_time"`
  42. CreateName string `json:"create_name"`
  43. RealName string `json:"real_name"`
  44. ModifyTime time.Time `json:"modify_time"`
  45. Cover string `json:"cover"`
  46. Theme string `json:"theme"`
  47. Label string `json:"label"`
  48. MemberId int `json:"member_id"`
  49. Editor string `json:"editor"`
  50. AutoRelease bool `json:"auto_release"`
  51. HistoryCount int `json:"history_count"`
  52. RelationshipId int `json:"relationship_id"`
  53. RoleId int `json:"role_id"`
  54. RoleName string `json:"role_name"`
  55. Status int `json:"status"`
  56. IsEnableShare bool `json:"is_enable_share"`
  57. IsUseFirstDocument bool `json:"is_use_first_document"`
  58. LastModifyText string `json:"last_modify_text"`
  59. IsDisplayComment bool `json:"is_display_comment"`
  60. IsDownload bool `json:"is_download"`
  61. }
  62. func NewBookResult() *BookResult {
  63. return &BookResult{}
  64. }
  65. // 根据项目标识查询项目以及指定用户权限的信息.
  66. func (m *BookResult) FindByIdentify(identify string, memberId int) (*BookResult, error) {
  67. if identify == "" || memberId <= 0 {
  68. return m, ErrInvalidParameter
  69. }
  70. o := orm.NewOrm()
  71. book := NewBook()
  72. err := o.QueryTable(book.TableNameWithPrefix()).Filter("identify", identify).One(book)
  73. if err != nil {
  74. return m, err
  75. }
  76. relationship := NewRelationship()
  77. err = o.QueryTable(relationship.TableNameWithPrefix()).Filter("book_id", book.BookId).Filter("member_id", memberId).One(relationship)
  78. if err != nil {
  79. return m, err
  80. }
  81. var relationship2 Relationship
  82. err = o.QueryTable(relationship.TableNameWithPrefix()).Filter("book_id", book.BookId).Filter("role_id", 0).One(&relationship2)
  83. if err != nil {
  84. logs.Error("根据项目标识查询项目以及指定用户权限的信息 => ", err)
  85. return m, ErrPermissionDenied
  86. }
  87. member, err := NewMember().Find(relationship2.MemberId)
  88. if err != nil {
  89. return m, err
  90. }
  91. m = NewBookResult().ToBookResult(*book)
  92. m.CreateName = member.Account
  93. if member.RealName != "" {
  94. m.RealName = member.RealName
  95. }
  96. m.MemberId = relationship.MemberId
  97. m.RoleId = relationship.RoleId
  98. m.RelationshipId = relationship.RelationshipId
  99. if m.RoleId == conf.BookFounder {
  100. m.RoleName = "创始人"
  101. } else if m.RoleId == conf.BookAdmin {
  102. m.RoleName = "管理员"
  103. } else if m.RoleId == conf.BookEditor {
  104. m.RoleName = "编辑者"
  105. } else if m.RoleId == conf.BookObserver {
  106. m.RoleName = "观察者"
  107. }
  108. doc := NewDocument()
  109. err = o.QueryTable(doc.TableNameWithPrefix()).Filter("book_id", book.BookId).OrderBy("modify_time").One(doc)
  110. if err == nil {
  111. member2 := NewMember()
  112. member2.Find(doc.ModifyAt)
  113. m.LastModifyText = member2.Account + " 于 " + doc.ModifyTime.Local().Format("2006-01-02 15:04:05")
  114. }
  115. return m, nil
  116. }
  117. func (m *BookResult) FindToPager(pageIndex, pageSize int) (books []*BookResult, totalCount int, err error) {
  118. o := orm.NewOrm()
  119. count, err := o.QueryTable(NewBook().TableNameWithPrefix()).Count()
  120. if err != nil {
  121. return
  122. }
  123. totalCount = int(count)
  124. sql := `SELECT
  125. book.*,rel.relationship_id,rel.role_id,m.account AS create_name,m.real_name
  126. FROM md_books AS book
  127. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.role_id = 0
  128. LEFT JOIN md_members AS m ON rel.member_id = m.member_id
  129. ORDER BY book.order_index DESC ,book.book_id DESC LIMIT ?,?`
  130. offset := (pageIndex - 1) * pageSize
  131. _, err = o.Raw(sql, offset, pageSize).QueryRows(&books)
  132. return
  133. }
  134. //实体转换
  135. func (m *BookResult) ToBookResult(book Book) *BookResult {
  136. m.BookId = book.BookId
  137. m.BookName = book.BookName
  138. m.Identify = book.Identify
  139. m.OrderIndex = book.OrderIndex
  140. m.Description = strings.Replace(book.Description, "\r\n", "<br/>", -1)
  141. m.PrivatelyOwned = book.PrivatelyOwned
  142. m.PrivateToken = book.PrivateToken
  143. m.DocCount = book.DocCount
  144. m.CommentStatus = book.CommentStatus
  145. m.CommentCount = book.CommentCount
  146. m.CreateTime = book.CreateTime
  147. m.ModifyTime = book.ModifyTime
  148. m.Cover = book.Cover
  149. m.Label = book.Label
  150. m.Status = book.Status
  151. m.Editor = book.Editor
  152. m.Theme = book.Theme
  153. m.AutoRelease = book.AutoRelease == 1
  154. m.IsEnableShare = book.IsEnableShare == 0
  155. m.IsUseFirstDocument = book.IsUseFirstDocument == 1
  156. m.Publisher = book.Publisher
  157. m.HistoryCount = book.HistoryCount
  158. m.IsDownload = book.IsDownload == 0
  159. if book.Theme == "" {
  160. m.Theme = "default"
  161. }
  162. if book.Editor == "" {
  163. m.Editor = "markdown"
  164. }
  165. doc := NewDocument()
  166. o := orm.NewOrm()
  167. err := o.QueryTable(doc.TableNameWithPrefix()).Filter("book_id", book.BookId).OrderBy("modify_time").One(doc)
  168. if err == nil {
  169. member2 := NewMember()
  170. member2.Find(doc.ModifyAt)
  171. m.LastModifyText = member2.Account + " 于 " + doc.ModifyTime.Local().Format("2006-01-02 15:04:05")
  172. }
  173. return m
  174. }
  175. //后台转换
  176. func BackgroupConvert(sessionId string,bookResult *BookResult){
  177. exportLimitWorkerChannel.LoadOrStore(bookResult.Identify, func() {
  178. bookResult.Converter(sessionId)
  179. })
  180. exportLimitWorkerChannel.Start()
  181. }
  182. //导出PDF、word等格式
  183. func (m *BookResult) Converter(sessionId string) (ConvertBookResult, error) {
  184. convertBookResult := ConvertBookResult{}
  185. outputPath := filepath.Join(conf.WorkingDirectory, "uploads", "books", strconv.Itoa(m.BookId))
  186. viewPath := beego.BConfig.WebConfig.ViewsPath
  187. pdfpath := filepath.Join(outputPath, "book.pdf")
  188. epubpath := filepath.Join(outputPath, "book.epub")
  189. mobipath := filepath.Join(outputPath, "book.mobi")
  190. docxpath := filepath.Join(outputPath, "book.docx")
  191. //先将转换的文件储存到临时目录
  192. tempOutputPath := filepath.Join(os.TempDir(), sessionId, m.Identify,"source") //filepath.Abs(filepath.Join("cache", sessionId))
  193. if err := os.MkdirAll(outputPath, 0766); err != nil {
  194. beego.Error("创建目录失败 => ",outputPath,err)
  195. }
  196. if err := os.MkdirAll(tempOutputPath, 0766);err != nil {
  197. beego.Error("创建目录失败 => ",tempOutputPath,err)
  198. }
  199. //defer os.RemoveAll(strings.TrimSuffix(tempOutputPath,"source"))
  200. sourceDir := strings.TrimSuffix(tempOutputPath,"source");
  201. if filetil.FileExists(sourceDir) {
  202. if err := os.MkdirAll(sourceDir,0755); err != nil {
  203. beego.Error("删除临时目录失败 ->", sourceDir , err)
  204. }
  205. }
  206. if filetil.FileExists(pdfpath) && filetil.FileExists(epubpath) && filetil.FileExists(mobipath) && filetil.FileExists(docxpath) {
  207. convertBookResult.EpubPath = epubpath
  208. convertBookResult.MobiPath = mobipath
  209. convertBookResult.PDFPath = pdfpath
  210. convertBookResult.WordPath = docxpath
  211. return convertBookResult, nil
  212. }
  213. docs, err := NewDocument().FindListByBookId(m.BookId)
  214. if err != nil {
  215. return convertBookResult, err
  216. }
  217. tocList := make([]converter.Toc, 0)
  218. for _, item := range docs {
  219. if item.ParentId == 0 {
  220. toc := converter.Toc{
  221. Id: item.DocumentId,
  222. Link: strconv.Itoa(item.DocumentId) + ".html",
  223. Pid: item.ParentId,
  224. Title: item.DocumentName,
  225. }
  226. tocList = append(tocList, toc)
  227. }
  228. }
  229. for _, item := range docs {
  230. if item.ParentId != 0 {
  231. toc := converter.Toc{
  232. Id: item.DocumentId,
  233. Link: strconv.Itoa(item.DocumentId) + ".html",
  234. Pid: item.ParentId,
  235. Title: item.DocumentName,
  236. }
  237. tocList = append(tocList, toc)
  238. }
  239. }
  240. ebookConfig := converter.Config{
  241. Charset: "utf-8",
  242. Cover: m.Cover,
  243. Timestamp: time.Now().Format("2006-01-02 15:04:05"),
  244. Description: string(blackfriday.Run([]byte(m.Description))),
  245. Footer: "<p style='color:#8E8E8E;font-size:12px;'>本文档使用 <a href='https://www.iminho.me' style='text-decoration:none;color:#1abc9c;font-weight:bold;'>MinDoc</a> 构建 <span style='float:right'>- _PAGENUM_ -</span></p>",
  246. Header: "<p style='color:#8E8E8E;font-size:12px;'>_SECTION_</p>",
  247. Identifier: "",
  248. Language: "zh-CN",
  249. Creator: m.CreateName,
  250. Publisher: m.Publisher,
  251. Contributor: m.Publisher,
  252. Title: m.BookName,
  253. Format: []string{"epub", "mobi", "pdf", "docx"},
  254. FontSize: "14",
  255. PaperSize: "a4",
  256. MarginLeft: "72",
  257. MarginRight: "72",
  258. MarginTop: "72",
  259. MarginBottom: "72",
  260. Toc: tocList,
  261. More: []string{},
  262. }
  263. if m.Publisher != "" {
  264. ebookConfig.Footer = "<p style='color:#8E8E8E;font-size:12px;'>本文档由 <span style='text-decoration:none;color:#1abc9c;font-weight:bold;'>" + m.Publisher + "</span> 生成<span style='float:right'>- _PAGENUM_ -</span></p>"
  265. }
  266. if m.RealName != "" {
  267. ebookConfig.Creator = m.RealName
  268. }
  269. if tempOutputPath, err = filepath.Abs(tempOutputPath); err != nil {
  270. beego.Error("导出目录配置错误:" + err.Error())
  271. return convertBookResult, err
  272. }
  273. for _, item := range docs {
  274. name := strconv.Itoa(item.DocumentId)
  275. fpath := filepath.Join(tempOutputPath, name+".html")
  276. f, err := os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0777)
  277. if err != nil {
  278. return convertBookResult, err
  279. }
  280. var buf bytes.Buffer
  281. if err := beego.ExecuteViewPathTemplate(&buf, "document/export.tpl", viewPath, map[string]interface{}{"Model": m, "Lists": item, "BaseUrl": conf.BaseUrl}); err != nil {
  282. return convertBookResult, err
  283. }
  284. html := buf.String()
  285. if err != nil {
  286. f.Close()
  287. return convertBookResult, err
  288. }
  289. bufio := bytes.NewReader(buf.Bytes())
  290. doc, err := goquery.NewDocumentFromReader(bufio)
  291. doc.Find("img").Each(func(i int, contentSelection *goquery.Selection) {
  292. if src, ok := contentSelection.Attr("src"); ok {
  293. var encodeString string
  294. //如果是本地路径则直接读取文件内容
  295. if strings.HasPrefix(src, "/") {
  296. spath := filepath.Join(conf.WorkingDirectory, src)
  297. if ff, e := ioutil.ReadFile(spath); e == nil {
  298. encodeString = base64.StdEncoding.EncodeToString(ff)
  299. }else{
  300. return
  301. }
  302. }else{
  303. client := &http.Client{}
  304. if req,err := http.NewRequest("GET",src,nil); err == nil {
  305. req.Header.Set("User-Agent","Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36")
  306. req.Header.Set("Referer",src)
  307. //10秒连接超时时间
  308. client.Timeout = time.Second * 10
  309. if resp ,err := client.Do(req);err == nil {
  310. defer resp.Body.Close()
  311. if body, err := ioutil.ReadAll(resp.Body);err == nil {
  312. encodeString = base64.StdEncoding.EncodeToString(body)
  313. }else{
  314. return
  315. }
  316. }else{
  317. return
  318. }
  319. }else{
  320. return
  321. }
  322. }
  323. src = "data:image/" + filepath.Ext(src) + ";base64," + encodeString
  324. contentSelection.SetAttr("src", src)
  325. }
  326. })
  327. html, err = doc.Html()
  328. if err != nil {
  329. f.Close()
  330. return convertBookResult, err
  331. }
  332. f.WriteString(html)
  333. f.Close()
  334. }
  335. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "css", "kancloud.css"), filepath.Join(tempOutputPath, "styles", "css", "kancloud.css")); err != nil {
  336. beego.Error("复制CSS样式出错 => static/css/kancloud.css",err)
  337. }
  338. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "css", "export.css"), filepath.Join(tempOutputPath, "styles", "css", "export.css"));err != nil {
  339. beego.Error("复制CSS样式出错 => static/css/export.css",err)
  340. }
  341. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "editor.md", "css", "editormd.preview.css"), filepath.Join(tempOutputPath, "styles", "editor.md", "css", "editormd.preview.css"));err != nil {
  342. beego.Error("复制CSS样式出错 => static/editor.md/css/editormd.preview.css",err)
  343. }
  344. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "prettify", "themes", "prettify.css"), filepath.Join(tempOutputPath, "styles", "prettify", "themes", "prettify.css")); err != nil {
  345. beego.Error("复制CSS样式出错 => static/prettify/themes/prettify.css",err)
  346. }
  347. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "css", "markdown.preview.css"), filepath.Join(tempOutputPath, "styles", "css", "markdown.preview.css"));err != nil {
  348. beego.Error("复制CSS样式出错 => static/css/markdown.preview.css",err)
  349. }
  350. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "highlight", "styles", "vs.css"), filepath.Join(tempOutputPath, "styles", "highlight", "styles", "vs.css")); err != nil {
  351. beego.Error("复制CSS样式出错 => static/highlight/styles/vs.css",err)
  352. }
  353. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "katex", "katex.min.css"), filepath.Join(tempOutputPath, "styles", "katex", "katex.min.css")); err != nil {
  354. beego.Error("复制CSS样式出错 => static/katex/katex.min.css",err)
  355. }
  356. eBookConverter := &converter.Converter{
  357. BasePath: tempOutputPath,
  358. OutputPath: filepath.Join(strings.TrimSuffix(tempOutputPath, "source"),"output"),
  359. Config: ebookConfig,
  360. Debug: true,
  361. ProcessNum: conf.GetExportProcessNum(),
  362. }
  363. os.MkdirAll(eBookConverter.OutputPath,0766)
  364. if err := eBookConverter.Convert(); err != nil {
  365. beego.Error("转换文件错误:" + m.BookName + " => " + err.Error())
  366. return convertBookResult, err
  367. }
  368. beego.Info("文档转换完成:" + m.BookName)
  369. if err := filetil.CopyFile(filepath.Join(eBookConverter.OutputPath,"output", "book.mobi"),mobipath,);err != nil {
  370. beego.Error("复制文档失败 => ",filepath.Join(eBookConverter.OutputPath,"output", "book.mobi"),err)
  371. }
  372. if err := filetil.CopyFile(filepath.Join(eBookConverter.OutputPath,"output", "book.pdf"),pdfpath);err != nil {
  373. beego.Error("复制文档失败 => ",filepath.Join(eBookConverter.OutputPath,"output", "book.pdf"),err)
  374. }
  375. if err := filetil.CopyFile(filepath.Join(eBookConverter.OutputPath,"output", "book.epub"),epubpath); err != nil{
  376. beego.Error("复制文档失败 => ",filepath.Join(eBookConverter.OutputPath,"output", "book.epub"),err)
  377. }
  378. if err := filetil.CopyFile(filepath.Join(eBookConverter.OutputPath,"output", "book.docx"),docxpath); err != nil {
  379. beego.Error("复制文档失败 => ",filepath.Join(eBookConverter.OutputPath,"output", "book.docx"),err)
  380. }
  381. convertBookResult.MobiPath = mobipath
  382. convertBookResult.PDFPath = pdfpath
  383. convertBookResult.EpubPath = epubpath
  384. convertBookResult.WordPath = docxpath
  385. return convertBookResult, nil
  386. }
  387. //导出Markdown原始文件
  388. func (m *BookResult) ExportMarkdown(sessionId string) (string, error) {
  389. outputPath := filepath.Join(conf.WorkingDirectory, "uploads", "books", strconv.Itoa(m.BookId), "book.zip")
  390. os.MkdirAll(filepath.Dir(outputPath), 0644)
  391. tempOutputPath := filepath.Join(os.TempDir(), sessionId, "markdown")
  392. defer os.RemoveAll(tempOutputPath)
  393. bookUrl := conf.URLFor("DocumentController.Index",":key" , m.Identify) + "/"
  394. err := exportMarkdown(tempOutputPath, 0, m.BookId,tempOutputPath,bookUrl)
  395. if err != nil {
  396. return "", err
  397. }
  398. if err := ziptil.Compress(outputPath, tempOutputPath); err != nil {
  399. beego.Error("导出Markdown失败=>", err)
  400. return "", err
  401. }
  402. return outputPath, nil
  403. }
  404. //递归导出Markdown文档
  405. func exportMarkdown(p string, parentId int, bookId int,baseDir string,bookUrl string) error {
  406. o := orm.NewOrm()
  407. var docs []*Document
  408. _, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", bookId).Filter("parent_id", parentId).All(&docs)
  409. if err != nil {
  410. beego.Error("导出Markdown失败=>", err)
  411. return err
  412. }
  413. for _, doc := range docs {
  414. //获取当前文档的子文档数量,如果数量不为0,则将当前文档命名为READMD.md并设置成目录。
  415. subDocCount, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("parent_id", doc.DocumentId).Count()
  416. if err != nil {
  417. beego.Error("导出Markdown失败=>", err)
  418. return err
  419. }
  420. var docPath string
  421. if subDocCount > 0 {
  422. if doc.Identify != "" {
  423. docPath = filepath.Join(p, doc.Identify, "README.md")
  424. } else {
  425. docPath = filepath.Join(p, strconv.Itoa(doc.DocumentId), "README.md")
  426. }
  427. } else {
  428. if doc.Identify != "" {
  429. if strings.HasSuffix(doc.Identify,".md") || strings.HasSuffix(doc.Identify,".markdown") {
  430. docPath = filepath.Join(p, doc.Identify)
  431. }else {
  432. docPath = filepath.Join(p, doc.Identify+".md")
  433. }
  434. } else {
  435. docPath = filepath.Join(p, strings.TrimSpace(doc.DocumentName)+".md")
  436. }
  437. }
  438. dirPath := filepath.Dir(docPath)
  439. os.MkdirAll(dirPath, 0766)
  440. markdown := doc.Markdown
  441. //如果当前文档不为空
  442. if strings.TrimSpace(doc.Markdown) != "" {
  443. re := regexp.MustCompile(`!\[(.*?)\]\((.*?)\)`)
  444. //处理文档中图片
  445. markdown = re.ReplaceAllStringFunc(doc.Markdown, func(image string) string {
  446. images := re.FindAllSubmatch([]byte(image), -1)
  447. if len(images) <= 0 || len(images[0]) < 3 {
  448. return image
  449. }
  450. originalImageUrl := string(images[0][2])
  451. imageUrl := strings.Replace(string(originalImageUrl), "\\", "/", -1)
  452. //如果是本地路径,则需要将图片复制到项目目录
  453. if strings.HasPrefix(imageUrl, "http://") || strings.HasPrefix(imageUrl, "https://") {
  454. imageExt := cryptil.Md5Crypt(imageUrl) + filepath.Ext(imageUrl)
  455. dstFile := filepath.Join(baseDir, "uploads", time.Now().Format("200601"), imageExt)
  456. if err := requests.DownloadAndSaveFile(imageUrl, dstFile); err == nil {
  457. imageUrl = strings.TrimPrefix(strings.Replace(dstFile, "\\", "/", -1), strings.Replace(baseDir, "\\", "/", -1))
  458. if !strings.HasPrefix(imageUrl, "/") && !strings.HasPrefix(imageUrl, "\\") {
  459. imageUrl = "/" + imageUrl
  460. }
  461. }
  462. } else if strings.HasPrefix(imageUrl, "/") {
  463. filetil.CopyFile(filepath.Join(conf.WorkingDirectory, imageUrl), filepath.Join(baseDir, imageUrl))
  464. }
  465. imageUrl = strings.Replace(strings.TrimSuffix(image, originalImageUrl+")")+imageUrl+")", "\\", "/", -1)
  466. return imageUrl
  467. })
  468. linkRe := regexp.MustCompile(`\[(.*?)\]\((.*?)\)`)
  469. markdown = linkRe.ReplaceAllStringFunc(markdown, func(link string) string {
  470. links := linkRe.FindAllStringSubmatch(link, -1)
  471. if len(links) > 0 && len(links[0]) >= 3 {
  472. originalLink := links[0][2]
  473. //如果当前链接位于当前项目内
  474. if strings.HasPrefix(originalLink,bookUrl) {
  475. docIdentify := strings.TrimSpace(strings.TrimPrefix(originalLink, bookUrl))
  476. tempDoc := NewDocument()
  477. if id,err := strconv.Atoi(docIdentify);err == nil && id > 0 {
  478. err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("document_id",id).One(tempDoc,"identify","parent_id","document_id")
  479. if err != nil {
  480. beego.Error(err)
  481. return link
  482. }
  483. }else{
  484. err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("identify",docIdentify).One(tempDoc,"identify","parent_id","document_id")
  485. if err != nil {
  486. beego.Error(err)
  487. return link
  488. }
  489. }
  490. tempLink := recursiveJoinDocumentIdentify(tempDoc.ParentId,"") + strings.TrimPrefix(originalLink, bookUrl)
  491. if !strings.HasSuffix(tempLink,".md") && !strings.HasSuffix(doc.Identify,".markdown") {
  492. tempLink = tempLink + ".md"
  493. }
  494. relative := strings.TrimPrefix(strings.Replace(p,"\\","/",-1),strings.Replace(baseDir,"\\","/",-1))
  495. repeat := 0
  496. if relative != "" {
  497. relative = strings.TrimSuffix(strings.TrimPrefix(relative,"/"),"/")
  498. repeat = strings.Count(relative,"/") + 1
  499. }
  500. beego.Info(repeat,"|",relative,"|",p,"|",baseDir)
  501. tempLink = strings.Repeat("../",repeat) + tempLink
  502. link = strings.TrimSuffix(link, originalLink+")") + tempLink + ")"
  503. }
  504. }
  505. return link
  506. })
  507. }else{
  508. markdown = "# " + doc.DocumentName + "\n"
  509. }
  510. if err := ioutil.WriteFile(docPath, []byte(markdown), 0644); err != nil {
  511. beego.Error("导出Markdown失败=>", err)
  512. return err
  513. }
  514. if subDocCount > 0 {
  515. if err = exportMarkdown(dirPath, doc.DocumentId, bookId,baseDir,bookUrl); err != nil {
  516. return err
  517. }
  518. }
  519. }
  520. return nil
  521. }
  522. func recursiveJoinDocumentIdentify(parentDocId int,identify string) string {
  523. o := orm.NewOrm()
  524. doc := NewDocument()
  525. err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("document_id",parentDocId).One(doc,"identify","parent_id","document_id")
  526. if err != nil {
  527. beego.Error(err)
  528. return identify
  529. }
  530. if doc.Identify == "" {
  531. identify = strconv.Itoa(doc.DocumentId) + "/" + identify
  532. }else{
  533. identify = doc.Identify + "/" + identify
  534. }
  535. if doc.ParentId > 0 {
  536. identify = recursiveJoinDocumentIdentify(doc.ParentId,identify)
  537. }
  538. return identify
  539. }
  540. //查询项目的第一篇文档
  541. func (m *BookResult) FindFirstDocumentByBookId(bookId int) (*Document, error) {
  542. o := orm.NewOrm()
  543. doc := NewDocument()
  544. err := o.QueryTable(doc.TableNameWithPrefix()).Filter("book_id", bookId).Filter("parent_id", 0).OrderBy("order_sort").One(doc)
  545. return doc, err
  546. }