1
0

BookResult.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678
  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.GetExportLimitNum(),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) error {
  177. if err := converter.CheckConvertCommand(); err != nil {
  178. beego.Error("检查转换程序失败 -> ",err)
  179. return err
  180. }
  181. err := exportLimitWorkerChannel.LoadOrStore(bookResult.Identify, func() {
  182. bookResult.Converter(sessionId)
  183. })
  184. if err != nil {
  185. beego.Error("将导出任务加入任务队列失败 -> ",err)
  186. return err
  187. }
  188. exportLimitWorkerChannel.Start()
  189. return nil
  190. }
  191. //导出PDF、word等格式
  192. func (m *BookResult) Converter(sessionId string) (ConvertBookResult, error) {
  193. convertBookResult := ConvertBookResult{}
  194. outputPath := filepath.Join(conf.GetExportOutputPath(), strconv.Itoa(m.BookId))
  195. viewPath := beego.BConfig.WebConfig.ViewsPath
  196. pdfpath := filepath.Join(outputPath, "book.pdf")
  197. epubpath := filepath.Join(outputPath, "book.epub")
  198. mobipath := filepath.Join(outputPath, "book.mobi")
  199. docxpath := filepath.Join(outputPath, "book.docx")
  200. //先将转换的文件储存到临时目录
  201. tempOutputPath := filepath.Join(os.TempDir(), sessionId, m.Identify,"source") //filepath.Abs(filepath.Join("cache", sessionId))
  202. if err := os.MkdirAll(outputPath, 0766); err != nil {
  203. beego.Error("创建目录失败 => ",outputPath,err)
  204. }
  205. if err := os.MkdirAll(tempOutputPath, 0766);err != nil {
  206. beego.Error("创建目录失败 => ",tempOutputPath,err)
  207. }
  208. //defer os.RemoveAll(strings.TrimSuffix(tempOutputPath,"source"))
  209. sourceDir := strings.TrimSuffix(tempOutputPath,"source");
  210. if filetil.FileExists(sourceDir) {
  211. if err := os.MkdirAll(sourceDir,0755); err != nil {
  212. beego.Error("删除临时目录失败 ->", sourceDir , err)
  213. }
  214. }
  215. if filetil.FileExists(pdfpath) && filetil.FileExists(epubpath) && filetil.FileExists(mobipath) && filetil.FileExists(docxpath) {
  216. convertBookResult.EpubPath = epubpath
  217. convertBookResult.MobiPath = mobipath
  218. convertBookResult.PDFPath = pdfpath
  219. convertBookResult.WordPath = docxpath
  220. return convertBookResult, nil
  221. }
  222. docs, err := NewDocument().FindListByBookId(m.BookId)
  223. if err != nil {
  224. return convertBookResult, err
  225. }
  226. tocList := make([]converter.Toc, 0)
  227. for _, item := range docs {
  228. if item.ParentId == 0 {
  229. toc := converter.Toc{
  230. Id: item.DocumentId,
  231. Link: strconv.Itoa(item.DocumentId) + ".html",
  232. Pid: item.ParentId,
  233. Title: item.DocumentName,
  234. }
  235. tocList = append(tocList, toc)
  236. }
  237. }
  238. for _, item := range docs {
  239. if item.ParentId != 0 {
  240. toc := converter.Toc{
  241. Id: item.DocumentId,
  242. Link: strconv.Itoa(item.DocumentId) + ".html",
  243. Pid: item.ParentId,
  244. Title: item.DocumentName,
  245. }
  246. tocList = append(tocList, toc)
  247. }
  248. }
  249. ebookConfig := converter.Config{
  250. Charset: "utf-8",
  251. Cover: m.Cover,
  252. Timestamp: time.Now().Format("2006-01-02 15:04:05"),
  253. Description: string(blackfriday.Run([]byte(m.Description))),
  254. 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>",
  255. Header: "<p style='color:#8E8E8E;font-size:12px;'>_SECTION_</p>",
  256. Identifier: "",
  257. Language: "zh-CN",
  258. Creator: m.CreateName,
  259. Publisher: m.Publisher,
  260. Contributor: m.Publisher,
  261. Title: m.BookName,
  262. Format: []string{"epub", "mobi", "pdf", "docx"},
  263. FontSize: "14",
  264. PaperSize: "a4",
  265. MarginLeft: "72",
  266. MarginRight: "72",
  267. MarginTop: "72",
  268. MarginBottom: "72",
  269. Toc: tocList,
  270. More: []string{},
  271. }
  272. if m.Publisher != "" {
  273. 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>"
  274. }
  275. if m.RealName != "" {
  276. ebookConfig.Creator = m.RealName
  277. }
  278. if tempOutputPath, err = filepath.Abs(tempOutputPath); err != nil {
  279. beego.Error("导出目录配置错误:" + err.Error())
  280. return convertBookResult, err
  281. }
  282. for _, item := range docs {
  283. name := strconv.Itoa(item.DocumentId)
  284. fpath := filepath.Join(tempOutputPath, name+".html")
  285. f, err := os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0777)
  286. if err != nil {
  287. return convertBookResult, err
  288. }
  289. var buf bytes.Buffer
  290. if err := beego.ExecuteViewPathTemplate(&buf, "document/export.tpl", viewPath, map[string]interface{}{"Model": m, "Lists": item, "BaseUrl": conf.BaseUrl}); err != nil {
  291. return convertBookResult, err
  292. }
  293. html := buf.String()
  294. if err != nil {
  295. f.Close()
  296. return convertBookResult, err
  297. }
  298. bufio := bytes.NewReader(buf.Bytes())
  299. doc, err := goquery.NewDocumentFromReader(bufio)
  300. doc.Find("img").Each(func(i int, contentSelection *goquery.Selection) {
  301. if src, ok := contentSelection.Attr("src"); ok {
  302. var encodeString string
  303. //如果是本地路径则直接读取文件内容
  304. if strings.HasPrefix(src, "/") {
  305. spath := filepath.Join(conf.WorkingDirectory, src)
  306. if ff, e := ioutil.ReadFile(spath); e == nil {
  307. encodeString = base64.StdEncoding.EncodeToString(ff)
  308. }else{
  309. return
  310. }
  311. }else{
  312. client := &http.Client{}
  313. if req,err := http.NewRequest("GET",src,nil); err == nil {
  314. 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")
  315. req.Header.Set("Referer",src)
  316. //10秒连接超时时间
  317. client.Timeout = time.Second * 10
  318. if resp ,err := client.Do(req);err == nil {
  319. defer resp.Body.Close()
  320. if body, err := ioutil.ReadAll(resp.Body);err == nil {
  321. encodeString = base64.StdEncoding.EncodeToString(body)
  322. }else{
  323. return
  324. }
  325. }else{
  326. return
  327. }
  328. }else{
  329. return
  330. }
  331. }
  332. src = "data:image/" + filepath.Ext(src) + ";base64," + encodeString
  333. contentSelection.SetAttr("src", src)
  334. }
  335. })
  336. html, err = doc.Html()
  337. if err != nil {
  338. f.Close()
  339. return convertBookResult, err
  340. }
  341. f.WriteString(html)
  342. f.Close()
  343. }
  344. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "css", "kancloud.css"), filepath.Join(tempOutputPath, "styles", "css", "kancloud.css")); err != nil {
  345. beego.Error("复制CSS样式出错 => static/css/kancloud.css",err)
  346. }
  347. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "css", "export.css"), filepath.Join(tempOutputPath, "styles", "css", "export.css"));err != nil {
  348. beego.Error("复制CSS样式出错 => static/css/export.css",err)
  349. }
  350. 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 {
  351. beego.Error("复制CSS样式出错 => static/editor.md/css/editormd.preview.css",err)
  352. }
  353. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "prettify", "themes", "prettify.css"), filepath.Join(tempOutputPath, "styles", "prettify", "themes", "prettify.css")); err != nil {
  354. beego.Error("复制CSS样式出错 => static/prettify/themes/prettify.css",err)
  355. }
  356. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "css", "markdown.preview.css"), filepath.Join(tempOutputPath, "styles", "css", "markdown.preview.css"));err != nil {
  357. beego.Error("复制CSS样式出错 => static/css/markdown.preview.css",err)
  358. }
  359. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "highlight", "styles", "vs.css"), filepath.Join(tempOutputPath, "styles", "highlight", "styles", "vs.css")); err != nil {
  360. beego.Error("复制CSS样式出错 => static/highlight/styles/vs.css",err)
  361. }
  362. if err := filetil.CopyFile(filepath.Join(conf.WorkingDirectory, "static", "katex", "katex.min.css"), filepath.Join(tempOutputPath, "styles", "katex", "katex.min.css")); err != nil {
  363. beego.Error("复制CSS样式出错 => static/katex/katex.min.css",err)
  364. }
  365. eBookConverter := &converter.Converter{
  366. BasePath: tempOutputPath,
  367. OutputPath: filepath.Join(strings.TrimSuffix(tempOutputPath, "source"),"output"),
  368. Config: ebookConfig,
  369. Debug: true,
  370. ProcessNum: conf.GetExportProcessNum(),
  371. }
  372. os.MkdirAll(eBookConverter.OutputPath,0766)
  373. if err := eBookConverter.Convert(); err != nil {
  374. beego.Error("转换文件错误:" + m.BookName + " => " + err.Error())
  375. return convertBookResult, err
  376. }
  377. beego.Info("文档转换完成:" + m.BookName)
  378. if err := filetil.CopyFile(filepath.Join(eBookConverter.OutputPath,"output", "book.mobi"),mobipath,);err != nil {
  379. beego.Error("复制文档失败 => ",filepath.Join(eBookConverter.OutputPath,"output", "book.mobi"),err)
  380. }
  381. if err := filetil.CopyFile(filepath.Join(eBookConverter.OutputPath,"output", "book.pdf"),pdfpath);err != nil {
  382. beego.Error("复制文档失败 => ",filepath.Join(eBookConverter.OutputPath,"output", "book.pdf"),err)
  383. }
  384. if err := filetil.CopyFile(filepath.Join(eBookConverter.OutputPath,"output", "book.epub"),epubpath); err != nil{
  385. beego.Error("复制文档失败 => ",filepath.Join(eBookConverter.OutputPath,"output", "book.epub"),err)
  386. }
  387. if err := filetil.CopyFile(filepath.Join(eBookConverter.OutputPath,"output", "book.docx"),docxpath); err != nil {
  388. beego.Error("复制文档失败 => ",filepath.Join(eBookConverter.OutputPath,"output", "book.docx"),err)
  389. }
  390. convertBookResult.MobiPath = mobipath
  391. convertBookResult.PDFPath = pdfpath
  392. convertBookResult.EpubPath = epubpath
  393. convertBookResult.WordPath = docxpath
  394. return convertBookResult, nil
  395. }
  396. //导出Markdown原始文件
  397. func (m *BookResult) ExportMarkdown(sessionId string) (string, error) {
  398. outputPath := filepath.Join(conf.WorkingDirectory, "uploads", "books", strconv.Itoa(m.BookId), "book.zip")
  399. os.MkdirAll(filepath.Dir(outputPath), 0644)
  400. tempOutputPath := filepath.Join(os.TempDir(), sessionId, "markdown")
  401. defer os.RemoveAll(tempOutputPath)
  402. bookUrl := conf.URLFor("DocumentController.Index",":key" , m.Identify) + "/"
  403. err := exportMarkdown(tempOutputPath, 0, m.BookId,tempOutputPath,bookUrl)
  404. if err != nil {
  405. return "", err
  406. }
  407. if err := ziptil.Compress(outputPath, tempOutputPath); err != nil {
  408. beego.Error("导出Markdown失败=>", err)
  409. return "", err
  410. }
  411. return outputPath, nil
  412. }
  413. //递归导出Markdown文档
  414. func exportMarkdown(p string, parentId int, bookId int,baseDir string,bookUrl string) error {
  415. o := orm.NewOrm()
  416. var docs []*Document
  417. _, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", bookId).Filter("parent_id", parentId).All(&docs)
  418. if err != nil {
  419. beego.Error("导出Markdown失败=>", err)
  420. return err
  421. }
  422. for _, doc := range docs {
  423. //获取当前文档的子文档数量,如果数量不为0,则将当前文档命名为READMD.md并设置成目录。
  424. subDocCount, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("parent_id", doc.DocumentId).Count()
  425. if err != nil {
  426. beego.Error("导出Markdown失败=>", err)
  427. return err
  428. }
  429. var docPath string
  430. if subDocCount > 0 {
  431. if doc.Identify != "" {
  432. docPath = filepath.Join(p, doc.Identify, "README.md")
  433. } else {
  434. docPath = filepath.Join(p, strconv.Itoa(doc.DocumentId), "README.md")
  435. }
  436. } else {
  437. if doc.Identify != "" {
  438. if strings.HasSuffix(doc.Identify,".md") || strings.HasSuffix(doc.Identify,".markdown") {
  439. docPath = filepath.Join(p, doc.Identify)
  440. }else {
  441. docPath = filepath.Join(p, doc.Identify+".md")
  442. }
  443. } else {
  444. docPath = filepath.Join(p, strings.TrimSpace(doc.DocumentName)+".md")
  445. }
  446. }
  447. dirPath := filepath.Dir(docPath)
  448. os.MkdirAll(dirPath, 0766)
  449. markdown := doc.Markdown
  450. //如果当前文档不为空
  451. if strings.TrimSpace(doc.Markdown) != "" {
  452. re := regexp.MustCompile(`!\[(.*?)\]\((.*?)\)`)
  453. //处理文档中图片
  454. markdown = re.ReplaceAllStringFunc(doc.Markdown, func(image string) string {
  455. images := re.FindAllSubmatch([]byte(image), -1)
  456. if len(images) <= 0 || len(images[0]) < 3 {
  457. return image
  458. }
  459. originalImageUrl := string(images[0][2])
  460. imageUrl := strings.Replace(string(originalImageUrl), "\\", "/", -1)
  461. //如果是本地路径,则需要将图片复制到项目目录
  462. if strings.HasPrefix(imageUrl, "http://") || strings.HasPrefix(imageUrl, "https://") {
  463. imageExt := cryptil.Md5Crypt(imageUrl) + filepath.Ext(imageUrl)
  464. dstFile := filepath.Join(baseDir, "uploads", time.Now().Format("200601"), imageExt)
  465. if err := requests.DownloadAndSaveFile(imageUrl, dstFile); err == nil {
  466. imageUrl = strings.TrimPrefix(strings.Replace(dstFile, "\\", "/", -1), strings.Replace(baseDir, "\\", "/", -1))
  467. if !strings.HasPrefix(imageUrl, "/") && !strings.HasPrefix(imageUrl, "\\") {
  468. imageUrl = "/" + imageUrl
  469. }
  470. }
  471. } else if strings.HasPrefix(imageUrl, "/") {
  472. filetil.CopyFile(filepath.Join(conf.WorkingDirectory, imageUrl), filepath.Join(baseDir, imageUrl))
  473. }
  474. imageUrl = strings.Replace(strings.TrimSuffix(image, originalImageUrl+")")+imageUrl+")", "\\", "/", -1)
  475. return imageUrl
  476. })
  477. linkRe := regexp.MustCompile(`\[(.*?)\]\((.*?)\)`)
  478. markdown = linkRe.ReplaceAllStringFunc(markdown, func(link string) string {
  479. links := linkRe.FindAllStringSubmatch(link, -1)
  480. if len(links) > 0 && len(links[0]) >= 3 {
  481. originalLink := links[0][2]
  482. //如果当前链接位于当前项目内
  483. if strings.HasPrefix(originalLink,bookUrl) {
  484. docIdentify := strings.TrimSpace(strings.TrimPrefix(originalLink, bookUrl))
  485. tempDoc := NewDocument()
  486. if id,err := strconv.Atoi(docIdentify);err == nil && id > 0 {
  487. err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("document_id",id).One(tempDoc,"identify","parent_id","document_id")
  488. if err != nil {
  489. beego.Error(err)
  490. return link
  491. }
  492. }else{
  493. err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("identify",docIdentify).One(tempDoc,"identify","parent_id","document_id")
  494. if err != nil {
  495. beego.Error(err)
  496. return link
  497. }
  498. }
  499. tempLink := recursiveJoinDocumentIdentify(tempDoc.ParentId,"") + strings.TrimPrefix(originalLink, bookUrl)
  500. if !strings.HasSuffix(tempLink,".md") && !strings.HasSuffix(doc.Identify,".markdown") {
  501. tempLink = tempLink + ".md"
  502. }
  503. relative := strings.TrimPrefix(strings.Replace(p,"\\","/",-1),strings.Replace(baseDir,"\\","/",-1))
  504. repeat := 0
  505. if relative != "" {
  506. relative = strings.TrimSuffix(strings.TrimPrefix(relative,"/"),"/")
  507. repeat = strings.Count(relative,"/") + 1
  508. }
  509. beego.Info(repeat,"|",relative,"|",p,"|",baseDir)
  510. tempLink = strings.Repeat("../",repeat) + tempLink
  511. link = strings.TrimSuffix(link, originalLink+")") + tempLink + ")"
  512. }
  513. }
  514. return link
  515. })
  516. }else{
  517. markdown = "# " + doc.DocumentName + "\n"
  518. }
  519. if err := ioutil.WriteFile(docPath, []byte(markdown), 0644); err != nil {
  520. beego.Error("导出Markdown失败=>", err)
  521. return err
  522. }
  523. if subDocCount > 0 {
  524. if err = exportMarkdown(dirPath, doc.DocumentId, bookId,baseDir,bookUrl); err != nil {
  525. return err
  526. }
  527. }
  528. }
  529. return nil
  530. }
  531. func recursiveJoinDocumentIdentify(parentDocId int,identify string) string {
  532. o := orm.NewOrm()
  533. doc := NewDocument()
  534. err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("document_id",parentDocId).One(doc,"identify","parent_id","document_id")
  535. if err != nil {
  536. beego.Error(err)
  537. return identify
  538. }
  539. if doc.Identify == "" {
  540. identify = strconv.Itoa(doc.DocumentId) + "/" + identify
  541. }else{
  542. identify = doc.Identify + "/" + identify
  543. }
  544. if doc.ParentId > 0 {
  545. identify = recursiveJoinDocumentIdentify(doc.ParentId,identify)
  546. }
  547. return identify
  548. }
  549. //查询项目的第一篇文档
  550. func (m *BookResult) FindFirstDocumentByBookId(bookId int) (*Document, error) {
  551. o := orm.NewOrm()
  552. doc := NewDocument()
  553. err := o.QueryTable(doc.TableNameWithPrefix()).Filter("book_id", bookId).Filter("parent_id", 0).OrderBy("order_sort").One(doc)
  554. return doc, err
  555. }