BookController.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "html/template"
  7. "os"
  8. "path/filepath"
  9. "regexp"
  10. "strconv"
  11. "strings"
  12. "time"
  13. "net/http"
  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/graphics"
  19. "github.com/lifei6671/mindoc/models"
  20. "github.com/lifei6671/mindoc/utils"
  21. "github.com/lifei6671/mindoc/utils/pagination"
  22. "gopkg.in/russross/blackfriday.v2"
  23. )
  24. type BookController struct {
  25. BaseController
  26. }
  27. func (c *BookController) Index() {
  28. c.Prepare()
  29. c.TplName = "book/index.tpl"
  30. pageIndex, _ := c.GetInt("page", 1)
  31. books, totalCount, err := models.NewBook().FindToPager(pageIndex, conf.PageSize, c.Member.MemberId)
  32. if err != nil {
  33. logs.Error("BookController.Index => ", err)
  34. c.Abort("500")
  35. }
  36. for i, book := range books {
  37. books[i].Description = utils.StripTags(string(blackfriday.Run([]byte(book.Description))))
  38. books[i].ModifyTime = book.ModifyTime.Local()
  39. books[i].CreateTime = book.CreateTime.Local()
  40. }
  41. if totalCount > 0 {
  42. pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl())
  43. c.Data["PageHtml"] = pager.HtmlPages()
  44. } else {
  45. c.Data["PageHtml"] = ""
  46. }
  47. b, err := json.Marshal(books)
  48. if err != nil || len(books) <= 0 {
  49. c.Data["Result"] = template.JS("[]")
  50. } else {
  51. c.Data["Result"] = template.JS(string(b))
  52. }
  53. }
  54. // Dashboard 项目概要 .
  55. func (c *BookController) Dashboard() {
  56. c.Prepare()
  57. c.TplName = "book/dashboard.tpl"
  58. key := c.Ctx.Input.Param(":key")
  59. if key == "" {
  60. c.Abort("404")
  61. }
  62. book, err := models.NewBookResult().FindByIdentify(key, c.Member.MemberId)
  63. if err != nil {
  64. if err == models.ErrPermissionDenied {
  65. c.Abort("403")
  66. }
  67. beego.Error(err)
  68. c.Abort("500")
  69. }
  70. c.Data["Description"] = template.HTML(blackfriday.Run([]byte(book.Description)))
  71. c.Data["Model"] = *book
  72. }
  73. // Setting 项目设置 .
  74. func (c *BookController) Setting() {
  75. c.Prepare()
  76. c.TplName = "book/setting.tpl"
  77. key := c.Ctx.Input.Param(":key")
  78. if key == "" {
  79. c.Abort("404")
  80. }
  81. book, err := models.NewBookResult().FindByIdentify(key, c.Member.MemberId)
  82. if err != nil {
  83. if err == orm.ErrNoRows {
  84. c.Abort("404")
  85. }
  86. if err == models.ErrPermissionDenied {
  87. c.Abort("403")
  88. }
  89. c.Abort("500")
  90. }
  91. //如果不是创始人也不是管理员则不能操作
  92. if book.RoleId != conf.BookFounder && book.RoleId != conf.BookAdmin {
  93. c.Abort("403")
  94. }
  95. if book.PrivateToken != "" {
  96. book.PrivateToken = conf.URLFor("DocumentController.Index", ":key", book.Identify, "token", book.PrivateToken)
  97. }
  98. c.Data["Model"] = book
  99. }
  100. //保存项目信息
  101. func (c *BookController) SaveBook() {
  102. bookResult, err := c.IsPermission()
  103. if err != nil {
  104. c.JsonResult(6001, err.Error())
  105. }
  106. book, err := models.NewBook().Find(bookResult.BookId)
  107. if err != nil {
  108. logs.Error("SaveBook => ", err)
  109. c.JsonResult(6002, err.Error())
  110. }
  111. bookName := strings.TrimSpace(c.GetString("book_name"))
  112. description := strings.TrimSpace(c.GetString("description", ""))
  113. commentStatus := c.GetString("comment_status")
  114. tag := strings.TrimSpace(c.GetString("label"))
  115. editor := strings.TrimSpace(c.GetString("editor"))
  116. autoRelease := strings.TrimSpace(c.GetString("auto_release")) == "on"
  117. publisher := strings.TrimSpace(c.GetString("publisher"))
  118. historyCount, _ := c.GetInt("history_count", 0)
  119. isDownload := strings.TrimSpace(c.GetString("is_download")) == "on"
  120. enableShare := strings.TrimSpace(c.GetString("enable_share")) == "on"
  121. isUseFirstDocument := strings.TrimSpace(c.GetString("is_use_first_document")) == "on"
  122. if strings.Count(description, "") > 500 {
  123. c.JsonResult(6004, "项目描述不能大于500字")
  124. }
  125. if commentStatus != "open" && commentStatus != "closed" && commentStatus != "group_only" && commentStatus != "registered_only" {
  126. commentStatus = "closed"
  127. }
  128. if tag != "" {
  129. tags := strings.Split(tag, ",")
  130. if len(tags) > 10 {
  131. c.JsonResult(6005, "最多允许添加10个标签")
  132. }
  133. }
  134. if editor != "markdown" && editor != "html" {
  135. editor = "markdown"
  136. }
  137. book.BookName = bookName
  138. book.Description = description
  139. book.CommentStatus = commentStatus
  140. book.Publisher = publisher
  141. book.Label = tag
  142. book.Editor = editor
  143. book.HistoryCount = historyCount
  144. book.IsDownload = 0
  145. if autoRelease {
  146. book.AutoRelease = 1
  147. } else {
  148. book.AutoRelease = 0
  149. }
  150. if isDownload {
  151. book.IsDownload = 0
  152. } else {
  153. book.IsDownload = 1
  154. }
  155. if enableShare {
  156. book.IsEnableShare = 0
  157. } else {
  158. book.IsEnableShare = 1
  159. }
  160. if isUseFirstDocument {
  161. book.IsUseFirstDocument = 1
  162. } else {
  163. book.IsUseFirstDocument = 0
  164. }
  165. if err := book.Update(); err != nil {
  166. c.JsonResult(6006, "保存失败")
  167. }
  168. bookResult.BookName = bookName
  169. bookResult.Description = description
  170. bookResult.CommentStatus = commentStatus
  171. bookResult.Label = tag
  172. beego.Info("用户 [", c.Member.Account, "] 修改了项目 ->", book)
  173. c.JsonResult(0, "ok", bookResult)
  174. }
  175. //设置项目私有状态.
  176. func (c *BookController) PrivatelyOwned() {
  177. status := c.GetString("status")
  178. if status != "open" && status != "close" {
  179. c.JsonResult(6003, "参数错误")
  180. }
  181. state := 0
  182. if status == "open" {
  183. state = 0
  184. } else {
  185. state = 1
  186. }
  187. bookResult, err := c.IsPermission()
  188. if err != nil {
  189. c.JsonResult(6001, err.Error())
  190. }
  191. //只有创始人才能变更私有状态
  192. if bookResult.RoleId != conf.BookFounder {
  193. c.JsonResult(6002, "权限不足")
  194. }
  195. book, err := models.NewBook().Find(bookResult.BookId)
  196. if err != nil {
  197. c.JsonResult(6005, "项目不存在")
  198. }
  199. book.PrivatelyOwned = state
  200. err = book.Update()
  201. if err != nil {
  202. logs.Error("PrivatelyOwned => ", err)
  203. c.JsonResult(6004, "保存失败")
  204. }
  205. beego.Info("用户 【", c.Member.Account, "]修改了项目权限 ->", state)
  206. c.JsonResult(0, "ok")
  207. }
  208. // Transfer 转让项目.
  209. func (c *BookController) Transfer() {
  210. c.Prepare()
  211. account := c.GetString("account")
  212. if account == "" {
  213. c.JsonResult(6004, "接受者账号不能为空")
  214. }
  215. member, err := models.NewMember().FindByAccount(account)
  216. if err != nil {
  217. logs.Error("FindByAccount => ", err)
  218. c.JsonResult(6005, "接受用户不存在")
  219. }
  220. if member.Status != 0 {
  221. c.JsonResult(6006, "接受用户已被禁用")
  222. }
  223. if member.MemberId == c.Member.MemberId {
  224. c.JsonResult(6007, "不能转让给自己")
  225. }
  226. bookResult, err := c.IsPermission()
  227. if err != nil {
  228. c.JsonResult(6001, err.Error())
  229. }
  230. err = models.NewRelationship().Transfer(bookResult.BookId, c.Member.MemberId, member.MemberId)
  231. if err != nil {
  232. logs.Error("Transfer => ", err)
  233. c.JsonResult(6008, err.Error())
  234. }
  235. c.JsonResult(0, "ok")
  236. }
  237. //上传项目封面.
  238. func (c *BookController) UploadCover() {
  239. bookResult, err := c.IsPermission()
  240. if err != nil {
  241. c.JsonResult(6001, err.Error())
  242. }
  243. book, err := models.NewBook().Find(bookResult.BookId)
  244. if err != nil {
  245. logs.Error("SaveBook => ", err)
  246. c.JsonResult(6002, err.Error())
  247. }
  248. file, moreFile, err := c.GetFile("image-file")
  249. if err != nil {
  250. logs.Error("获取上传文件失败 ->", err.Error())
  251. c.JsonResult(500, "读取文件异常")
  252. }
  253. defer file.Close()
  254. ext := filepath.Ext(moreFile.Filename)
  255. if !strings.EqualFold(ext, ".png") && !strings.EqualFold(ext, ".jpg") && !strings.EqualFold(ext, ".gif") && !strings.EqualFold(ext, ".jpeg") {
  256. c.JsonResult(500, "不支持的图片格式")
  257. }
  258. x1, _ := strconv.ParseFloat(c.GetString("x"), 10)
  259. y1, _ := strconv.ParseFloat(c.GetString("y"), 10)
  260. w1, _ := strconv.ParseFloat(c.GetString("width"), 10)
  261. h1, _ := strconv.ParseFloat(c.GetString("height"), 10)
  262. x := int(x1)
  263. y := int(y1)
  264. width := int(w1)
  265. height := int(h1)
  266. fileName := "cover_" + strconv.FormatInt(time.Now().UnixNano(), 16)
  267. //附件路径按照项目组织
  268. filePath := filepath.Join("uploads", book.Identify,"images", fileName+ext)
  269. path := filepath.Dir(filePath)
  270. os.MkdirAll(path, os.ModePerm)
  271. err = c.SaveToFile("image-file", filePath)
  272. if err != nil {
  273. logs.Error("", err)
  274. c.JsonResult(500, "图片保存失败")
  275. }
  276. defer func(filePath string) {
  277. os.Remove(filePath)
  278. }(filePath)
  279. //剪切图片
  280. subImg, err := graphics.ImageCopyFromFile(filePath, x, y, width, height)
  281. if err != nil {
  282. logs.Error("graphics.ImageCopyFromFile => ", err)
  283. c.JsonResult(500, "图片剪切")
  284. }
  285. filePath = filepath.Join(conf.WorkingDirectory, "uploads", time.Now().Format("200601"), fileName+"_small"+ext)
  286. //生成缩略图并保存到磁盘
  287. err = graphics.ImageResizeSaveFile(subImg, 350, 460, filePath)
  288. if err != nil {
  289. logs.Error("ImageResizeSaveFile => ", err.Error())
  290. c.JsonResult(500, "保存图片失败")
  291. }
  292. url := "/" + strings.Replace(strings.TrimPrefix(filePath, conf.WorkingDirectory), "\\", "/", -1)
  293. if strings.HasPrefix(url, "//") {
  294. url = string(url[1:])
  295. }
  296. oldCover := book.Cover
  297. book.Cover = conf.URLForWithCdnImage(url)
  298. if err := book.Update(); err != nil {
  299. c.JsonResult(6001, "保存图片失败")
  300. }
  301. //如果原封面不是默认封面则删除
  302. if oldCover != conf.GetDefaultCover() {
  303. os.Remove("." + oldCover)
  304. }
  305. beego.Info("用户[", c.Member.Account, "]上传了项目封面 ->", book.BookName, book.BookId, book.Cover)
  306. c.JsonResult(0, "ok", url)
  307. }
  308. // Users 用户列表.
  309. func (c *BookController) Users() {
  310. c.Prepare()
  311. c.TplName = "book/users.tpl"
  312. key := c.Ctx.Input.Param(":key")
  313. pageIndex, _ := c.GetInt("page", 1)
  314. if key == "" {
  315. c.Abort("404")
  316. }
  317. book, err := models.NewBookResult().FindByIdentify(key, c.Member.MemberId)
  318. if err != nil {
  319. if err == models.ErrPermissionDenied {
  320. c.Abort("403")
  321. }
  322. c.Abort("500")
  323. }
  324. c.Data["Model"] = *book
  325. members, totalCount, err := models.NewMemberRelationshipResult().FindForUsersByBookId(book.BookId, pageIndex, 15)
  326. if totalCount > 0 {
  327. pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl())
  328. c.Data["PageHtml"] = pager.HtmlPages()
  329. } else {
  330. c.Data["PageHtml"] = ""
  331. }
  332. b, err := json.Marshal(members)
  333. if err != nil {
  334. c.Data["Result"] = template.JS("[]")
  335. } else {
  336. c.Data["Result"] = template.JS(string(b))
  337. }
  338. }
  339. // Create 创建项目.
  340. func (c *BookController) Create() {
  341. if c.Ctx.Input.IsPost() {
  342. bookName := strings.TrimSpace(c.GetString("book_name", ""))
  343. identify := strings.TrimSpace(c.GetString("identify", ""))
  344. description := strings.TrimSpace(c.GetString("description", ""))
  345. privatelyOwned, _ := strconv.Atoi(c.GetString("privately_owned"))
  346. commentStatus := c.GetString("comment_status")
  347. if bookName == "" {
  348. c.JsonResult(6001, "项目名称不能为空")
  349. }
  350. if identify == "" {
  351. c.JsonResult(6002, "项目标识不能为空")
  352. }
  353. if ok, err := regexp.MatchString(`^[a-z]+[a-zA-Z0-9_\-]*$`, identify); !ok || err != nil {
  354. c.JsonResult(6003, "项目标识只能包含小写字母、数字,以及“-”和“_”符号,并且只能小写字母开头")
  355. }
  356. if strings.Count(identify, "") > 50 {
  357. c.JsonResult(6004, "文档标识不能超过50字")
  358. }
  359. if strings.Count(description, "") > 500 {
  360. c.JsonResult(6004, "项目描述不能大于500字")
  361. }
  362. if privatelyOwned != 0 && privatelyOwned != 1 {
  363. privatelyOwned = 1
  364. }
  365. if commentStatus != "open" && commentStatus != "closed" && commentStatus != "group_only" && commentStatus != "registered_only" {
  366. commentStatus = "closed"
  367. }
  368. book := models.NewBook()
  369. book.Cover = conf.GetDefaultCover()
  370. //如果客户端上传了项目封面则直接保存
  371. if file, moreFile, err := c.GetFile("image-file"); err == nil {
  372. defer file.Close()
  373. ext := filepath.Ext(moreFile.Filename)
  374. //如果上传的是图片
  375. if strings.EqualFold(ext, ".png") || strings.EqualFold(ext, ".jpg") || strings.EqualFold(ext, ".gif") || strings.EqualFold(ext, ".jpeg") {
  376. fileName := "cover_" + strconv.FormatInt(time.Now().UnixNano(), 16)
  377. filePath := filepath.Join("uploads", time.Now().Format("200601"), fileName+ext)
  378. path := filepath.Dir(filePath)
  379. os.MkdirAll(path, os.ModePerm)
  380. if err := c.SaveToFile("image-file", filePath); err == nil {
  381. url := "/" + strings.Replace(strings.TrimPrefix(filePath, conf.WorkingDirectory), "\\", "/", -1)
  382. if strings.HasPrefix(url, "//") {
  383. url = string(url[1:])
  384. }
  385. book.Cover = url
  386. }
  387. }
  388. }
  389. if books, _ := book.FindByField("identify", identify, "book_id"); len(books) > 0 {
  390. c.JsonResult(6006, "项目标识已存在")
  391. }
  392. book.BookName = bookName
  393. book.Description = description
  394. book.CommentCount = 0
  395. book.PrivatelyOwned = privatelyOwned
  396. book.CommentStatus = commentStatus
  397. book.Identify = identify
  398. book.DocCount = 0
  399. book.MemberId = c.Member.MemberId
  400. book.CommentCount = 0
  401. book.Version = time.Now().Unix()
  402. book.IsEnableShare = 0
  403. book.IsUseFirstDocument = 1
  404. book.IsDownload = 1
  405. book.AutoRelease = 0
  406. book.Editor = "markdown"
  407. book.Theme = "default"
  408. if err := book.Insert(); err != nil {
  409. logs.Error("Insert => ", err)
  410. c.JsonResult(6005, "保存项目失败")
  411. }
  412. bookResult, err := models.NewBookResult().FindByIdentify(book.Identify, c.Member.MemberId)
  413. if err != nil {
  414. beego.Error(err)
  415. }
  416. beego.Info("用户[", c.Member.Account, "]创建了项目 ->", book)
  417. c.JsonResult(0, "ok", bookResult)
  418. }
  419. c.JsonResult(6001, "error")
  420. }
  421. //复制项目
  422. func (c *BookController) Copy() {
  423. if c.Ctx.Input.IsPost() {
  424. //检查是否有复制项目的权限
  425. if _, err := c.IsPermission(); err != nil {
  426. c.JsonResult(500, err.Error())
  427. }
  428. identify := strings.TrimSpace(c.GetString("identify", ""))
  429. if identify == "" {
  430. c.JsonResult(6001, "参数错误")
  431. }
  432. book := models.NewBook()
  433. err := book.Copy(identify)
  434. if err != nil {
  435. c.JsonResult(6002, "复制项目出错")
  436. } else {
  437. bookResult, err := models.NewBookResult().FindByIdentify(book.Identify, c.Member.MemberId)
  438. if err != nil {
  439. beego.Error("查询失败")
  440. }
  441. c.JsonResult(0, "ok", bookResult)
  442. }
  443. }
  444. }
  445. //导入zip压缩包
  446. func (c *BookController) Import() {
  447. file, moreFile, err := c.GetFile("import-file")
  448. if err == http.ErrMissingFile {
  449. c.JsonResult(6003, "没有发现需要上传的文件")
  450. }
  451. defer file.Close()
  452. bookName := strings.TrimSpace(c.GetString("book_name"))
  453. identify := strings.TrimSpace(c.GetString("identify"))
  454. description := strings.TrimSpace(c.GetString("description", ""))
  455. privatelyOwned, _ := strconv.Atoi(c.GetString("privately_owned"))
  456. if bookName == "" {
  457. c.JsonResult(6001, "项目名称不能为空")
  458. }
  459. if len([]rune(bookName)) > 500 {
  460. c.JsonResult(6002, "项目名称不能大于500字")
  461. }
  462. if identify == "" {
  463. c.JsonResult(6002, "项目标识不能为空")
  464. }
  465. if ok, err := regexp.MatchString(`^[a-z]+[a-zA-Z0-9_\-]*$`, identify); !ok || err != nil {
  466. c.JsonResult(6003, "项目标识只能包含小写字母、数字,以及“-”和“_”符号,并且只能小写字母开头")
  467. }
  468. if strings.Count(identify, "") > 50 {
  469. c.JsonResult(6004, "文档标识不能超过50字")
  470. }
  471. ext := filepath.Ext(moreFile.Filename)
  472. if !strings.EqualFold(ext, ".zip") {
  473. c.JsonResult(6004, "不支持的文件类型")
  474. }
  475. if books, _ := models.NewBook().FindByField("identify", identify, "book_id"); len(books) > 0 {
  476. c.JsonResult(6006, "项目标识已存在")
  477. }
  478. tempPath := filepath.Join(os.TempDir(), c.CruSession.SessionID())
  479. os.MkdirAll(tempPath, 0766)
  480. tempPath = filepath.Join(tempPath, moreFile.Filename)
  481. err = c.SaveToFile("import-file", tempPath)
  482. book := models.NewBook()
  483. book.MemberId = c.Member.MemberId
  484. book.Cover = conf.GetDefaultCover()
  485. book.BookName = bookName
  486. book.Description = description
  487. book.CommentCount = 0
  488. book.PrivatelyOwned = privatelyOwned
  489. book.CommentStatus = "closed"
  490. book.Identify = identify
  491. book.DocCount = 0
  492. book.MemberId = c.Member.MemberId
  493. book.CommentCount = 0
  494. book.Version = time.Now().Unix()
  495. book.Editor = "markdown"
  496. book.Theme = "default"
  497. go book.ImportBook(tempPath)
  498. beego.Info("用户[", c.Member.Account, "]导入了项目 ->", book)
  499. c.JsonResult(0, "项目正在后台转换中,请稍后查看")
  500. }
  501. // CreateToken 创建访问来令牌.
  502. func (c *BookController) CreateToken() {
  503. action := c.GetString("action")
  504. bookResult, err := c.IsPermission()
  505. if err != nil {
  506. if err == models.ErrPermissionDenied {
  507. c.JsonResult(403, "权限不足")
  508. }
  509. if err == orm.ErrNoRows {
  510. c.JsonResult(404, "项目不存在")
  511. }
  512. logs.Error("生成阅读令牌失败 =>", err)
  513. c.JsonResult(6002, err.Error())
  514. }
  515. book := models.NewBook()
  516. if _, err := book.Find(bookResult.BookId); err != nil {
  517. c.JsonResult(6001, "项目不存在")
  518. }
  519. if action == "create" {
  520. if bookResult.PrivatelyOwned == 0 {
  521. c.JsonResult(6001, "公开项目不能创建阅读令牌")
  522. }
  523. book.PrivateToken = string(utils.Krand(conf.GetTokenSize(), utils.KC_RAND_KIND_ALL))
  524. if err := book.Update(); err != nil {
  525. logs.Error("生成阅读令牌失败 => ", err)
  526. c.JsonResult(6003, "生成阅读令牌失败")
  527. }
  528. beego.Info("用户[", c.Member.Account, "]创建项目令牌 ->", book.PrivateToken)
  529. c.JsonResult(0, "ok", conf.URLFor("DocumentController.Index", ":key", book.Identify, "token", book.PrivateToken))
  530. } else {
  531. book.PrivateToken = ""
  532. if err := book.Update(); err != nil {
  533. logs.Error("CreateToken => ", err)
  534. c.JsonResult(6004, "删除令牌失败")
  535. }
  536. beego.Info("用户[", c.Member.Account, "]创建项目令牌 ->", book.PrivateToken)
  537. c.JsonResult(0, "ok", "")
  538. }
  539. }
  540. // Delete 删除项目.
  541. func (c *BookController) Delete() {
  542. c.Prepare()
  543. bookResult, err := c.IsPermission()
  544. if err != nil {
  545. c.JsonResult(6001, err.Error())
  546. }
  547. if bookResult.RoleId != conf.BookFounder {
  548. c.JsonResult(6002, "只有创始人才能删除项目")
  549. }
  550. err = models.NewBook().ThoroughDeleteBook(bookResult.BookId)
  551. if err == orm.ErrNoRows {
  552. c.JsonResult(6002, "项目不存在")
  553. }
  554. if err != nil {
  555. logs.Error("删除项目 => ", err)
  556. c.JsonResult(6003, "删除失败")
  557. }
  558. beego.Info("用户[", c.Member.Account, "]删除了项目 ->", bookResult)
  559. c.JsonResult(0, "ok")
  560. }
  561. //发布项目.
  562. func (c *BookController) Release() {
  563. c.Prepare()
  564. identify := c.GetString("identify")
  565. bookId := 0
  566. if c.Member.IsAdministrator() {
  567. book, err := models.NewBook().FindByFieldFirst("identify", identify)
  568. if err != nil {
  569. }
  570. bookId = book.BookId
  571. } else {
  572. book, err := models.NewBookResult().FindByIdentify(identify, c.Member.MemberId)
  573. if err != nil {
  574. if err == models.ErrPermissionDenied {
  575. c.JsonResult(6001, "权限不足")
  576. }
  577. if err == orm.ErrNoRows {
  578. c.JsonResult(6002, "项目不存在")
  579. }
  580. beego.Error(err)
  581. c.JsonResult(6003, "未知错误")
  582. }
  583. if book.RoleId != conf.BookAdmin && book.RoleId != conf.BookFounder && book.RoleId != conf.BookEditor {
  584. c.JsonResult(6003, "权限不足")
  585. }
  586. bookId = book.BookId
  587. }
  588. go func(identify string) {
  589. models.NewBook().ReleaseContent(bookId)
  590. //当文档发布后,需要删除已缓存的转换项目
  591. outputPath := filepath.Join(conf.GetExportOutputPath(), strconv.Itoa(bookId))
  592. os.RemoveAll(outputPath)
  593. }(identify)
  594. c.JsonResult(0, "发布任务已推送到任务队列,稍后将在后台执行。")
  595. }
  596. //文档排序.
  597. func (c *BookController) SaveSort() {
  598. c.Prepare()
  599. identify := c.Ctx.Input.Param(":key")
  600. if identify == "" {
  601. c.Abort("404")
  602. }
  603. book_id := 0
  604. if c.Member.IsAdministrator() {
  605. book, err := models.NewBook().FindByFieldFirst("identify", identify)
  606. if err != nil {
  607. }
  608. book_id = book.BookId
  609. } else {
  610. bookResult, err := models.NewBookResult().FindByIdentify(identify, c.Member.MemberId)
  611. if err != nil {
  612. beego.Error("DocumentController.Edit => ", err)
  613. c.Abort("403")
  614. }
  615. if bookResult.RoleId == conf.BookObserver {
  616. c.JsonResult(6002, "项目不存在或权限不足")
  617. }
  618. book_id = bookResult.BookId
  619. }
  620. content := c.Ctx.Input.RequestBody
  621. var docs []map[string]interface{}
  622. err := json.Unmarshal(content, &docs)
  623. if err != nil {
  624. beego.Error(err)
  625. c.JsonResult(6003, "数据错误")
  626. }
  627. for _, item := range docs {
  628. if doc_id, ok := item["id"].(float64); ok {
  629. doc, err := models.NewDocument().Find(int(doc_id))
  630. if err != nil {
  631. beego.Error(err)
  632. continue
  633. }
  634. if doc.BookId != book_id {
  635. logs.Info("%s", "权限错误")
  636. continue
  637. }
  638. sort, ok := item["sort"].(float64)
  639. if !ok {
  640. beego.Info("排序数字转换失败 => ", item)
  641. continue
  642. }
  643. parent_id, ok := item["parent"].(float64)
  644. if !ok {
  645. beego.Info("父分类转换失败 => ", item)
  646. continue
  647. }
  648. if parent_id > 0 {
  649. if parent, err := models.NewDocument().Find(int(parent_id)); err != nil || parent.BookId != book_id {
  650. continue
  651. }
  652. }
  653. doc.OrderSort = int(sort)
  654. doc.ParentId = int(parent_id)
  655. if err := doc.InsertOrUpdate(); err != nil {
  656. fmt.Printf("%s", err.Error())
  657. beego.Error(err)
  658. }
  659. } else {
  660. fmt.Printf("文档ID转换失败 => %+v", item)
  661. }
  662. }
  663. c.JsonResult(0, "ok")
  664. }
  665. func (c *BookController) IsPermission() (*models.BookResult, error) {
  666. identify := c.GetString("identify")
  667. book, err := models.NewBookResult().FindByIdentify(identify, c.Member.MemberId)
  668. if err != nil {
  669. if err == models.ErrPermissionDenied {
  670. return book, errors.New("权限不足")
  671. }
  672. if err == orm.ErrNoRows {
  673. return book, errors.New("项目不存在")
  674. }
  675. return book, err
  676. }
  677. if book.RoleId != conf.BookAdmin && book.RoleId != conf.BookFounder {
  678. return book, errors.New("权限不足")
  679. }
  680. return book, nil
  681. }