report_v2.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839
  1. package services
  2. import (
  3. "archive/zip"
  4. "errors"
  5. "eta/eta_api/models"
  6. "eta/eta_api/models/report"
  7. "eta/eta_api/models/system"
  8. "eta/eta_api/utils"
  9. "fmt"
  10. "github.com/rdlucklib/rdluck_tools/file"
  11. "github.com/rdlucklib/rdluck_tools/http"
  12. "os"
  13. "path"
  14. "strconv"
  15. "time"
  16. )
  17. // AddReportAndChapter
  18. // @Description: 新增报告(包含章节)
  19. // @author: Roc
  20. // @datetime 2024-06-04 11:23:20
  21. // @param reportInfo *models.Report
  22. // @param inheritReportId int
  23. // @return err error
  24. // @return errMsg string
  25. func AddReportAndChapter(reportInfo *models.Report, inheritReportId int, grantAdminIdList []int) (err error, errMsg string) {
  26. // 获取最小分类id
  27. minClassifyId := reportInfo.ClassifyIdThird
  28. if minClassifyId <= 0 {
  29. minClassifyId = reportInfo.ClassifyIdSecond
  30. }
  31. if minClassifyId <= 0 {
  32. minClassifyId = reportInfo.ClassifyIdFirst
  33. }
  34. if minClassifyId <= 0 {
  35. errMsg = "分类异常"
  36. err = errors.New(errMsg)
  37. return
  38. }
  39. errMsg = "生成报告失败"
  40. // 报告继承
  41. if inheritReportId > 0 {
  42. inheritReport, tmpErr := models.GetReportByReportId(inheritReportId)
  43. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  44. errMsg = "获取待继承的报告失败"
  45. err = tmpErr
  46. return
  47. }
  48. if inheritReport != nil {
  49. // 判断当前的报告分类与继承的报告分类是否一致
  50. if inheritReport.ClassifyIdFirst != reportInfo.ClassifyIdFirst || inheritReport.ClassifyIdSecond != reportInfo.ClassifyIdSecond || inheritReport.ClassifyIdThird != reportInfo.ClassifyIdThird {
  51. errMsg = "分类异常,与继承的报告分类不一致"
  52. err = tmpErr
  53. return
  54. }
  55. reportInfo.ChapterType = inheritReport.ChapterType
  56. reportInfo.Content = inheritReport.Content
  57. reportInfo.ContentSub = inheritReport.ContentSub
  58. reportInfo.ContentStruct = inheritReport.ContentStruct
  59. reportInfo.HeadImg = inheritReport.HeadImg
  60. reportInfo.EndImg = inheritReport.EndImg
  61. reportInfo.CanvasColor = inheritReport.CanvasColor
  62. reportInfo.NeedSplice = inheritReport.NeedSplice
  63. reportInfo.HeadResourceId = inheritReport.HeadResourceId
  64. reportInfo.EndResourceId = inheritReport.EndResourceId
  65. reportInfo.InheritReportId = inheritReport.Id
  66. }
  67. }
  68. // 获取待生成的报告章节
  69. addChapterList, allGrantUserList, err, errMsg := getAddChapter(reportInfo, minClassifyId, inheritReportId, grantAdminIdList)
  70. // 新增报告及章节
  71. var reportId int64
  72. reportId, err = models.AddReportAndChapter(reportInfo, allGrantUserList, addChapterList)
  73. if err != nil {
  74. err = errors.New("新增报告及章节失败, Err: " + err.Error())
  75. return
  76. }
  77. reportCode := utils.MD5(strconv.Itoa(reportInfo.Id))
  78. reportInfo.ReportCode = reportCode
  79. // 修改唯一编码
  80. {
  81. go models.ModifyReportCode(reportId, reportCode)
  82. }
  83. // TODO 报告权限处理
  84. //处理权限
  85. //go func() {
  86. // permissionItems, e := models.GetPermission(req.ClassifyIdSecond)
  87. // if e != nil {
  88. // alarm_msg.SendAlarmMsg("获取权限失败,Err:"+e.Error(), 3)
  89. // return
  90. // }
  91. // for _, v := range permissionItems {
  92. // e = models.AddChartPermissionChapterMapping(v.ChartPermissionId, int64(item.Id))
  93. // if e != nil {
  94. // alarm_msg.SendAlarmMsg("新增权限失败,Err:"+e.Error(), 3)
  95. // return
  96. // }
  97. // }
  98. // // 同步crm权限
  99. // _ = EditReportPermissionSync(int64(item.Id), req.ClassifyIdSecond)
  100. //}()
  101. return
  102. }
  103. // EditReport
  104. // @Description: 修改报告的基础信息、授权用户权限
  105. // @author: Roc
  106. // @datetime 2024-06-06 17:16:58
  107. // @param reportInfo *models.Report
  108. // @param req models.EditReq
  109. // @param sysUser *system.Admin
  110. // @return err error
  111. // @return errMsg string
  112. func EditReport(reportInfo *models.Report, req models.EditReq, sysUser *system.Admin) (err error, errMsg string) {
  113. errMsg = `保存失败`
  114. var stage int
  115. if reportInfo.ClassifyNameFirst != req.ClassifyNameFirst || reportInfo.ClassifyNameSecond != req.ClassifyNameSecond {
  116. // 报告期数
  117. maxStage, _ := models.GetReportStageEdit(req.ClassifyIdFirst, req.ClassifyIdSecond, req.ClassifyIdThird, int(req.ReportId))
  118. maxStage = maxStage + 1
  119. stage = maxStage
  120. } else {
  121. stage = reportInfo.Stage
  122. }
  123. //if req.State != reportInfo.State {
  124. // recordItem := &models.ReportStateRecord{
  125. // ReportId: int(req.ReportId),
  126. // ReportType: 1,
  127. // State: req.State,
  128. // AdminId: this.SysUser.AdminId,
  129. // AdminName: this.SysUser.AdminName,
  130. // CreateTime: time.Now(),
  131. // }
  132. // go func() {
  133. // _, _ = models.AddReportStateRecord(recordItem)
  134. // }()
  135. //}
  136. //item := new(models.Report)
  137. reportInfo.ClassifyIdFirst = req.ClassifyIdFirst
  138. reportInfo.ClassifyNameFirst = req.ClassifyNameFirst
  139. reportInfo.ClassifyIdSecond = req.ClassifyIdSecond
  140. reportInfo.ClassifyNameSecond = req.ClassifyNameSecond
  141. reportInfo.ClassifyIdThird = req.ClassifyIdThird
  142. reportInfo.ClassifyNameThird = req.ClassifyNameThird
  143. reportInfo.Title = req.Title
  144. reportInfo.Abstract = req.Abstract
  145. reportInfo.Author = req.Author
  146. reportInfo.Frequency = req.Frequency
  147. //reportInfo.State = reportInfo.State // 编辑不变更状态
  148. reportInfo.Stage = stage // 编辑不变更期数
  149. //reportInfo.Content = html.EscapeString(req.Content) // 编辑不变更具体内容
  150. //reportInfo.ContentSub = html.EscapeString(contentSub) // 编辑不变更具体内容
  151. reportInfo.CreateTime = req.CreateTime
  152. //reportInfo.CollaborateType = req.CollaborateType
  153. //reportInfo.ReportLayout = req.ReportLayout
  154. if req.IsPublicPublish <= 0 {
  155. req.IsPublicPublish = 1
  156. }
  157. reportInfo.IsPublicPublish = req.IsPublicPublish
  158. reportInfo.LastModifyAdminId = sysUser.AdminId
  159. reportInfo.LastModifyAdminName = sysUser.RealName
  160. reportInfo.ModifyTime = time.Now()
  161. updateCols := []string{"ClassifyIdFirst", "ClassifyNameFirst", "ClassifyIdSecond", "ClassifyNameSecond", "ClassifyIdThird", "ClassifyNameThird", "Title", "Abstract", "Author", "Frequency", "Stage", "CreateTime", "IsPublicPublish", "LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  162. if req.HeadResourceId > 0 {
  163. reportInfo.HeadResourceId = req.HeadResourceId
  164. updateCols = append(updateCols, "HeadResourceId")
  165. }
  166. if req.EndResourceId > 0 {
  167. reportInfo.EndResourceId = req.EndResourceId
  168. updateCols = append(updateCols, "EndResourceId")
  169. }
  170. // 需要添加的报告授权数据
  171. addReportAdminList := make([]*report.ReportGrant, 0)
  172. // 待移除的报告授权数据id
  173. delReportGrantIdList := make([]int, 0)
  174. // 处理当前报告需要新增/移除的授权信息
  175. {
  176. reportGrantObj := report.ReportGrant{}
  177. // 获取当前报告已经授权的用户信息
  178. reportGrantList, tmpErr := reportGrantObj.GetGrantListById(reportInfo.Id)
  179. if tmpErr != nil {
  180. err = tmpErr
  181. return
  182. }
  183. // 当前报告已经授权的用户信息
  184. currReportAdminMap := make(map[int]*report.ReportGrant)
  185. // 需要删除的报告授权数据
  186. delReportAdminMap := make(map[int]*report.ReportGrant)
  187. for _, v := range reportGrantList {
  188. currReportAdminMap[v.AdminId] = v
  189. delReportAdminMap[v.AdminId] = v
  190. }
  191. // 先看需要新增哪些用户
  192. for _, tmpAdminId := range req.GrantAdminIdList {
  193. _, ok := currReportAdminMap[tmpAdminId]
  194. // 如果章节中需要新增的用户 已经在 报告授权用户里面,那么就忽略,可以不用新增了
  195. if ok {
  196. delete(delReportAdminMap, tmpAdminId)
  197. continue
  198. }
  199. // 如果不存在,那么就新增授权
  200. addReportAdminList = append(addReportAdminList, &report.ReportGrant{
  201. //GrantId: 0,
  202. ReportId: reportInfo.Id,
  203. AdminId: tmpAdminId,
  204. CreateTime: time.Now(),
  205. })
  206. }
  207. // 查出需要移除的授权id
  208. for _, v := range delReportAdminMap {
  209. delReportGrantIdList = append(delReportGrantIdList, v.GrantId)
  210. }
  211. }
  212. // 修改报告的基本信息,以及报告的授权用户
  213. err = models.EditReportAndPermission(reportInfo, updateCols, addReportAdminList, delReportGrantIdList)
  214. if err != nil {
  215. return
  216. }
  217. // TODO 权限处理
  218. //处理权限
  219. //go func() {
  220. // e := models.RemoveChartPermissionChapterMapping(req.ReportId)
  221. // if e != nil {
  222. // alarm_msg.SendAlarmMsg("修改删除报告权限失败,Err:"+e.Error(), 3)
  223. // return
  224. // }
  225. // permissionItems, e := models.GetPermission(req.ClassifyIdSecond)
  226. // if e != nil {
  227. // alarm_msg.SendAlarmMsg("获取权限失败,Err:"+e.Error(), 3)
  228. // return
  229. // }
  230. // for _, v := range permissionItems {
  231. // e = models.AddChartPermissionChapterMapping(v.ChartPermissionId, req.ReportId)
  232. // if e != nil {
  233. // alarm_msg.SendAlarmMsg("新增权限失败,Err:"+e.Error(), 3)
  234. // return
  235. // }
  236. // }
  237. // // 同步crm权限
  238. // _ = services.EditReportPermissionSync(req.ReportId, req.ClassifyIdSecond)
  239. //}()
  240. return
  241. }
  242. // getAddChapter
  243. // @Description: 获取待新增的报告章节列表
  244. // @author: Roc
  245. // @datetime 2024-06-04 13:10:58
  246. // @param reportInfo *models.Report
  247. // @param minClassifyId int
  248. // @param inheritReportId int
  249. // @param grantAdminIdList []int
  250. // @return chapterList []*models.ReportChapter
  251. // @return err error
  252. // @return errMsg string
  253. func getAddChapter(reportInfo *models.Report, minClassifyId, inheritReportId int, grantAdminIdList []int) (chapterList []models.AddReportChapter, allGrantUserList []*report.ReportGrant, err error, errMsg string) {
  254. // 待生成的报告章节内容
  255. chapterList = make([]models.AddReportChapter, 0)
  256. // 所有的授权用户
  257. allGrantUserList = make([]*report.ReportGrant, 0)
  258. // 报告授权的用户ID
  259. needAdminIdMap := make(map[int]bool)
  260. for _, adminId := range grantAdminIdList {
  261. needAdminIdMap[adminId] = true
  262. allGrantUserList = append(allGrantUserList, &report.ReportGrant{
  263. //GrantId: 0,
  264. //ReportId: 0,
  265. AdminId: adminId,
  266. CreateTime: time.Now(),
  267. })
  268. }
  269. if reportInfo.HasChapter != 1 {
  270. return
  271. }
  272. // TODO 弘则得单独处理启用禁用的情况
  273. //if utils.BusinessCode != utils.BusinessCodeRelease && utils.BusinessCode != utils.BusinessCodeSandbox && utils.BusinessCode != utils.BusinessCodeDebug {
  274. // br.Ret = 200
  275. // br.Success = true
  276. // br.Msg = "操作成功"
  277. // return
  278. //}
  279. errMsg = "生成报告章节失败"
  280. // 章节类型列表
  281. typeList, err := models.GetReportChapterTypeListByClassifyId(minClassifyId)
  282. if err != nil {
  283. err = errors.New(errMsg)
  284. return
  285. }
  286. // 分类章节的授权用户
  287. typeGrantListMap := make(map[int][]*report.ReportChapterGrant)
  288. // 自定义章节列表
  289. customAddChapterList := make([]models.AddReportChapter, 0)
  290. // 报告继承
  291. inheritChapterMap := make(map[int]*models.ReportChapter)
  292. if inheritReportId > 0 {
  293. // 继承待继承的报告章节内容
  294. inheritReportChapters, tmpErr := models.GetChapterListByReportId(inheritReportId)
  295. if tmpErr != nil {
  296. errMsg = "获取待继承的报告章节失败"
  297. err = tmpErr
  298. return
  299. }
  300. reportChaptersIdList := make([]int, 0)
  301. for _, v := range inheritReportChapters {
  302. reportChaptersIdList = append(reportChaptersIdList, v.ReportChapterId)
  303. }
  304. // 继承的报告章节用户map
  305. grantListMap := make(map[int][]*report.ReportChapterGrant)
  306. // 授权数据列表
  307. if len(reportChaptersIdList) > 0 {
  308. obj := report.ReportChapterGrant{}
  309. grantList, tmpErr := obj.GetGrantListByIdList(reportChaptersIdList)
  310. if tmpErr != nil {
  311. errMsg = "获取待继承的报告章节的授权用户列表失败"
  312. err = tmpErr
  313. return
  314. }
  315. for _, v := range grantList {
  316. // 如果不在报告授权的用户ID里面,那么该章节就不继承该授权用户
  317. if _, ok := needAdminIdMap[v.AdminId]; !ok {
  318. continue
  319. }
  320. currReportChapterId := v.ReportChapterId
  321. tmpGrantList, ok := grantListMap[currReportChapterId]
  322. if !ok {
  323. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  324. }
  325. v.ReportChapterId = 0
  326. v.GrantId = 0
  327. tmpGrantList = append(tmpGrantList, v)
  328. grantListMap[currReportChapterId] = tmpGrantList
  329. }
  330. }
  331. // 继承的报告章节内容
  332. for i := 0; i < len(inheritReportChapters); i++ {
  333. customChapter := inheritReportChapters[i]
  334. // 授权用户列表
  335. tmpGrantList, ok := grantListMap[customChapter.ReportChapterId]
  336. if !ok {
  337. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  338. }
  339. typeGrantListMap[customChapter.TypeId] = tmpGrantList
  340. // 判断该章节是否是系统章节,如果是的话,那就是需要额外创建的
  341. if customChapter.TypeId > 0 {
  342. inheritChapterMap[customChapter.TypeId] = customChapter
  343. continue
  344. }
  345. // 自定义的报告内容
  346. customChapter.ReportId = reportInfo.Id
  347. customChapter.ReportChapterId = 0
  348. customChapter.ClassifyIdFirst = reportInfo.ClassifyIdFirst
  349. customChapter.ClassifyNameFirst = reportInfo.ClassifyNameFirst
  350. customChapter.ClassifyIdSecond = reportInfo.ClassifyIdSecond
  351. customChapter.ClassifyNameSecond = reportInfo.ClassifyNameSecond
  352. customChapter.ClassifyIdThird = reportInfo.ClassifyIdThird
  353. customChapter.ClassifyNameThird = reportInfo.ClassifyNameThird
  354. customChapter.Stage = reportInfo.Stage
  355. customChapter.PublishState = 1
  356. customChapter.CreateTime = reportInfo.CreateTime
  357. customChapter.ModifyTime = time.Now()
  358. customChapter.LastModifyAdminId = reportInfo.LastModifyAdminId
  359. customChapter.LastModifyAdminName = reportInfo.LastModifyAdminName
  360. customChapter.ContentModifyTime = time.Now()
  361. customAddChapter := models.AddReportChapter{
  362. ReportChapter: customChapter,
  363. GrantList: tmpGrantList,
  364. }
  365. customAddChapterList = append(customAddChapterList, customAddChapter)
  366. }
  367. }
  368. // 最大排序
  369. var maxSort int
  370. for _, typeItem := range typeList {
  371. v := inheritChapterMap[typeItem.ReportChapterTypeId]
  372. chapterItem := new(models.ReportChapter)
  373. if v != nil {
  374. chapterItem.AddType = 2
  375. chapterItem.Title = v.Title
  376. chapterItem.ReportType = v.ReportType
  377. chapterItem.ClassifyIdFirst = reportInfo.ClassifyIdFirst
  378. chapterItem.ClassifyNameFirst = reportInfo.ClassifyNameFirst
  379. chapterItem.ClassifyIdSecond = reportInfo.ClassifyIdSecond
  380. chapterItem.ClassifyNameSecond = reportInfo.ClassifyNameSecond
  381. chapterItem.ClassifyIdThird = reportInfo.ClassifyIdThird
  382. chapterItem.ClassifyNameThird = reportInfo.ClassifyNameThird
  383. chapterItem.TypeId = typeItem.ReportChapterTypeId
  384. chapterItem.TypeName = typeItem.ReportChapterTypeName
  385. chapterItem.Content = v.Content
  386. chapterItem.ContentSub = v.ContentSub
  387. chapterItem.Stage = reportInfo.Stage
  388. chapterItem.PublishState = 1
  389. chapterItem.Sort = typeItem.Sort
  390. chapterItem.CreateTime = reportInfo.CreateTime
  391. chapterItem.ModifyTime = time.Now()
  392. chapterItem.LastModifyAdminId = reportInfo.LastModifyAdminId
  393. chapterItem.LastModifyAdminName = reportInfo.LastModifyAdminName
  394. chapterItem.ContentModifyTime = time.Now()
  395. chapterItem.ContentStruct = v.ContentStruct
  396. chapterItem.ReportLayout = v.ReportLayout
  397. chapterItem.ReportCreateTime = time.Now()
  398. } else {
  399. chapterItem.AddType = 1
  400. chapterItem.Title = typeItem.ReportChapterTypeName
  401. //chapterItem.ReportType = reportType
  402. chapterItem.ClassifyIdFirst = reportInfo.ClassifyIdFirst
  403. chapterItem.ClassifyNameFirst = reportInfo.ClassifyNameFirst
  404. chapterItem.ClassifyIdSecond = reportInfo.ClassifyIdSecond
  405. chapterItem.ClassifyNameSecond = reportInfo.ClassifyNameSecond
  406. chapterItem.ClassifyIdThird = reportInfo.ClassifyIdThird
  407. chapterItem.ClassifyNameThird = reportInfo.ClassifyNameThird
  408. chapterItem.TypeId = typeItem.ReportChapterTypeId
  409. chapterItem.TypeName = typeItem.ReportChapterTypeName
  410. chapterItem.Stage = reportInfo.Stage
  411. chapterItem.PublishState = 1
  412. chapterItem.Sort = typeItem.Sort
  413. chapterItem.CreateTime = reportInfo.CreateTime
  414. chapterItem.ModifyTime = time.Now()
  415. chapterItem.LastModifyAdminId = reportInfo.LastModifyAdminId
  416. chapterItem.LastModifyAdminName = reportInfo.LastModifyAdminName
  417. chapterItem.ContentModifyTime = time.Now()
  418. //chapterItem.ContentStruct = v.ContentStruct
  419. chapterItem.ReportLayout = reportInfo.ReportLayout
  420. chapterItem.ReportCreateTime = time.Now()
  421. }
  422. if typeItem.Sort > maxSort {
  423. maxSort = typeItem.Sort
  424. }
  425. // 授权用户列表
  426. tmpGrantList, ok := typeGrantListMap[chapterItem.TypeId]
  427. if !ok {
  428. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  429. }
  430. addChapter := models.AddReportChapter{
  431. ReportChapter: chapterItem,
  432. GrantList: tmpGrantList,
  433. }
  434. chapterList = append(chapterList, addChapter)
  435. }
  436. // 将自定义的章节内容添加到待生成的章节内容中
  437. for _, addChapterItem := range customAddChapterList {
  438. maxSort++
  439. addChapterItem.ReportChapter.Sort = maxSort
  440. chapterList = append(chapterList, addChapterItem)
  441. }
  442. hasGrantUserMap := make(map[int]bool)
  443. for _, grantList := range typeGrantListMap {
  444. for _, grant := range grantList {
  445. if _, ok := hasGrantUserMap[grant.AdminId]; !ok {
  446. allGrantUserList = append(allGrantUserList, &report.ReportGrant{
  447. //GrantId: 0,
  448. //ReportId: 0,
  449. AdminId: grant.AdminId,
  450. CreateTime: time.Now(),
  451. })
  452. }
  453. }
  454. }
  455. return
  456. }
  457. // AddChapterBaseInfoAndPermission
  458. // @Description: 添加章节基本信息及权限
  459. // @author: Roc
  460. // @datetime 2024-06-11 15:35:23
  461. // @param reportInfo *models.Report
  462. // @param reportChapterInfo *models.ReportChapter
  463. // @param permissionIdList []int
  464. // @param adminIdList []int
  465. // @return err error
  466. // @return errMsg string
  467. func AddChapterBaseInfoAndPermission(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, permissionIdList []int, adminIdList []int) (err error, errMsg string) {
  468. errMsg = "新增失败"
  469. if reportInfo.State == 2 {
  470. errMsg = "该报告已发布,不允许编辑"
  471. err = errors.New(errMsg)
  472. return
  473. }
  474. // 需要添加的报告章节授权数据
  475. addChapterAdminList := make([]*report.ReportChapterGrant, 0)
  476. for _, adminId := range adminIdList {
  477. //新增授权
  478. addChapterAdminList = append(addChapterAdminList, &report.ReportChapterGrant{
  479. //GrantId: 0,
  480. //ReportChapterId: reportChapterInfo.ReportChapterId,
  481. AdminId: adminId,
  482. CreateTime: time.Now(),
  483. })
  484. }
  485. // 需要添加的报告章节品种权限数据
  486. addChapterPermissionList := make([]*report.ReportChapterPermissionMapping, 0)
  487. // 品种权限
  488. for _, permissionId := range permissionIdList {
  489. // 如果不存在,那么就新增品种权限配置
  490. addChapterPermissionList = append(addChapterPermissionList, &report.ReportChapterPermissionMapping{
  491. //ReportChapterPermissionMappingId: 0,
  492. //ReportChapterId: reportChapterInfo.ReportChapterId,
  493. ChartPermissionId: permissionId,
  494. CreateTime: time.Now(),
  495. })
  496. }
  497. err = models.AddChapterBaseInfoAndPermission(reportChapterInfo, addChapterAdminList, addChapterPermissionList)
  498. return
  499. }
  500. // EditChapterBaseInfoAndPermission
  501. // @Description: 修改报告章节的基础信息、授权用户权限、品种权限
  502. // @author: Roc
  503. // @datetime 2024-06-05 11:49:11
  504. // @param reportInfo *models.Report
  505. // @param reportChapterInfo *models.ReportChapter
  506. // @param title string
  507. // @param permissionIdList []int
  508. // @param adminIdList []int
  509. // @return err error
  510. // @return errMsg string
  511. func EditChapterBaseInfoAndPermission(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, title string, permissionIdList []int, adminIdList []int) (err error, errMsg string) {
  512. errMsg = "修改失败"
  513. if reportInfo.State == 2 {
  514. errMsg = "该报告已发布,不允许编辑"
  515. err = errors.New(errMsg)
  516. return
  517. }
  518. updateCols := make([]string, 0)
  519. // 如果标题内容,那么就修改
  520. if title != `` {
  521. reportChapterInfo.Title = title
  522. reportChapterInfo.ModifyTime = time.Now()
  523. updateCols = append(updateCols, "Title", "ModifyTime")
  524. reportChapterInfo.UpdateChapter(updateCols)
  525. }
  526. reportGrantObj := report.ReportGrant{}
  527. chapterGrantObj := report.ReportChapterGrant{}
  528. chapterPermissionObj := report.ReportChapterPermissionMapping{}
  529. // 报告授权的用户map
  530. reportGrantAdminIdMap := make(map[int]bool)
  531. // 获取报告授权的用户信息
  532. {
  533. // 获取当前报告已经授权的用户信息
  534. reportGrantList, tmpErr := reportGrantObj.GetGrantListById(reportInfo.Id)
  535. if tmpErr != nil {
  536. err = tmpErr
  537. return
  538. }
  539. for _, v := range reportGrantList {
  540. reportGrantAdminIdMap[v.AdminId] = true
  541. }
  542. }
  543. // 需要添加的报告章节授权数据
  544. addChapterAdminList := make([]*report.ReportChapterGrant, 0)
  545. // 待移除的报告章节授权数据id
  546. delReportChapterGrantIdList := make([]int, 0)
  547. // 处理当前报告章节需要新增/移除的授权信息
  548. {
  549. // 获取当前章节已经授权的用户信息
  550. chapterGrantList, tmpErr := chapterGrantObj.GetGrantListById(reportChapterInfo.ReportChapterId)
  551. if tmpErr != nil {
  552. err = tmpErr
  553. return
  554. }
  555. // 当前章节已经授权的用户信息
  556. currChapterAdminMap := make(map[int]*report.ReportChapterGrant)
  557. // 需要删除的报告章节授权数据
  558. delChapterAdminMap := make(map[int]*report.ReportChapterGrant)
  559. for _, v := range chapterGrantList {
  560. currChapterAdminMap[v.AdminId] = v
  561. delChapterAdminMap[v.AdminId] = v
  562. }
  563. for _, adminId := range adminIdList {
  564. // 如果该用户 不在 报告授权的用户map 里面,说明这个用户是要移除的
  565. if _, ok := reportGrantAdminIdMap[adminId]; !ok {
  566. continue
  567. }
  568. _, ok := currChapterAdminMap[adminId]
  569. // 如果存在,那么从 “需要删除的报告章节授权数据” 的map中移除
  570. if ok {
  571. delete(delChapterAdminMap, adminId)
  572. continue
  573. }
  574. // 如果不存在,那么就新增授权
  575. addChapterAdminList = append(addChapterAdminList, &report.ReportChapterGrant{
  576. //GrantId: 0,
  577. ReportChapterId: reportChapterInfo.ReportChapterId,
  578. AdminId: adminId,
  579. CreateTime: time.Now(),
  580. })
  581. }
  582. // 查出需要移除的授权id
  583. for _, v := range delChapterAdminMap {
  584. delReportChapterGrantIdList = append(delReportChapterGrantIdList, v.GrantId)
  585. }
  586. }
  587. // 其他章节授权的用户
  588. otherChapterAdminMap := make(map[int]bool)
  589. {
  590. // 获取报告所有的章节id
  591. reportChapterIdList, tmpErr := models.GetReportChapterIdList(reportInfo.Id)
  592. if tmpErr != nil {
  593. err = tmpErr
  594. return
  595. }
  596. if len(reportChapterIdList) > 0 {
  597. list, tmpErr := chapterGrantObj.GetGrantListByIdList(reportChapterIdList)
  598. if tmpErr != nil {
  599. err = tmpErr
  600. return
  601. }
  602. for _, v := range list {
  603. // 如果是当前章节,因为涉及到重新授权,所以得过滤
  604. if v.ReportChapterId == reportChapterInfo.ReportChapterId {
  605. continue
  606. }
  607. otherChapterAdminMap[v.AdminId] = true
  608. }
  609. }
  610. }
  611. // 需要添加的报告章节品种权限数据
  612. addChapterPermissionList := make([]*report.ReportChapterPermissionMapping, 0)
  613. // 待移除的报告章节品种权限数据id
  614. delChapterPermissionMappingIdList := make([]int, 0)
  615. // 处理当前报告章节需要新增/移除的品种权限信息
  616. {
  617. // 获取当前章节已经配置的品种权限信息
  618. chapterPermissionList, tmpErr := chapterPermissionObj.GetPermissionListById(reportChapterInfo.ReportChapterId)
  619. if tmpErr != nil {
  620. err = tmpErr
  621. return
  622. }
  623. // 当前章节已经配置的品种权限信息
  624. currChapterPermissionMap := make(map[int]*report.ReportChapterPermissionMapping)
  625. // 需要删除的报告章节品种权限配置
  626. delChapterPermissionMap := make(map[int]*report.ReportChapterPermissionMapping)
  627. for _, v := range chapterPermissionList {
  628. currChapterPermissionMap[v.ChartPermissionId] = v
  629. delChapterPermissionMap[v.ChartPermissionId] = v
  630. }
  631. for _, permissionId := range permissionIdList {
  632. _, ok := currChapterPermissionMap[permissionId]
  633. // 如果存在,那么从 “需要删除的报告章节品种权限配置” 的map中移除
  634. if ok {
  635. delete(delChapterPermissionMap, permissionId)
  636. continue
  637. }
  638. // 如果不存在,那么就新增品种权限配置
  639. addChapterPermissionList = append(addChapterPermissionList, &report.ReportChapterPermissionMapping{
  640. //ReportChapterPermissionMappingId: 0,
  641. ReportChapterId: reportChapterInfo.ReportChapterId,
  642. ChartPermissionId: permissionId,
  643. CreateTime: time.Now(),
  644. })
  645. }
  646. // 查出需要移除的品种权限配置
  647. for _, v := range delChapterPermissionMap {
  648. delChapterPermissionMappingIdList = append(delChapterPermissionMappingIdList, v.ReportChapterPermissionMappingId)
  649. }
  650. }
  651. err = models.EditChapterBaseInfoAndPermission(reportChapterInfo, updateCols, addChapterAdminList, addChapterPermissionList, delReportChapterGrantIdList, delChapterPermissionMappingIdList)
  652. return
  653. }
  654. // DelChapter
  655. // @Description: 删除报告章节、授权用户权限、品种权限
  656. // @author: Roc
  657. // @datetime 2024-06-06 17:28:37
  658. // @param reportInfo *models.Report
  659. // @param reportChapterInfo *models.ReportChapter
  660. // @param sysUser *system.Admin
  661. // @return err error
  662. // @return errMsg string
  663. func DelChapter(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, sysUser *system.Admin) (err error, errMsg string) {
  664. errMsg = "删除失败"
  665. if reportInfo.State == 2 {
  666. errMsg = "该报告已发布,不允许删除"
  667. err = errors.New(errMsg)
  668. return
  669. }
  670. reportInfo.LastModifyAdminId = sysUser.AdminId
  671. reportInfo.LastModifyAdminName = sysUser.RealName
  672. reportInfo.ModifyTime = time.Now()
  673. updateReportCols := []string{"LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  674. err = models.DelChapterAndPermission(reportInfo, updateReportCols, reportChapterInfo)
  675. return
  676. }
  677. // DownloadVoice
  678. // @Description: 下载报告音频文件
  679. // @author: Roc
  680. // @datetime 2024-06-13 15:36:46
  681. // @param reportInfo *models.ReportDetail
  682. // @return savePath string
  683. // @return fileName string
  684. // @return err error
  685. // @return errMsg string
  686. func DownloadVoice(reportInfo *models.ReportDetail) (savePath, fileName string, err error, errMsg string) {
  687. errMsg = `下载失败`
  688. // 如果报告有音频,那么下载音频
  689. if reportInfo.VideoUrl != `` {
  690. savePath = time.Now().Format(utils.FormatDateTimeUnSpace) + utils.GetRandString(5) + ".mp3"
  691. fileName = reportInfo.VideoName + ".mp3"
  692. fileBody, tmpErr := http.Get(reportInfo.VideoUrl)
  693. if tmpErr != nil {
  694. err = tmpErr
  695. errMsg = "音频下载失败"
  696. return
  697. }
  698. err = file.SaveFile(fileBody, savePath)
  699. if err != nil {
  700. errMsg = "音频保存失败"
  701. return
  702. }
  703. }
  704. // 如果是章节报告,那么就下载压缩包
  705. if reportInfo.HasChapter == 1 {
  706. videoList, tmpErr := models.GetReportChapterVideoList(reportInfo.Id)
  707. if tmpErr != nil {
  708. err = tmpErr
  709. errMsg = "获取音频列表失败"
  710. return
  711. }
  712. if len(videoList) > 0 {
  713. // 创建zip
  714. savePath = time.Now().Format(utils.FormatDateTimeUnSpace) + utils.GetRandString(5) + ".zip"
  715. fileName = reportInfo.VideoName + ".zip"
  716. zipFile, tmpErr := os.Create(savePath)
  717. if tmpErr != nil {
  718. err = tmpErr
  719. return
  720. }
  721. zipWriter := zip.NewWriter(zipFile)
  722. // 生成zip过程中报错关闭
  723. defer func() {
  724. if err != nil {
  725. zipWriter.Close()
  726. zipFile.Close()
  727. }
  728. }()
  729. // 获取音频,写入zip
  730. for i := 0; i < len(videoList); i++ {
  731. item := videoList[i]
  732. if item.VideoName == "" || item.VideoUrl == "" {
  733. continue
  734. }
  735. // 音频文件后缀
  736. ext := path.Ext(item.VideoUrl)
  737. ioWriter, tmpErr := zipWriter.Create(fmt.Sprintf("%s%s", item.VideoName, ext))
  738. if tmpErr != nil {
  739. err = tmpErr
  740. if os.IsPermission(err) {
  741. fmt.Println("权限不足: ", err)
  742. return
  743. }
  744. return
  745. }
  746. var content []byte
  747. content, err = http.Get(item.VideoUrl)
  748. if err != nil {
  749. content = []byte("")
  750. }
  751. ioWriter.Write(content)
  752. }
  753. // 生成zip后关闭,否则下载文件会损坏
  754. zipWriter.Close()
  755. zipFile.Close()
  756. }
  757. }
  758. return
  759. }