report_v2.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931
  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. typeList, err := models.GetReportChapterTypeListByClassifyId(minClassifyId)
  292. if err != nil {
  293. err = errors.New(errMsg)
  294. return
  295. }
  296. // 分类章节的授权用户
  297. typeGrantListMap := make(map[int][]*report.ReportChapterGrant)
  298. // 自定义章节列表
  299. customAddChapterList := make([]models.AddReportChapter, 0)
  300. // 报告继承
  301. inheritChapterMap := make(map[int]*models.ReportChapter)
  302. if inheritReportId > 0 {
  303. // 继承待继承的报告章节内容
  304. inheritReportChapters, tmpErr := models.GetChapterListByReportId(inheritReportId)
  305. if tmpErr != nil {
  306. errMsg = "获取待继承的报告章节失败"
  307. err = tmpErr
  308. return
  309. }
  310. reportChaptersIdList := make([]int, 0)
  311. for _, v := range inheritReportChapters {
  312. reportChaptersIdList = append(reportChaptersIdList, v.ReportChapterId)
  313. }
  314. // 继承的报告章节用户map
  315. grantListMap := make(map[int][]*report.ReportChapterGrant)
  316. // 授权数据列表
  317. if len(reportChaptersIdList) > 0 {
  318. obj := report.ReportChapterGrant{}
  319. grantList, tmpErr := obj.GetGrantListByIdList(reportChaptersIdList)
  320. if tmpErr != nil {
  321. errMsg = "获取待继承的报告章节的授权用户列表失败"
  322. err = tmpErr
  323. return
  324. }
  325. for _, v := range grantList {
  326. // 如果不在报告授权的用户ID里面,那么该章节就不继承该授权用户
  327. if _, ok := needAdminIdMap[v.AdminId]; !ok {
  328. continue
  329. }
  330. currReportChapterId := v.ReportChapterId
  331. tmpGrantList, ok := grantListMap[currReportChapterId]
  332. if !ok {
  333. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  334. }
  335. v.ReportChapterId = 0
  336. v.GrantId = 0
  337. tmpGrantList = append(tmpGrantList, v)
  338. grantListMap[currReportChapterId] = tmpGrantList
  339. }
  340. }
  341. // 继承的报告章节内容
  342. for i := 0; i < len(inheritReportChapters); i++ {
  343. customChapter := inheritReportChapters[i]
  344. // 授权用户列表
  345. tmpGrantList, ok := grantListMap[customChapter.ReportChapterId]
  346. if !ok {
  347. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  348. }
  349. typeGrantListMap[customChapter.TypeId] = tmpGrantList
  350. // 判断该章节是否是系统章节,如果是的话,那就是需要额外创建的
  351. if customChapter.TypeId > 0 {
  352. inheritChapterMap[customChapter.TypeId] = customChapter
  353. continue
  354. }
  355. // 自定义的报告内容
  356. customChapter.ReportId = reportInfo.Id
  357. customChapter.ReportChapterId = 0
  358. customChapter.ClassifyIdFirst = minClassifyId
  359. customChapter.ClassifyNameFirst = minClassifyName
  360. customChapter.Stage = reportInfo.Stage
  361. customChapter.PublishState = 1
  362. customChapter.CreateTime = reportInfo.CreateTime
  363. customChapter.ModifyTime = time.Now()
  364. customChapter.LastModifyAdminId = reportInfo.LastModifyAdminId
  365. customChapter.LastModifyAdminName = reportInfo.LastModifyAdminName
  366. customChapter.ContentModifyTime = time.Now()
  367. customAddChapter := models.AddReportChapter{
  368. ReportChapter: customChapter,
  369. GrantList: tmpGrantList,
  370. }
  371. customAddChapterList = append(customAddChapterList, customAddChapter)
  372. }
  373. }
  374. // 最大排序
  375. var maxSort int
  376. for _, typeItem := range typeList {
  377. v := inheritChapterMap[typeItem.ReportChapterTypeId]
  378. chapterItem := new(models.ReportChapter)
  379. if v != nil {
  380. chapterItem.AddType = 2
  381. chapterItem.Title = v.Title
  382. chapterItem.ReportType = v.ReportType
  383. chapterItem.ClassifyIdFirst = minClassifyId
  384. chapterItem.ClassifyNameFirst = minClassifyName
  385. chapterItem.TypeId = typeItem.ReportChapterTypeId
  386. chapterItem.TypeName = typeItem.ReportChapterTypeName
  387. chapterItem.Content = v.Content
  388. chapterItem.ContentSub = v.ContentSub
  389. chapterItem.Stage = reportInfo.Stage
  390. chapterItem.PublishState = 1
  391. chapterItem.Sort = typeItem.Sort
  392. chapterItem.CreateTime = reportInfo.CreateTime
  393. chapterItem.ModifyTime = time.Now()
  394. chapterItem.LastModifyAdminId = reportInfo.LastModifyAdminId
  395. chapterItem.LastModifyAdminName = reportInfo.LastModifyAdminName
  396. chapterItem.ContentModifyTime = time.Now()
  397. chapterItem.ContentStruct = v.ContentStruct
  398. chapterItem.ReportLayout = v.ReportLayout
  399. chapterItem.ReportCreateTime = time.Now()
  400. } else {
  401. chapterItem.AddType = 1
  402. chapterItem.Title = typeItem.ReportChapterTypeName
  403. chapterItem.ReportType = typeItem.ResearchType
  404. chapterItem.ClassifyIdFirst = minClassifyId
  405. chapterItem.ClassifyNameFirst = minClassifyName
  406. chapterItem.TypeId = typeItem.ReportChapterTypeId
  407. chapterItem.TypeName = typeItem.ReportChapterTypeName
  408. chapterItem.Stage = reportInfo.Stage
  409. chapterItem.PublishState = 1
  410. chapterItem.Sort = typeItem.Sort
  411. chapterItem.CreateTime = reportInfo.CreateTime
  412. chapterItem.ModifyTime = time.Now()
  413. chapterItem.LastModifyAdminId = reportInfo.LastModifyAdminId
  414. chapterItem.LastModifyAdminName = reportInfo.LastModifyAdminName
  415. chapterItem.ContentModifyTime = time.Now()
  416. //chapterItem.ContentStruct = v.ContentStruct
  417. chapterItem.ReportLayout = reportInfo.ReportLayout
  418. chapterItem.ReportCreateTime = time.Now()
  419. }
  420. if typeItem.Sort > maxSort {
  421. maxSort = typeItem.Sort
  422. }
  423. // 授权用户列表
  424. tmpGrantList, ok := typeGrantListMap[chapterItem.TypeId]
  425. if !ok {
  426. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  427. }
  428. addChapter := models.AddReportChapter{
  429. ReportChapter: chapterItem,
  430. GrantList: tmpGrantList,
  431. }
  432. chapterList = append(chapterList, addChapter)
  433. }
  434. // 将自定义的章节内容添加到待生成的章节内容中
  435. for _, addChapterItem := range customAddChapterList {
  436. maxSort++
  437. addChapterItem.ReportChapter.Sort = maxSort
  438. chapterList = append(chapterList, addChapterItem)
  439. }
  440. hasGrantUserMap := make(map[int]bool)
  441. for _, grantList := range typeGrantListMap {
  442. for _, grant := range grantList {
  443. if _, ok := hasGrantUserMap[grant.AdminId]; !ok {
  444. allGrantUserList = append(allGrantUserList, &report.ReportGrant{
  445. //GrantId: 0,
  446. //ReportId: 0,
  447. AdminId: grant.AdminId,
  448. CreateTime: time.Now(),
  449. })
  450. }
  451. }
  452. }
  453. return
  454. }
  455. // AddChapterBaseInfoAndPermission
  456. // @Description: 添加章节基本信息及权限
  457. // @author: Roc
  458. // @datetime 2024-06-11 15:35:23
  459. // @param reportInfo *models.Report
  460. // @param reportChapterInfo *models.ReportChapter
  461. // @param permissionIdList []int
  462. // @param adminIdList []int
  463. // @return err error
  464. // @return errMsg string
  465. func AddChapterBaseInfoAndPermission(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, permissionIdList []int, adminIdList []int) (err error, errMsg string) {
  466. errMsg = "新增失败"
  467. if reportInfo.State == 2 {
  468. errMsg = "该报告已发布,不允许编辑"
  469. err = errors.New(errMsg)
  470. return
  471. }
  472. // 需要添加的报告章节授权数据
  473. addChapterAdminList := make([]*report.ReportChapterGrant, 0)
  474. for _, adminId := range adminIdList {
  475. //新增授权
  476. addChapterAdminList = append(addChapterAdminList, &report.ReportChapterGrant{
  477. //GrantId: 0,
  478. //ReportChapterId: reportChapterInfo.ReportChapterId,
  479. AdminId: adminId,
  480. CreateTime: time.Now(),
  481. })
  482. }
  483. // 需要添加的报告章节品种权限数据
  484. addChapterPermissionList := make([]*report.ReportChapterPermissionMapping, 0)
  485. // 品种权限
  486. for _, permissionId := range permissionIdList {
  487. // 如果不存在,那么就新增品种权限配置
  488. addChapterPermissionList = append(addChapterPermissionList, &report.ReportChapterPermissionMapping{
  489. //ReportChapterPermissionMappingId: 0,
  490. //ReportChapterId: reportChapterInfo.ReportChapterId,
  491. ChartPermissionId: permissionId,
  492. CreateTime: time.Now(),
  493. })
  494. }
  495. err = models.AddChapterBaseInfoAndPermission(reportChapterInfo, addChapterAdminList, addChapterPermissionList)
  496. return
  497. }
  498. // EditChapterBaseInfoAndPermission
  499. // @Description: 修改报告章节的基础信息、授权用户权限、品种权限
  500. // @author: Roc
  501. // @datetime 2024-06-05 11:49:11
  502. // @param reportInfo *models.Report
  503. // @param reportChapterInfo *models.ReportChapter
  504. // @param title string
  505. // @param permissionIdList []int
  506. // @param adminIdList []int
  507. // @return err error
  508. // @return errMsg string
  509. func EditChapterBaseInfoAndPermission(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, title string, permissionIdList []int, adminIdList []int) (err error, errMsg string) {
  510. errMsg = "修改失败"
  511. if reportInfo.State == 2 {
  512. errMsg = "该报告已发布,不允许编辑"
  513. err = errors.New(errMsg)
  514. return
  515. }
  516. updateCols := make([]string, 0)
  517. // 如果标题内容,那么就修改
  518. if title != `` {
  519. reportChapterInfo.Title = title
  520. reportChapterInfo.ModifyTime = time.Now()
  521. updateCols = append(updateCols, "Title", "ModifyTime")
  522. reportChapterInfo.UpdateChapter(updateCols)
  523. }
  524. reportGrantObj := report.ReportGrant{}
  525. chapterGrantObj := report.ReportChapterGrant{}
  526. chapterPermissionObj := report.ReportChapterPermissionMapping{}
  527. // 报告授权的用户map
  528. reportGrantAdminIdMap := make(map[int]bool)
  529. // 获取报告授权的用户信息
  530. {
  531. // 获取当前报告已经授权的用户信息
  532. reportGrantList, tmpErr := reportGrantObj.GetGrantListById(reportInfo.Id)
  533. if tmpErr != nil {
  534. err = tmpErr
  535. return
  536. }
  537. for _, v := range reportGrantList {
  538. reportGrantAdminIdMap[v.AdminId] = true
  539. }
  540. }
  541. // 需要添加的报告章节授权数据
  542. addChapterAdminList := make([]*report.ReportChapterGrant, 0)
  543. // 待移除的报告章节授权数据id
  544. delReportChapterGrantIdList := make([]int, 0)
  545. // 处理当前报告章节需要新增/移除的授权信息
  546. {
  547. // 获取当前章节已经授权的用户信息
  548. chapterGrantList, tmpErr := chapterGrantObj.GetGrantListById(reportChapterInfo.ReportChapterId)
  549. if tmpErr != nil {
  550. err = tmpErr
  551. return
  552. }
  553. // 当前章节已经授权的用户信息
  554. currChapterAdminMap := make(map[int]*report.ReportChapterGrant)
  555. // 需要删除的报告章节授权数据
  556. delChapterAdminMap := make(map[int]*report.ReportChapterGrant)
  557. for _, v := range chapterGrantList {
  558. currChapterAdminMap[v.AdminId] = v
  559. delChapterAdminMap[v.AdminId] = v
  560. }
  561. for _, adminId := range adminIdList {
  562. // 如果该用户 不在 报告授权的用户map 里面,说明这个用户是要移除的
  563. if _, ok := reportGrantAdminIdMap[adminId]; !ok {
  564. continue
  565. }
  566. _, ok := currChapterAdminMap[adminId]
  567. // 如果存在,那么从 “需要删除的报告章节授权数据” 的map中移除
  568. if ok {
  569. delete(delChapterAdminMap, adminId)
  570. continue
  571. }
  572. // 如果不存在,那么就新增授权
  573. addChapterAdminList = append(addChapterAdminList, &report.ReportChapterGrant{
  574. //GrantId: 0,
  575. ReportChapterId: reportChapterInfo.ReportChapterId,
  576. AdminId: adminId,
  577. CreateTime: time.Now(),
  578. })
  579. }
  580. // 查出需要移除的授权id
  581. for _, v := range delChapterAdminMap {
  582. delReportChapterGrantIdList = append(delReportChapterGrantIdList, v.GrantId)
  583. }
  584. }
  585. // 其他章节授权的用户
  586. otherChapterAdminMap := make(map[int]bool)
  587. {
  588. // 获取报告所有的章节id
  589. reportChapterIdList, tmpErr := models.GetReportChapterIdList(reportInfo.Id)
  590. if tmpErr != nil {
  591. err = tmpErr
  592. return
  593. }
  594. if len(reportChapterIdList) > 0 {
  595. list, tmpErr := chapterGrantObj.GetGrantListByIdList(reportChapterIdList)
  596. if tmpErr != nil {
  597. err = tmpErr
  598. return
  599. }
  600. for _, v := range list {
  601. // 如果是当前章节,因为涉及到重新授权,所以得过滤
  602. if v.ReportChapterId == reportChapterInfo.ReportChapterId {
  603. continue
  604. }
  605. otherChapterAdminMap[v.AdminId] = true
  606. }
  607. }
  608. }
  609. // 需要添加的报告章节品种权限数据
  610. addChapterPermissionList := make([]*report.ReportChapterPermissionMapping, 0)
  611. // 待移除的报告章节品种权限数据id
  612. delChapterPermissionMappingIdList := make([]int, 0)
  613. // 处理当前报告章节需要新增/移除的品种权限信息
  614. {
  615. // 获取当前章节已经配置的品种权限信息
  616. chapterPermissionList, tmpErr := chapterPermissionObj.GetPermissionListById(reportChapterInfo.ReportChapterId)
  617. if tmpErr != nil {
  618. err = tmpErr
  619. return
  620. }
  621. // 当前章节已经配置的品种权限信息
  622. currChapterPermissionMap := make(map[int]*report.ReportChapterPermissionMapping)
  623. // 需要删除的报告章节品种权限配置
  624. delChapterPermissionMap := make(map[int]*report.ReportChapterPermissionMapping)
  625. for _, v := range chapterPermissionList {
  626. currChapterPermissionMap[v.ChartPermissionId] = v
  627. delChapterPermissionMap[v.ChartPermissionId] = v
  628. }
  629. for _, permissionId := range permissionIdList {
  630. _, ok := currChapterPermissionMap[permissionId]
  631. // 如果存在,那么从 “需要删除的报告章节品种权限配置” 的map中移除
  632. if ok {
  633. delete(delChapterPermissionMap, permissionId)
  634. continue
  635. }
  636. // 如果不存在,那么就新增品种权限配置
  637. addChapterPermissionList = append(addChapterPermissionList, &report.ReportChapterPermissionMapping{
  638. //ReportChapterPermissionMappingId: 0,
  639. ReportChapterId: reportChapterInfo.ReportChapterId,
  640. ChartPermissionId: permissionId,
  641. CreateTime: time.Now(),
  642. })
  643. }
  644. // 查出需要移除的品种权限配置
  645. for _, v := range delChapterPermissionMap {
  646. delChapterPermissionMappingIdList = append(delChapterPermissionMappingIdList, v.ReportChapterPermissionMappingId)
  647. }
  648. }
  649. err = models.EditChapterBaseInfoAndPermission(reportChapterInfo, updateCols, addChapterAdminList, addChapterPermissionList, delReportChapterGrantIdList, delChapterPermissionMappingIdList)
  650. return
  651. }
  652. // DelChapter
  653. // @Description: 删除报告章节、授权用户权限、品种权限
  654. // @author: Roc
  655. // @datetime 2024-06-06 17:28:37
  656. // @param reportInfo *models.Report
  657. // @param reportChapterInfo *models.ReportChapter
  658. // @param sysUser *system.Admin
  659. // @return err error
  660. // @return errMsg string
  661. func DelChapter(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, sysUser *system.Admin) (err error, errMsg string) {
  662. errMsg = "删除失败"
  663. if reportInfo.State == 2 {
  664. errMsg = "该报告已发布,不允许删除"
  665. err = errors.New(errMsg)
  666. return
  667. }
  668. reportInfo.LastModifyAdminId = sysUser.AdminId
  669. reportInfo.LastModifyAdminName = sysUser.RealName
  670. reportInfo.ModifyTime = time.Now()
  671. updateReportCols := []string{"LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  672. err = models.DelChapterAndPermission(reportInfo, updateReportCols, reportChapterInfo)
  673. return
  674. }
  675. // DownloadVoice
  676. // @Description: 下载报告音频文件
  677. // @author: Roc
  678. // @datetime 2024-06-13 15:36:46
  679. // @param reportInfo *models.ReportDetail
  680. // @return savePath string
  681. // @return fileName string
  682. // @return err error
  683. // @return errMsg string
  684. func DownloadVoice(reportInfo *models.ReportDetail) (savePath, fileName string, err error, errMsg string) {
  685. errMsg = `下载失败`
  686. // 如果报告有音频,那么下载音频
  687. if reportInfo.VideoUrl != `` {
  688. savePath = time.Now().Format(utils.FormatDateTimeUnSpace) + utils.GetRandString(5) + ".mp3"
  689. fileName = reportInfo.VideoName + ".mp3"
  690. fileBody, tmpErr := http.Get(reportInfo.VideoUrl)
  691. if tmpErr != nil {
  692. err = tmpErr
  693. errMsg = "音频下载失败"
  694. return
  695. }
  696. err = file.SaveFile(fileBody, savePath)
  697. if err != nil {
  698. errMsg = "音频保存失败"
  699. return
  700. }
  701. }
  702. // 如果是章节报告,那么就下载压缩包
  703. if reportInfo.HasChapter == 1 {
  704. videoList, tmpErr := models.GetReportChapterVideoList(reportInfo.Id)
  705. if tmpErr != nil {
  706. err = tmpErr
  707. errMsg = "获取音频列表失败"
  708. return
  709. }
  710. if len(videoList) > 0 {
  711. // 创建zip
  712. savePath = time.Now().Format(utils.FormatDateTimeUnSpace) + utils.GetRandString(5) + ".zip"
  713. fileName = reportInfo.VideoName + ".zip"
  714. zipFile, tmpErr := os.Create(savePath)
  715. if tmpErr != nil {
  716. err = tmpErr
  717. return
  718. }
  719. zipWriter := zip.NewWriter(zipFile)
  720. // 生成zip过程中报错关闭
  721. defer func() {
  722. if err != nil {
  723. zipWriter.Close()
  724. zipFile.Close()
  725. }
  726. }()
  727. // 获取音频,写入zip
  728. for i := 0; i < len(videoList); i++ {
  729. item := videoList[i]
  730. if item.VideoName == "" || item.VideoUrl == "" {
  731. continue
  732. }
  733. // 音频文件后缀
  734. ext := path.Ext(item.VideoUrl)
  735. ioWriter, tmpErr := zipWriter.Create(fmt.Sprintf("%s%s", item.VideoName, ext))
  736. if tmpErr != nil {
  737. err = tmpErr
  738. if os.IsPermission(err) {
  739. fmt.Println("权限不足: ", err)
  740. return
  741. }
  742. return
  743. }
  744. var content []byte
  745. content, err = http.Get(item.VideoUrl)
  746. if err != nil {
  747. content = []byte("")
  748. }
  749. ioWriter.Write(content)
  750. }
  751. // 生成zip后关闭,否则下载文件会损坏
  752. zipWriter.Close()
  753. zipFile.Close()
  754. }
  755. }
  756. return
  757. }
  758. // CheckReportAuthByAdminIdList
  759. // @Description: 根据管理员id列表,判断当前用户是否有报告权限
  760. // @author: Roc
  761. // @datetime 2024-06-13 11:03:10
  762. // @param adminId int
  763. // @param createAdminId int
  764. // @param grantAdminIdList []int
  765. // @return isAuth bool
  766. func CheckReportAuthByAdminIdList(adminId, createAdminId int, grantAdminIdList []int) (isAuth bool) {
  767. // 如果是自己创建的报告,那么就有权限
  768. if adminId == createAdminId {
  769. isAuth = true
  770. return
  771. }
  772. // 如果是授权用户,那么就有权限
  773. if utils.IsCheckInList(grantAdminIdList, adminId) {
  774. isAuth = true
  775. return
  776. }
  777. return
  778. }
  779. // CheckReportAuthByReportChapterInfo
  780. // @Description: 根据报告ID,判断当前用户是否有报告权限
  781. // @author: Roc
  782. // @datetime 2024-06-13 16:21:28
  783. // @param adminId int
  784. // @param reportInfoId int
  785. // @return isAuth bool
  786. // @return err error
  787. func CheckReportAuthByReportChapterInfo(adminId, createAdminId int, reportInfoId int) (isAuth bool, err error) {
  788. // 如果是自己创建的报告,那么就有权限
  789. if adminId == createAdminId {
  790. isAuth = true
  791. return
  792. }
  793. obj := report.ReportGrant{}
  794. chapterGrantList, err := obj.GetGrantListById(reportInfoId)
  795. if err != nil {
  796. return
  797. }
  798. for _, v := range chapterGrantList {
  799. if v.AdminId == adminId {
  800. isAuth = true
  801. return
  802. }
  803. }
  804. return
  805. }
  806. // EditReportLayoutImg
  807. // @Description: 修改报告的版图信息
  808. // @author: Roc
  809. // @datetime 2024-06-18 11:35:00
  810. // @param reportInfo *models.Report
  811. // @param req models.EditLayoutImgReq
  812. // @param sysUser *system.Admin
  813. // @return err error
  814. // @return errMsg string
  815. func EditReportLayoutImg(reportInfo *models.Report, req models.EditLayoutImgReq, sysUser *system.Admin) (err error, errMsg string) {
  816. errMsg = `保存失败`
  817. reportInfo.CanvasColor = req.CanvasColor
  818. reportInfo.LastModifyAdminId = sysUser.AdminId
  819. reportInfo.LastModifyAdminName = sysUser.RealName
  820. reportInfo.ModifyTime = time.Now()
  821. updateCols := []string{"CanvasColor", "LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  822. if req.HeadResourceId > 0 {
  823. reportInfo.HeadResourceId = req.HeadResourceId
  824. reportInfo.HeadImg = req.HeadImg
  825. updateCols = append(updateCols, "HeadResourceId", "HeadImg")
  826. }
  827. if req.EndResourceId > 0 {
  828. reportInfo.EndResourceId = req.EndResourceId
  829. reportInfo.EndImg = req.EndImg
  830. updateCols = append(updateCols, "EndResourceId", "EndImg")
  831. }
  832. err = reportInfo.UpdateReport(updateCols)
  833. if err != nil {
  834. return
  835. }
  836. return
  837. }