jl_service.go 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179
  1. /*
  2. * @description: 计量业务相关
  3. * @Author: CP
  4. * @Date: 2021-12-25 14:37:01
  5. * @FilePath: \construction_management\services\jl_service.go
  6. */
  7. package services
  8. import (
  9. "errors"
  10. "fmt"
  11. "net/http"
  12. "net/url"
  13. "strconv"
  14. "go.mod/comm"
  15. "go.mod/conf"
  16. "go.mod/dao"
  17. "go.mod/datasource"
  18. "go.mod/lib"
  19. "go.mod/models"
  20. "go.mod/web/viewmodels"
  21. )
  22. type JlService struct {
  23. dao *dao.ProjectDao
  24. accountDao *dao.ProjectAccountDao
  25. }
  26. // 创建项目用户service
  27. func NewJlService() *JlService {
  28. return &JlService{
  29. dao: dao.NewProjectDao(datasource.InstanceDbMaster()),
  30. accountDao: dao.NewProjectAccountDao(datasource.InstanceDbMaster()),
  31. }
  32. }
  33. // 是否有项目
  34. func (s *JlService) IsCode(code string) map[string]interface{} {
  35. exist := 0
  36. if code != "" {
  37. projectData := s.dao.GetCode(code)
  38. if projectData.Id != 0 {
  39. exist = 1
  40. }
  41. }
  42. data := map[string]interface{}{
  43. "exist": exist,
  44. }
  45. return data
  46. }
  47. // 从计量来的项目和管理员账号,添加入项目管理
  48. func (s *JlService) ProjectAdd(claimsData *viewmodels.Jl) (map[string]interface{}, error) {
  49. var (
  50. err error
  51. JlResultData map[string]interface{}
  52. accountData *models.CmProjectAccount
  53. )
  54. Jlzf := lib.NewJlzf()
  55. // 1.获得计量 账号和项目 信息
  56. if JlResultData, err = Jlzf.GetProjectAndAccount(claimsData.Code, claimsData.Account); err != nil {
  57. return nil, err
  58. }
  59. fmt.Println(JlResultData)
  60. // 2.获得本地项目
  61. projectData := s.dao.GetCode(claimsData.Code)
  62. // 项目不存在
  63. if projectData.Id == 0 {
  64. // 2-1. 创建项目和账号 TODO
  65. project := &models.CmProject{}
  66. account := &models.CmProjectAccount{}
  67. if err = s.dao.AddProjectAndAccount(project, account); err != nil {
  68. return nil, err
  69. }
  70. // project := &models.CmProject{}
  71. // project.Name = projectVM.Name
  72. // project.Code = projectVM.Code
  73. // project.CategoryId = projectVM.CategoryId
  74. // project.Category = projectVM.Category
  75. // project.InsideCategoryid = projectVM.InsideCategoryid
  76. // project.InsideCategory = projectVM.InsideCategory
  77. // project.StaffId = projectVM.StaffId
  78. // project.StaffName = projectVM.StaffName
  79. // project.CreateTime = time.Now()
  80. // project.CreateName = projectVM.CreateName
  81. // project.CreateCategory = projectVM.CreateCategory
  82. // project.Remark = projectVM.Remark
  83. } else {
  84. // 2-2. 是否有该账号
  85. accountData = s.accountDao.GetAccountProjectId(claimsData.Account, projectData.Id)
  86. // 2-2-1 账号不存在
  87. if accountData.Id == 0 {
  88. // 2-2-2. 创建 账号 TODO
  89. account := &models.CmProjectAccount{}
  90. if err = s.accountDao.Add(account); err != nil {
  91. return nil, err
  92. }
  93. }
  94. }
  95. data := map[string]interface{}{
  96. "sync": 1,
  97. }
  98. return data, nil
  99. }
  100. func (s *JlService) AuthLogin(claimsData *viewmodels.Jl, writer http.ResponseWriter) (*viewmodels.ProjectAccount, error) {
  101. // 1-1.工程项目是否存在
  102. projectInfo := s.dao.GetCode(claimsData.Code)
  103. if projectInfo.Id == 0 {
  104. return nil, errors.New("工程建设管理员还未创建项目,禁止登录")
  105. }
  106. // 1-2.获得项目管理员
  107. projectAdminInfo := s.accountDao.Get(projectInfo.UserId, projectInfo.Id)
  108. if projectAdminInfo.Id == 0 {
  109. return nil, errors.New("工程建设管理员不存在,禁止登录")
  110. }
  111. // 2.验证登陆-临时
  112. projectAccountInfo := s.accountDao.GetAccountProjectId(claimsData.Account, projectInfo.Id)
  113. // 未账号改账号
  114. if projectAccountInfo.Id == 0 {
  115. // 到计量拉取普通账号 - TODO
  116. // Jlzf := lib.NewJlzf()
  117. // // 1.获得计量 账号和项目 信息
  118. // if JlResultData, err := Jlzf.GetProjectAndAccount(claimsData.Code, claimsData.Account); err != nil {
  119. // return nil, err
  120. // }
  121. // // 2-. 创建 账号 TODO
  122. // account := &models.CmProjectAccount{}
  123. // if err := s.accountDao.Add(account); err != nil {
  124. // return nil, err
  125. // }
  126. }
  127. // 2-1.账号停用
  128. if projectAccountInfo.Enable == 0 {
  129. return nil, errors.New("该账号已被停用")
  130. }
  131. // 加密用户标识
  132. identity, err := comm.AesEncrypt(strconv.Itoa(projectAccountInfo.Id), conf.CookieSecret)
  133. if err != nil {
  134. return nil, err
  135. }
  136. // 加密项目标识
  137. projectId, err := comm.AesEncrypt(strconv.Itoa(projectAccountInfo.ProjectId), conf.CookieSecret)
  138. if err != nil {
  139. return nil, err
  140. }
  141. digitalToken := comm.CreateSign(conf.CookieSecret + strconv.Itoa(projectAccountInfo.Id))
  142. // 设置cookie
  143. maxAge := 60 * 60 * 24 * 7
  144. params := url.Values{}
  145. params.Add("identity", identity)
  146. params.Add("attachedIdentity", projectId)
  147. params.Add("digitalToken", digitalToken)
  148. c := &http.Cookie{
  149. Name: "cm",
  150. Value: params.Encode(),
  151. Path: "/",
  152. MaxAge: maxAge,
  153. HttpOnly: true,
  154. }
  155. http.SetCookie(writer, c)
  156. viewAccountData := comm.MakeProjectAccountVM(projectAccountInfo)
  157. return &viewAccountData, nil
  158. }