user_login.go 30 KB

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