BookResult.go 22 KB

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