eta_api/controllers/sys_user.go

package controllers

import (
	"encoding/json"
	"eta/eta_api/models"
	"eta/eta_api/models/system"
	"eta/eta_api/services"
	"eta/eta_api/utils"
	"fmt"
	"time"
)

// 系统相关
type SysUserController struct {
	BaseCommonController
}

type SysUserAuthController struct {
	BaseAuthController
}

// @Title 修改密码
// @Description 修改密码
// @Param	request	body models.ModifyPwdReq true "type json string"
// @Success 200 {object} models.LoginResp
// @router /modify/pwd [post]
func (this *SysUserAuthController) ModifyPwd() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	var req system.ModifyPwdReq
	err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}
	sysUser := this.SysUser
	if sysUser == nil {
		br.Msg = "登录失败,账号或密码错误"
		br.ErrMsg = "登录失败，sysUser is nil "
		return
	}
	if req.OldPwd == "" {
		br.Msg = "原始密码不能为空"
		return
	}

	if req.NewPwd == "" {
		br.Msg = "新密码不能为空"
		return
	}

	if sysUser.Password != req.OldPwd {
		br.Msg = "旧密码错误，请重新输入"
		return
	}
	err = system.ModifyPwd(sysUser.AdminId, req.NewPwd)
	if err != nil {
		br.Msg = "修改失败"
		br.ErrMsg = "修改失败，Err:" + err.Error()
		return
	}
	br.Ret = 200
	br.Success = true
	br.Msg = "修改成功"
}

// CheckPwd
// @Title 校验是否初始密码
// @Description 校验是否初始密码
// @Success 200 {bool} true "是否初始密码；false代表不是初始密码，直接进去；true代表是初始密码，需要强制修改密码"
// @router /check_pwd [get]
func (this *SysUserAuthController) CheckPwd() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	sysUser := this.SysUser
	if sysUser == nil {
		br.Msg = "登录失败,账号或密码错误"
		br.ErrMsg = "登录失败，sysUser is nil "
		return
	}

	br.Data = false
	if sysUser.Password == utils.MD5("123456a") {
		br.Data = true
	}
	br.Ret = 200
	br.Success = true
	br.Msg = "校验成功"
}

// @Title 获取系统用户UUID
// @Description 获取系统用户UUID
// @Success 200 {object} system.SysuserListResp
// @router /uuid [get]
func (this *SysUserController) Uuid() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	nowTime := time.Now().Format(utils.FormatDateTimeUnSpace)
	randBigint := utils.GetRandDigit(4)
	randStr := utils.GetRandStringNoSpecialChar(4)
	uuid := nowTime + randBigint + randStr
	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
	br.Data = uuid
}

// GetCrmAuthCode
// @Title 免密登录-获取登录CRM系统的编码
// @Description 免密登录-获取登录CRM系统的编码
// @Success 200 Ret=200 获取成功
// @router /get_crm_auth_code [get]
func (this *SysUserAuthController) GetCrmAuthCode() {
	br := new(models.BaseResponse).Init()
	defer func() {
		if br.ErrMsg == "" {
			br.IsSendEmail = false
		}
		this.Data["json"] = br
		this.ServeJSON()
	}()
	sysUser := this.SysUser
	if sysUser == nil {
		br.Msg = "请登录"
		br.ErrMsg = "请登录，SysUser Is Empty"
		br.Ret = 408
		return
	}

	code, e := services.GetAuthCodeFromMiddleServer(sysUser.AdminName)
	if e != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "获取登录编码失败, Err: " + e.Error()
		return
	}

	br.Data = code
	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
}

// AuthCodeLogin
// @Title 编码登录
// @Description 编码登录
// @Success 200 Ret=200 操作成功
// @router /auth_code_login [get]
func (this *SysUserController) AuthCodeLogin() {
	br := new(models.BaseResponse).Init()
	defer func() {
		if br.ErrMsg == "" {
			br.IsSendEmail = false
		}
		this.Data["json"] = br
		this.ServeJSON()
	}()
	//authCode := this.GetString("AuthCode", "")
	//if authCode == "" {
	//	br.Msg = "参数有误"
	//	br.ErrMsg = "参数缺失, AuthCode"
	//	return
	//}

	req := make(map[string]interface{})
	query := this.Ctx.Request.URL.Query()
	for key, value := range query {
		req[key] = value[0]
	}
	fmt.Println(req)

	data, e, errMsg := services.ThirdLogin(req)
	if e != nil {
		br.Msg = "获取失败"
		if errMsg != `` {
			br.Msg = errMsg
		}
		br.ErrMsg = "获取登录Token失败, Err: " + e.Error()
		return
	}

	// 查询一下用户是否被禁用
	sysAdmin, e := system.GetSysUserById(data.AdminId)
	if e != nil && !utils.IsErrNoRow(e) {
		br.Msg = "获取失败"
		br.ErrMsg = fmt.Sprintf("获取用户信息失败, %v", e)
		return
	}
	if sysAdmin != nil && sysAdmin.Enabled != 1 {
		br.Ret = 408
		br.Msg = "您的账号已被禁用，如需登录，请联系管理员"
		j, _ := json.Marshal(data)
		br.ErrMsg = fmt.Sprintf("AuthCodeLogin, 账户信息异常：%s", j)
		return
	}

	br.Data = data
	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
}

// SystemConfig
// @Title 系统配置列表
// @Description 系统配置列表
// @Success 200 {object} []system.BusinessConf
// @router /public_config [get]
func (this *SysUserController) SystemConfig() {
	br := new(models.BaseResponse).Init()
	defer func() {
		if br.ErrMsg == "" {
			br.IsSendEmail = false
		}
		this.Data["json"] = br
		this.ServeJSON()
	}()
	list := make([]system.BusinessConf, 0)

	// 获取基础配置, 若未配置则直接返回
	conf, e := models.GetBusinessConf()
	if e != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "获取基础配置失败, Err: " + e.Error()
		return
	}

	list = append(list, system.BusinessConf{
		ConfKey: "LoginUrl",
		ConfVal: conf["LoginUrl"],
	}, system.BusinessConf{
		ConfKey: "LogoutUrl",
		ConfVal: conf["LogoutUrl"],
	}, system.BusinessConf{
		ConfKey: "UserLogoutUrl",
		ConfVal: conf["UserLogoutUrl"],
	})

	br.Data = list
	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
}

// Logout
// @Title 登出接口
// @Description 登出接口
// @Success 200 Ret=200 获取成功
// @router /logout [post]
func (this *SysUserAuthController) Logout() {
	br := new(models.BaseResponse).Init()
	defer func() {
		if br.ErrMsg == "" {
			br.IsSendEmail = false
		}
		this.Data["json"] = br
		this.ServeJSON()
	}()
	sysUser := this.SysUser
	if sysUser == nil {
		br.Msg = "请登录"
		br.ErrMsg = "请登录，SysUser Is Empty"
		br.Ret = 408
		return
	}

	services.ThirdLogout(this.Session.AccessToken)
	//e := services.ThirdLogout(this.Session.AccessToken)
	//if e != nil {
	//	br.Msg = "登出失败"
	//	br.ErrMsg = "登出失败, Err: " + e.Error()
	//	return
	//}

	br.Ret = 200
	br.Success = true
	br.Msg = "登出成功"
}