user_login.go 29 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007
  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. "github.com/mojocn/base64Captcha"
  13. "image/color"
  14. "strings"
  15. "time"
  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 e.Error() == utils.ErrNoRow() {
  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 e.Error() == utils.ErrNoRow() {
  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. }
  248. var req UserLoginReq
  249. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  250. if err != nil {
  251. br.Msg = "参数解析异常!"
  252. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  253. return
  254. }
  255. req.Username = strings.TrimSpace(req.Username)
  256. req.Mobile = strings.TrimSpace(req.Mobile)
  257. if req.Mobile != "" {
  258. if !utils.ValidateMobileFormatat(req.Mobile) {
  259. br.Msg = "您的手机号输入有误, 请检查"
  260. return
  261. }
  262. }
  263. req.Email = strings.TrimSpace(req.Email)
  264. if req.Email != "" {
  265. if !utils.ValidateEmailFormatat(req.Email) {
  266. br.Msg = "您的邮箱输入有误, 请检查"
  267. return
  268. }
  269. }
  270. req.VerifyCode = strings.TrimSpace(req.VerifyCode)
  271. if req.LoginType != 1 && req.LoginType != 2 && req.LoginType != 3 {
  272. br.Msg = "登录方式有误"
  273. br.ErrMsg = fmt.Sprintf("登录方式有误, Type: %d", req.LoginType)
  274. return
  275. }
  276. sysUser := new(system.Admin)
  277. // 账号密码登录
  278. if req.LoginType == 1 {
  279. if req.Username == "" {
  280. br.Msg = "请输入账号"
  281. return
  282. }
  283. if req.Password == "" {
  284. br.Msg = "请输入密码"
  285. return
  286. }
  287. // 判断账号是否为异常登录, 算作异常的话就需要换另外的方式去登录了
  288. abnormalKey := fmt.Sprint(utils.CACHE_ABNORMAL_LOGIN, req.Username)
  289. isAbnormal, _ := utils.Rc.RedisString(abnormalKey)
  290. if isAbnormal != "" {
  291. br.Ret = models.BaseRespCodeAbnormalLogin
  292. br.Msg = "账号异常, 请进行手机号/邮箱校验"
  293. return
  294. }
  295. // 查询账号信息
  296. errPassKey := fmt.Sprint(utils.CACHE_LOGIN_ERR_PASS, req.Username)
  297. accountUser, e := system.GetSysUserByAdminName(req.Username)
  298. if e != nil {
  299. br.Ret = models.BaseRespCodeLoginErr
  300. br.Msg = "登录失败, 账号或密码错误"
  301. // 账号查询异常均进行标记, 避免一直尝试进行登录
  302. if !utils.Rc.IsExist(errPassKey) {
  303. _ = utils.Rc.Put(errPassKey, 1, utils.GetTodayLastSecond())
  304. return
  305. }
  306. errNum, _ := utils.Rc.RedisInt(errPassKey)
  307. errNum += 1
  308. if errNum < 6 {
  309. _ = utils.Rc.Put(errPassKey, errNum, utils.GetTodayLastSecond())
  310. return
  311. }
  312. // 标记异常登录, 重置计数
  313. br.Ret = models.BaseRespCodeAbnormalLogin
  314. br.Msg = "账号异常, 请进行手机号/邮箱校验"
  315. _ = utils.Rc.Put(abnormalKey, "true", utils.GetTodayLastSecond())
  316. _ = utils.Rc.Delete(errPassKey)
  317. return
  318. }
  319. // 系统用户-账号密码校验
  320. if accountUser.IsLdap == 0 {
  321. dbPass := utils.MD5(fmt.Sprintf("%s%s%s", accountUser.Password, utils.UserLoginSalt, req.ReqTime))
  322. if req.Password != dbPass {
  323. br.Ret = models.BaseRespCodeLoginErr
  324. br.Msg = "登录失败, 账号或密码错误"
  325. // 错误密码计数, 超过6次标记异常
  326. if !utils.Rc.IsExist(errPassKey) {
  327. _ = utils.Rc.Put(errPassKey, 1, utils.GetTodayLastSecond())
  328. return
  329. }
  330. errNum, _ := utils.Rc.RedisInt(errPassKey)
  331. errNum += 1
  332. if errNum < 6 {
  333. _ = utils.Rc.Put(errPassKey, errNum, utils.GetTodayLastSecond())
  334. return
  335. }
  336. // 标记异常登录, 重置计数
  337. br.Ret = models.BaseRespCodeAbnormalLogin
  338. br.Msg = "账号异常, 请进行手机号/邮箱校验"
  339. _ = utils.Rc.Put(abnormalKey, "true", utils.GetTodayLastSecond())
  340. _ = utils.Rc.Delete(errPassKey)
  341. return
  342. }
  343. if accountUser.Enabled == 0 {
  344. br.Msg = "您的账号已被禁用, 如需登录, 请联系管理员"
  345. br.ErrMsg = fmt.Sprintf("账号已被禁用, 登录账号: %s, 账户名称: %s", accountUser.AdminName, accountUser.RealName)
  346. return
  347. }
  348. // 异常登录-是否登录间隔大于60天
  349. if isAbnormal == "" {
  350. abnormalTime := time.Now().AddDate(0, 0, -60)
  351. lastLogin, _ := time.ParseInLocation(utils.FormatDateTime, accountUser.LastLoginTime, time.Local)
  352. if !lastLogin.IsZero() && lastLogin.Before(abnormalTime) {
  353. br.Msg = "请进行异常登录校验"
  354. br.Ret = models.BaseRespCodeAbnormalLogin
  355. // 标记异常登录
  356. _ = utils.Rc.Put(abnormalKey, "true", utils.GetTodayLastSecond())
  357. return
  358. }
  359. }
  360. }
  361. // 如果是域账号, 那么取出原始密码走AD域校验
  362. if accountUser.IsLdap == 1 {
  363. passDecode, e := base64.StdEncoding.DecodeString(req.Password)
  364. if e != nil {
  365. br.Ret = models.BaseRespCodeLoginErr
  366. br.Msg = "登录失败, 账号或密码错误"
  367. br.ErrMsg = "Pass Decode err: " + e.Error()
  368. return
  369. }
  370. originPass := strings.Replace(string(passDecode), utils.UserLoginSalt, "", 1)
  371. fmt.Println("originPass: ", originPass)
  372. pass, e := services.LdapUserCheck(req.Username, originPass)
  373. if e != nil {
  374. br.Ret = models.BaseRespCodeLoginErr
  375. br.Msg = "登录失败, 账号或密码错误"
  376. br.ErrMsg = "LdapLogin err: " + e.Error()
  377. return
  378. }
  379. if !pass {
  380. br.Ret = models.BaseRespCodeLoginErr
  381. br.Msg = "登录失败, 账号或密码错误"
  382. return
  383. }
  384. }
  385. sysUser = accountUser
  386. }
  387. // 手机号
  388. if req.LoginType == 2 {
  389. if req.Mobile == "" {
  390. br.Msg = "请输入手机号"
  391. return
  392. }
  393. // 大陆校验区号
  394. if req.TelAreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Mobile) {
  395. br.Msg = "您的手机号输入有误, 请检查"
  396. return
  397. }
  398. if req.VerifyCode == "" {
  399. br.Msg = "请输入验证码"
  400. return
  401. }
  402. expiredTime := time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute).Format(utils.FormatDateTime)
  403. recordCond := ` AND source = ? AND mobile = ? AND code = ? AND expired_time <= ?`
  404. recordPars := make([]interface{}, 0)
  405. recordPars = append(recordPars, system.AdminVerifyCodeRecordSourceLogin, req.Mobile, req.VerifyCode, expiredTime)
  406. recordOb := new(system.AdminVerifyCodeRecord)
  407. _, e := recordOb.GetItemByCondition(recordCond, recordPars)
  408. if e != nil {
  409. if e.Error() == utils.ErrNoRow() {
  410. br.Msg = "验证码错误, 请重新输入"
  411. return
  412. }
  413. br.Msg = "验证码错误, 请重新输入"
  414. br.ErrMsg = "获取手机号登陆验证码失败, Err: " + e.Error()
  415. return
  416. }
  417. mobileUser, e := system.GetSysUserByMobile(req.Mobile)
  418. if e != nil {
  419. if e.Error() == utils.ErrNoRow() {
  420. br.Msg = "您的手机号未绑定账号, 请检查"
  421. return
  422. }
  423. br.Msg = "您的手机号未绑定账号, 请检查"
  424. br.ErrMsg = "手机号获取用户失败, Err:" + e.Error()
  425. return
  426. }
  427. if mobileUser.Enabled == 0 {
  428. br.Msg = "您的账号已被禁用, 如需登录, 请联系管理员"
  429. br.ErrMsg = fmt.Sprintf("账号已被禁用, 登录手机号: %s", req.Mobile)
  430. return
  431. }
  432. sysUser = mobileUser
  433. }
  434. // 邮箱登录
  435. if req.LoginType == 3 {
  436. if req.Email == "" {
  437. br.Msg = "请输入邮箱"
  438. return
  439. }
  440. if !utils.ValidateEmailFormatat(req.Email) {
  441. br.Msg = "您的邮箱输入有误, 请检查"
  442. return
  443. }
  444. if req.VerifyCode == "" {
  445. br.Msg = "请输入验证码"
  446. return
  447. }
  448. expiredTime := time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute).Format(utils.FormatDateTime)
  449. recordCond := ` AND source = ? AND email = ? AND code = ? AND expired_time <= ?`
  450. recordPars := make([]interface{}, 0)
  451. recordPars = append(recordPars, system.AdminVerifyCodeRecordSourceLogin, req.Email, req.VerifyCode, expiredTime)
  452. recordOb := new(system.AdminVerifyCodeRecord)
  453. _, e := recordOb.GetItemByCondition(recordCond, recordPars)
  454. if e != nil {
  455. if e.Error() == utils.ErrNoRow() {
  456. br.Msg = "验证码错误, 请重新输入"
  457. return
  458. }
  459. br.Msg = "验证码错误, 请重新输入"
  460. br.ErrMsg = "获取手机号登陆验证码失败, Err: " + e.Error()
  461. return
  462. }
  463. emailUser, e := system.GetSysUserByEmail(req.Email)
  464. if e != nil {
  465. if e.Error() == utils.ErrNoRow() {
  466. br.Msg = "您的邮箱未绑定账号, 请检查"
  467. return
  468. }
  469. br.Msg = "您的邮箱未绑定账号, 请检查"
  470. br.ErrMsg = "邮箱获取用户失败, Err:" + e.Error()
  471. return
  472. }
  473. if emailUser.Enabled == 0 {
  474. br.Msg = "您的账号已被禁用, 如需登录, 请联系管理员"
  475. br.ErrMsg = fmt.Sprintf("账号已被禁用, 登录邮箱: %s", req.Email)
  476. return
  477. }
  478. sysUser = emailUser
  479. }
  480. // 登录成功(无论哪种方式登录), 清除异常标记
  481. abnormalCache := fmt.Sprint(utils.CACHE_ABNORMAL_LOGIN, sysUser.AdminName)
  482. errPassCache := fmt.Sprint(utils.CACHE_LOGIN_ERR_PASS, sysUser.AdminName)
  483. _ = utils.Rc.Delete(abnormalCache)
  484. _ = utils.Rc.Delete(errPassCache)
  485. account := utils.MD5(sysUser.AdminName)
  486. token := utils.GenToken(account)
  487. sysSession := new(system.SysSession)
  488. sysSession.UserName = req.Username
  489. sysSession.SysUserId = sysUser.AdminId
  490. sysSession.ExpiredTime = time.Now().AddDate(0, 0, 90)
  491. sysSession.IsRemember = 0 // 均需要做过期校验
  492. sysSession.CreatedTime = time.Now()
  493. sysSession.LastUpdatedTime = time.Now()
  494. sysSession.AccessToken = token
  495. if e := system.AddSysSession(sysSession); e != nil {
  496. br.Msg = "登录失败"
  497. br.ErrMsg = "新增session信息失败, Err:" + e.Error()
  498. return
  499. }
  500. // 修改最后登录时间
  501. {
  502. sysUser.LastLoginTime = time.Now().Format(utils.FormatDateTime)
  503. sysUser.LastUpdatedTime = time.Now().Format(utils.FormatDateTime)
  504. _ = sysUser.Update([]string{"LastLoginTime", "LastUpdatedTime"})
  505. }
  506. resp := new(system.LoginResp)
  507. resp.Authorization = token
  508. resp.Authorization = "authorization=" + token + "$account=" + account
  509. resp.RealName = sysUser.RealName
  510. resp.AdminName = sysUser.AdminName
  511. resp.RoleName = sysUser.RoleName
  512. resp.SysRoleTypeCode = sysUser.RoleTypeCode //系统角色编码
  513. resp.RoleTypeCode = sysUser.RoleTypeCode
  514. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_GROUP {
  515. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  516. }
  517. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_TEAM {
  518. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  519. }
  520. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_DEPARTMENT {
  521. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  522. }
  523. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_RAI_GROUP {
  524. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_RAI_SELLER
  525. }
  526. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_RAI_DEPARTMENT {
  527. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_RAI_SELLER
  528. }
  529. if sysUser.RoleName == utils.ROLE_NAME_FICC_DIRECTOR {
  530. resp.RoleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  531. }
  532. resp.AdminId = sysUser.AdminId
  533. var productName string
  534. productId := services.GetProductId(sysUser.RoleTypeCode)
  535. if productId == 1 {
  536. productName = utils.COMPANY_PRODUCT_FICC_NAME
  537. } else if productId == 2 {
  538. productName = utils.COMPANY_PRODUCT_RAI_NAME
  539. } else {
  540. productName = "admin"
  541. }
  542. resp.ProductName = productName
  543. resp.Authority = sysUser.Authority
  544. // 设置redis缓存
  545. {
  546. // 获取不可信的登录态,并将该登录态重置掉,不允许多次登录
  547. noTrustLoginKey := fmt.Sprint(utils.CACHE_ACCESS_TOKEN_LOGIN_NO_TRUST, sysUser.AdminId)
  548. noTrustLoginId, _ := utils.Rc.RedisString(noTrustLoginKey)
  549. if noTrustLoginId != `` { // 如果存在不可信设备,那么将其下架
  550. oldNoTrustLoginKey := fmt.Sprint(utils.CACHE_ACCESS_TOKEN_LOGIN, noTrustLoginId)
  551. _ = utils.Rc.Put(oldNoTrustLoginKey, "0", utils.LoginCacheTime*time.Minute)
  552. }
  553. // 如果当前是不可信设备,那么将其加入到不可信名单
  554. loginKey := fmt.Sprint(utils.CACHE_ACCESS_TOKEN_LOGIN, sysSession.Id)
  555. _ = utils.Rc.Put(loginKey, "1", utils.LoginCacheTime*time.Minute)
  556. _ = utils.Rc.Put(noTrustLoginKey, sysSession.Id, utils.LoginCacheTime*time.Minute)
  557. }
  558. // 新增登录记录
  559. go func() {
  560. record := new(system.SysUserLoginRecord)
  561. record.Uid = sysUser.AdminId
  562. record.UserName = req.Username
  563. record.Ip = this.Ctx.Input.IP()
  564. record.Stage = "login"
  565. record.CreateTime = time.Now()
  566. _ = system.AddSysUserLoginRecord(record)
  567. }()
  568. // ETA试用平台-请求中间服务更新用户最后登录时间和次数
  569. if utils.BusinessCode == utils.BusinessCodeSandbox {
  570. go func() {
  571. var r eta_trial.EtaTrialUserReq
  572. r.Mobile = sysUser.Mobile
  573. _, _ = eta_trial.UpdateEtaTrialUserLogin(r)
  574. }()
  575. }
  576. br.Data = resp
  577. br.Ret = 200
  578. br.Success = true
  579. br.Msg = "登录成功"
  580. }
  581. // ForgetAccountGet
  582. // @Title 忘记密码-账号校验
  583. // @Description 忘记密码-账号校验
  584. // @Param request body ForgetAccountGetReq true "type json string"
  585. // @Success 200 Ret=200 获取成功
  586. // @router /forget/account_get [post]
  587. func (this *UserLoginController) ForgetAccountGet() {
  588. br := new(models.BaseResponse).Init()
  589. defer func() {
  590. if br.ErrMsg == "" {
  591. br.IsSendEmail = false
  592. }
  593. this.Data["json"] = br
  594. this.ServeJSON()
  595. }()
  596. type ForgetAccountGetReq struct {
  597. CaptchaId string `description:"验证码ID"`
  598. CaptchaCode string `description:"图形验证码"`
  599. UserName string `description:"用户名"`
  600. }
  601. var req ForgetAccountGetReq
  602. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  603. if err != nil {
  604. br.Msg = "参数解析异常!"
  605. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  606. return
  607. }
  608. req.UserName = strings.TrimSpace(req.UserName)
  609. if req.UserName == "" {
  610. br.Msg = "请输入账号"
  611. return
  612. }
  613. if req.CaptchaId == "" || req.CaptchaCode == "" {
  614. br.Msg = "请输入图形验证码"
  615. return
  616. }
  617. store := services.CaptchaRedis{}
  618. ok := store.Verify(req.CaptchaId, req.CaptchaCode, true)
  619. if !ok {
  620. br.Msg = "验证码错误, 请重新输入"
  621. return
  622. }
  623. sysUser, e := system.GetSysUserByAdminName(req.UserName)
  624. if e != nil {
  625. if e.Error() == utils.ErrNoRow() {
  626. br.Msg = "用户不存在, 请检查"
  627. return
  628. }
  629. br.Msg = "账号错误, 请重新输入"
  630. br.ErrMsg = "用户名获取用户失败, Err: " + e.Error()
  631. return
  632. }
  633. type ForgetAccountCheckResp struct {
  634. Mobile string `description:"手机号"`
  635. Email string `description:"邮箱"`
  636. TelAreaCode string `description:"手机区号"`
  637. }
  638. resp := ForgetAccountCheckResp{
  639. Mobile: sysUser.Mobile,
  640. Email: sysUser.Email,
  641. TelAreaCode: sysUser.TelAreaCode,
  642. }
  643. br.Data = resp
  644. br.Ret = 200
  645. br.Success = true
  646. br.Msg = "获取成功"
  647. }
  648. // ForgetCodeVerify
  649. // @Title 忘记密码-验证码校验
  650. // @Description 忘记密码-验证码校验
  651. // @Param request body ForgetCodeVerifyReq true "type json string"
  652. // @Success 200 Ret=200 获取成功
  653. // @router /forget/code_verify [post]
  654. func (this *UserLoginController) ForgetCodeVerify() {
  655. br := new(models.BaseResponse).Init()
  656. defer func() {
  657. if br.ErrMsg == "" {
  658. br.IsSendEmail = false
  659. }
  660. this.Data["json"] = br
  661. this.ServeJSON()
  662. }()
  663. type ForgetCodeVerifyReq struct {
  664. FindType int `description:"密码找回方式: 1-手机号; 2-邮箱"`
  665. VerifyCode string `description:"验证码"`
  666. UserName string `description:"用户名"`
  667. Mobile string `description:"手机号"`
  668. Email string `description:"邮箱"`
  669. TelAreaCode string `description:"区号"`
  670. }
  671. var req ForgetCodeVerifyReq
  672. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  673. if err != nil {
  674. br.Msg = "参数解析异常!"
  675. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  676. return
  677. }
  678. req.UserName = strings.TrimSpace(req.UserName)
  679. if req.UserName == "" {
  680. br.Msg = "请输入账号"
  681. return
  682. }
  683. if req.VerifyCode == "" {
  684. br.Msg = "请输入验证码"
  685. return
  686. }
  687. // 手机号找回
  688. var verifyRes bool
  689. if req.FindType == 1 {
  690. req.Mobile = strings.TrimSpace(req.Mobile)
  691. if req.Mobile == "" {
  692. br.Msg = "请输入手机号"
  693. return
  694. }
  695. if req.TelAreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Mobile) {
  696. br.Msg = "您的手机号输入有误, 请检查"
  697. return
  698. }
  699. expiredTime := time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute).Format(utils.FormatDateTime)
  700. recordCond := ` AND source = ? AND mobile = ? AND code = ? AND expired_time <= ?`
  701. recordPars := make([]interface{}, 0)
  702. recordPars = append(recordPars, system.AdminVerifyCodeRecordSourceForget, req.Mobile, req.VerifyCode, expiredTime)
  703. recordOb := new(system.AdminVerifyCodeRecord)
  704. _, e := recordOb.GetItemByCondition(recordCond, recordPars)
  705. if e != nil {
  706. if e.Error() == utils.ErrNoRow() {
  707. br.Msg = "验证码错误, 请重新输入"
  708. return
  709. }
  710. br.Msg = "验证码错误, 请重新输入"
  711. br.ErrMsg = "获取手机号登陆验证码失败, Err: " + e.Error()
  712. return
  713. }
  714. verifyRes = true
  715. }
  716. // 邮箱找回
  717. if req.FindType == 2 {
  718. req.Email = strings.TrimSpace(req.Email)
  719. if req.Email == "" {
  720. br.Msg = "请输入邮箱"
  721. return
  722. }
  723. if !utils.ValidateEmailFormatat(req.Email) {
  724. br.Msg = "您的邮箱输入有误, 请检查"
  725. return
  726. }
  727. expiredTime := time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute).Format(utils.FormatDateTime)
  728. recordCond := ` AND source = ? AND email = ? AND code = ? AND expired_time <= ?`
  729. recordPars := make([]interface{}, 0)
  730. recordPars = append(recordPars, system.AdminVerifyCodeRecordSourceForget, req.Email, req.VerifyCode, expiredTime)
  731. recordOb := new(system.AdminVerifyCodeRecord)
  732. _, e := recordOb.GetItemByCondition(recordCond, recordPars)
  733. if e != nil {
  734. if e.Error() == utils.ErrNoRow() {
  735. br.Msg = "验证码错误, 请重新输入"
  736. return
  737. }
  738. br.Msg = "验证码错误, 请重新输入"
  739. br.ErrMsg = "获取手机号登陆验证码失败, Err: " + e.Error()
  740. return
  741. }
  742. verifyRes = true
  743. }
  744. if verifyRes {
  745. successKey := fmt.Sprint(utils.CACHE_FIND_PASS_VERIFY, req.UserName)
  746. _ = utils.Rc.Put(successKey, "true", utils.GetTodayLastSecond())
  747. }
  748. br.Data = verifyRes
  749. br.Ret = 200
  750. br.Success = true
  751. br.Msg = "操作成功"
  752. }
  753. // ForgetResetPass
  754. // @Title 忘记密码-重置密码
  755. // @Description 忘记密码-重置密码
  756. // @Param request body ForgetResetPassReq true "type json string"
  757. // @Success 200 Ret=200 获取成功
  758. // @router /forget/reset_pass [post]
  759. func (this *UserLoginController) ForgetResetPass() {
  760. br := new(models.BaseResponse).Init()
  761. defer func() {
  762. if br.ErrMsg == "" {
  763. br.IsSendEmail = false
  764. }
  765. this.Data["json"] = br
  766. this.ServeJSON()
  767. }()
  768. type ForgetResetPassReq struct {
  769. UserName string `description:"用户名"`
  770. Password string `description:"密码"`
  771. RePassword string `description:"重复密码"`
  772. }
  773. var req ForgetResetPassReq
  774. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  775. if err != nil {
  776. br.Msg = "参数解析异常!"
  777. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  778. return
  779. }
  780. req.UserName = strings.TrimSpace(req.UserName)
  781. if req.UserName == "" {
  782. br.Msg = "请输入账号"
  783. return
  784. }
  785. req.Password = strings.TrimSpace(req.Password)
  786. if req.Password == "" {
  787. br.Msg = "请输入新密码"
  788. return
  789. }
  790. req.RePassword = strings.TrimSpace(req.RePassword)
  791. if req.RePassword == "" {
  792. br.Msg = "请确认新密码"
  793. return
  794. }
  795. if req.Password != req.RePassword {
  796. br.Msg = "两次密码输入不一致, 请检查"
  797. return
  798. }
  799. // 接上一步-验证码校验成功后才允许修改
  800. successKey := fmt.Sprint(utils.CACHE_FIND_PASS_VERIFY, req.UserName)
  801. verifyOk, _ := utils.Rc.RedisString(successKey)
  802. if verifyOk != "true" {
  803. br.Msg = "验证码校验失效, 请重新验证"
  804. br.ErrMsg = "重置密码异常, 验证码校验已失效"
  805. return
  806. }
  807. sysUser, e := system.GetSysAdminByName(req.UserName)
  808. if e != nil {
  809. if e.Error() == utils.ErrNoRow() {
  810. br.Msg = "用户不存在, 请检查"
  811. return
  812. }
  813. br.Msg = "操作失败"
  814. br.ErrMsg = "密码找回获取用户失败, Err:" + e.Error()
  815. return
  816. }
  817. b, e := base64.StdEncoding.DecodeString(req.Password)
  818. if e != nil {
  819. br.Msg = "操作失败"
  820. br.ErrMsg = "解析密码失败, Err:" + e.Error()
  821. return
  822. }
  823. pwd := string(b)
  824. pwd = utils.MD5(pwd)
  825. sysUser.Password = pwd
  826. sysUser.LastUpdatedPasswordTime = time.Now().Format(utils.FormatDateTime)
  827. // 重置密码后更新一下登录时间, 不然账号密码登录如果超过60天还会被异常校验再校验一次, 影响体验
  828. sysUser.LastLoginTime = time.Now().Format(utils.FormatDateTime)
  829. sysUser.LastUpdatedTime = time.Now().Format(utils.FormatDateTime)
  830. if e = sysUser.Update([]string{"Password", "LastUpdatedPasswordTime", "LastLoginTime", "LastUpdatedTime"}); e != nil {
  831. br.Msg = "操作失败"
  832. br.ErrMsg = "更新密码失败, Err: " + e.Error()
  833. return
  834. }
  835. // 如果有异常标记也清除掉
  836. abnormalKey := fmt.Sprint(utils.CACHE_ABNORMAL_LOGIN, req.UserName)
  837. _ = utils.Rc.Delete(abnormalKey)
  838. _ = utils.Rc.Delete(successKey)
  839. // 同步ETA用户缓存
  840. if utils.BusinessCode == utils.BusinessCodeRelease || utils.BusinessCode == utils.BusinessCodeSandbox {
  841. var syncData system.SyncAdminData
  842. syncData.Source = utils.SOURCE_ETA_FLAG
  843. syncData.AdminName = sysUser.AdminName
  844. _ = utils.Rc.LPush(utils.CACHE_SYNC_ADMIN, syncData)
  845. }
  846. br.Data = true
  847. br.Ret = 200
  848. br.Success = true
  849. br.Msg = "操作成功"
  850. }
  851. // AreaCodeList
  852. // @Title 手机号区号列表
  853. // @Description 手机号区号列表
  854. // @Success 200 Ret=200 获取成功
  855. // @router /area_code/list [get]
  856. func (this *UserLoginController) AreaCodeList() {
  857. br := new(models.BaseResponse).Init()
  858. defer func() {
  859. if br.ErrMsg == "" {
  860. br.IsSendEmail = false
  861. }
  862. this.Data["json"] = br
  863. this.ServeJSON()
  864. }()
  865. type AreaCodeListResp struct {
  866. Name string `description:"地区"`
  867. Value string `description:"区号"`
  868. }
  869. resp := make([]AreaCodeListResp, 0)
  870. confAuth, e := company.GetConfigDetailByCode(company.ConfAreaCodeListKey)
  871. if e != nil {
  872. br.Msg = "获取失败"
  873. br.ErrMsg = "获取手机号区号配置失败, Err: " + e.Error()
  874. return
  875. }
  876. if confAuth.ConfigValue == "" {
  877. br.Msg = "获取失败"
  878. br.ErrMsg = "手机号区号配置为空"
  879. return
  880. }
  881. if e := json.Unmarshal([]byte(confAuth.ConfigValue), &resp); e != nil {
  882. br.Msg = "获取失败"
  883. br.ErrMsg = "手机号区号配置有误"
  884. return
  885. }
  886. br.Data = resp
  887. br.Ret = 200
  888. br.Success = true
  889. br.Msg = "获取成功"
  890. }
  891. // CheckUserLdap
  892. // @Title 校验用户是否为域用户
  893. // @Description 校验用户是否为域用户
  894. // @Param request body CheckUserLdapReq true "type json string"
  895. // @Success 200 Ret=200 获取成功
  896. // @router /ldap/user_check [post]
  897. func (this *UserLoginController) CheckUserLdap() {
  898. br := new(models.BaseResponse).Init()
  899. defer func() {
  900. if br.ErrMsg == "" {
  901. br.IsSendEmail = false
  902. }
  903. this.Data["json"] = br
  904. this.ServeJSON()
  905. }()
  906. type CheckUserLdapReq struct {
  907. UserName string `description:"用户名"`
  908. }
  909. var req CheckUserLdapReq
  910. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  911. if err != nil {
  912. br.Msg = "参数解析异常!"
  913. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  914. return
  915. }
  916. req.UserName = strings.TrimSpace(req.UserName)
  917. if req.UserName == "" {
  918. br.Msg = "请输入账号"
  919. return
  920. }
  921. isLdap := false
  922. accountUser, e := system.GetSysUserByAdminName(req.UserName)
  923. if e != nil {
  924. // 无该用户视作普通用户
  925. if e.Error() == utils.ErrNoRow() {
  926. br.Data = false
  927. br.Success = true
  928. br.Msg = "操作成功"
  929. return
  930. }
  931. br.Msg = "操作失败"
  932. br.ErrMsg = "获取用户信息失败, Err: " + e.Error()
  933. return
  934. }
  935. if accountUser.IsLdap == 1 {
  936. isLdap = true
  937. }
  938. br.Data = isLdap
  939. br.Ret = 200
  940. br.Success = true
  941. br.Msg = "操作成功"
  942. }