report_v2.go 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960
  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. updateCols := []string{"Title", "Abstract", "Author", "Frequency", "CreateTime", "IsPublicPublish", "LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  163. if req.HeadResourceId > 0 {
  164. reportInfo.HeadResourceId = req.HeadResourceId
  165. updateCols = append(updateCols, "HeadResourceId")
  166. }
  167. if req.EndResourceId > 0 {
  168. reportInfo.EndResourceId = req.EndResourceId
  169. updateCols = append(updateCols, "EndResourceId")
  170. }
  171. // 需要添加的报告授权数据
  172. addReportAdminList := make([]*report.ReportGrant, 0)
  173. // 待移除的报告授权数据id
  174. delReportGrantIdList := make([]int, 0)
  175. // 处理当前报告需要新增/移除的授权信息
  176. {
  177. reportGrantObj := report.ReportGrant{}
  178. // 获取当前报告已经授权的用户信息
  179. reportGrantList, tmpErr := reportGrantObj.GetGrantListById(reportInfo.Id)
  180. if tmpErr != nil {
  181. err = tmpErr
  182. return
  183. }
  184. // 当前报告已经授权的用户信息
  185. currReportAdminMap := make(map[int]*report.ReportGrant)
  186. // 需要删除的报告授权数据
  187. delReportAdminMap := make(map[int]*report.ReportGrant)
  188. for _, v := range reportGrantList {
  189. currReportAdminMap[v.AdminId] = v
  190. delReportAdminMap[v.AdminId] = v
  191. }
  192. // 先看需要新增哪些用户
  193. for _, tmpAdminId := range req.GrantAdminIdList {
  194. _, ok := currReportAdminMap[tmpAdminId]
  195. // 如果章节中需要新增的用户 已经在 报告授权用户里面,那么就忽略,可以不用新增了
  196. if ok {
  197. delete(delReportAdminMap, tmpAdminId)
  198. continue
  199. }
  200. // 如果不存在,那么就新增授权
  201. addReportAdminList = append(addReportAdminList, &report.ReportGrant{
  202. //GrantId: 0,
  203. ReportId: reportInfo.Id,
  204. AdminId: tmpAdminId,
  205. CreateTime: time.Now(),
  206. })
  207. }
  208. // 查出需要移除的授权id
  209. for _, v := range delReportAdminMap {
  210. delReportGrantIdList = append(delReportGrantIdList, v.GrantId)
  211. }
  212. }
  213. // 修改报告的基本信息,以及报告的授权用户
  214. err = models.EditReportAndPermission(reportInfo, updateCols, addReportAdminList, delReportGrantIdList)
  215. if err != nil {
  216. return
  217. }
  218. // TODO 权限处理
  219. //处理权限
  220. //go func() {
  221. // e := models.RemoveChartPermissionChapterMapping(req.ReportId)
  222. // if e != nil {
  223. // alarm_msg.SendAlarmMsg("修改删除报告权限失败,Err:"+e.Error(), 3)
  224. // return
  225. // }
  226. // permissionItems, e := models.GetPermission(req.ClassifyIdSecond)
  227. // if e != nil {
  228. // alarm_msg.SendAlarmMsg("获取权限失败,Err:"+e.Error(), 3)
  229. // return
  230. // }
  231. // for _, v := range permissionItems {
  232. // e = models.AddChartPermissionChapterMapping(v.ChartPermissionId, req.ReportId)
  233. // if e != nil {
  234. // alarm_msg.SendAlarmMsg("新增权限失败,Err:"+e.Error(), 3)
  235. // return
  236. // }
  237. // }
  238. // // 同步crm权限
  239. // _ = services.EditReportPermissionSync(req.ReportId, req.ClassifyIdSecond)
  240. //}()
  241. return
  242. }
  243. // getAddChapter
  244. // @Description: 获取待新增的报告章节列表
  245. // @author: Roc
  246. // @datetime 2024-06-04 13:10:58
  247. // @param reportInfo *models.Report
  248. // @param minClassifyId int
  249. // @param inheritReportId int
  250. // @param grantAdminIdList []int
  251. // @return chapterList []*models.ReportChapter
  252. // @return err error
  253. // @return errMsg string
  254. func getAddChapter(reportInfo *models.Report, minClassifyId, inheritReportId int, grantAdminIdList []int) (chapterList []models.AddReportChapter, allGrantUserList []*report.ReportGrant, err error, errMsg string) {
  255. // 待生成的报告章节内容
  256. chapterList = make([]models.AddReportChapter, 0)
  257. // 所有的授权用户
  258. allGrantUserList = make([]*report.ReportGrant, 0)
  259. // 报告授权的用户ID
  260. needAdminIdMap := make(map[int]bool)
  261. for _, adminId := range grantAdminIdList {
  262. needAdminIdMap[adminId] = true
  263. allGrantUserList = append(allGrantUserList, &report.ReportGrant{
  264. //GrantId: 0,
  265. //ReportId: 0,
  266. AdminId: adminId,
  267. CreateTime: time.Now(),
  268. })
  269. }
  270. if reportInfo.HasChapter != 1 {
  271. return
  272. }
  273. // 最小单元的分类
  274. var minClassifyName string
  275. if reportInfo.ClassifyIdThird == minClassifyId {
  276. minClassifyName = reportInfo.ClassifyNameThird
  277. } else if reportInfo.ClassifyIdSecond == minClassifyId {
  278. minClassifyName = reportInfo.ClassifyNameSecond
  279. } else {
  280. minClassifyName = reportInfo.ClassifyNameFirst
  281. }
  282. // TODO 弘则得单独处理启用禁用的情况
  283. //if utils.BusinessCode != utils.BusinessCodeRelease && utils.BusinessCode != utils.BusinessCodeSandbox && utils.BusinessCode != utils.BusinessCodeDebug {
  284. // br.Ret = 200
  285. // br.Success = true
  286. // br.Msg = "操作成功"
  287. // return
  288. //}
  289. errMsg = "生成报告章节失败"
  290. // 章节类型列表
  291. allTypeList, err := models.GetReportChapterTypeListByClassifyId(minClassifyId)
  292. if err != nil {
  293. err = errors.New(errMsg)
  294. return
  295. }
  296. // 待添加的章节
  297. typeList := make([]*models.ReportChapterType, 0)
  298. nowTime := time.Now().Local()
  299. for _, chapterType := range allTypeList {
  300. // 如果被永久暂停更新了
  301. if chapterType.Enabled == 0 { //该章节已被永久禁用,那么就不允许继承或者新增该章节
  302. continue
  303. }
  304. if chapterType.PauseStartTime != "" && chapterType.PauseEndTime != "" && chapterType.PauseStartTime != utils.EmptyDateStr && chapterType.PauseEndTime != utils.EmptyDateStr {
  305. startTime, timeErr := time.ParseInLocation(utils.FormatDate, chapterType.PauseStartTime, time.Local)
  306. if timeErr != nil {
  307. utils.FileLog.Error("更新规则时间转换失败4001, Err: " + timeErr.Error())
  308. continue
  309. }
  310. endTime, timeErr := time.ParseInLocation(utils.FormatDate, chapterType.PauseEndTime, time.Local)
  311. if timeErr != nil {
  312. utils.FileLog.Error("更新规则时间转换失败4002, Err: " + timeErr.Error())
  313. continue
  314. }
  315. // 暂停更新
  316. if nowTime.After(startTime) && nowTime.Before(endTime.AddDate(0, 0, 1)) {
  317. continue
  318. }
  319. typeList = append(typeList, chapterType)
  320. }
  321. }
  322. // 分类章节的授权用户
  323. typeGrantListMap := make(map[int][]*report.ReportChapterGrant)
  324. // 自定义章节列表
  325. customAddChapterList := make([]models.AddReportChapter, 0)
  326. // 报告继承
  327. inheritChapterMap := make(map[int]*models.ReportChapter)
  328. if inheritReportId > 0 {
  329. // 继承待继承的报告章节内容
  330. inheritReportChapters, tmpErr := models.GetChapterListByReportId(inheritReportId)
  331. if tmpErr != nil {
  332. errMsg = "获取待继承的报告章节失败"
  333. err = tmpErr
  334. return
  335. }
  336. reportChaptersIdList := make([]int, 0)
  337. for _, v := range inheritReportChapters {
  338. reportChaptersIdList = append(reportChaptersIdList, v.ReportChapterId)
  339. }
  340. // 继承的报告章节用户map
  341. grantListMap := make(map[int][]*report.ReportChapterGrant)
  342. // 授权数据列表
  343. if len(reportChaptersIdList) > 0 {
  344. obj := report.ReportChapterGrant{}
  345. grantList, tmpErr := obj.GetGrantListByIdList(reportChaptersIdList)
  346. if tmpErr != nil {
  347. errMsg = "获取待继承的报告章节的授权用户列表失败"
  348. err = tmpErr
  349. return
  350. }
  351. for _, v := range grantList {
  352. // 如果不在报告授权的用户ID里面,那么该章节就不继承该授权用户
  353. if _, ok := needAdminIdMap[v.AdminId]; !ok {
  354. continue
  355. }
  356. currReportChapterId := v.ReportChapterId
  357. tmpGrantList, ok := grantListMap[currReportChapterId]
  358. if !ok {
  359. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  360. }
  361. v.ReportChapterId = 0
  362. v.GrantId = 0
  363. tmpGrantList = append(tmpGrantList, v)
  364. grantListMap[currReportChapterId] = tmpGrantList
  365. }
  366. }
  367. // 继承的报告章节内容
  368. for i := 0; i < len(inheritReportChapters); i++ {
  369. customChapter := inheritReportChapters[i]
  370. // 授权用户列表
  371. tmpGrantList, ok := grantListMap[customChapter.ReportChapterId]
  372. if !ok {
  373. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  374. }
  375. typeGrantListMap[customChapter.TypeId] = tmpGrantList
  376. // 判断该章节是否是系统章节,如果是的话,那就是需要额外创建的
  377. if customChapter.TypeId > 0 {
  378. inheritChapterMap[customChapter.TypeId] = customChapter
  379. continue
  380. }
  381. // 自定义的报告内容
  382. customChapter.ReportId = reportInfo.Id
  383. customChapter.ReportChapterId = 0
  384. customChapter.ClassifyIdFirst = minClassifyId
  385. customChapter.ClassifyNameFirst = minClassifyName
  386. customChapter.Stage = reportInfo.Stage
  387. customChapter.PublishState = 1
  388. customChapter.CreateTime = reportInfo.CreateTime
  389. customChapter.ModifyTime = time.Now()
  390. customChapter.LastModifyAdminId = reportInfo.LastModifyAdminId
  391. customChapter.LastModifyAdminName = reportInfo.LastModifyAdminName
  392. customChapter.ContentModifyTime = time.Now()
  393. customAddChapter := models.AddReportChapter{
  394. ReportChapter: customChapter,
  395. GrantList: tmpGrantList,
  396. }
  397. customAddChapterList = append(customAddChapterList, customAddChapter)
  398. }
  399. }
  400. // 最大排序
  401. var maxSort int
  402. for _, typeItem := range typeList {
  403. v := inheritChapterMap[typeItem.ReportChapterTypeId]
  404. chapterItem := new(models.ReportChapter)
  405. if v != nil {
  406. chapterItem.AddType = 2
  407. chapterItem.Title = v.Title
  408. chapterItem.ReportType = v.ReportType
  409. chapterItem.ClassifyIdFirst = minClassifyId
  410. chapterItem.ClassifyNameFirst = minClassifyName
  411. chapterItem.TypeId = typeItem.ReportChapterTypeId
  412. chapterItem.TypeName = typeItem.ReportChapterTypeName
  413. chapterItem.Content = v.Content
  414. chapterItem.ContentSub = v.ContentSub
  415. chapterItem.Stage = reportInfo.Stage
  416. chapterItem.PublishState = 1
  417. chapterItem.Sort = typeItem.Sort
  418. chapterItem.CreateTime = reportInfo.CreateTime
  419. chapterItem.ModifyTime = time.Now()
  420. chapterItem.LastModifyAdminId = reportInfo.LastModifyAdminId
  421. chapterItem.LastModifyAdminName = reportInfo.LastModifyAdminName
  422. chapterItem.ContentModifyTime = time.Now()
  423. chapterItem.ContentStruct = v.ContentStruct
  424. chapterItem.ReportLayout = v.ReportLayout
  425. chapterItem.ReportCreateTime = time.Now()
  426. } else {
  427. chapterItem.AddType = 1
  428. chapterItem.Title = typeItem.ReportChapterTypeName
  429. chapterItem.ReportType = typeItem.ResearchType
  430. chapterItem.ClassifyIdFirst = minClassifyId
  431. chapterItem.ClassifyNameFirst = minClassifyName
  432. chapterItem.TypeId = typeItem.ReportChapterTypeId
  433. chapterItem.TypeName = typeItem.ReportChapterTypeName
  434. chapterItem.Stage = reportInfo.Stage
  435. chapterItem.PublishState = 1
  436. chapterItem.Sort = typeItem.Sort
  437. chapterItem.CreateTime = reportInfo.CreateTime
  438. chapterItem.ModifyTime = time.Now()
  439. chapterItem.LastModifyAdminId = reportInfo.LastModifyAdminId
  440. chapterItem.LastModifyAdminName = reportInfo.LastModifyAdminName
  441. chapterItem.ContentModifyTime = time.Now()
  442. //chapterItem.ContentStruct = v.ContentStruct
  443. chapterItem.ReportLayout = reportInfo.ReportLayout
  444. chapterItem.ReportCreateTime = time.Now()
  445. }
  446. if typeItem.Sort > maxSort {
  447. maxSort = typeItem.Sort
  448. }
  449. // 授权用户列表
  450. tmpGrantList, ok := typeGrantListMap[chapterItem.TypeId]
  451. if !ok {
  452. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  453. }
  454. addChapter := models.AddReportChapter{
  455. ReportChapter: chapterItem,
  456. GrantList: tmpGrantList,
  457. }
  458. chapterList = append(chapterList, addChapter)
  459. }
  460. // 将自定义的章节内容添加到待生成的章节内容中
  461. for _, addChapterItem := range customAddChapterList {
  462. maxSort++
  463. addChapterItem.ReportChapter.Sort = maxSort
  464. chapterList = append(chapterList, addChapterItem)
  465. }
  466. hasGrantUserMap := make(map[int]bool)
  467. for _, grantList := range typeGrantListMap {
  468. for _, grant := range grantList {
  469. if _, ok := hasGrantUserMap[grant.AdminId]; !ok {
  470. allGrantUserList = append(allGrantUserList, &report.ReportGrant{
  471. //GrantId: 0,
  472. //ReportId: 0,
  473. AdminId: grant.AdminId,
  474. CreateTime: time.Now(),
  475. })
  476. }
  477. }
  478. }
  479. return
  480. }
  481. // AddChapterBaseInfoAndPermission
  482. // @Description: 添加章节基本信息及权限
  483. // @author: Roc
  484. // @datetime 2024-06-11 15:35:23
  485. // @param reportInfo *models.Report
  486. // @param reportChapterInfo *models.ReportChapter
  487. // @param permissionIdList []int
  488. // @param adminIdList []int
  489. // @return err error
  490. // @return errMsg string
  491. func AddChapterBaseInfoAndPermission(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, permissionIdList []int, adminIdList []int) (err error, errMsg string) {
  492. errMsg = "新增失败"
  493. if reportInfo.State == 2 {
  494. errMsg = "该报告已发布,不允许编辑"
  495. err = errors.New(errMsg)
  496. return
  497. }
  498. // 需要添加的报告章节授权数据
  499. addChapterAdminList := make([]*report.ReportChapterGrant, 0)
  500. for _, adminId := range adminIdList {
  501. //新增授权
  502. addChapterAdminList = append(addChapterAdminList, &report.ReportChapterGrant{
  503. //GrantId: 0,
  504. //ReportChapterId: reportChapterInfo.ReportChapterId,
  505. AdminId: adminId,
  506. CreateTime: time.Now(),
  507. })
  508. }
  509. // 需要添加的报告章节品种权限数据
  510. addChapterPermissionList := make([]*report.ReportChapterPermissionMapping, 0)
  511. // 品种权限
  512. for _, permissionId := range permissionIdList {
  513. // 如果不存在,那么就新增品种权限配置
  514. addChapterPermissionList = append(addChapterPermissionList, &report.ReportChapterPermissionMapping{
  515. //ReportChapterPermissionMappingId: 0,
  516. //ReportChapterId: reportChapterInfo.ReportChapterId,
  517. ChartPermissionId: permissionId,
  518. CreateTime: time.Now(),
  519. })
  520. }
  521. err = models.AddChapterBaseInfoAndPermission(reportChapterInfo, addChapterAdminList, addChapterPermissionList)
  522. return
  523. }
  524. // EditChapterBaseInfoAndPermission
  525. // @Description: 修改报告章节的基础信息、授权用户权限、品种权限
  526. // @author: Roc
  527. // @datetime 2024-06-05 11:49:11
  528. // @param reportInfo *models.Report
  529. // @param reportChapterInfo *models.ReportChapter
  530. // @param title string
  531. // @param permissionIdList []int
  532. // @param adminIdList []int
  533. // @return err error
  534. // @return errMsg string
  535. func EditChapterBaseInfoAndPermission(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, title string, permissionIdList []int, adminIdList []int) (err error, errMsg string) {
  536. errMsg = "修改失败"
  537. if reportInfo.State == 2 {
  538. errMsg = "该报告已发布,不允许编辑"
  539. err = errors.New(errMsg)
  540. return
  541. }
  542. updateCols := make([]string, 0)
  543. // 如果标题内容,那么就修改
  544. if title != `` {
  545. reportChapterInfo.Title = title
  546. reportChapterInfo.ModifyTime = time.Now()
  547. updateCols = append(updateCols, "Title", "ModifyTime")
  548. reportChapterInfo.UpdateChapter(updateCols)
  549. }
  550. reportGrantObj := report.ReportGrant{}
  551. chapterGrantObj := report.ReportChapterGrant{}
  552. chapterPermissionObj := report.ReportChapterPermissionMapping{}
  553. // 报告授权的用户map
  554. reportGrantAdminIdMap := make(map[int]bool)
  555. // 获取报告授权的用户信息
  556. {
  557. // 获取当前报告已经授权的用户信息
  558. reportGrantList, tmpErr := reportGrantObj.GetGrantListById(reportInfo.Id)
  559. if tmpErr != nil {
  560. err = tmpErr
  561. return
  562. }
  563. for _, v := range reportGrantList {
  564. reportGrantAdminIdMap[v.AdminId] = true
  565. }
  566. }
  567. // 需要添加的报告章节授权数据
  568. addChapterAdminList := make([]*report.ReportChapterGrant, 0)
  569. // 待移除的报告章节授权数据id
  570. delReportChapterGrantIdList := make([]int, 0)
  571. // 处理当前报告章节需要新增/移除的授权信息
  572. {
  573. // 获取当前章节已经授权的用户信息
  574. chapterGrantList, tmpErr := chapterGrantObj.GetGrantListById(reportChapterInfo.ReportChapterId)
  575. if tmpErr != nil {
  576. err = tmpErr
  577. return
  578. }
  579. // 当前章节已经授权的用户信息
  580. currChapterAdminMap := make(map[int]*report.ReportChapterGrant)
  581. // 需要删除的报告章节授权数据
  582. delChapterAdminMap := make(map[int]*report.ReportChapterGrant)
  583. for _, v := range chapterGrantList {
  584. currChapterAdminMap[v.AdminId] = v
  585. delChapterAdminMap[v.AdminId] = v
  586. }
  587. for _, adminId := range adminIdList {
  588. // 如果该用户 不在 报告授权的用户map 里面,说明这个用户是要移除的
  589. if _, ok := reportGrantAdminIdMap[adminId]; !ok {
  590. continue
  591. }
  592. _, ok := currChapterAdminMap[adminId]
  593. // 如果存在,那么从 “需要删除的报告章节授权数据” 的map中移除
  594. if ok {
  595. delete(delChapterAdminMap, adminId)
  596. continue
  597. }
  598. // 如果不存在,那么就新增授权
  599. addChapterAdminList = append(addChapterAdminList, &report.ReportChapterGrant{
  600. //GrantId: 0,
  601. ReportChapterId: reportChapterInfo.ReportChapterId,
  602. AdminId: adminId,
  603. CreateTime: time.Now(),
  604. })
  605. }
  606. // 查出需要移除的授权id
  607. for _, v := range delChapterAdminMap {
  608. delReportChapterGrantIdList = append(delReportChapterGrantIdList, v.GrantId)
  609. }
  610. }
  611. // 其他章节授权的用户
  612. otherChapterAdminMap := make(map[int]bool)
  613. {
  614. // 获取报告所有的章节id
  615. reportChapterIdList, tmpErr := models.GetReportChapterIdList(reportInfo.Id)
  616. if tmpErr != nil {
  617. err = tmpErr
  618. return
  619. }
  620. if len(reportChapterIdList) > 0 {
  621. list, tmpErr := chapterGrantObj.GetGrantListByIdList(reportChapterIdList)
  622. if tmpErr != nil {
  623. err = tmpErr
  624. return
  625. }
  626. for _, v := range list {
  627. // 如果是当前章节,因为涉及到重新授权,所以得过滤
  628. if v.ReportChapterId == reportChapterInfo.ReportChapterId {
  629. continue
  630. }
  631. otherChapterAdminMap[v.AdminId] = true
  632. }
  633. }
  634. }
  635. // 需要添加的报告章节品种权限数据
  636. addChapterPermissionList := make([]*report.ReportChapterPermissionMapping, 0)
  637. // 待移除的报告章节品种权限数据id
  638. delChapterPermissionMappingIdList := make([]int, 0)
  639. // 处理当前报告章节需要新增/移除的品种权限信息
  640. {
  641. // 获取当前章节已经配置的品种权限信息
  642. chapterPermissionList, tmpErr := chapterPermissionObj.GetPermissionListById(reportChapterInfo.ReportChapterId)
  643. if tmpErr != nil {
  644. err = tmpErr
  645. return
  646. }
  647. // 当前章节已经配置的品种权限信息
  648. currChapterPermissionMap := make(map[int]*report.ReportChapterPermissionMapping)
  649. // 需要删除的报告章节品种权限配置
  650. delChapterPermissionMap := make(map[int]*report.ReportChapterPermissionMapping)
  651. for _, v := range chapterPermissionList {
  652. currChapterPermissionMap[v.ChartPermissionId] = v
  653. delChapterPermissionMap[v.ChartPermissionId] = v
  654. }
  655. for _, permissionId := range permissionIdList {
  656. _, ok := currChapterPermissionMap[permissionId]
  657. // 如果存在,那么从 “需要删除的报告章节品种权限配置” 的map中移除
  658. if ok {
  659. delete(delChapterPermissionMap, permissionId)
  660. continue
  661. }
  662. // 如果不存在,那么就新增品种权限配置
  663. addChapterPermissionList = append(addChapterPermissionList, &report.ReportChapterPermissionMapping{
  664. //ReportChapterPermissionMappingId: 0,
  665. ReportChapterId: reportChapterInfo.ReportChapterId,
  666. ChartPermissionId: permissionId,
  667. CreateTime: time.Now(),
  668. })
  669. }
  670. // 查出需要移除的品种权限配置
  671. for _, v := range delChapterPermissionMap {
  672. delChapterPermissionMappingIdList = append(delChapterPermissionMappingIdList, v.ReportChapterPermissionMappingId)
  673. }
  674. }
  675. err = models.EditChapterBaseInfoAndPermission(reportChapterInfo, updateCols, addChapterAdminList, addChapterPermissionList, delReportChapterGrantIdList, delChapterPermissionMappingIdList)
  676. return
  677. }
  678. // DelChapter
  679. // @Description: 删除报告章节、授权用户权限、品种权限
  680. // @author: Roc
  681. // @datetime 2024-06-06 17:28:37
  682. // @param reportInfo *models.Report
  683. // @param reportChapterInfo *models.ReportChapter
  684. // @param sysUser *system.Admin
  685. // @return err error
  686. // @return errMsg string
  687. func DelChapter(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, sysUser *system.Admin) (err error, errMsg string) {
  688. errMsg = "删除失败"
  689. if reportInfo.State == 2 {
  690. errMsg = "该报告已发布,不允许删除"
  691. err = errors.New(errMsg)
  692. return
  693. }
  694. reportInfo.LastModifyAdminId = sysUser.AdminId
  695. reportInfo.LastModifyAdminName = sysUser.RealName
  696. reportInfo.ModifyTime = time.Now()
  697. updateReportCols := []string{"LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  698. err = models.DelChapterAndPermission(reportInfo, updateReportCols, reportChapterInfo)
  699. return
  700. }
  701. // DownloadVoice
  702. // @Description: 下载报告音频文件
  703. // @author: Roc
  704. // @datetime 2024-06-13 15:36:46
  705. // @param reportInfo *models.ReportDetail
  706. // @return savePath string
  707. // @return fileName string
  708. // @return err error
  709. // @return errMsg string
  710. func DownloadVoice(reportInfo *models.ReportDetail) (savePath, fileName string, err error, errMsg string) {
  711. errMsg = `下载失败`
  712. // 如果报告有音频,那么下载音频
  713. if reportInfo.VideoUrl != `` {
  714. savePath = time.Now().Format(utils.FormatDateTimeUnSpace) + utils.GetRandString(5) + ".mp3"
  715. fileName = reportInfo.VideoName + ".mp3"
  716. fileBody, tmpErr := http.Get(reportInfo.VideoUrl)
  717. if tmpErr != nil {
  718. err = tmpErr
  719. errMsg = "音频下载失败"
  720. return
  721. }
  722. err = file.SaveFile(fileBody, savePath)
  723. if err != nil {
  724. errMsg = "音频保存失败"
  725. return
  726. }
  727. }
  728. // 如果是章节报告,那么就下载压缩包
  729. if reportInfo.HasChapter == 1 {
  730. videoList, tmpErr := models.GetReportChapterVideoList(reportInfo.Id)
  731. if tmpErr != nil {
  732. err = tmpErr
  733. errMsg = "获取音频列表失败"
  734. return
  735. }
  736. if len(videoList) > 0 {
  737. // 创建zip
  738. savePath = time.Now().Format(utils.FormatDateTimeUnSpace) + utils.GetRandString(5) + ".zip"
  739. fileName = reportInfo.VideoName + ".zip"
  740. zipFile, tmpErr := os.Create(savePath)
  741. if tmpErr != nil {
  742. err = tmpErr
  743. return
  744. }
  745. zipWriter := zip.NewWriter(zipFile)
  746. // 生成zip过程中报错关闭
  747. defer func() {
  748. if err != nil {
  749. zipWriter.Close()
  750. zipFile.Close()
  751. }
  752. }()
  753. // 获取音频,写入zip
  754. for i := 0; i < len(videoList); i++ {
  755. item := videoList[i]
  756. if item.VideoName == "" || item.VideoUrl == "" {
  757. continue
  758. }
  759. // 音频文件后缀
  760. ext := path.Ext(item.VideoUrl)
  761. ioWriter, tmpErr := zipWriter.Create(fmt.Sprintf("%s%s", item.VideoName, ext))
  762. if tmpErr != nil {
  763. err = tmpErr
  764. if os.IsPermission(err) {
  765. fmt.Println("权限不足: ", err)
  766. return
  767. }
  768. return
  769. }
  770. var content []byte
  771. content, err = http.Get(item.VideoUrl)
  772. if err != nil {
  773. content = []byte("")
  774. }
  775. ioWriter.Write(content)
  776. }
  777. // 生成zip后关闭,否则下载文件会损坏
  778. zipWriter.Close()
  779. zipFile.Close()
  780. }
  781. }
  782. return
  783. }
  784. // CheckReportAuthByAdminIdList
  785. // @Description: 根据管理员id列表,判断当前用户是否有报告权限
  786. // @author: Roc
  787. // @datetime 2024-06-13 11:03:10
  788. // @param adminId int
  789. // @param createAdminId int
  790. // @param grantAdminIdList []int
  791. // @return isAuth bool
  792. func CheckReportAuthByAdminIdList(adminId, createAdminId int, grantAdminIdList []int) (isAuth bool) {
  793. // 如果是自己创建的报告,那么就有权限
  794. if adminId == createAdminId {
  795. isAuth = true
  796. return
  797. }
  798. // 如果是授权用户,那么就有权限
  799. if utils.IsCheckInList(grantAdminIdList, adminId) {
  800. isAuth = true
  801. return
  802. }
  803. return
  804. }
  805. // CheckReportAuthByReportChapterInfo
  806. // @Description: 根据报告ID,判断当前用户是否有报告权限
  807. // @author: Roc
  808. // @datetime 2024-06-13 16:21:28
  809. // @param adminId int
  810. // @param reportInfoId int
  811. // @return isAuth bool
  812. // @return err error
  813. func CheckReportAuthByReportChapterInfo(adminId, createAdminId int, reportInfoId int) (isAuth bool, err error) {
  814. // 如果是自己创建的报告,那么就有权限
  815. if adminId == createAdminId {
  816. isAuth = true
  817. return
  818. }
  819. obj := report.ReportGrant{}
  820. chapterGrantList, err := obj.GetGrantListById(reportInfoId)
  821. if err != nil {
  822. return
  823. }
  824. for _, v := range chapterGrantList {
  825. if v.AdminId == adminId {
  826. isAuth = true
  827. return
  828. }
  829. }
  830. return
  831. }
  832. // EditReportLayoutImg
  833. // @Description: 修改报告的版图信息
  834. // @author: Roc
  835. // @datetime 2024-06-18 11:35:00
  836. // @param reportInfo *models.Report
  837. // @param req models.EditLayoutImgReq
  838. // @param sysUser *system.Admin
  839. // @return err error
  840. // @return errMsg string
  841. func EditReportLayoutImg(reportInfo *models.Report, req models.EditLayoutImgReq, sysUser *system.Admin) (err error, errMsg string) {
  842. errMsg = `保存失败`
  843. reportInfo.CanvasColor = req.CanvasColor
  844. reportInfo.LastModifyAdminId = sysUser.AdminId
  845. reportInfo.LastModifyAdminName = sysUser.RealName
  846. reportInfo.ModifyTime = time.Now()
  847. updateCols := []string{"CanvasColor", "LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  848. if req.HeadResourceId > 0 {
  849. reportInfo.HeadResourceId = req.HeadResourceId
  850. reportInfo.HeadImg = req.HeadImg
  851. updateCols = append(updateCols, "HeadResourceId", "HeadImg")
  852. }
  853. if req.EndResourceId > 0 {
  854. reportInfo.EndResourceId = req.EndResourceId
  855. reportInfo.EndImg = req.EndImg
  856. updateCols = append(updateCols, "EndResourceId", "EndImg")
  857. }
  858. err = reportInfo.UpdateReport(updateCols)
  859. if err != nil {
  860. return
  861. }
  862. return
  863. }