BookModel.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852
  1. package models
  2. import (
  3. "crypto/md5"
  4. "errors"
  5. "fmt"
  6. "io"
  7. "io/ioutil"
  8. "os"
  9. "path/filepath"
  10. "regexp"
  11. "strconv"
  12. "strings"
  13. "time"
  14. "github.com/astaxie/beego"
  15. "github.com/astaxie/beego/logs"
  16. "github.com/astaxie/beego/orm"
  17. "github.com/lifei6671/mindoc/conf"
  18. "github.com/lifei6671/mindoc/utils/cryptil"
  19. "github.com/lifei6671/mindoc/utils/filetil"
  20. "github.com/lifei6671/mindoc/utils/requests"
  21. "github.com/lifei6671/mindoc/utils/ziptil"
  22. "gopkg.in/russross/blackfriday.v2"
  23. "encoding/json"
  24. "github.com/lifei6671/mindoc/utils"
  25. )
  26. // Book struct .
  27. type Book struct {
  28. BookId int `orm:"pk;auto;unique;column(book_id)" json:"book_id"`
  29. // BookName 项目名称.
  30. BookName string `orm:"column(book_name);size(500)" json:"book_name"`
  31. // Identify 项目唯一标识.
  32. Identify string `orm:"column(identify);size(100);unique" json:"identify"`
  33. //是否是自动发布 0 否/1 是
  34. AutoRelease int `orm:"column(auto_release);type(int);default(0)" json:"auto_release"`
  35. //是否开启下载功能 0 是/1 否
  36. IsDownload int `orm:"column(is_download);type(int);default(0)" json:"is_download"`
  37. OrderIndex int `orm:"column(order_index);type(int);default(0)" json:"order_index"`
  38. // Description 项目描述.
  39. Description string `orm:"column(description);size(2000)" json:"description"`
  40. //发行公司
  41. Publisher string `orm:"column(publisher);size(500)" json:"publisher"`
  42. Label string `orm:"column(label);size(500)" json:"label"`
  43. // PrivatelyOwned 项目私有: 0 公开/ 1 私有
  44. PrivatelyOwned int `orm:"column(privately_owned);type(int);default(0)" json:"privately_owned"`
  45. // 当项目是私有时的访问Token.
  46. PrivateToken string `orm:"column(private_token);size(500);null" json:"private_token"`
  47. //状态:0 正常/1 已删除
  48. Status int `orm:"column(status);type(int);default(0)" json:"status"`
  49. //默认的编辑器.
  50. Editor string `orm:"column(editor);size(50)" json:"editor"`
  51. // DocCount 包含文档数量.
  52. DocCount int `orm:"column(doc_count);type(int)" json:"doc_count"`
  53. // CommentStatus 评论设置的状态:open 为允许所有人评论,closed 为不允许评论, group_only 仅允许参与者评论 ,registered_only 仅允许注册者评论.
  54. CommentStatus string `orm:"column(comment_status);size(20);default(open)" json:"comment_status"`
  55. CommentCount int `orm:"column(comment_count);type(int)" json:"comment_count"`
  56. //封面地址
  57. Cover string `orm:"column(cover);size(1000)" json:"cover"`
  58. //主题风格
  59. Theme string `orm:"column(theme);size(255);default(default)" json:"theme"`
  60. // CreateTime 创建时间 .
  61. CreateTime time.Time `orm:"type(datetime);column(create_time);auto_now_add" json:"create_time"`
  62. //每个文档保存的历史记录数量,0 为不限制
  63. HistoryCount int `orm:"column(history_count);type(int);default(0)" json:"history_count"`
  64. //是否启用分享,0启用/1不启用
  65. IsEnableShare int `orm:"column(is_enable_share);type(int);default(0)" json:"is_enable_share"`
  66. MemberId int `orm:"column(member_id);size(100)" json:"member_id"`
  67. ModifyTime time.Time `orm:"type(datetime);column(modify_time);null;auto_now" json:"modify_time"`
  68. Version int64 `orm:"type(bigint);column(version)" json:"version"`
  69. //是否使用第一篇文章项目为默认首页,0 否/1 是
  70. IsUseFirstDocument int `orm:"column(is_use_first_document);type(int);default(0)" json:"is_use_first_document"`
  71. //是否开启自动保存:0 否/1 是
  72. AutoSave int `orm:"column(auto_save);type(tinyint);default(0)" json:"auto_save"`
  73. }
  74. func (book *Book) String() string {
  75. ret, err := json.Marshal(*book)
  76. if err != nil {
  77. return ""
  78. }
  79. return string(ret)
  80. }
  81. // TableName 获取对应数据库表名.
  82. func (book *Book) TableName() string {
  83. return "books"
  84. }
  85. // TableEngine 获取数据使用的引擎.
  86. func (book *Book) TableEngine() string {
  87. return "INNODB"
  88. }
  89. func (book *Book) TableNameWithPrefix() string {
  90. return conf.GetDatabasePrefix() + book.TableName()
  91. }
  92. func (book *Book) QueryTable() orm.QuerySeter {
  93. return orm.NewOrm().QueryTable(book.TableNameWithPrefix())
  94. }
  95. func NewBook() *Book {
  96. return &Book{}
  97. }
  98. //添加一个项目
  99. func (book *Book) Insert() error {
  100. o := orm.NewOrm()
  101. // o.Begin()
  102. book.BookName = utils.StripTags(book.BookName)
  103. _, err := o.Insert(book)
  104. if err == nil {
  105. if book.Label != "" {
  106. NewLabel().InsertOrUpdateMulti(book.Label)
  107. }
  108. relationship := NewRelationship()
  109. relationship.BookId = book.BookId
  110. relationship.RoleId = 0
  111. relationship.MemberId = book.MemberId
  112. err = relationship.Insert()
  113. if err != nil {
  114. logs.Error("插入项目与用户关联 => ", err)
  115. //o.Rollback()
  116. return err
  117. }
  118. document := NewDocument()
  119. document.BookId = book.BookId
  120. document.DocumentName = "空白文档"
  121. document.MemberId = book.MemberId
  122. err = document.InsertOrUpdate()
  123. if err != nil {
  124. //o.Rollback()
  125. return err
  126. }
  127. //o.Commit()
  128. return nil
  129. }
  130. //o.Rollback()
  131. return err
  132. }
  133. func (book *Book) Find(id int,cols ...string) (*Book, error) {
  134. if id <= 0 {
  135. return book, ErrInvalidParameter
  136. }
  137. o := orm.NewOrm()
  138. err := o.QueryTable(book.TableNameWithPrefix()).Filter("book_id", id).One(book,cols...)
  139. return book, err
  140. }
  141. //更新一个项目
  142. func (book *Book) Update(cols ...string) error {
  143. o := orm.NewOrm()
  144. book.BookName = utils.StripTags(book.BookName)
  145. temp := NewBook()
  146. temp.BookId = book.BookId
  147. if err := o.Read(temp); err != nil {
  148. return err
  149. }
  150. if book.Label != "" || temp.Label != "" {
  151. go NewLabel().InsertOrUpdateMulti(book.Label + "," + temp.Label)
  152. }
  153. _, err := o.Update(book, cols...)
  154. return err
  155. }
  156. //复制项目
  157. func (book *Book) Copy(identify string) error {
  158. o := orm.NewOrm()
  159. err := o.QueryTable(book.TableNameWithPrefix()).Filter("identify",identify).One(book)
  160. if err != nil {
  161. beego.Error("查询项目时出错 -> ",err)
  162. return err
  163. }
  164. if err := o.Begin();err != nil {
  165. beego.Error("开启事物时出错 -> ",err)
  166. return err
  167. }
  168. bookId := book.BookId
  169. book.BookId = 0
  170. book.Identify = book.Identify + fmt.Sprintf("%s-%s",identify,strconv.FormatInt(time.Now().UnixNano(), 32))
  171. book.BookName = book.BookName + "[副本]"
  172. book.CreateTime = time.Now()
  173. book.CommentCount = 0
  174. book.HistoryCount = 0
  175. if _,err := o.Insert(book);err != nil {
  176. beego.Error("复制项目时出错 -> ",err)
  177. o.Rollback()
  178. return err
  179. }
  180. var rels []*Relationship
  181. if _,err := o.QueryTable(NewRelationship().TableNameWithPrefix()).Filter("book_id",bookId).All(&rels); err != nil {
  182. beego.Error("复制项目关系时出错 -> ",err)
  183. o.Rollback()
  184. return err
  185. }
  186. for _,rel := range rels {
  187. rel.BookId = book.BookId
  188. rel.RelationshipId = 0
  189. if _,err := o.Insert(rel);err != nil {
  190. beego.Error("复制项目关系时出错 -> ",err)
  191. o.Rollback()
  192. return err
  193. }
  194. }
  195. var docs []*Document
  196. if _,err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id",bookId).Filter("parent_id",0).All(&docs);err != nil && err != orm.ErrNoRows {
  197. beego.Error("读取项目文档时出错 -> ",err)
  198. o.Rollback()
  199. return err
  200. }
  201. if len(docs) > 0 {
  202. if err := recursiveInsertDocument(docs,o, book.BookId,0);err != nil {
  203. beego.Error("复制项目时出错 -> ",err)
  204. o.Rollback()
  205. return err
  206. }
  207. }
  208. return o.Commit()
  209. }
  210. //递归的复制文档
  211. func recursiveInsertDocument(docs []*Document,o orm.Ormer,bookId int,parentId int) error {
  212. for _,doc := range docs {
  213. docId := doc.DocumentId
  214. doc.DocumentId = 0
  215. doc.ParentId = parentId
  216. doc.BookId = bookId
  217. doc.Version = time.Now().Unix()
  218. if _,err := o.Insert(doc);err != nil {
  219. beego.Error("插入项目时出错 -> ",err)
  220. return err
  221. }
  222. var attachList []*Attachment
  223. //读取所有附件列表
  224. if _,err := o.QueryTable(NewAttachment().TableNameWithPrefix()).Filter("document_id",docId).All(&attachList); err == nil {
  225. for _,attach := range attachList {
  226. attach.BookId = bookId
  227. attach.DocumentId = doc.DocumentId
  228. attach.AttachmentId = 0
  229. if _,err := o.Insert(attach);err != nil {
  230. return err
  231. }
  232. }
  233. }
  234. var subDocs []*Document
  235. if _,err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("parent_id",docId).All(&subDocs);err != nil && err != orm.ErrNoRows {
  236. beego.Error("读取文档时出错 -> ",err)
  237. return err
  238. }
  239. if len(subDocs) > 0{
  240. if err := recursiveInsertDocument(subDocs,o,bookId,doc.DocumentId);err != nil {
  241. return err
  242. }
  243. }
  244. }
  245. return nil
  246. }
  247. //根据指定字段查询结果集.
  248. func (book *Book) FindByField(field string, value interface{},cols ...string) ([]*Book, error) {
  249. o := orm.NewOrm()
  250. var books []*Book
  251. _, err := o.QueryTable(book.TableNameWithPrefix()).Filter(field, value).All(&books,cols...)
  252. return books, err
  253. }
  254. //根据指定字段查询一个结果.
  255. func (book *Book) FindByFieldFirst(field string, value interface{}) (*Book, error) {
  256. o := orm.NewOrm()
  257. err := o.QueryTable(book.TableNameWithPrefix()).Filter(field, value).One(book)
  258. return book, err
  259. }
  260. //根据项目标识查询项目
  261. func (book *Book) FindByIdentify(identify string,cols ...string) (*Book, error) {
  262. o := orm.NewOrm()
  263. err := o.QueryTable(book.TableNameWithPrefix()).Filter("identify", identify).One(book,cols...)
  264. return book, err
  265. }
  266. //分页查询指定用户的项目
  267. func (book *Book) FindToPager(pageIndex, pageSize, memberId int) (books []*BookResult, totalCount int, err error) {
  268. relationship := NewRelationship()
  269. o := orm.NewOrm()
  270. sql1 := "SELECT COUNT(book.book_id) AS total_count FROM " + book.TableNameWithPrefix() + " AS book LEFT JOIN " +
  271. relationship.TableNameWithPrefix() + " AS rel ON book.book_id=rel.book_id AND rel.member_id = ? WHERE rel.relationship_id > 0 "
  272. err = o.Raw(sql1, memberId).QueryRow(&totalCount)
  273. if err != nil {
  274. return
  275. }
  276. offset := (pageIndex - 1) * pageSize
  277. sql2 := "SELECT book.*,rel.member_id,rel.role_id,m.account as create_name FROM " + book.TableNameWithPrefix() + " AS book" +
  278. " LEFT JOIN " + relationship.TableNameWithPrefix() + " AS rel ON book.book_id=rel.book_id AND rel.member_id = ?" +
  279. " LEFT JOIN " + relationship.TableNameWithPrefix() + " AS rel1 ON book.book_id=rel1.book_id AND rel1.role_id=0" +
  280. " LEFT JOIN " + NewMember().TableNameWithPrefix() + " AS m ON rel1.member_id=m.member_id " +
  281. " WHERE rel.relationship_id > 0 ORDER BY book.order_index DESC,book.book_id DESC LIMIT " + fmt.Sprintf("%d,%d", offset, pageSize)
  282. _, err = o.Raw(sql2, memberId).QueryRows(&books)
  283. if err != nil {
  284. logs.Error("分页查询项目列表 => ", err)
  285. return
  286. }
  287. 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"
  288. if err == nil && len(books) > 0 {
  289. for index, book := range books {
  290. var text struct {
  291. Account string
  292. ModifyTime time.Time
  293. }
  294. err1 := o.Raw(sql, book.BookId).QueryRow(&text)
  295. if err1 == nil {
  296. books[index].LastModifyText = text.Account + " 于 " + text.ModifyTime.Format("2006-01-02 15:04:05")
  297. }
  298. if book.RoleId == 0 {
  299. book.RoleName = "创始人"
  300. } else if book.RoleId == 1 {
  301. book.RoleName = "管理员"
  302. } else if book.RoleId == 2 {
  303. book.RoleName = "编辑者"
  304. } else if book.RoleId == 3 {
  305. book.RoleName = "观察者"
  306. }
  307. }
  308. }
  309. return
  310. }
  311. // 彻底删除项目.
  312. func (book *Book) ThoroughDeleteBook(id int) error {
  313. if id <= 0 {
  314. return ErrInvalidParameter
  315. }
  316. o := orm.NewOrm()
  317. book, err := book.Find(id)
  318. if err != nil {
  319. return err
  320. }
  321. o.Begin()
  322. //删除附件,这里没有删除实际物理文件
  323. _,err = o.Raw("DELETE FROM " + NewAttachment().TableNameWithPrefix() + " WHERE book_id=?",book.BookId).Exec()
  324. if err != nil {
  325. o.Rollback()
  326. return err
  327. }
  328. //删除文档
  329. _, err = o.Raw( "DELETE FROM " + NewDocument().TableNameWithPrefix() + " WHERE book_id = ?", book.BookId).Exec()
  330. if err != nil {
  331. o.Rollback()
  332. return err
  333. }
  334. //删除项目
  335. _, err = o.Raw("DELETE FROM " + book.TableNameWithPrefix() + " WHERE book_id = ?", book.BookId).Exec()
  336. if err != nil {
  337. o.Rollback()
  338. return err
  339. }
  340. //删除关系
  341. _, err = o.Raw("DELETE FROM " + NewRelationship().TableNameWithPrefix() + " WHERE book_id = ?", book.BookId).Exec()
  342. if err != nil {
  343. o.Rollback()
  344. return err
  345. }
  346. //删除模板
  347. _,err = o.Raw("DELETE FROM " + NewTemplate().TableNameWithPrefix() + " WHERE book_id = ?",book.BookId).Exec()
  348. if err != nil {
  349. o.Rollback()
  350. return err
  351. }
  352. if book.Label != "" {
  353. NewLabel().InsertOrUpdateMulti(book.Label)
  354. }
  355. //删除导出缓存
  356. if err := os.RemoveAll(filepath.Join(conf.GetExportOutputPath(), strconv.Itoa(id))); err != nil {
  357. beego.Error("删除项目缓存失败 ->",err)
  358. }
  359. //删除附件和图片
  360. if err := os.RemoveAll(filepath.Join(conf.WorkingDirectory,"uploads",book.Identify)); err != nil {
  361. beego.Error("删除项目附件和图片失败 ->",err)
  362. }
  363. return o.Commit()
  364. }
  365. //分页查找系统首页数据.
  366. func (book *Book) FindForHomeToPager(pageIndex, pageSize, member_id int) (books []*BookResult, totalCount int, err error) {
  367. o := orm.NewOrm()
  368. offset := (pageIndex - 1) * pageSize
  369. //如果是登录用户
  370. if member_id > 0 {
  371. 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"
  372. err = o.Raw(sql1, member_id).QueryRow(&totalCount)
  373. if err != nil {
  374. return
  375. }
  376. sql2 := `SELECT book.*,rel1.*,member.account AS create_name,member.real_name FROM md_books AS book
  377. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?
  378. LEFT JOIN md_relationship AS rel1 ON rel1.book_id = book.book_id AND rel1.role_id = 0
  379. LEFT JOIN md_members AS member ON rel1.member_id = member.member_id
  380. WHERE rel.relationship_id > 0 OR book.privately_owned = 0 ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
  381. _, err = o.Raw(sql2, member_id, offset, pageSize).QueryRows(&books)
  382. } else {
  383. count, err1 := o.QueryTable(book.TableNameWithPrefix()).Filter("privately_owned", 0).Count()
  384. if err1 != nil {
  385. err = err1
  386. return
  387. }
  388. totalCount = int(count)
  389. sql := `SELECT book.*,rel.*,member.account AS create_name,member.real_name FROM md_books AS book
  390. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.role_id = 0
  391. LEFT JOIN md_members AS member ON rel.member_id = member.member_id
  392. WHERE book.privately_owned = 0 ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
  393. _, err = o.Raw(sql, offset, pageSize).QueryRows(&books)
  394. }
  395. return
  396. }
  397. //分页全局搜索.
  398. func (book *Book) FindForLabelToPager(keyword string, pageIndex, pageSize, memberId int) (books []*BookResult, totalCount int, err error) {
  399. o := orm.NewOrm()
  400. keyword = "%" + keyword + "%"
  401. offset := (pageIndex - 1) * pageSize
  402. //如果是登录用户
  403. if memberId > 0 {
  404. 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 ?"
  405. err = o.Raw(sql1, memberId, keyword).QueryRow(&totalCount)
  406. if err != nil {
  407. return
  408. }
  409. sql2 := `SELECT book.*,rel1.*,member.account AS create_name FROM md_books AS book
  410. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?
  411. LEFT JOIN md_relationship AS rel1 ON rel1.book_id = book.book_id AND rel1.role_id = 0
  412. LEFT JOIN md_members AS member ON rel1.member_id = member.member_id
  413. WHERE (rel.relationship_id > 0 OR book.privately_owned = 0) AND book.label LIKE ? ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
  414. _, err = o.Raw(sql2, memberId, keyword, offset, pageSize).QueryRows(&books)
  415. return
  416. } else {
  417. count, err1 := o.QueryTable(NewBook().TableNameWithPrefix()).Filter("privately_owned", 0).Filter("label__icontains", keyword).Count()
  418. if err1 != nil {
  419. err = err1
  420. return
  421. }
  422. totalCount = int(count)
  423. sql := `SELECT book.*,rel.*,member.account AS create_name FROM md_books AS book
  424. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.role_id = 0
  425. LEFT JOIN md_members AS member ON rel.member_id = member.member_id
  426. WHERE book.privately_owned = 0 AND book.label LIKE ? ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
  427. _, err = o.Raw(sql, keyword, offset, pageSize).QueryRows(&books)
  428. return
  429. }
  430. }
  431. //发布文档
  432. func (book *Book) ReleaseContent(bookId int) {
  433. o := orm.NewOrm()
  434. var docs []*Document
  435. _, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", bookId).All(&docs)
  436. if err != nil {
  437. beego.Error("发布失败 =>",bookId, err)
  438. return
  439. }
  440. for _, item := range docs {
  441. item.BookId = bookId
  442. item.ReleaseContent()
  443. }
  444. }
  445. //重置文档数量
  446. func (book *Book) ResetDocumentNumber(bookId int) {
  447. o := orm.NewOrm()
  448. totalCount, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", bookId).Count()
  449. if err == nil {
  450. _,err = o.Raw("UPDATE md_books SET doc_count = ? WHERE book_id = ?", int(totalCount), bookId).Exec()
  451. if err != nil {
  452. beego.Error("重置文档数量失败 =>",bookId,err)
  453. }
  454. } else {
  455. beego.Error("获取文档数量失败 =>",bookId,err)
  456. }
  457. }
  458. //导入项目
  459. func (book *Book) ImportBook(zipPath string) error {
  460. if !filetil.FileExists(zipPath) {
  461. return errors.New("文件不存在 => " + zipPath)
  462. }
  463. w := md5.New()
  464. io.WriteString(w, zipPath) //将str写入到w中
  465. io.WriteString(w, time.Now().String())
  466. io.WriteString(w, book.BookName)
  467. md5str := fmt.Sprintf("%x", w.Sum(nil)) //w.Sum(nil)将w的hash转成[]byte格式
  468. tempPath := filepath.Join(os.TempDir(), md5str)
  469. if err := os.MkdirAll(tempPath, 0766); err != nil {
  470. beego.Error("创建导入目录出错 => ",err)
  471. }
  472. //如果加压缩失败
  473. if err := ziptil.Unzip(zipPath, tempPath); err != nil {
  474. return err
  475. }
  476. //当导入结束后,删除临时文件
  477. //defer os.RemoveAll(tempPath)
  478. for {
  479. //如果当前目录下只有一个目录,则重置根目录
  480. if entries, err := ioutil.ReadDir(tempPath); err == nil && len(entries) == 1 {
  481. dir := entries[0]
  482. if dir.IsDir() && dir.Name() != "." && dir.Name() != ".." {
  483. tempPath = filepath.Join(tempPath, dir.Name())
  484. break
  485. }
  486. } else {
  487. break
  488. }
  489. }
  490. tempPath = strings.Replace(tempPath, "\\", "/", -1)
  491. docMap := make(map[string]int, 0)
  492. o := orm.NewOrm()
  493. o.Insert(book)
  494. relationship := NewRelationship()
  495. relationship.BookId = book.BookId
  496. relationship.RoleId = 0
  497. relationship.MemberId = book.MemberId
  498. relationship.Insert()
  499. err := filepath.Walk(tempPath, func(path string, info os.FileInfo, err error) error {
  500. path = strings.Replace(path, "\\", "/", -1)
  501. if path == tempPath {
  502. return nil
  503. }
  504. if !info.IsDir() {
  505. ext := filepath.Ext(info.Name())
  506. //如果是Markdown文件
  507. if strings.EqualFold(ext, ".md") || strings.EqualFold(ext, ".markdown") {
  508. beego.Info("正在处理 =>",path,info.Name())
  509. doc := NewDocument()
  510. doc.BookId = book.BookId
  511. doc.MemberId = book.MemberId
  512. docIdentify := strings.Replace(strings.TrimPrefix(path, tempPath+"/"), "/", "-", -1)
  513. if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, docIdentify); !ok || err != nil {
  514. docIdentify = "import-" + docIdentify
  515. }
  516. doc.Identify = docIdentify
  517. //匹配图片,如果图片语法是在代码块中,这里同样会处理
  518. re := regexp.MustCompile(`!\[(.*?)\]\((.*?)\)`)
  519. markdown, err := filetil.ReadFileAndIgnoreUTF8BOM(path)
  520. if err != nil {
  521. return err
  522. }
  523. //处理图片
  524. doc.Markdown = re.ReplaceAllStringFunc(string(markdown), func(image string) string {
  525. images := re.FindAllSubmatch([]byte(image), -1)
  526. if len(images) <= 0 || len(images[0]) < 3 {
  527. return image
  528. }
  529. originalImageUrl := string(images[0][2])
  530. imageUrl := strings.Replace(string(originalImageUrl), "\\", "/", -1)
  531. //如果是本地路径,则需要将图片复制到项目目录
  532. if !strings.HasPrefix(imageUrl, "http://") &&
  533. !strings.HasPrefix(imageUrl, "https://") &&
  534. !strings.HasPrefix(imageUrl,"ftp://") {
  535. //如果路径中存在参数
  536. if l := strings.Index(imageUrl,"?"); l > 0 {
  537. imageUrl = imageUrl[:l]
  538. }
  539. if strings.HasPrefix(imageUrl, "/") {
  540. imageUrl = filepath.Join(tempPath, imageUrl)
  541. } else if strings.HasPrefix(imageUrl, "./") {
  542. imageUrl = filepath.Join(filepath.Dir(path), strings.TrimPrefix(imageUrl, "./"))
  543. } else if strings.HasPrefix(imageUrl, "../") {
  544. imageUrl = filepath.Join(filepath.Dir(path), imageUrl)
  545. } else {
  546. imageUrl = filepath.Join(filepath.Dir(path), imageUrl)
  547. }
  548. imageUrl = strings.Replace(imageUrl, "\\", "/", -1)
  549. dstFile := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), strings.TrimPrefix(imageUrl, tempPath))
  550. if filetil.FileExists(imageUrl) {
  551. filetil.CopyFile(imageUrl, dstFile)
  552. imageUrl = strings.TrimPrefix(strings.Replace(dstFile,"\\","/",-1), strings.Replace(conf.WorkingDirectory,"\\","/",-1))
  553. if !strings.HasPrefix(imageUrl, "/") && !strings.HasPrefix(imageUrl, "\\") {
  554. imageUrl = "/" + imageUrl
  555. }
  556. }
  557. } else {
  558. imageExt := cryptil.Md5Crypt(imageUrl) + filepath.Ext(imageUrl)
  559. dstFile := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), imageExt)
  560. if err := requests.DownloadAndSaveFile(imageUrl, dstFile); err == nil {
  561. imageUrl = strings.TrimPrefix(strings.Replace(dstFile, "\\", "/", -1), strings.Replace(conf.WorkingDirectory, "\\", "/", -1))
  562. if !strings.HasPrefix(imageUrl, "/") && !strings.HasPrefix(imageUrl, "\\") {
  563. imageUrl = "/" + imageUrl
  564. }
  565. }
  566. }
  567. imageUrl = strings.Replace(strings.TrimSuffix(image, originalImageUrl+")")+ conf.URLForWithCdnImage(imageUrl) +")", "\\", "/", -1)
  568. return imageUrl
  569. })
  570. linkRegexp := regexp.MustCompile(`\[(.*?)\]\((.*?)\)`)
  571. //处理链接
  572. doc.Markdown = linkRegexp.ReplaceAllStringFunc(doc.Markdown, func(link string) string {
  573. links := linkRegexp.FindAllStringSubmatch(link, -1)
  574. originalLink := links[0][2]
  575. var linkPath string
  576. var err error
  577. if strings.HasPrefix(originalLink,"<") {
  578. originalLink = strings.TrimPrefix(originalLink,"<")
  579. }
  580. if strings.HasSuffix(originalLink,">") {
  581. originalLink = strings.TrimSuffix(originalLink,">")
  582. }
  583. //如果是从根目录开始,
  584. if strings.HasPrefix(originalLink, "/") {
  585. linkPath, err = filepath.Abs(filepath.Join(tempPath, originalLink))
  586. } else if strings.HasPrefix(originalLink, "./") {
  587. linkPath, err = filepath.Abs(filepath.Join(filepath.Dir(path), originalLink[1:]))
  588. } else{
  589. linkPath, err = filepath.Abs(filepath.Join(filepath.Dir(path), originalLink))
  590. }
  591. if err == nil {
  592. //如果本地存在该链接
  593. if filetil.FileExists(linkPath) {
  594. ext := filepath.Ext(linkPath)
  595. //beego.Info("当前后缀 -> ",ext)
  596. //如果链接是Markdown文件,则生成文档标识,否则,将目标文件复制到项目目录
  597. if strings.EqualFold(ext, ".md") || strings.EqualFold(ext, ".markdown") {
  598. docIdentify := strings.Replace(strings.TrimPrefix(strings.Replace(linkPath, "\\", "/", -1), tempPath+"/"), "/", "-", -1)
  599. //beego.Info(originalLink, "|", linkPath, "|", docIdentify)
  600. if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, docIdentify); !ok || err != nil {
  601. docIdentify = "import-" + docIdentify
  602. }
  603. docIdentify = strings.TrimSuffix(docIdentify, "-README.md")
  604. link = strings.TrimSuffix(link, originalLink+")") + conf.URLFor("DocumentController.Read", ":key", book.Identify, ":id", docIdentify) + ")"
  605. } else {
  606. dstPath := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), originalLink)
  607. filetil.CopyFile(linkPath, dstPath)
  608. tempLink := conf.BaseUrl + strings.TrimPrefix(strings.Replace(dstPath, "\\", "/", -1), strings.Replace(conf.WorkingDirectory, "\\", "/", -1))
  609. link = strings.TrimSuffix(link, originalLink+")") + tempLink + ")"
  610. }
  611. }else{
  612. beego.Info("文件不存在 ->",linkPath)
  613. }
  614. }
  615. return link
  616. })
  617. //codeRe := regexp.MustCompile("```\\w+")
  618. //doc.Markdown = codeRe.ReplaceAllStringFunc(doc.Markdown, func(s string) string {
  619. // //beego.Info(s)
  620. // return strings.Replace(s,"```","``` ",-1)
  621. //})
  622. doc.Content = string(blackfriday.Run([]byte(doc.Markdown)))
  623. doc.Version = time.Now().Unix()
  624. //解析文档名称,默认使用第一个h标签为标题
  625. docName := strings.TrimSuffix(info.Name(), ext)
  626. for _, line := range strings.Split(doc.Markdown, "\n") {
  627. if strings.HasPrefix(line, "#") {
  628. docName = strings.TrimLeft(line, "#")
  629. break
  630. }
  631. }
  632. doc.DocumentName = strings.TrimSpace(docName)
  633. parentId := 0
  634. parentIdentify := strings.Replace(strings.Trim(strings.TrimSuffix(strings.TrimPrefix(path, tempPath), info.Name()), "/"), "/", "-", -1)
  635. if parentIdentify != "" {
  636. if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, parentIdentify); !ok || err != nil {
  637. parentIdentify = "import-" + parentIdentify
  638. }
  639. if id, ok := docMap[parentIdentify]; ok {
  640. parentId = id
  641. }
  642. }
  643. if strings.EqualFold(info.Name(), "README.md") {
  644. beego.Info(path, "|", info.Name(), "|", parentIdentify, "|", parentId)
  645. }
  646. isInsert := false
  647. //如果当前文件是README.md,则将内容更新到父级
  648. if strings.EqualFold(info.Name(), "README.md") && parentId != 0 {
  649. doc.DocumentId = parentId
  650. //beego.Info(path,"|",parentId)
  651. } else {
  652. //beego.Info(path,"|",parentIdentify)
  653. doc.ParentId = parentId
  654. isInsert = true
  655. }
  656. if err := doc.InsertOrUpdate("document_name", "markdown", "content"); err != nil {
  657. beego.Error(doc.DocumentId, err)
  658. }
  659. if isInsert {
  660. docMap[docIdentify] = doc.DocumentId
  661. }
  662. }
  663. } else {
  664. //如果当前目录下存在Markdown文件,则需要创建此节点
  665. if filetil.HasFileOfExt(path, []string{".md", ".markdown"}) {
  666. beego.Info("正在处理 =>",path,info.Name())
  667. identify := strings.Replace(strings.Trim(strings.TrimPrefix(path, tempPath), "/"), "/", "-", -1)
  668. if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, identify); !ok || err != nil {
  669. identify = "import-" + identify
  670. }
  671. parentDoc := NewDocument()
  672. parentDoc.MemberId = book.MemberId
  673. parentDoc.BookId = book.BookId
  674. parentDoc.Identify = identify
  675. parentDoc.Version = time.Now().Unix()
  676. parentDoc.DocumentName = "空白文档"
  677. parentId := 0
  678. parentIdentify := strings.TrimSuffix(identify, "-"+info.Name())
  679. if id, ok := docMap[parentIdentify]; ok {
  680. parentId = id
  681. }
  682. parentDoc.ParentId = parentId
  683. if err := parentDoc.InsertOrUpdate(); err != nil {
  684. beego.Error(err)
  685. }
  686. docMap[identify] = parentDoc.DocumentId
  687. //beego.Info(path,"|",parentDoc.DocumentId,"|",identify,"|",info.Name(),"|",parentIdentify)
  688. }
  689. }
  690. return nil
  691. })
  692. if err != nil {
  693. beego.Error("导入项目异常 => ", err)
  694. book.Description = "【项目导入存在错误:" + err.Error() + "】"
  695. }
  696. beego.Info("项目导入完毕 => ", book.BookName)
  697. book.ReleaseContent(book.BookId)
  698. return err
  699. }