user_login.go 30 KB

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