BookModel.go 31 KB

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