BookModel.go 35 KB

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