user.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "eta/eta_mini_api/models"
  5. "eta/eta_mini_api/models/request"
  6. "eta/eta_mini_api/models/response"
  7. "eta/eta_mini_api/services"
  8. "eta/eta_mini_api/services/go_redis"
  9. "eta/eta_mini_api/utils"
  10. "fmt"
  11. "strings"
  12. "time"
  13. )
  14. type UserController struct {
  15. BaseCommonController
  16. }
  17. type UserAuthController struct {
  18. BaseAuthController
  19. }
  20. // @Title 用户登录接口
  21. // @Description 用户登录
  22. // @Param request body models.LoginReq true "type json string"
  23. // @Success 200 {object} models.LoginResp
  24. // @router /login [post]
  25. func (this *UserAuthController) Login() {
  26. br := new(models.BaseResponse).Init()
  27. defer func() {
  28. this.Data["json"] = br
  29. this.ServeJSON()
  30. }()
  31. session := this.Session
  32. var req request.LoginReq
  33. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  34. if err != nil {
  35. br.Msg = "参数解析失败"
  36. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  37. return
  38. }
  39. switch req.LoginType {
  40. case 1:
  41. if req.Phone == "" {
  42. br.Msg = "请输入手机号"
  43. br.ErrMsg = "请输入手机号"
  44. return
  45. }
  46. if req.SmsCode == "" {
  47. br.Msg = "请输入验证码"
  48. br.ErrMsg = "请输入验证码"
  49. return
  50. }
  51. phone := req.AreaCode + req.Phone
  52. item, err := models.GetMsgCode(phone, req.SmsCode)
  53. if err != nil {
  54. if err.Error() == utils.ErrNoRow() {
  55. br.Msg = "验证码错误,请重新输入"
  56. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  57. return
  58. } else {
  59. br.Msg = "验证码错误,请重新输入"
  60. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  61. return
  62. }
  63. }
  64. if item == nil {
  65. br.Msg = "验证码错误,请重新输入"
  66. return
  67. }
  68. case 2:
  69. if req.Email == "" {
  70. br.Msg = "请输入邮箱"
  71. br.ErrMsg = "请输入邮箱"
  72. return
  73. }
  74. if req.SmsCode == "" {
  75. br.Msg = "请输入验证码"
  76. br.ErrMsg = "请输入验证码"
  77. return
  78. }
  79. item, err := models.GetMsgCode(req.Email, req.SmsCode)
  80. if err != nil {
  81. if err.Error() == utils.ErrNoRow() {
  82. br.Msg = "验证码错误,请重新输入"
  83. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  84. return
  85. } else {
  86. br.Msg = "验证码错误,请重新输入"
  87. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  88. return
  89. }
  90. }
  91. if item == nil {
  92. br.Msg = "验证码错误,请重新输入"
  93. return
  94. }
  95. }
  96. _, errMsg, err := services.BindUser(session.UnionId, session.OpenId, req.Phone, req.Email, req.AreaCode)
  97. if err != nil && err.Error() != utils.ErrNoRow() {
  98. br.Msg = "登录失败,系统处理中,请稍后重试"
  99. if errMsg != "" {
  100. br.Msg = errMsg
  101. }
  102. br.ErrMsg = "登录失败:" + err.Error()
  103. return
  104. }
  105. br.Msg = "登录成功"
  106. br.Success = true
  107. br.Ret = 200
  108. }
  109. // @Title 获取短信/邮箱验证码
  110. // @Description 用户登录
  111. // @Param request body models.LoginReq true "type json string"
  112. // @Success 200 {object} models.LoginResp
  113. // @router /getVerifyCode [post]
  114. func (this *UserController) GetVerifyCode() {
  115. br := new(models.BaseResponse).Init()
  116. defer func() {
  117. this.Data["json"] = br
  118. this.ServeJSON()
  119. }()
  120. var req request.VerifyCodeReq
  121. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  122. if err != nil {
  123. br.Msg = "参数解析异常!"
  124. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  125. return
  126. }
  127. if req.VerifyType != 1 && req.VerifyType != 2 {
  128. br.Msg = "验证方式有误"
  129. br.ErrMsg = fmt.Sprintf("验证方式异常<%d>", req.VerifyType)
  130. }
  131. code := utils.GetRandDigit(6)
  132. fmt.Println(code)
  133. switch req.VerifyType {
  134. case 1:
  135. if req.AreaCode == "" {
  136. br.Msg = "请选择区号"
  137. return
  138. }
  139. if req.Phone == "" {
  140. br.Msg = "请输入手机号"
  141. return
  142. }
  143. if req.AreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Phone) {
  144. br.Msg = "您的手机号输入有误, 请检查"
  145. return
  146. }
  147. phoneKey := utils.CACHE_ACCESS_PHONE_LOGIN_CODE + req.AreaCode + req.Phone
  148. res, _ := go_redis.RedisInt(phoneKey)
  149. if res > 5 {
  150. br.Msg = "验证码发送太频繁,请稍后重试"
  151. return
  152. }
  153. var ok bool
  154. if req.AreaCode == "86" {
  155. ok = services.SendSmsCode(req.Phone, code)
  156. }
  157. if !ok {
  158. br.Msg = "请检查手机号和区号"
  159. br.ErrMsg = "短信验证码发送失败"
  160. return
  161. } else {
  162. item := new(models.MsgCode)
  163. item.OpenId = ""
  164. item.Code = code
  165. item.Mobile = req.AreaCode + req.Phone
  166. item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
  167. item.CreateTime = time.Now()
  168. err = item.Insert()
  169. if err != nil {
  170. br.Msg = "发送失败"
  171. br.ErrMsg = "发送失败,Err:" + err.Error()
  172. return
  173. }
  174. br.Msg = "发送成功"
  175. isExist := go_redis.IsExist(phoneKey)
  176. if isExist {
  177. go_redis.Incr(phoneKey)
  178. } else {
  179. go_redis.SetNX(phoneKey, 1, time.Minute*15)
  180. }
  181. }
  182. case 2:
  183. if req.Email == "" {
  184. br.Msg = "请输入邮箱"
  185. }
  186. if !utils.ValidateEmailFormatat(req.Email) {
  187. br.Msg = "您的邮箱格式输入有误, 请检查"
  188. return
  189. }
  190. emailKey := utils.CACHE_ACCESS_EMAIL_LOGIN_CODE + req.Email
  191. res, _ := go_redis.RedisInt(emailKey)
  192. if res > 5 {
  193. br.Msg = "验证码发送太频繁,请稍后重试"
  194. return
  195. }
  196. date := time.Now()
  197. content := "尊敬的用户:</br>本次请求的验证码为:" + code + "(为了保障您账号的安全性,请在15分钟内完成验证。)</br>东吴期货研究团队 </br>" + fmt.Sprintf("%d年%02d月%02d日", date.Year(), date.Month(), date.Day())
  198. title := "东吴期货登录验证"
  199. result, err := utils.SendEmailByDw(title, content, req.Email)
  200. if err != nil {
  201. br.Msg = "发送失败"
  202. br.ErrMsg = "发送失败,Err:" + err.Error()
  203. return
  204. }
  205. if result {
  206. item := new(models.MsgCode)
  207. item.OpenId = ""
  208. item.Code = code
  209. item.Mobile = req.Email
  210. item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
  211. item.CreateTime = time.Now()
  212. err = item.Insert()
  213. if err != nil {
  214. br.Msg = "发送失败"
  215. br.ErrMsg = "发送失败,Err:" + err.Error()
  216. return
  217. }
  218. br.Msg = "发送成功"
  219. isExist := go_redis.IsExist(emailKey)
  220. if isExist {
  221. go_redis.Incr(emailKey)
  222. } else {
  223. go_redis.SetNX(emailKey, 1, time.Minute*15)
  224. }
  225. } else {
  226. br.Msg = "发送失败"
  227. }
  228. }
  229. br.Ret = 200
  230. br.Success = true
  231. br.Msg = "发送成功"
  232. }
  233. // @Title 新增报告浏览记录
  234. // @Description 新增报告浏览记录接口
  235. // @Param request body models.ReportRecordReq true "type json string"
  236. // @Success 200 新增成功
  237. // @router /addReportRecord [post]
  238. func (this *UserAuthController) AddReportRecord() {
  239. br := new(models.BaseResponse).Init()
  240. defer func() {
  241. this.Data["json"] = br
  242. this.ServeJSON()
  243. }()
  244. user := this.User
  245. if user == nil {
  246. br.Msg = "请登录"
  247. br.ErrMsg = "请登录,用户信息为空"
  248. br.Ret = 403
  249. return
  250. }
  251. var req request.ReportRecordReq
  252. if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
  253. br.Msg = "参数解析异常!"
  254. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  255. return
  256. }
  257. if req.ReportId <= 0 {
  258. br.Msg = "参数错误"
  259. br.ErrMsg = "参数错误,报告id小于等于0"
  260. return
  261. }
  262. reportResp, err := services.GetReportDetail(req.ReportId, user.UserId)
  263. if err != nil {
  264. br.Msg = "添加阅读记录失败"
  265. br.ErrMsg = "获取研报错误,Err:" + err.Error()
  266. return
  267. }
  268. if reportResp.Ret != 200 {
  269. br.Msg = reportResp.Msg
  270. br.ErrMsg = reportResp.ErrMsg
  271. return
  272. }
  273. reportDetail := reportResp.Data.Report
  274. reportPermissionResp, err := services.GetReportChartPermissionList(req.ReportId)
  275. if err != nil {
  276. br.Msg = "添加阅读记录失败"
  277. br.ErrMsg = "获取研报品种错误,Err:" + err.Error()
  278. return
  279. }
  280. if reportPermissionResp.Ret != 200 {
  281. br.Msg = reportPermissionResp.Msg
  282. br.ErrMsg = reportPermissionResp.ErrMsg
  283. return
  284. }
  285. reportPermissionList := reportPermissionResp.Data
  286. chartPermissionList, err := services.GetChartPermissionAllList()
  287. if err != nil {
  288. br.Msg = "添加阅读记录失败"
  289. br.ErrMsg = "获取研报品种列表失败,Err:" + err.Error()
  290. return
  291. }
  292. chartMap := make(map[int]*models.ChartPermission)
  293. for _, permission := range chartPermissionList.Data {
  294. chartMap[permission.ChartPermissionId] = permission
  295. }
  296. curTime := time.Now()
  297. var insertId int64
  298. if req.RecordId == 0 {
  299. // 如果不存在就新增一条记录
  300. permission1Ids := make([]int, 0)
  301. permission2Ids := make([]int, 0)
  302. permissionNames := make([]string, 0)
  303. for _, item := range reportPermissionList {
  304. curPermission := chartMap[item.ChartPermissionId]
  305. permission1Ids = append(permission1Ids, curPermission.ParentId)
  306. permission2Ids = append(permission2Ids, curPermission.ChartPermissionId)
  307. permissionNames = append(permissionNames, curPermission.PermissionName)
  308. }
  309. permission1Ids = utils.Unique(permission1Ids)
  310. permission2Ids = utils.Unique(permission2Ids)
  311. permissionNames = utils.Unique(permissionNames)
  312. userReadRecord := &models.UserReadRecord{
  313. UserId: user.UserId,
  314. ReportId: req.ReportId,
  315. ReportTitle: reportDetail.Title,
  316. ChartPermissionName: strings.Join(permissionNames, ","),
  317. ClassifyId1: reportDetail.ClassifyIdFirst,
  318. ClassifyName1: reportDetail.ClassifyNameFirst,
  319. ClassifyId2: reportDetail.ClassifyIdSecond,
  320. ClassifyName2: reportDetail.ClassifyNameSecond,
  321. Timestamp: int(curTime.Unix()),
  322. CreateTime: curTime,
  323. CreateDate: curTime.Format(utils.FormatDate),
  324. ReportType: utils.ReportTypeEta,
  325. }
  326. insertId, err = userReadRecord.Insert()
  327. if err != nil {
  328. br.Msg = "添加阅读记录失败"
  329. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  330. return
  331. }
  332. userReadPermission1 := make([]*models.UserReadPermission1, 0)
  333. for _, id := range permission1Ids {
  334. userReadPermission1 = append(userReadPermission1, &models.UserReadPermission1{
  335. UserReadRecordId: int(insertId),
  336. ChartPermissionId: id,
  337. PermissionName: chartMap[id].PermissionName,
  338. })
  339. }
  340. err = models.UserReadPermission1MultiInsert(userReadPermission1)
  341. if err != nil {
  342. br.Msg = "添加阅读记录失败"
  343. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  344. return
  345. }
  346. userReadPermission2 := make([]*models.UserReadPermission2, 0)
  347. for _, id := range permission2Ids {
  348. userReadPermission2 = append(userReadPermission2, &models.UserReadPermission2{
  349. UserReadRecordId: int(insertId),
  350. ChartPermissionId: id,
  351. })
  352. }
  353. err = models.UserReadPermission2MultiInsert(userReadPermission2)
  354. if err != nil {
  355. br.Msg = "添加阅读记录失败"
  356. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  357. return
  358. }
  359. } else {
  360. // 如果存在就计算停留时间
  361. userRecord, err := models.GetUserReadRecordListById(req.RecordId)
  362. if err != nil {
  363. if err.Error() == utils.ErrNoRow() {
  364. br.Msg = "更新阅读记录不存在"
  365. return
  366. }
  367. br.Msg = "更新阅读记录失败"
  368. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  369. return
  370. }
  371. stayTime := curTime.Unix() - int64(userRecord.Timestamp)
  372. stayTimeStr := utils.SecondsToHMS(stayTime)
  373. err = models.UpdateUserReadRecordById(req.RecordId, int(curTime.Unix()), int(stayTime), stayTimeStr)
  374. if err != nil {
  375. br.Msg = "更新阅读记录失败"
  376. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  377. return
  378. }
  379. }
  380. resp := new(response.UserReadRecordResp)
  381. resp.RecordId = insertId
  382. br.Msg = "添加阅读记录成功"
  383. br.Ret = 200
  384. br.Success = true
  385. br.Data = resp
  386. }
  387. // @Title 新增pdf报告浏览记录
  388. // @Description 新增pdf报告浏览记录接口
  389. // @Param request body models.ReportPdfRecordReq true "type json string"
  390. // @Success 200 新增成功
  391. // @router /pdf/addReportRecord [post]
  392. func (this *UserAuthController) AddReportPdfRecord() {
  393. br := new(models.BaseResponse).Init()
  394. defer func() {
  395. this.Data["json"] = br
  396. this.ServeJSON()
  397. }()
  398. user := this.User
  399. if user == nil {
  400. br.Msg = "请登录"
  401. br.ErrMsg = "请登录,用户信息为空"
  402. br.Ret = 403
  403. return
  404. }
  405. var req request.ReportPdfRecordReq
  406. if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
  407. br.Msg = "参数解析异常!"
  408. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  409. return
  410. }
  411. if req.ReportPdfId <= 0 {
  412. br.Msg = "参数错误"
  413. br.ErrMsg = "参数错误,报告id小于等于0"
  414. return
  415. }
  416. reportPdf, err := models.GetReportPdfById(req.ReportPdfId)
  417. if err != nil {
  418. if err.Error() == utils.ErrNoRow() {
  419. br.Msg = "报告不存在或已删除"
  420. return
  421. }
  422. br.Msg = "添加阅读记录失败"
  423. br.ErrMsg = "获取研报错误,Err:" + err.Error()
  424. return
  425. }
  426. reportPermissionResp, err := services.GetChartPermissionListByClassifyId(reportPdf.ClassifyIdSecond, 0)
  427. if err != nil {
  428. br.Msg = "添加阅读记录失败"
  429. br.ErrMsg = "获取研报品种错误,Err:" + err.Error()
  430. return
  431. }
  432. if reportPermissionResp.Ret != 200 {
  433. br.Msg = reportPermissionResp.Msg
  434. br.ErrMsg = reportPermissionResp.ErrMsg
  435. return
  436. }
  437. reportPermissionList := reportPermissionResp.Data
  438. chartPermissionList, err := services.GetChartPermissionAllList()
  439. if err != nil {
  440. br.Msg = "添加阅读记录失败"
  441. br.ErrMsg = "获取研报品种列表失败,Err:" + err.Error()
  442. return
  443. }
  444. chartMap := make(map[int]*models.ChartPermission)
  445. for _, permission := range chartPermissionList.Data {
  446. chartMap[permission.ChartPermissionId] = permission
  447. }
  448. curTime := time.Now()
  449. var insertId int64
  450. if req.RecordId == 0 {
  451. // 如果不存在就新增一条记录
  452. permission1Ids := make([]int, 0)
  453. permission2Ids := make([]int, 0)
  454. permissionNames := make([]string, 0)
  455. for _, item := range reportPermissionList {
  456. curPermission := chartMap[item.ChartPermissionId]
  457. permission1Ids = append(permission1Ids, curPermission.ParentId)
  458. permission2Ids = append(permission2Ids, curPermission.ChartPermissionId)
  459. permissionNames = append(permissionNames, curPermission.PermissionName)
  460. }
  461. permission1Ids = utils.Unique(permission1Ids)
  462. permission2Ids = utils.Unique(permission2Ids)
  463. permissionNames = utils.Unique(permissionNames)
  464. userReadRecord := &models.UserReadRecord{
  465. UserId: user.UserId,
  466. ReportId: reportPdf.ReportPdfId,
  467. ReportTitle: reportPdf.Title,
  468. ChartPermissionName: strings.Join(permissionNames, ","),
  469. ClassifyId1: reportPdf.ClassifyIdFirst,
  470. ClassifyName1: reportPdf.ClassifyNameFirst,
  471. ClassifyId2: reportPdf.ClassifyIdSecond,
  472. ClassifyName2: reportPdf.ClassifyNameSecond,
  473. Timestamp: int(curTime.Unix()),
  474. CreateTime: curTime,
  475. CreateDate: curTime.Format(utils.FormatDate),
  476. ReportType: utils.ReportTypePdf,
  477. }
  478. insertId, err = userReadRecord.Insert()
  479. if err != nil {
  480. br.Msg = "添加阅读记录失败"
  481. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  482. return
  483. }
  484. userReadPermission1 := make([]*models.UserReadPermission1, 0)
  485. for _, id := range permission1Ids {
  486. userReadPermission1 = append(userReadPermission1, &models.UserReadPermission1{
  487. UserReadRecordId: int(insertId),
  488. ChartPermissionId: id,
  489. PermissionName: chartMap[id].PermissionName,
  490. })
  491. }
  492. err = models.UserReadPermission1MultiInsert(userReadPermission1)
  493. if err != nil {
  494. br.Msg = "添加阅读记录失败"
  495. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  496. return
  497. }
  498. userReadPermission2 := make([]*models.UserReadPermission2, 0)
  499. for _, id := range permission2Ids {
  500. userReadPermission2 = append(userReadPermission2, &models.UserReadPermission2{
  501. UserReadRecordId: int(insertId),
  502. ChartPermissionId: id,
  503. })
  504. }
  505. err = models.UserReadPermission2MultiInsert(userReadPermission2)
  506. if err != nil {
  507. br.Msg = "添加阅读记录失败"
  508. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  509. return
  510. }
  511. count, err := models.GetUserReadRecordCountByReportPdfIdAndUserId(req.ReportPdfId, user.UserId)
  512. if err != nil {
  513. br.Msg = "添加阅读记录失败"
  514. br.ErrMsg = "获取阅读记录失败,Err:" + err.Error()
  515. return
  516. }
  517. if count > 1 {
  518. err = models.UpdateReportPdfPv(req.ReportPdfId)
  519. if err != nil {
  520. br.Msg = "添加研报阅读记录失败"
  521. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  522. return
  523. }
  524. } else {
  525. err = models.UpdateReportPdfUvAndPv(req.ReportPdfId)
  526. if err != nil {
  527. br.Msg = "添加研报阅读记录失败"
  528. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  529. return
  530. }
  531. }
  532. } else {
  533. // 如果存在就计算停留时间
  534. userRecord, err := models.GetUserReadRecordListById(req.RecordId)
  535. if err != nil {
  536. if err.Error() == utils.ErrNoRow() {
  537. br.Msg = "更新阅读记录不存在"
  538. return
  539. }
  540. br.Msg = "更新阅读记录失败"
  541. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  542. return
  543. }
  544. stayTime := curTime.Unix() - int64(userRecord.Timestamp)
  545. stayTimeStr := utils.SecondsToHMS(stayTime)
  546. err = models.UpdateUserReadRecordById(req.RecordId, int(curTime.Unix()), int(stayTime), stayTimeStr)
  547. if err != nil {
  548. br.Msg = "更新阅读记录失败"
  549. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  550. return
  551. }
  552. }
  553. resp := new(response.UserReadRecordResp)
  554. resp.RecordId = insertId
  555. br.Msg = "添加阅读记录成功"
  556. br.Ret = 200
  557. br.Success = true
  558. br.Data = resp
  559. }
  560. // AreaCodeList
  561. // @Title 手机号区号列表
  562. // @Description 手机号区号列表
  563. // @Success 200 Ret=200 获取成功
  564. // @router /area_code/list [get]
  565. func (this *UserController) AreaCodeList() {
  566. br := new(models.BaseResponse).Init()
  567. defer func() {
  568. if br.ErrMsg == "" {
  569. br.IsSendEmail = false
  570. }
  571. this.Data["json"] = br
  572. this.ServeJSON()
  573. }()
  574. resp := make([]response.AreaCodeListResp, 0)
  575. confValue, e := models.GetConfigDetailByCode(models.ConfAreaCodeListKey)
  576. if e != nil {
  577. br.Msg = "获取失败"
  578. br.ErrMsg = "获取手机号区号配置失败, Err: " + e.Error()
  579. return
  580. }
  581. if confValue == "" {
  582. br.Msg = "获取失败"
  583. br.ErrMsg = "手机号区号配置为空"
  584. return
  585. }
  586. if e := json.Unmarshal([]byte(confValue), &resp); e != nil {
  587. br.Msg = "获取失败"
  588. br.ErrMsg = "手机号区号配置有误"
  589. return
  590. }
  591. br.Data = resp
  592. br.Ret = 200
  593. br.Success = true
  594. br.Msg = "获取成功"
  595. }
  596. // 用户品种信息
  597. // @Title 用户信息
  598. // @Description 用户信息
  599. // @Success 200 Ret=200 获取成功
  600. // @router /info [get]
  601. func (this *UserAuthController) Info() {
  602. br := new(models.BaseResponse).Init()
  603. defer func() {
  604. this.Data["json"] = br
  605. this.ServeJSON()
  606. }()
  607. user := this.User
  608. if user == nil {
  609. user = &models.User{}
  610. }
  611. seller, err := models.GetSellerById(user.SellerId)
  612. if err != nil && err.Error() != utils.ErrNoRow() {
  613. br.Msg = "查询失败"
  614. br.ErrMsg = "查询失败,系统异常,Err:" + err.Error()
  615. return
  616. }
  617. private, err := services.GetPrivateChartPermissionList()
  618. if err != nil {
  619. br.Msg = "查询失败"
  620. br.ErrMsg = "查询失败,系统异常,Err:" + err.Error()
  621. return
  622. }
  623. if private.Ret != 200 {
  624. br.Msg = private.Msg
  625. br.ErrMsg = private.ErrMsg
  626. return
  627. }
  628. userPermissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  629. if err != nil {
  630. br.Msg = "查看权限失败"
  631. br.ErrMsg = "查看权限失败,系统异常,Err:" + err.Error()
  632. return
  633. }
  634. userPermissionMap := make(map[int]struct{})
  635. for _, v := range userPermissionIds {
  636. userPermissionMap[v] = struct{}{}
  637. }
  638. if seller == nil {
  639. // 添加默认的销售电话
  640. seller = &models.SysUser{
  641. Phone: utils.DefaultPhone,
  642. AreaCode: utils.DefaultAreaCode,
  643. }
  644. }
  645. var hasPermission string
  646. for _, v := range private.Data {
  647. for _, vv := range v.Child {
  648. if _, ok := userPermissionMap[vv.ChartPermissionId]; ok {
  649. hasPermission = "私有权限"
  650. break
  651. }
  652. }
  653. }
  654. if hasPermission == "" || user.Status == utils.UserStatusNo {
  655. hasPermission = "公有权限"
  656. }
  657. miniconf, err := models.GetMiniConf()
  658. if err != nil {
  659. br.Msg = "获取配置失败"
  660. br.ErrMsg = "获取配置失败,系统异常,Err:" + err.Error()
  661. return
  662. }
  663. userView := &models.UserView{
  664. Headimgurl: miniconf["Logo"],
  665. RealName: user.RealName,
  666. Phone: user.Phone,
  667. AreaCode: user.AreaCode,
  668. SellerName: seller.SysRealName,
  669. SellerPhone: seller.Phone,
  670. SellerAreaCode: seller.AreaCode,
  671. HasPermission: hasPermission,
  672. Email: user.Email,
  673. Componey: user.Company,
  674. IsRegistered: user.IsRegistered,
  675. Status: user.Status,
  676. ValidEndTime: user.ValidEndTime.Format(utils.FormatDate),
  677. }
  678. if user.ValidEndTime.IsZero() {
  679. userView.ValidEndTime = ""
  680. }
  681. br.Data = userView
  682. br.Msg = "查询成功"
  683. br.Success = true
  684. br.Ret = 200
  685. }
  686. // PermissionList
  687. // @Title 用户个人权限列表
  688. // @Description 用户个人权限列表
  689. // @Param request body models.LoginReq true "type json string"
  690. // @Success 200 {object} []models.ChartPermissionTreeView
  691. // @router /permission/list [get]
  692. func (this *UserAuthController) PermissionList() {
  693. br := new(models.BaseResponse).Init()
  694. defer func() {
  695. this.Data["json"] = br
  696. this.ServeJSON()
  697. }()
  698. user := this.User
  699. permissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  700. if err != nil {
  701. br.Msg = "查询用户权限失败"
  702. br.ErrMsg = "查询用户权限失败,系统异常,Err:" + err.Error()
  703. return
  704. }
  705. userPermissionMap := make(map[int]struct{})
  706. for _, v := range permissionIds {
  707. userPermissionMap[v] = struct{}{}
  708. }
  709. // 查询公有权限
  710. var publicView []*models.ChartPermissionTreeView
  711. publicResp, err := services.GetPublicChartPermissionList()
  712. if err != nil {
  713. br.Msg = "查询公有权限失败"
  714. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  715. }
  716. if publicResp.Ret != 200 {
  717. br.Msg = publicResp.Msg
  718. br.ErrMsg = publicResp.ErrMsg
  719. return
  720. }
  721. publicView = publicResp.Data
  722. // 如果是正式用户,则查询私有权限
  723. var privateView []*models.ChartPermissionTreeView
  724. if user.Status == utils.UserStatusFormal {
  725. privateResp, err := services.GetPrivateChartPermissionList()
  726. if err != nil {
  727. br.Msg = "查询私有权限失败"
  728. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  729. }
  730. if privateResp.Ret != 200 {
  731. br.Msg = privateResp.Msg
  732. br.ErrMsg = privateResp.ErrMsg
  733. return
  734. }
  735. for _, v := range privateResp.Data {
  736. IsAdd := false
  737. curPermissionView := &models.ChartPermissionTreeView{
  738. ChartPermissionId: v.ChartPermissionId,
  739. PermissionName: v.PermissionName,
  740. IsPublic: v.IsPublic,
  741. Sort: v.Sort,
  742. Child: make([]*models.ChartPermissionTreeView, 0),
  743. }
  744. for _, vv := range v.Child {
  745. if _, ok := userPermissionMap[vv.ChartPermissionId]; ok {
  746. curPermissionView.Child = append(curPermissionView.Child, vv)
  747. IsAdd = true
  748. }
  749. }
  750. if IsAdd {
  751. privateView = append(privateView, curPermissionView)
  752. }
  753. }
  754. }
  755. privateAloneView := make([]*models.ChartPermissionTreeView, 0)
  756. for _, vi := range privateView {
  757. isHas := false
  758. for _, vb := range publicView {
  759. if vi.ChartPermissionId == vb.ChartPermissionId {
  760. isHas = true
  761. vb.Child = append(vb.Child, vi.Child...)
  762. break
  763. }
  764. }
  765. if !isHas {
  766. privateAloneView = append(privateAloneView, vi)
  767. }
  768. }
  769. publicView = append(publicView, privateAloneView...)
  770. for _, vi := range publicView {
  771. vi.ChartPermissionType = utils.PermissionTypeEta
  772. for _, vi := range vi.Child {
  773. vi.ChartPermissionType = utils.PermissionTypeEta
  774. }
  775. }
  776. br.Data = publicView
  777. br.Msg = "查询成功"
  778. br.Success = true
  779. br.Ret = 200
  780. }
  781. // PermissionList
  782. // @Title 用户已购权限列表
  783. // @Description 用户已购权限列表
  784. // @Param request body models.LoginReq true "type json string"
  785. // @Success 200 {object} []models.ChartPermissionTreeView
  786. // @router /purchased/list [get]
  787. func (this *UserAuthController) PurchasedPermission() {
  788. br := new(models.BaseResponse).Init()
  789. defer func() {
  790. this.Data["json"] = br
  791. this.ServeJSON()
  792. }()
  793. user := this.User
  794. permissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  795. if err != nil {
  796. br.Msg = "查询用户权限失败"
  797. br.ErrMsg = "查询用户权限失败,系统异常,Err:" + err.Error()
  798. return
  799. }
  800. permissionMap := make(map[int]struct{})
  801. for _, v := range permissionIds {
  802. permissionMap[v] = struct{}{}
  803. }
  804. // 查询私有权限
  805. privateResp, err := services.GetPrivateChartPermissionList()
  806. if err != nil {
  807. br.Msg = "查询私有权限失败"
  808. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  809. }
  810. if privateResp.Ret != 200 {
  811. br.Msg = privateResp.Msg
  812. br.ErrMsg = privateResp.ErrMsg
  813. return
  814. }
  815. var privateView []*models.ChartPermissionTreeView
  816. for _, v := range privateResp.Data {
  817. IsAdd := false
  818. curPermissionView := &models.ChartPermissionTreeView{
  819. ChartPermissionId: v.ChartPermissionId,
  820. PermissionName: v.PermissionName,
  821. IsPublic: v.IsPublic,
  822. Sort: v.Sort,
  823. Child: make([]*models.ChartPermissionTreeView, 0),
  824. ChartPermissionType: utils.PermissionTypeEta,
  825. }
  826. for _, vv := range v.Child {
  827. if _, ok := permissionMap[vv.ChartPermissionId]; ok && vv.ParentId != 0 {
  828. vv.ChartPermissionType = utils.PermissionTypeEta
  829. curPermissionView.Child = append(curPermissionView.Child, vv)
  830. IsAdd = true
  831. }
  832. }
  833. if IsAdd {
  834. privateView = append(privateView, curPermissionView)
  835. }
  836. }
  837. for _, vi := range privateView {
  838. vi.Child = append(vi.Child, &models.ChartPermissionTreeView{
  839. ChartPermissionId: -1,
  840. PermissionName: utils.MINI_CUSTOM_PERMISSION_NAME,
  841. IsPublic: 0,
  842. Sort: 0,
  843. ChartPermissionType: utils.ReportTypePdf,
  844. })
  845. }
  846. br.Data = privateView
  847. br.Msg = "查询成功"
  848. br.Success = true
  849. br.Ret = 200
  850. }