user_login.go 31 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066
  1. package controllers
  2. import (
  3. "encoding/base64"
  4. "encoding/json"
  5. "eta/eta_api/models"
  6. "eta/eta_api/models/company"
  7. "eta/eta_api/models/system"
  8. "eta/eta_api/services"
  9. "eta/eta_api/services/eta_trial"
  10. "eta/eta_api/utils"
  11. "fmt"
  12. "image/color"
  13. "strings"
  14. "time"
  15. "github.com/mojocn/base64Captcha"
  16. )
  17. // UserLoginController 登录-无需Token
  18. type UserLoginController struct {
  19. BaseCommonController
  20. }
  21. // UserLoginAuthController 登录-需Token
  22. type UserLoginAuthController struct {
  23. BaseAuthController
  24. }
  25. // GenerateCaptcha
  26. // @Title 生成图形验证码
  27. // @Description 生成图形验证码
  28. // @Success 200 Ret=200 获取成功
  29. // @router /get_captcha [get]
  30. func (this *UserLoginController) GenerateCaptcha() {
  31. br := new(models.BaseResponse).Init()
  32. defer func() {
  33. if br.ErrMsg == "" {
  34. br.IsSendEmail = false
  35. }
  36. this.Data["json"] = br
  37. this.ServeJSON()
  38. }()
  39. //driver := base64Captcha.DefaultDriverDigit
  40. // 自定义验证码样式
  41. var driver base64Captcha.Driver
  42. driverString := base64Captcha.DriverString{
  43. Height: 60, //高度
  44. Width: 120, //宽度
  45. NoiseCount: 0, //干扰数
  46. ShowLineOptions: 2 | 4, //展示个数
  47. Length: 4, //长度
  48. //Source: "1234567890qwertyuioplkjhgfdsazxcvbnm", //验证码随机字符串来源
  49. Source: "1234567890", //验证码随机字符串来源
  50. BgColor: &color.RGBA{ // 背景颜色
  51. R: 0,
  52. G: 0,
  53. B: 0,
  54. A: 0,
  55. },
  56. Fonts: []string{"wqy-microhei.ttc"}, // 字体
  57. }
  58. driver = driverString.ConvertFonts()
  59. // 生成验证码
  60. store := services.CaptchaRedis{}
  61. captcha := base64Captcha.NewCaptcha(driver, store)
  62. id, b64s, _, err := captcha.Generate()
  63. if err != nil {
  64. br.Msg = "生成失败"
  65. br.ErrMsg = "生成验证码失败, Err: " + err.Error()
  66. return
  67. }
  68. type CaptchaResult struct {
  69. Id string
  70. Base64Blob string
  71. }
  72. res := new(CaptchaResult)
  73. res.Id = id
  74. res.Base64Blob = b64s
  75. br.Ret = 200
  76. br.Success = true
  77. br.Msg = "获取成功"
  78. br.Data = res
  79. }
  80. // GetVerifyCode
  81. // @Title 获取短信/邮箱验证码
  82. // @Description 获取短信/邮箱验证码
  83. // @Param request body VerifyCodeReq true "type json string"
  84. // @Success 200 Ret=200 获取成功
  85. // @router /verify_code [post]
  86. func (this *UserLoginController) GetVerifyCode() {
  87. br := new(models.BaseResponse).Init()
  88. defer func() {
  89. if br.ErrMsg == "" {
  90. br.IsSendEmail = false
  91. }
  92. this.Data["json"] = br
  93. this.ServeJSON()
  94. }()
  95. type VerifyCodeReq struct {
  96. VerifyType int `description:"验证方式: 1-手机号; 2-邮箱"`
  97. CaptchaId string `description:"验证码ID"`
  98. CaptchaCode string `description:"图形验证码"`
  99. Mobile string `description:"手机号"`
  100. TelAreaCode string `description:"手机区号"`
  101. Email string `description:"邮箱"`
  102. Source int `description:"来源:1-登录;2-异常登录校验;3-忘记密码"`
  103. }
  104. var req VerifyCodeReq
  105. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  106. if err != nil {
  107. br.Msg = "参数解析异常!"
  108. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  109. return
  110. }
  111. if req.VerifyType != 1 && req.VerifyType != 2 {
  112. br.Msg = "验证方式有误"
  113. br.ErrMsg = "验证方式有误"
  114. return
  115. }
  116. if req.Source != 1 && req.Source != 2 && req.Source != 3 {
  117. br.Msg = "来源有误"
  118. br.ErrMsg = "来源有误"
  119. return
  120. }
  121. // 忘记密码图形校验在前一步, 这一步不再校验图形验证码
  122. if req.Source != 3 {
  123. if req.CaptchaId == "" || req.CaptchaCode == "" {
  124. br.Msg = "请输入图形验证码"
  125. return
  126. }
  127. }
  128. if req.VerifyType == 1 {
  129. if req.TelAreaCode == "" {
  130. br.Msg = "请选择区号"
  131. return
  132. }
  133. if req.Mobile == "" {
  134. br.Msg = "请输入手机号"
  135. return
  136. }
  137. if req.TelAreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Mobile) {
  138. br.Msg = "您的手机号输入有误, 请检查"
  139. return
  140. }
  141. _, e := system.GetSysUserByMobile(req.Mobile)
  142. if e != nil {
  143. if utils.IsErrNoRow(e) {
  144. br.Msg = "您的手机号未绑定账号, 请检查"
  145. return
  146. }
  147. br.Msg = "您的手机号未绑定账号, 请检查"
  148. br.ErrMsg = "手机号获取用户失败, Err:" + e.Error()
  149. return
  150. }
  151. }
  152. if req.VerifyType == 2 {
  153. if req.Email == "" {
  154. br.Msg = "请输入邮箱"
  155. return
  156. }
  157. if !utils.ValidateEmailFormatat(req.Email) {
  158. br.Msg = "您的邮箱输入有误, 请检查"
  159. return
  160. }
  161. _, e := system.GetSysUserByEmail(req.Email)
  162. if e != nil {
  163. if utils.IsErrNoRow(e) {
  164. br.Msg = "您的邮箱未绑定账号, 请检查"
  165. return
  166. }
  167. br.Msg = "您的邮箱未绑定账号, 请检查"
  168. br.ErrMsg = "邮箱获取用户失败, Err:" + e.Error()
  169. return
  170. }
  171. }
  172. if req.Source != 3 {
  173. store := services.CaptchaRedis{}
  174. ok := store.Verify(req.CaptchaId, req.CaptchaCode, true)
  175. if !ok {
  176. br.Msg = "图形验证码错误, 请重新输入"
  177. return
  178. }
  179. }
  180. // 限制最多60s获取一次
  181. var lockKey string
  182. if req.VerifyType == 1 {
  183. lockKey = fmt.Sprint(utils.CaptchaCachePrefix, req.Mobile)
  184. }
  185. if req.VerifyType == 2 {
  186. lockKey = fmt.Sprint(utils.CaptchaCachePrefix, req.Email)
  187. }
  188. locked := utils.Rc.SetNX(lockKey, 1, time.Minute)
  189. if !locked {
  190. br.Msg = "请勿频繁发送"
  191. return
  192. }
  193. // 发送验证码
  194. sendRes := false
  195. if req.VerifyType == 1 {
  196. r, e := services.SendAdminMobileVerifyCode(req.Source, req.Mobile, req.TelAreaCode)
  197. if e != nil {
  198. br.Msg = "发送失败"
  199. br.ErrMsg = "发送短信验证码失败, Err: " + e.Error()
  200. return
  201. }
  202. sendRes = r
  203. }
  204. if req.VerifyType == 2 {
  205. r, e := services.SendAdminEmailVerifyCode(req.Source, req.Email)
  206. if e != nil {
  207. br.Msg = "发送失败"
  208. br.ErrMsg = "发送邮箱验证码失败, Err: " + e.Error()
  209. return
  210. }
  211. sendRes = r
  212. }
  213. if !sendRes {
  214. br.Msg = "发送失败"
  215. br.ErrMsg = "发送短信验证码失败"
  216. return
  217. }
  218. br.Ret = 200
  219. br.Success = true
  220. br.Msg = "发送成功"
  221. }
  222. // Login
  223. // @Title 用户登录
  224. // @Description 用户登录
  225. // @Param request body UserLoginReq true "type json string"
  226. // @Success 200 {object} models.LoginResp
  227. // @router /login [post]
  228. func (this *UserLoginController) Login() {
  229. br := new(models.BaseResponse).Init()
  230. defer func() {
  231. if br.ErrMsg == "" {
  232. br.IsSendEmail = false
  233. }
  234. this.Data["json"] = br
  235. this.ServeJSON()
  236. }()
  237. // 入参
  238. type UserLoginReq struct {
  239. LoginType int `description:"登录方式: 1-账号; 2-手机号; 3-邮箱"`
  240. Username string `description:"账号"`
  241. Password string `description:"密码"`
  242. Mobile string `description:"手机号"`
  243. Email string `description:"邮箱"`
  244. VerifyCode string `description:"验证码"`
  245. ReqTime string `description:"登录时间戳"`
  246. TelAreaCode string `description:"区号"`
  247. CaptchaId string `description:"验证码ID"`
  248. CaptchaCode string `description:"图形验证码"`
  249. }
  250. var req UserLoginReq
  251. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  252. if err != nil {
  253. br.Msg = "参数解析异常!"
  254. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  255. return
  256. }
  257. req.Username = strings.TrimSpace(req.Username)
  258. req.Mobile = strings.TrimSpace(req.Mobile)
  259. req.Email = strings.TrimSpace(req.Email)
  260. req.VerifyCode = strings.TrimSpace(req.VerifyCode)
  261. if req.LoginType != 1 && req.LoginType != 2 && req.LoginType != 3 {
  262. br.Msg = "登录方式有误"
  263. br.ErrMsg = fmt.Sprintf("登录方式有误, Type: %d", req.LoginType)
  264. return
  265. }
  266. sysUser := new(system.Admin)
  267. // 账号密码登录
  268. if req.LoginType == 1 {
  269. if req.Username == "" {
  270. br.Msg = "请输入账号"
  271. return
  272. }
  273. if req.Password == "" {
  274. br.Msg = "请输入密码"
  275. return
  276. }
  277. // 获取基本配置信息
  278. picVerifyConfInfo, e := models.GetBusinessConfByKey(models.BusinessConfLoginPictureVerifyCode)
  279. if e != nil && !utils.IsErrNoRow(e) {
  280. br.Msg = "查询配置信息失败"
  281. br.ErrMsg = e.Error()
  282. return
  283. }
  284. if picVerifyConfInfo != nil && picVerifyConfInfo.ConfVal == "1" {
  285. if req.CaptchaCode == "" || req.CaptchaId == ""{
  286. br.Msg = "请输入验证码"
  287. return
  288. }
  289. }
  290. // 判断账号是否为异常登录, 算作异常的话就需要换另外的方式去登录了
  291. abnormalKey := fmt.Sprint(utils.CACHE_ABNORMAL_LOGIN, req.Username)
  292. isAbnormal, _ := utils.Rc.RedisString(abnormalKey)
  293. if isAbnormal != "" {
  294. br.Ret = models.BaseRespCodeAbnormalLogin
  295. br.Msg = "账号异常, 请进行手机号/邮箱校验"
  296. return
  297. }
  298. // 校验图形验证码
  299. if picVerifyConfInfo != nil && picVerifyConfInfo.ConfVal == "1" {
  300. store := services.CaptchaRedis{}
  301. ok := store.Verify(req.CaptchaId, req.CaptchaCode, true)
  302. if !ok {
  303. br.Msg = "图形验证码错误, 请重新输入"
  304. br.Ret = models.BaseRespCodeLoginCaptchErr
  305. return
  306. }
  307. }
  308. // 查询账号信息
  309. errPassKey := fmt.Sprint(utils.CACHE_LOGIN_ERR_PASS, req.Username)
  310. accountUser, e := system.GetSysUserByAdminName(req.Username)
  311. if e != nil {
  312. br.Ret = models.BaseRespCodeLoginErr
  313. br.Msg = "登录失败, 账号或密码错误"
  314. // 账号查询异常均进行标记, 避免一直尝试进行登录
  315. if !utils.Rc.IsExist(errPassKey) {
  316. _ = utils.Rc.Put(errPassKey, 1, utils.GetTodayLastSecond())
  317. return
  318. }
  319. errNum, _ := utils.Rc.RedisInt(errPassKey)
  320. errNum += 1
  321. if errNum < 6 {
  322. _ = utils.Rc.Put(errPassKey, errNum, utils.GetTodayLastSecond())
  323. return
  324. }
  325. // 标记异常登录, 重置计数
  326. br.Ret = models.BaseRespCodeAbnormalLogin
  327. br.Msg = "账号异常, 请进行手机号/邮箱校验"
  328. _ = utils.Rc.Put(abnormalKey, "true", utils.GetTodayLastSecond())
  329. _ = utils.Rc.Delete(errPassKey)
  330. return
  331. }
  332. // 系统用户-账号密码校验
  333. if accountUser.IsLdap == 0 {
  334. dbPass := utils.MD5(fmt.Sprintf("%s%s%s", accountUser.Password, utils.UserLoginSalt, req.ReqTime))
  335. if req.Password != dbPass {
  336. br.Ret = models.BaseRespCodeLoginErr
  337. br.Msg = "登录失败, 账号或密码错误"
  338. // 错误密码计数, 超过6次标记异常
  339. if !utils.Rc.IsExist(errPassKey) {
  340. _ = utils.Rc.Put(errPassKey, 1, utils.GetTodayLastSecond())
  341. return
  342. }
  343. errNum, _ := utils.Rc.RedisInt(errPassKey)
  344. errNum += 1
  345. if errNum < 6 {
  346. _ = utils.Rc.Put(errPassKey, errNum, utils.GetTodayLastSecond())
  347. return
  348. }
  349. // 标记异常登录, 重置计数
  350. br.Ret = models.BaseRespCodeAbnormalLogin
  351. br.Msg = "账号异常, 请进行手机号/邮箱校验"
  352. _ = utils.Rc.Put(abnormalKey, "true", utils.GetTodayLastSecond())
  353. _ = utils.Rc.Delete(errPassKey)
  354. return
  355. }
  356. if accountUser.Enabled == 0 {
  357. br.Msg = "您的账号已被禁用, 如需登录, 请联系管理员"
  358. br.ErrMsg = fmt.Sprintf("账号已被禁用, 登录账号: %s, 账户名称: %s", accountUser.AdminName, accountUser.RealName)
  359. return
  360. }
  361. // 异常登录-是否登录间隔大于60天
  362. if isAbnormal == "" {
  363. abnormalTime := time.Now().AddDate(0, 0, -60)
  364. lastLogin, _ := time.ParseInLocation(utils.FormatDateTime, accountUser.LastLoginTime, time.Local)
  365. if !lastLogin.IsZero() && lastLogin.Before(abnormalTime) {
  366. br.Msg = "请进行异常登录校验"
  367. br.Ret = models.BaseRespCodeAbnormalLogin
  368. // 标记异常登录
  369. _ = utils.Rc.Put(abnormalKey, "true", utils.GetTodayLastSecond())
  370. return
  371. }
  372. }
  373. }
  374. // 如果是域账号, 那么取出原始密码走AD域校验
  375. if accountUser.IsLdap == 1 {
  376. passDecode, e := base64.StdEncoding.DecodeString(req.Password)
  377. if e != nil {
  378. br.Ret = models.BaseRespCodeLoginErr
  379. br.Msg = "登录失败, 账号或密码错误"
  380. br.ErrMsg = "Pass Decode err: " + e.Error()
  381. return
  382. }
  383. originPass := strings.Replace(string(passDecode), utils.UserLoginSalt, "", 1)
  384. pass, e := services.LdapUserCheck(req.Username, originPass)
  385. if e != nil {
  386. br.Ret = models.BaseRespCodeLoginErr
  387. br.Msg = "登录失败, 账号或密码错误"
  388. br.ErrMsg = "LdapLogin err: " + e.Error()
  389. return
  390. }
  391. if !pass {
  392. br.Ret = models.BaseRespCodeLoginErr
  393. br.Msg = "登录失败, 账号或密码错误"
  394. return
  395. }
  396. }
  397. sysUser = accountUser
  398. }
  399. // 手机号
  400. if req.LoginType == 2 {
  401. if req.Mobile == "" {
  402. br.Msg = "请输入手机号"
  403. return
  404. }
  405. // 大陆校验区号
  406. if req.TelAreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Mobile) {
  407. br.Msg = "您的手机号输入有误, 请检查"
  408. return
  409. }
  410. if req.VerifyCode == "" {
  411. br.Msg = "请输入验证码"
  412. return
  413. }
  414. expiredTime := time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute).Format(utils.FormatDateTime)
  415. recordCond := ` AND source = ? AND mobile = ? AND code = ? AND expired_time <= ?`
  416. recordPars := make([]interface{}, 0)
  417. recordPars = append(recordPars, system.AdminVerifyCodeRecordSourceLogin, req.Mobile, req.VerifyCode, expiredTime)
  418. recordOb := new(system.AdminVerifyCodeRecord)
  419. _, e := recordOb.GetItemByCondition(recordCond, recordPars)
  420. if e != nil {
  421. if utils.IsErrNoRow(e) {
  422. br.Msg = "验证码错误, 请重新输入"
  423. return
  424. }
  425. br.Msg = "验证码错误, 请重新输入"
  426. br.ErrMsg = "获取手机号登陆验证码失败, Err: " + e.Error()
  427. return
  428. }
  429. mobileUser, e := system.GetSysUserByMobile(req.Mobile)
  430. if e != nil {
  431. if utils.IsErrNoRow(e) {
  432. br.Msg = "您的手机号未绑定账号, 请检查"
  433. return
  434. }
  435. br.Msg = "您的手机号未绑定账号, 请检查"
  436. br.ErrMsg = "手机号获取用户失败, Err:" + e.Error()
  437. return
  438. }
  439. if mobileUser.Enabled == 0 {
  440. br.Msg = "您的账号已被禁用, 如需登录, 请联系管理员"
  441. br.ErrMsg = fmt.Sprintf("账号已被禁用, 登录手机号: %s", req.Mobile)
  442. return
  443. }
  444. sysUser = mobileUser
  445. }
  446. // 邮箱登录
  447. if req.LoginType == 3 {
  448. if req.Email == "" {
  449. br.Msg = "请输入邮箱"
  450. return
  451. }
  452. if !utils.ValidateEmailFormatat(req.Email) {
  453. br.Msg = "您的邮箱输入有误, 请检查"
  454. return
  455. }
  456. if req.VerifyCode == "" {
  457. br.Msg = "请输入验证码"
  458. return
  459. }
  460. expiredTime := time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute).Format(utils.FormatDateTime)
  461. recordCond := ` AND source = ? AND email = ? AND code = ? AND expired_time <= ?`
  462. recordPars := make([]interface{}, 0)
  463. recordPars = append(recordPars, system.AdminVerifyCodeRecordSourceLogin, req.Email, req.VerifyCode, expiredTime)
  464. recordOb := new(system.AdminVerifyCodeRecord)
  465. _, e := recordOb.GetItemByCondition(recordCond, recordPars)
  466. if e != nil {
  467. if utils.IsErrNoRow(e) {
  468. br.Msg = "验证码错误, 请重新输入"
  469. return
  470. }
  471. br.Msg = "验证码错误, 请重新输入"
  472. br.ErrMsg = "获取手机号登陆验证码失败, Err: " + e.Error()
  473. return
  474. }
  475. emailUser, e := system.GetSysUserByEmail(req.Email)
  476. if e != nil {
  477. if utils.IsErrNoRow(e) {
  478. br.Msg = "您的邮箱未绑定账号, 请检查"
  479. return
  480. }
  481. br.Msg = "您的邮箱未绑定账号, 请检查"
  482. br.ErrMsg = "邮箱获取用户失败, Err:" + e.Error()
  483. return
  484. }
  485. if emailUser.Enabled == 0 {
  486. br.Msg = "您的账号已被禁用, 如需登录, 请联系管理员"
  487. br.ErrMsg = fmt.Sprintf("账号已被禁用, 登录邮箱: %s", req.Email)
  488. return
  489. }
  490. sysUser = emailUser
  491. }
  492. // 登录成功(无论哪种方式登录), 清除异常标记
  493. abnormalCache := fmt.Sprint(utils.CACHE_ABNORMAL_LOGIN, sysUser.AdminName)
  494. errPassCache := fmt.Sprint(utils.CACHE_LOGIN_ERR_PASS, sysUser.AdminName)
  495. _ = utils.Rc.Delete(abnormalCache)
  496. _ = utils.Rc.Delete(errPassCache)
  497. account := utils.MD5(sysUser.AdminName)
  498. token := utils.GenToken(account)
  499. sysSession := new(system.SysSession)
  500. sysSession.UserName = sysUser.AdminName
  501. sysSession.SysUserId = sysUser.AdminId
  502. sysSession.ExpiredTime = time.Now().AddDate(0, 0, 90)
  503. sysSession.IsRemember = 0 // 均需要做过期校验
  504. sysSession.CreatedTime = time.Now()
  505. sysSession.LastUpdatedTime = time.Now()
  506. sysSession.AccessToken = token
  507. if e := system.AddSysSession(sysSession); e != nil {
  508. br.Msg = "登录失败"
  509. br.ErrMsg = "新增session信息失败, Err:" + e.Error()
  510. return
  511. }
  512. // 修改最后登录时间
  513. {
  514. sysUser.LastLoginTime = time.Now().Format(utils.FormatDateTime)
  515. sysUser.LastUpdatedTime = time.Now().Format(utils.FormatDateTime)
  516. _ = sysUser.Update([]string{"LastLoginTime", "LastUpdatedTime"})
  517. }
  518. resp := new(system.LoginResp)
  519. resp.Authorization = token
  520. resp.Authorization = "authorization=" + token
  521. resp.RealName = sysUser.RealName
  522. resp.AdminName = sysUser.AdminName
  523. resp.RoleName = sysUser.RoleName
  524. resp.SysRoleTypeCode = sysUser.RoleTypeCode //系统角色编码
  525. resp.RoleTypeCode = sysUser.RoleTypeCode
  526. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_GROUP {
  527. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  528. }
  529. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_TEAM {
  530. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  531. }
  532. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_DEPARTMENT {
  533. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  534. }
  535. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_RAI_GROUP {
  536. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_RAI_SELLER
  537. }
  538. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_RAI_DEPARTMENT {
  539. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_RAI_SELLER
  540. }
  541. if sysUser.RoleName == utils.ROLE_NAME_FICC_DIRECTOR {
  542. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  543. }
  544. resp.AdminId = sysUser.AdminId
  545. var productName string
  546. productId := services.GetProductId(sysUser.RoleTypeCode)
  547. if productId == 1 {
  548. productName = utils.COMPANY_PRODUCT_FICC_NAME
  549. } else if productId == 2 {
  550. productName = utils.COMPANY_PRODUCT_RAI_NAME
  551. } else {
  552. productName = "admin"
  553. }
  554. resp.ProductName = productName
  555. resp.Authority = sysUser.Authority
  556. resp.Mobile = sysUser.Mobile
  557. // 设置redis缓存
  558. {
  559. // 获取不可信的登录态,并将该登录态重置掉,不允许多次登录
  560. noTrustLoginKey := fmt.Sprint(utils.CACHE_ACCESS_TOKEN_LOGIN_NO_TRUST, sysUser.AdminId)
  561. noTrustLoginId, _ := utils.Rc.RedisString(noTrustLoginKey)
  562. if noTrustLoginId != `` { // 如果存在不可信设备,那么将其下架
  563. oldNoTrustLoginKey := fmt.Sprint(utils.CACHE_ACCESS_TOKEN_LOGIN, noTrustLoginId)
  564. _ = utils.Rc.Put(oldNoTrustLoginKey, "0", utils.LoginCacheTime*time.Minute)
  565. }
  566. // 如果当前是不可信设备,那么将其加入到不可信名单
  567. loginKey := fmt.Sprint(utils.CACHE_ACCESS_TOKEN_LOGIN, sysSession.Id)
  568. _ = utils.Rc.Put(loginKey, "1", utils.LoginCacheTime*time.Minute)
  569. _ = utils.Rc.Put(noTrustLoginKey, sysSession.Id, utils.LoginCacheTime*time.Minute)
  570. }
  571. // 新增登录记录
  572. go func() {
  573. record := new(system.SysUserLoginRecord)
  574. record.Uid = sysUser.AdminId
  575. record.UserName = req.Username
  576. record.Ip = this.Ctx.Input.IP()
  577. record.Stage = "login"
  578. record.CreateTime = time.Now()
  579. _ = system.AddSysUserLoginRecord(record)
  580. }()
  581. // ETA试用平台-请求中间服务更新用户最后登录时间和次数
  582. if utils.BusinessCode == utils.BusinessCodeSandbox {
  583. go func() {
  584. var r eta_trial.EtaTrialUserReq
  585. r.Mobile = sysUser.Mobile
  586. _, _ = eta_trial.UpdateEtaTrialUserLogin(r)
  587. }()
  588. }
  589. br.Data = resp
  590. br.Ret = 200
  591. br.Success = true
  592. br.Msg = "登录成功"
  593. }
  594. // ForgetAccountGet
  595. // @Title 忘记密码-账号校验
  596. // @Description 忘记密码-账号校验
  597. // @Param request body ForgetAccountGetReq true "type json string"
  598. // @Success 200 Ret=200 获取成功
  599. // @router /forget/account_get [post]
  600. func (this *UserLoginController) ForgetAccountGet() {
  601. br := new(models.BaseResponse).Init()
  602. defer func() {
  603. if br.ErrMsg == "" {
  604. br.IsSendEmail = false
  605. }
  606. this.Data["json"] = br
  607. this.ServeJSON()
  608. }()
  609. type ForgetAccountGetReq struct {
  610. CaptchaId string `description:"验证码ID"`
  611. CaptchaCode string `description:"图形验证码"`
  612. UserName string `description:"用户名"`
  613. }
  614. var req ForgetAccountGetReq
  615. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  616. if err != nil {
  617. br.Msg = "参数解析异常!"
  618. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  619. return
  620. }
  621. req.UserName = strings.TrimSpace(req.UserName)
  622. if req.UserName == "" {
  623. br.Msg = "请输入账号"
  624. return
  625. }
  626. if req.CaptchaId == "" || req.CaptchaCode == "" {
  627. br.Msg = "请输入图形验证码"
  628. return
  629. }
  630. store := services.CaptchaRedis{}
  631. ok := store.Verify(req.CaptchaId, req.CaptchaCode, true)
  632. if !ok {
  633. br.Msg = "验证码错误, 请重新输入"
  634. return
  635. }
  636. sysUser, e := system.GetSysUserByAdminName(req.UserName)
  637. if e != nil {
  638. if utils.IsErrNoRow(e) {
  639. br.Msg = "用户不存在, 请检查"
  640. return
  641. }
  642. br.Msg = "账号错误, 请重新输入"
  643. br.ErrMsg = "用户名获取用户失败, Err: " + e.Error()
  644. return
  645. }
  646. type ForgetAccountCheckResp struct {
  647. Mobile string `description:"手机号"`
  648. Email string `description:"邮箱"`
  649. TelAreaCode string `description:"手机区号"`
  650. }
  651. resp := ForgetAccountCheckResp{
  652. Mobile: sysUser.Mobile,
  653. Email: sysUser.Email,
  654. TelAreaCode: sysUser.TelAreaCode,
  655. }
  656. br.Data = resp
  657. br.Ret = 200
  658. br.Success = true
  659. br.Msg = "获取成功"
  660. }
  661. // ForgetCodeVerify
  662. // @Title 忘记密码-验证码校验
  663. // @Description 忘记密码-验证码校验
  664. // @Param request body ForgetCodeVerifyReq true "type json string"
  665. // @Success 200 Ret=200 获取成功
  666. // @router /forget/code_verify [post]
  667. func (this *UserLoginController) ForgetCodeVerify() {
  668. br := new(models.BaseResponse).Init()
  669. defer func() {
  670. if br.ErrMsg == "" {
  671. br.IsSendEmail = false
  672. }
  673. this.Data["json"] = br
  674. this.ServeJSON()
  675. }()
  676. type ForgetCodeVerifyReq struct {
  677. FindType int `description:"密码找回方式: 1-手机号; 2-邮箱"`
  678. VerifyCode string `description:"验证码"`
  679. UserName string `description:"用户名"`
  680. Mobile string `description:"手机号"`
  681. Email string `description:"邮箱"`
  682. TelAreaCode string `description:"区号"`
  683. }
  684. var req ForgetCodeVerifyReq
  685. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  686. if err != nil {
  687. br.Msg = "参数解析异常!"
  688. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  689. return
  690. }
  691. req.UserName = strings.TrimSpace(req.UserName)
  692. if req.UserName == "" {
  693. br.Msg = "请输入账号"
  694. return
  695. }
  696. if req.VerifyCode == "" {
  697. br.Msg = "请输入验证码"
  698. return
  699. }
  700. // 手机号找回
  701. var verifyRes bool
  702. if req.FindType == 1 {
  703. req.Mobile = strings.TrimSpace(req.Mobile)
  704. if req.Mobile == "" {
  705. br.Msg = "请输入手机号"
  706. return
  707. }
  708. if req.TelAreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Mobile) {
  709. br.Msg = "您的手机号输入有误, 请检查"
  710. return
  711. }
  712. expiredTime := time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute).Format(utils.FormatDateTime)
  713. recordCond := ` AND source = ? AND mobile = ? AND code = ? AND expired_time <= ?`
  714. recordPars := make([]interface{}, 0)
  715. recordPars = append(recordPars, system.AdminVerifyCodeRecordSourceForget, req.Mobile, req.VerifyCode, expiredTime)
  716. recordOb := new(system.AdminVerifyCodeRecord)
  717. _, e := recordOb.GetItemByCondition(recordCond, recordPars)
  718. if e != nil {
  719. if utils.IsErrNoRow(e) {
  720. br.Msg = "验证码错误, 请重新输入"
  721. return
  722. }
  723. br.Msg = "验证码错误, 请重新输入"
  724. br.ErrMsg = "获取手机号登陆验证码失败, Err: " + e.Error()
  725. return
  726. }
  727. verifyRes = true
  728. }
  729. // 邮箱找回
  730. if req.FindType == 2 {
  731. req.Email = strings.TrimSpace(req.Email)
  732. if req.Email == "" {
  733. br.Msg = "请输入邮箱"
  734. return
  735. }
  736. if !utils.ValidateEmailFormatat(req.Email) {
  737. br.Msg = "您的邮箱输入有误, 请检查"
  738. return
  739. }
  740. expiredTime := time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute).Format(utils.FormatDateTime)
  741. recordCond := ` AND source = ? AND email = ? AND code = ? AND expired_time <= ?`
  742. recordPars := make([]interface{}, 0)
  743. recordPars = append(recordPars, system.AdminVerifyCodeRecordSourceForget, req.Email, req.VerifyCode, expiredTime)
  744. recordOb := new(system.AdminVerifyCodeRecord)
  745. _, e := recordOb.GetItemByCondition(recordCond, recordPars)
  746. if e != nil {
  747. if utils.IsErrNoRow(e) {
  748. br.Msg = "验证码错误, 请重新输入"
  749. return
  750. }
  751. br.Msg = "验证码错误, 请重新输入"
  752. br.ErrMsg = "获取手机号登陆验证码失败, Err: " + e.Error()
  753. return
  754. }
  755. verifyRes = true
  756. }
  757. if verifyRes {
  758. successKey := fmt.Sprint(utils.CACHE_FIND_PASS_VERIFY, req.UserName)
  759. _ = utils.Rc.Put(successKey, "true", utils.GetTodayLastSecond())
  760. }
  761. br.Data = verifyRes
  762. br.Ret = 200
  763. br.Success = true
  764. br.Msg = "操作成功"
  765. }
  766. // ForgetResetPass
  767. // @Title 忘记密码-重置密码
  768. // @Description 忘记密码-重置密码
  769. // @Param request body ForgetResetPassReq true "type json string"
  770. // @Success 200 Ret=200 获取成功
  771. // @router /forget/reset_pass [post]
  772. func (this *UserLoginController) ForgetResetPass() {
  773. br := new(models.BaseResponse).Init()
  774. defer func() {
  775. if br.ErrMsg == "" {
  776. br.IsSendEmail = false
  777. }
  778. this.Data["json"] = br
  779. this.ServeJSON()
  780. }()
  781. type ForgetResetPassReq struct {
  782. UserName string `description:"用户名"`
  783. Password string `description:"密码"`
  784. RePassword string `description:"重复密码"`
  785. }
  786. var req ForgetResetPassReq
  787. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  788. if err != nil {
  789. br.Msg = "参数解析异常!"
  790. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  791. return
  792. }
  793. req.UserName = strings.TrimSpace(req.UserName)
  794. if req.UserName == "" {
  795. br.Msg = "请输入账号"
  796. return
  797. }
  798. req.Password = strings.TrimSpace(req.Password)
  799. if req.Password == "" {
  800. br.Msg = "请输入新密码"
  801. return
  802. }
  803. req.RePassword = strings.TrimSpace(req.RePassword)
  804. if req.RePassword == "" {
  805. br.Msg = "请确认新密码"
  806. return
  807. }
  808. if req.Password != req.RePassword {
  809. br.Msg = "两次密码输入不一致, 请检查"
  810. return
  811. }
  812. // 接上一步-验证码校验成功后才允许修改
  813. successKey := fmt.Sprint(utils.CACHE_FIND_PASS_VERIFY, req.UserName)
  814. verifyOk, _ := utils.Rc.RedisString(successKey)
  815. if verifyOk != "true" {
  816. br.Msg = "验证码校验失效, 请重新验证"
  817. br.ErrMsg = "重置密码异常, 验证码校验已失效"
  818. return
  819. }
  820. sysUser, e := system.GetSysAdminByName(req.UserName)
  821. if e != nil {
  822. if utils.IsErrNoRow(e) {
  823. br.Msg = "用户不存在, 请检查"
  824. return
  825. }
  826. br.Msg = "操作失败"
  827. br.ErrMsg = "密码找回获取用户失败, Err:" + e.Error()
  828. return
  829. }
  830. b, e := base64.StdEncoding.DecodeString(req.Password)
  831. if e != nil {
  832. br.Msg = "操作失败"
  833. br.ErrMsg = "解析密码失败, Err:" + e.Error()
  834. return
  835. }
  836. pwd := string(b)
  837. pwd = utils.MD5(pwd)
  838. sysUser.Password = pwd
  839. sysUser.LastUpdatedPasswordTime = time.Now().Format(utils.FormatDateTime)
  840. // 重置密码后更新一下登录时间, 不然账号密码登录如果超过60天还会被异常校验再校验一次, 影响体验
  841. sysUser.LastLoginTime = time.Now().Format(utils.FormatDateTime)
  842. sysUser.LastUpdatedTime = time.Now().Format(utils.FormatDateTime)
  843. if e = sysUser.Update([]string{"Password", "LastUpdatedPasswordTime", "LastLoginTime", "LastUpdatedTime"}); e != nil {
  844. br.Msg = "操作失败"
  845. br.ErrMsg = "更新密码失败, Err: " + e.Error()
  846. return
  847. }
  848. // 如果有异常标记也清除掉
  849. abnormalKey := fmt.Sprint(utils.CACHE_ABNORMAL_LOGIN, req.UserName)
  850. _ = utils.Rc.Delete(abnormalKey)
  851. _ = utils.Rc.Delete(successKey)
  852. mobileAbnormalKey := fmt.Sprint(utils.MOBILE_CACHE_ABNORMAL_LOGIN, req.UserName)
  853. _ = utils.Rc.Delete(mobileAbnormalKey)
  854. // 同步ETA用户缓存
  855. if utils.BusinessCode == utils.BusinessCodeRelease || utils.BusinessCode == utils.BusinessCodeSandbox {
  856. var syncData system.SyncAdminData
  857. syncData.Source = utils.SOURCE_ETA_FLAG
  858. syncData.AdminName = sysUser.AdminName
  859. _ = utils.Rc.LPush(utils.CACHE_SYNC_ADMIN, syncData)
  860. }
  861. br.Data = true
  862. br.Ret = 200
  863. br.Success = true
  864. br.Msg = "操作成功"
  865. }
  866. // AreaCodeList
  867. // @Title 手机号区号列表
  868. // @Description 手机号区号列表
  869. // @Success 200 Ret=200 获取成功
  870. // @router /area_code/list [get]
  871. func (this *UserLoginController) AreaCodeList() {
  872. br := new(models.BaseResponse).Init()
  873. defer func() {
  874. if br.ErrMsg == "" {
  875. br.IsSendEmail = false
  876. }
  877. this.Data["json"] = br
  878. this.ServeJSON()
  879. }()
  880. type AreaCodeListResp struct {
  881. Name string `description:"地区"`
  882. Value string `description:"区号"`
  883. }
  884. resp := make([]AreaCodeListResp, 0)
  885. confAuth, e := company.GetConfigDetailByCode(company.ConfAreaCodeListKey)
  886. if e != nil {
  887. br.Msg = "获取失败"
  888. br.ErrMsg = "获取手机号区号配置失败, Err: " + e.Error()
  889. return
  890. }
  891. if confAuth.ConfigValue == "" {
  892. br.Msg = "获取失败"
  893. br.ErrMsg = "手机号区号配置为空"
  894. return
  895. }
  896. if e := json.Unmarshal([]byte(confAuth.ConfigValue), &resp); e != nil {
  897. br.Msg = "获取失败"
  898. br.ErrMsg = "手机号区号配置有误"
  899. return
  900. }
  901. br.Data = resp
  902. br.Ret = 200
  903. br.Success = true
  904. br.Msg = "获取成功"
  905. }
  906. // CheckUserLdap
  907. // @Title 校验用户是否为域用户
  908. // @Description 校验用户是否为域用户
  909. // @Param request body CheckUserLdapReq true "type json string"
  910. // @Success 200 Ret=200 获取成功
  911. // @router /ldap/user_check [post]
  912. func (this *UserLoginController) CheckUserLdap() {
  913. br := new(models.BaseResponse).Init()
  914. defer func() {
  915. if br.ErrMsg == "" {
  916. br.IsSendEmail = false
  917. }
  918. this.Data["json"] = br
  919. this.ServeJSON()
  920. }()
  921. type CheckUserLdapReq struct {
  922. UserName string `description:"用户名"`
  923. }
  924. var req CheckUserLdapReq
  925. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  926. if err != nil {
  927. br.Msg = "参数解析异常!"
  928. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  929. return
  930. }
  931. req.UserName = strings.TrimSpace(req.UserName)
  932. if req.UserName == "" {
  933. br.Msg = "请输入账号"
  934. return
  935. }
  936. isLdap := false
  937. accountUser, e := system.GetSysUserByAdminName(req.UserName)
  938. if e != nil {
  939. // 无该用户视作普通用户
  940. if utils.IsErrNoRow(e) {
  941. br.Ret = 200
  942. br.Data = false
  943. br.Success = true
  944. br.Msg = "操作成功"
  945. return
  946. }
  947. br.Msg = "操作失败"
  948. br.ErrMsg = "获取用户信息失败, Err: " + e.Error()
  949. return
  950. }
  951. if accountUser.IsLdap == 1 {
  952. isLdap = true
  953. }
  954. br.Data = isLdap
  955. br.Ret = 200
  956. br.Success = true
  957. br.Msg = "操作成功"
  958. }
  959. // BaseInfo
  960. // @Title 基础信息
  961. // @Description 基础信息
  962. // @Success 200 Ret=200 获取成功
  963. // @router /base_info [get]
  964. func (this *UserLoginController) BaseInfo() {
  965. br := new(models.BaseResponse).Init()
  966. defer func() {
  967. if br.ErrMsg == "" {
  968. br.IsSendEmail = false
  969. }
  970. this.Data["json"] = br
  971. this.ServeJSON()
  972. }()
  973. configKeys := []string{
  974. "ICPLicense", "ETATitle", "TabName", "LogoCN", "LogoEN", "LogoCNMini", "LogoENMini", "LoginLeftImg",
  975. "ETASubTitleCN", "ETASubTitleEN","IsLoginPictureVerifyCode",
  976. }
  977. configOb := new(models.BusinessConf)
  978. list, e := configOb.GetItemsByCondition(``, make([]interface{}, 0), []string{}, "")
  979. if e != nil {
  980. br.Msg = "获取失败"
  981. br.ErrMsg = fmt.Sprintf("获取配置失败, %v", e)
  982. return
  983. }
  984. resp := make(map[string]*models.BusinessConf)
  985. for _, v := range list {
  986. if !utils.InArrayByStr(configKeys, v.ConfKey) {
  987. continue
  988. }
  989. resp[v.ConfKey] = v
  990. }
  991. br.Data = resp
  992. br.Ret = 200
  993. br.Success = true
  994. br.Msg = "获取成功"
  995. }