login_service.go 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261
  1. /*
  2. * @description:登陆相关数据操作
  3. * @Author: CP
  4. * @Date: 2020-09-02 09:56:28
  5. * @FilePath: \construction_management\services\login_service.go
  6. */
  7. package services
  8. import (
  9. "errors"
  10. "fmt"
  11. "log"
  12. "net/http"
  13. "strconv"
  14. "time"
  15. "github.com/iris-contrib/middleware/jwt"
  16. "github.com/kataras/iris/v12"
  17. "go.mod/comm"
  18. "go.mod/conf"
  19. "go.mod/dao"
  20. "go.mod/datasource"
  21. "go.mod/lib"
  22. "go.mod/models"
  23. "go.mod/web/viewmodels"
  24. )
  25. //定义项目用户Service接口
  26. type LoginService interface {
  27. ValidRule(ctx iris.Context) (viewmodels.Login, error)
  28. ValidProjectAccount(viewmodels.Login, http.ResponseWriter) (error, string)
  29. Out(ctx iris.Context) error
  30. CrtateAccount(viewmodels.Login) error
  31. }
  32. //返回service操作类
  33. type loginService struct {
  34. projectAccountDao *dao.ProjectAccountDao
  35. projectDao *dao.ProjectDao
  36. }
  37. //创建项目用户service
  38. func NewLoginService() LoginService {
  39. return &loginService{
  40. projectAccountDao: dao.NewProjectAccountDao(datasource.InstanceDbMaster()),
  41. projectDao: dao.NewProjectDao(datasource.InstanceDbMaster()),
  42. }
  43. }
  44. // 登陆验证
  45. func (s *loginService) ValidRule(ctx iris.Context) (viewmodels.Login, error) {
  46. loginVaild := viewmodels.Login{}
  47. err := ctx.ReadJSON(&loginVaild)
  48. fmt.Println(loginVaild)
  49. if err != nil {
  50. log.Println("ReadForm转换异常, error=", err)
  51. return loginVaild, err
  52. }
  53. err = loginVaild.Validate()
  54. if err != nil {
  55. log.Println("登录验证, error=", err)
  56. return loginVaild, err
  57. }
  58. return loginVaild, nil
  59. }
  60. // 验证项目用户登陆相关
  61. // func (s *loginService) ValidProjectAccount(loginData viewmodels.Login, writer http.ResponseWriter) (*viewmodels.ProjectAccount, error) {
  62. // // 1-1.工程项目是否存在
  63. // projectInfo := s.projectDao.GetCode(loginData.Code)
  64. // if projectInfo.Id == 0 {
  65. // return nil, errors.New("工程建设管理员还未创建项目,禁止登录")
  66. // }
  67. // // 1-2.获得项目管理员
  68. // projectAdminInfo := s.projectAccountDao.Get(projectInfo.UserId, projectInfo.Id)
  69. // if projectAdminInfo.Id == 0 {
  70. // return nil, errors.New("工程建设管理员不存在,禁止登录")
  71. // }
  72. // 2.验证登陆-临时
  73. // projectAccountInfo := s.projectAccountDao.GetAccount(loginData.Account)
  74. // if projectAccountInfo.Id == 0 {
  75. // return nil, errors.New("未找到该账号")
  76. // }
  77. // // 2.账号停用
  78. // if projectAccountInfo.Enable == 0 {
  79. // return nil, errors.New("该账号已被停用")
  80. // }
  81. // password := comm.CreatePasswordSign(loginData.Password, loginData.Account)
  82. // if projectAccountInfo.Password != password {
  83. // return nil, errors.New("密码不正确")
  84. // }
  85. // // 验证密码-TODO
  86. // Jlzf := lib.NewJlzf()
  87. // err := Jlzf.LoginValid(loginData)
  88. // if err != nil {
  89. // return nil, err
  90. // }
  91. // // 加密用户标识
  92. // identity, err := comm.AesEncrypt(strconv.Itoa(projectAccountInfo.Id), conf.CookieSecret)
  93. // if err != nil {
  94. // return nil, err
  95. // }
  96. // // 加密项目标识
  97. // projectId, err := comm.AesEncrypt(strconv.Itoa(projectAccountInfo.ProjectId), conf.CookieSecret)
  98. // if err != nil {
  99. // return nil, err
  100. // }
  101. // digitalToken := comm.CreateSign(conf.CookieSecret + strconv.Itoa(projectAccountInfo.Id))
  102. // // 设置cookie
  103. // maxAge := 60 * 60 * 24 * 7
  104. // params := url.Values{}
  105. // params.Add("identity", identity)
  106. // params.Add("attachedIdentity", projectId)
  107. // params.Add("digitalToken", digitalToken)
  108. // c := &http.Cookie{
  109. // Name: "cm",
  110. // Value: params.Encode(),
  111. // Path: "/",
  112. // MaxAge: maxAge,
  113. // HttpOnly: true,
  114. // }
  115. // http.SetCookie(writer, c)
  116. // // TODO--分布式会话--后续需要在加入
  117. // viewAccountData := comm.MakeProjectAccountVM(projectAccountInfo)
  118. // return &viewAccountData, nil
  119. // }
  120. func (s *loginService) CrtateAccount(loginData viewmodels.Login) error {
  121. newPsw := comm.CreatePasswordSign(loginData.Password, loginData.Account)
  122. account := &models.CmProjectAccount{Account: loginData.Account, Password: newPsw}
  123. err := s.projectAccountDao.Add(account)
  124. return err
  125. }
  126. // TODO -替换jwt
  127. // 验证项目用户登陆相关
  128. func (s *loginService) ValidProjectAccount(loginData viewmodels.Login, writer http.ResponseWriter) (error, string) {
  129. // projectInfo := models.CmProject{}
  130. // 工程项目是否存在
  131. // projectInfo.Code = loginData.Code
  132. // s.projectDao.Get(&projectInfo)
  133. // if projectInfo.Id == 0 {
  134. // return errors.New("工程建设管理员还未创建项目,禁止登录")
  135. // }
  136. // 获得项目用户
  137. projectAccountInfo := s.projectAccountDao.GetAccount(loginData.Account)
  138. if projectAccountInfo.Id == 0 {
  139. return errors.New("账号/密码不正确"), ""
  140. }
  141. // 验证密码-TODO
  142. Jlzf := lib.NewJlzf()
  143. err := Jlzf.LoginValid(loginData)
  144. if err != nil {
  145. return err, ""
  146. }
  147. // 加密用户标识
  148. identity, err := comm.AesEncrypt(strconv.Itoa(projectAccountInfo.Id), conf.CookieSecret)
  149. if err != nil {
  150. return err, ""
  151. }
  152. // 加密项目标识
  153. // projectId, err := comm.AesEncrypt(strconv.Itoa(projectAccountInfo.ProjectId), conf.CookieSecret)
  154. // if err != nil {
  155. // return err
  156. // }
  157. // jwt token
  158. token := jwt.NewTokenWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
  159. // 根据需求,可以存一些必要的数据
  160. "identity": identity,
  161. // 签发人
  162. "iss": "cm",
  163. // 签发时间
  164. "iat": time.Now().Unix(),
  165. // 设定过期时间,便于测试,设置7天过期
  166. "exp": time.Now().Add(72 * time.Hour * time.Duration(1)).Unix(),
  167. })
  168. // 使用设置的秘钥,签名生成jwt字符串
  169. tokenString, _ := token.SignedString([]byte(conf.SignSecret))
  170. // maxAge := 60 * 60 * 24 * 7
  171. // params := url.Values{}
  172. // params.Add("token", tokenString)
  173. // c := &http.Cookie{
  174. // Name: "cm",
  175. // Value: tokenString, //params.Encode(),
  176. // Path: "/",
  177. // MaxAge: maxAge,
  178. // //HttpOnly: true,
  179. // }
  180. // http.SetCookie(writer, c)
  181. // TODO--分布式会话--后续需要在加入
  182. return nil, tokenString
  183. }
  184. // 登出
  185. func (s *loginService) Out(ctx iris.Context) error {
  186. // 作废jwt token
  187. // 移除cookie
  188. ctx.RemoveCookie("cm")
  189. return nil
  190. }
  191. //fmt.Println(ctx.FormValues())
  192. // list := ctx.FormValues()
  193. // loginVaild := viewmodels.Login{}
  194. // v := reflect.ValueOf(&loginVaild).Elem()
  195. // for path, values := range list {
  196. // //fmt.Println(path)
  197. // //fmt.Println(values)
  198. // if path != "csrf.Token" {
  199. // v.FieldByName(Ucfirst(path)).Set(reflect.ValueOf(values[0]))
  200. // }
  201. // }
  202. //fmt.Println(loginVaild)
  203. // loginVaild := viewmodels.Login{
  204. // Code: ctx.FormValue("code"),
  205. // Account: ctx.FormValue("account"),
  206. // Password: ctx.FormValue("password"),
  207. // }
  208. // // 加密用户标识 生成数字证书
  209. // identity, err := comm.AesEncrypt(strconv.Itoa(projectAccountInfo.Id), conf.CookieSecret)
  210. // if err != nil {
  211. // return err
  212. // }
  213. // digitalToken := comm.CreateSign(conf.CookieSecret + strconv.Itoa(projectAccountInfo.Id))
  214. // // 设置cookie
  215. // maxAge := 60 * 60 * 24 * 7
  216. // params := url.Values{}
  217. // params.Add("identity", identity)
  218. // params.Add("digitalToken", digitalToken)
  219. // c := &http.Cookie{
  220. // Name: "cm",
  221. // Value: params.Encode(),
  222. // Path: "/",
  223. // MaxAge: maxAge,
  224. // }
  225. // http.SetCookie(writer, c)