1
0

BookModel.go 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925
  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. o := orm.NewOrm()
  269. //sql1 := "SELECT COUNT(book.book_id) AS total_count FROM " + book.TableNameWithPrefix() + " AS book LEFT JOIN " +
  270. // relationship.TableNameWithPrefix() + " AS rel ON book.book_id=rel.book_id AND rel.member_id = ? WHERE rel.relationship_id > 0 "
  271. sql1 := `SELECT
  272. count(*) AS total_count
  273. FROM md_books AS book
  274. LEFT JOIN md_relationship AS rel ON book.book_id = rel.book_id AND rel.member_id = ?
  275. left join (select *
  276. from (select book_id,team_member_id,role_id
  277. from md_team_relationship as mtr
  278. 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.book_id)
  279. as team on team.book_id=book.book_id
  280. WHERE rel.relationship_id > 0 or team.team_member_id > 0`
  281. err = o.Raw(sql1, memberId, memberId).QueryRow(&totalCount)
  282. if err != nil {
  283. return
  284. }
  285. offset := (pageIndex - 1) * pageSize
  286. //sql2 := "SELECT book.*,rel.member_id,rel.role_id,m.account as create_name FROM " + book.TableNameWithPrefix() + " AS book" +
  287. // " LEFT JOIN " + relationship.TableNameWithPrefix() + " AS rel ON book.book_id=rel.book_id AND rel.member_id = ?" +
  288. // " LEFT JOIN " + relationship.TableNameWithPrefix() + " AS rel1 ON book.book_id=rel1.book_id AND rel1.role_id=0" +
  289. // " LEFT JOIN " + NewMember().TableNameWithPrefix() + " AS m ON rel1.member_id=m.member_id " +
  290. // " WHERE rel.relationship_id > 0 ORDER BY book.order_index DESC,book.book_id DESC LIMIT " + fmt.Sprintf("%d,%d", offset, pageSize)
  291. sql2 := `SELECT
  292. book.*,
  293. case when rel.relationship_id is null then team.role_id else rel.role_id end as role_id,
  294. m.account as create_name
  295. FROM md_books AS book
  296. LEFT JOIN md_relationship AS rel ON book.book_id = rel.book_id AND rel.member_id = ?
  297. left join (select *
  298. from (select book_id,team_member_id,role_id
  299. from md_team_relationship as mtr
  300. 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.book_id) as team
  301. on team.book_id=book.book_id
  302. LEFT JOIN md_relationship AS rel1 ON book.book_id = rel1.book_id AND rel1.role_id = 0
  303. LEFT JOIN md_members AS m ON rel1.member_id = m.member_id
  304. WHERE rel.relationship_id > 0 or team.team_member_id > 0
  305. ORDER BY book.order_index, book.book_id DESC limit ?,?`
  306. _, err = o.Raw(sql2, memberId, memberId, offset, pageSize).QueryRows(&books)
  307. if err != nil {
  308. logs.Error("分页查询项目列表 => ", err)
  309. return
  310. }
  311. 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"
  312. if err == nil && len(books) > 0 {
  313. for index, book := range books {
  314. var text struct {
  315. Account string
  316. ModifyTime time.Time
  317. }
  318. err1 := o.Raw(sql, book.BookId).QueryRow(&text)
  319. if err1 == nil {
  320. books[index].LastModifyText = text.Account + " 于 " + text.ModifyTime.Format("2006-01-02 15:04:05")
  321. }
  322. if book.RoleId == 0 {
  323. book.RoleName = "创始人"
  324. } else if book.RoleId == 1 {
  325. book.RoleName = "管理员"
  326. } else if book.RoleId == 2 {
  327. book.RoleName = "编辑者"
  328. } else if book.RoleId == 3 {
  329. book.RoleName = "观察者"
  330. }
  331. }
  332. }
  333. return
  334. }
  335. // 彻底删除项目.
  336. func (book *Book) ThoroughDeleteBook(id int) error {
  337. if id <= 0 {
  338. return ErrInvalidParameter
  339. }
  340. o := orm.NewOrm()
  341. book, err := book.Find(id)
  342. if err != nil {
  343. return err
  344. }
  345. o.Begin()
  346. //删除附件,这里没有删除实际物理文件
  347. _, err = o.Raw("DELETE FROM "+NewAttachment().TableNameWithPrefix()+" WHERE book_id=?", book.BookId).Exec()
  348. if err != nil {
  349. o.Rollback()
  350. return err
  351. }
  352. //删除文档
  353. _, err = o.Raw("DELETE FROM "+NewDocument().TableNameWithPrefix()+" WHERE book_id = ?", book.BookId).Exec()
  354. if err != nil {
  355. o.Rollback()
  356. return err
  357. }
  358. //删除项目
  359. _, err = o.Raw("DELETE FROM "+book.TableNameWithPrefix()+" WHERE book_id = ?", book.BookId).Exec()
  360. if err != nil {
  361. o.Rollback()
  362. return err
  363. }
  364. //删除关系
  365. _, err = o.Raw("DELETE FROM "+NewRelationship().TableNameWithPrefix()+" WHERE book_id = ?", book.BookId).Exec()
  366. if err != nil {
  367. o.Rollback()
  368. return err
  369. }
  370. //删除模板
  371. _, err = o.Raw("DELETE FROM "+NewTemplate().TableNameWithPrefix()+" WHERE book_id = ?", book.BookId).Exec()
  372. if err != nil {
  373. o.Rollback()
  374. return err
  375. }
  376. if book.Label != "" {
  377. NewLabel().InsertOrUpdateMulti(book.Label)
  378. }
  379. //删除导出缓存
  380. if err := os.RemoveAll(filepath.Join(conf.GetExportOutputPath(), strconv.Itoa(id))); err != nil {
  381. beego.Error("删除项目缓存失败 ->", err)
  382. }
  383. //删除附件和图片
  384. if err := os.RemoveAll(filepath.Join(conf.WorkingDirectory, "uploads", book.Identify)); err != nil {
  385. beego.Error("删除项目附件和图片失败 ->", err)
  386. }
  387. return o.Commit()
  388. }
  389. //分页查找系统首页数据.
  390. func (book *Book) FindForHomeToPager(pageIndex, pageSize, memberId int) (books []*BookResult, totalCount int, err error) {
  391. o := orm.NewOrm()
  392. offset := (pageIndex - 1) * pageSize
  393. //如果是登录用户
  394. if memberId > 0 {
  395. sql1 := `SELECT COUNT(*)
  396. FROM md_books AS book
  397. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?
  398. left join (select *
  399. from (select book_id,team_member_id,role_id
  400. from md_team_relationship as mtr
  401. 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.book_id) as team on team.book_id=book.book_id
  402. WHERE relationship_id > 0 OR book.privately_owned = 0 or team.team_member_id > 0`
  403. err = o.Raw(sql1, memberId, memberId).QueryRow(&totalCount)
  404. if err != nil {
  405. return
  406. }
  407. sql2 := `SELECT book.*,rel1.*,member.account AS create_name,member.real_name FROM md_books AS book
  408. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?
  409. left join (select *
  410. from (select book_id,team_member_id,role_id
  411. from md_team_relationship as mtr
  412. 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.book_id) as team on team.book_id=book.book_id
  413. LEFT JOIN md_relationship AS rel1 ON rel1.book_id = book.book_id AND rel1.role_id = 0
  414. LEFT JOIN md_members AS member ON rel1.member_id = member.member_id
  415. WHERE rel.relationship_id > 0 OR book.privately_owned = 0 or team.team_member_id > 0 ORDER BY order_index ,book.book_id DESC LIMIT ?,?`
  416. _, err = o.Raw(sql2, memberId, memberId, offset, pageSize).QueryRows(&books)
  417. } else {
  418. count, err1 := o.QueryTable(book.TableNameWithPrefix()).Filter("privately_owned", 0).Count()
  419. if err1 != nil {
  420. err = err1
  421. return
  422. }
  423. totalCount = int(count)
  424. sql := `SELECT book.*,rel.*,member.account AS create_name,member.real_name FROM md_books AS book
  425. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.role_id = 0
  426. LEFT JOIN md_members AS member ON rel.member_id = member.member_id
  427. WHERE book.privately_owned = 0 ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
  428. _, err = o.Raw(sql, offset, pageSize).QueryRows(&books)
  429. }
  430. return
  431. }
  432. //分页全局搜索.
  433. func (book *Book) FindForLabelToPager(keyword string, pageIndex, pageSize, memberId int) (books []*BookResult, totalCount int, err error) {
  434. o := orm.NewOrm()
  435. keyword = "%" + keyword + "%"
  436. offset := (pageIndex - 1) * pageSize
  437. //如果是登录用户
  438. if memberId > 0 {
  439. sql1 := `SELECT COUNT(*)
  440. FROM md_books AS book
  441. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?
  442. left join (select *
  443. from (select book_id,team_member_id,role_id
  444. from md_team_relationship as mtr
  445. 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.book_id) as team on team.book_id = book.book_id
  446. WHERE (relationship_id > 0 OR book.privately_owned = 0 or team.team_member_id > 0) AND book.label LIKE ?`
  447. err = o.Raw(sql1, memberId, memberId, keyword).QueryRow(&totalCount)
  448. if err != nil {
  449. return
  450. }
  451. sql2 := `SELECT book.*,rel1.*,member.account AS create_name FROM md_books AS book
  452. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?
  453. left join (select * from (select book_id,team_member_id,role_id
  454. from md_team_relationship as mtr
  455. 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.book_id) as team
  456. on team.book_id = book.book_id
  457. LEFT JOIN md_relationship AS rel1 ON rel1.book_id = book.book_id AND rel1.role_id = 0
  458. LEFT JOIN md_members AS member ON rel1.member_id = member.member_id
  459. WHERE (rel.relationship_id > 0 OR book.privately_owned = 0 or team.team_member_id > 0)
  460. AND book.label LIKE ? ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
  461. _, err = o.Raw(sql2, memberId, memberId, keyword, offset, pageSize).QueryRows(&books)
  462. return
  463. } else {
  464. count, err1 := o.QueryTable(NewBook().TableNameWithPrefix()).Filter("privately_owned", 0).Filter("label__icontains", keyword).Count()
  465. if err1 != nil {
  466. err = err1
  467. return
  468. }
  469. totalCount = int(count)
  470. sql := `SELECT book.*,rel.*,member.account AS create_name FROM md_books AS book
  471. LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.role_id = 0
  472. LEFT JOIN md_members AS member ON rel.member_id = member.member_id
  473. WHERE book.privately_owned = 0 AND book.label LIKE ? ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
  474. _, err = o.Raw(sql, keyword, offset, pageSize).QueryRows(&books)
  475. return
  476. }
  477. }
  478. //发布文档
  479. func (book *Book) ReleaseContent(bookId int) {
  480. o := orm.NewOrm()
  481. var docs []*Document
  482. _, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", bookId).All(&docs)
  483. if err != nil {
  484. beego.Error("发布失败 =>", bookId, err)
  485. return
  486. }
  487. for _, item := range docs {
  488. item.BookId = bookId
  489. item.ReleaseContent()
  490. }
  491. }
  492. //重置文档数量
  493. func (book *Book) ResetDocumentNumber(bookId int) {
  494. o := orm.NewOrm()
  495. totalCount, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", bookId).Count()
  496. if err == nil {
  497. _, err = o.Raw("UPDATE md_books SET doc_count = ? WHERE book_id = ?", int(totalCount), bookId).Exec()
  498. if err != nil {
  499. beego.Error("重置文档数量失败 =>", bookId, err)
  500. }
  501. } else {
  502. beego.Error("获取文档数量失败 =>", bookId, err)
  503. }
  504. }
  505. //导入项目
  506. func (book *Book) ImportBook(zipPath string) error {
  507. if !filetil.FileExists(zipPath) {
  508. return errors.New("文件不存在 => " + zipPath)
  509. }
  510. w := md5.New()
  511. io.WriteString(w, zipPath) //将str写入到w中
  512. io.WriteString(w, time.Now().String())
  513. io.WriteString(w, book.BookName)
  514. md5str := fmt.Sprintf("%x", w.Sum(nil)) //w.Sum(nil)将w的hash转成[]byte格式
  515. tempPath := filepath.Join(os.TempDir(), md5str)
  516. if err := os.MkdirAll(tempPath, 0766); err != nil {
  517. beego.Error("创建导入目录出错 => ", err)
  518. }
  519. //如果加压缩失败
  520. if err := ziptil.Unzip(zipPath, tempPath); err != nil {
  521. return err
  522. }
  523. //当导入结束后,删除临时文件
  524. //defer os.RemoveAll(tempPath)
  525. for {
  526. //如果当前目录下只有一个目录,则重置根目录
  527. if entries, err := ioutil.ReadDir(tempPath); err == nil && len(entries) == 1 {
  528. dir := entries[0]
  529. if dir.IsDir() && dir.Name() != "." && dir.Name() != ".." {
  530. tempPath = filepath.Join(tempPath, dir.Name())
  531. break
  532. }
  533. } else {
  534. break
  535. }
  536. }
  537. tempPath = strings.Replace(tempPath, "\\", "/", -1)
  538. docMap := make(map[string]int, 0)
  539. o := orm.NewOrm()
  540. o.Insert(book)
  541. relationship := NewRelationship()
  542. relationship.BookId = book.BookId
  543. relationship.RoleId = 0
  544. relationship.MemberId = book.MemberId
  545. relationship.Insert()
  546. err := filepath.Walk(tempPath, func(path string, info os.FileInfo, err error) error {
  547. path = strings.Replace(path, "\\", "/", -1)
  548. if path == tempPath {
  549. return nil
  550. }
  551. if !info.IsDir() {
  552. ext := filepath.Ext(info.Name())
  553. //如果是Markdown文件
  554. if strings.EqualFold(ext, ".md") || strings.EqualFold(ext, ".markdown") {
  555. beego.Info("正在处理 =>", path, info.Name())
  556. doc := NewDocument()
  557. doc.BookId = book.BookId
  558. doc.MemberId = book.MemberId
  559. docIdentify := strings.Replace(strings.TrimPrefix(path, tempPath+"/"), "/", "-", -1)
  560. if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, docIdentify); !ok || err != nil {
  561. docIdentify = "import-" + docIdentify
  562. }
  563. doc.Identify = docIdentify
  564. //匹配图片,如果图片语法是在代码块中,这里同样会处理
  565. re := regexp.MustCompile(`!\[(.*?)\]\((.*?)\)`)
  566. markdown, err := filetil.ReadFileAndIgnoreUTF8BOM(path)
  567. if err != nil {
  568. return err
  569. }
  570. //处理图片
  571. doc.Markdown = re.ReplaceAllStringFunc(string(markdown), func(image string) string {
  572. images := re.FindAllSubmatch([]byte(image), -1)
  573. if len(images) <= 0 || len(images[0]) < 3 {
  574. return image
  575. }
  576. originalImageUrl := string(images[0][2])
  577. imageUrl := strings.Replace(string(originalImageUrl), "\\", "/", -1)
  578. //如果是本地路径,则需要将图片复制到项目目录
  579. if !strings.HasPrefix(imageUrl, "http://") &&
  580. !strings.HasPrefix(imageUrl, "https://") &&
  581. !strings.HasPrefix(imageUrl, "ftp://") {
  582. //如果路径中存在参数
  583. if l := strings.Index(imageUrl, "?"); l > 0 {
  584. imageUrl = imageUrl[:l]
  585. }
  586. if strings.HasPrefix(imageUrl, "/") {
  587. imageUrl = filepath.Join(tempPath, imageUrl)
  588. } else if strings.HasPrefix(imageUrl, "./") {
  589. imageUrl = filepath.Join(filepath.Dir(path), strings.TrimPrefix(imageUrl, "./"))
  590. } else if strings.HasPrefix(imageUrl, "../") {
  591. imageUrl = filepath.Join(filepath.Dir(path), imageUrl)
  592. } else {
  593. imageUrl = filepath.Join(filepath.Dir(path), imageUrl)
  594. }
  595. imageUrl = strings.Replace(imageUrl, "\\", "/", -1)
  596. dstFile := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), strings.TrimPrefix(imageUrl, tempPath))
  597. if filetil.FileExists(imageUrl) {
  598. filetil.CopyFile(imageUrl, dstFile)
  599. imageUrl = strings.TrimPrefix(strings.Replace(dstFile, "\\", "/", -1), strings.Replace(conf.WorkingDirectory, "\\", "/", -1))
  600. if !strings.HasPrefix(imageUrl, "/") && !strings.HasPrefix(imageUrl, "\\") {
  601. imageUrl = "/" + imageUrl
  602. }
  603. }
  604. } else {
  605. imageExt := cryptil.Md5Crypt(imageUrl) + filepath.Ext(imageUrl)
  606. dstFile := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), imageExt)
  607. if err := requests.DownloadAndSaveFile(imageUrl, dstFile); err == nil {
  608. imageUrl = strings.TrimPrefix(strings.Replace(dstFile, "\\", "/", -1), strings.Replace(conf.WorkingDirectory, "\\", "/", -1))
  609. if !strings.HasPrefix(imageUrl, "/") && !strings.HasPrefix(imageUrl, "\\") {
  610. imageUrl = "/" + imageUrl
  611. }
  612. }
  613. }
  614. imageUrl = strings.Replace(strings.TrimSuffix(image, originalImageUrl+")")+conf.URLForWithCdnImage(imageUrl)+")", "\\", "/", -1)
  615. return imageUrl
  616. })
  617. linkRegexp := regexp.MustCompile(`\[(.*?)\]\((.*?)\)`)
  618. //处理链接
  619. doc.Markdown = linkRegexp.ReplaceAllStringFunc(doc.Markdown, func(link string) string {
  620. links := linkRegexp.FindAllStringSubmatch(link, -1)
  621. originalLink := links[0][2]
  622. var linkPath string
  623. var err error
  624. if strings.HasPrefix(originalLink, "<") {
  625. originalLink = strings.TrimPrefix(originalLink, "<")
  626. }
  627. if strings.HasSuffix(originalLink, ">") {
  628. originalLink = strings.TrimSuffix(originalLink, ">")
  629. }
  630. //如果是从根目录开始,
  631. if strings.HasPrefix(originalLink, "/") {
  632. linkPath, err = filepath.Abs(filepath.Join(tempPath, originalLink))
  633. } else if strings.HasPrefix(originalLink, "./") {
  634. linkPath, err = filepath.Abs(filepath.Join(filepath.Dir(path), originalLink[1:]))
  635. } else {
  636. linkPath, err = filepath.Abs(filepath.Join(filepath.Dir(path), originalLink))
  637. }
  638. if err == nil {
  639. //如果本地存在该链接
  640. if filetil.FileExists(linkPath) {
  641. ext := filepath.Ext(linkPath)
  642. //beego.Info("当前后缀 -> ",ext)
  643. //如果链接是Markdown文件,则生成文档标识,否则,将目标文件复制到项目目录
  644. if strings.EqualFold(ext, ".md") || strings.EqualFold(ext, ".markdown") {
  645. docIdentify := strings.Replace(strings.TrimPrefix(strings.Replace(linkPath, "\\", "/", -1), tempPath+"/"), "/", "-", -1)
  646. //beego.Info(originalLink, "|", linkPath, "|", docIdentify)
  647. if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, docIdentify); !ok || err != nil {
  648. docIdentify = "import-" + docIdentify
  649. }
  650. docIdentify = strings.TrimSuffix(docIdentify, "-README.md")
  651. link = strings.TrimSuffix(link, originalLink+")") + conf.URLFor("DocumentController.Read", ":key", book.Identify, ":id", docIdentify) + ")"
  652. } else {
  653. dstPath := filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), originalLink)
  654. filetil.CopyFile(linkPath, dstPath)
  655. tempLink := conf.BaseUrl + strings.TrimPrefix(strings.Replace(dstPath, "\\", "/", -1), strings.Replace(conf.WorkingDirectory, "\\", "/", -1))
  656. link = strings.TrimSuffix(link, originalLink+")") + tempLink + ")"
  657. }
  658. } else {
  659. beego.Info("文件不存在 ->", linkPath)
  660. }
  661. }
  662. return link
  663. })
  664. //codeRe := regexp.MustCompile("```\\w+")
  665. //doc.Markdown = codeRe.ReplaceAllStringFunc(doc.Markdown, func(s string) string {
  666. // //beego.Info(s)
  667. // return strings.Replace(s,"```","``` ",-1)
  668. //})
  669. doc.Content = string(blackfriday.Run([]byte(doc.Markdown)))
  670. doc.Version = time.Now().Unix()
  671. //解析文档名称,默认使用第一个h标签为标题
  672. docName := strings.TrimSuffix(info.Name(), ext)
  673. for _, line := range strings.Split(doc.Markdown, "\n") {
  674. if strings.HasPrefix(line, "#") {
  675. docName = strings.TrimLeft(line, "#")
  676. break
  677. }
  678. }
  679. doc.DocumentName = strings.TrimSpace(docName)
  680. parentId := 0
  681. parentIdentify := strings.Replace(strings.Trim(strings.TrimSuffix(strings.TrimPrefix(path, tempPath), info.Name()), "/"), "/", "-", -1)
  682. if parentIdentify != "" {
  683. if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, parentIdentify); !ok || err != nil {
  684. parentIdentify = "import-" + parentIdentify
  685. }
  686. if id, ok := docMap[parentIdentify]; ok {
  687. parentId = id
  688. }
  689. }
  690. if strings.EqualFold(info.Name(), "README.md") {
  691. beego.Info(path, "|", info.Name(), "|", parentIdentify, "|", parentId)
  692. }
  693. isInsert := false
  694. //如果当前文件是README.md,则将内容更新到父级
  695. if strings.EqualFold(info.Name(), "README.md") && parentId != 0 {
  696. doc.DocumentId = parentId
  697. //beego.Info(path,"|",parentId)
  698. } else {
  699. //beego.Info(path,"|",parentIdentify)
  700. doc.ParentId = parentId
  701. isInsert = true
  702. }
  703. if err := doc.InsertOrUpdate("document_name", "markdown", "content"); err != nil {
  704. beego.Error(doc.DocumentId, err)
  705. }
  706. if isInsert {
  707. docMap[docIdentify] = doc.DocumentId
  708. }
  709. }
  710. } else {
  711. //如果当前目录下存在Markdown文件,则需要创建此节点
  712. if filetil.HasFileOfExt(path, []string{".md", ".markdown"}) {
  713. beego.Info("正在处理 =>", path, info.Name())
  714. identify := strings.Replace(strings.Trim(strings.TrimPrefix(path, tempPath), "/"), "/", "-", -1)
  715. if ok, err := regexp.MatchString(`[a-z]+[a-zA-Z0-9_.\-]*$`, identify); !ok || err != nil {
  716. identify = "import-" + identify
  717. }
  718. parentDoc := NewDocument()
  719. parentDoc.MemberId = book.MemberId
  720. parentDoc.BookId = book.BookId
  721. parentDoc.Identify = identify
  722. parentDoc.Version = time.Now().Unix()
  723. parentDoc.DocumentName = "空白文档"
  724. parentId := 0
  725. parentIdentify := strings.TrimSuffix(identify, "-"+info.Name())
  726. if id, ok := docMap[parentIdentify]; ok {
  727. parentId = id
  728. }
  729. parentDoc.ParentId = parentId
  730. if err := parentDoc.InsertOrUpdate(); err != nil {
  731. beego.Error(err)
  732. }
  733. docMap[identify] = parentDoc.DocumentId
  734. //beego.Info(path,"|",parentDoc.DocumentId,"|",identify,"|",info.Name(),"|",parentIdentify)
  735. }
  736. }
  737. return nil
  738. })
  739. if err != nil {
  740. beego.Error("导入项目异常 => ", err)
  741. book.Description = "【项目导入存在错误:" + err.Error() + "】"
  742. }
  743. beego.Info("项目导入完毕 => ", book.BookName)
  744. book.ReleaseContent(book.BookId)
  745. return err
  746. }
  747. func (book *Book) FindForRoleId(bookId, memberId int) (conf.BookRole, error) {
  748. o := orm.NewOrm()
  749. var relationship Relationship
  750. err := NewRelationship().QueryTable().Filter("book_id", bookId).Filter("member_id", memberId).One(&relationship)
  751. if err != nil && err != orm.ErrNoRows {
  752. return 0, err
  753. }
  754. if err == nil {
  755. return relationship.RoleId, nil
  756. }
  757. sql := `select role_id
  758. from md_team_relationship as mtr
  759. left join md_team_member as mtm using (team_id)
  760. where mtr.book_id = ? and mtm.member_id = ? order by mtm.role_id asc limit 1;`
  761. var roleId int
  762. err = o.Raw(sql, bookId, memberId).QueryRow(&roleId)
  763. if err != nil {
  764. beego.Error("查询用户项目角色出错 -> book_id=", bookId, " member_id=", memberId, err)
  765. return 0, nil
  766. }
  767. return conf.BookRole(roleId), nil
  768. }