eta_mini_crm/controllers/user.go

package controllers

import (
	"encoding/json"
	"eta/eta_mini_crm/models"
	"eta/eta_mini_crm/models/request"
	"eta/eta_mini_crm/models/response"
	"eta/eta_mini_crm/utils"
	"strings"
	"time"

	"github.com/beego/beego/v2/client/orm"
	"github.com/rdlucklib/rdluck_tools/paging"
)

type UserController struct {
	BaseAuthController
}

// Add
// @Title 添加新客户
// @Description 添加新客户
// @Param   request	body request.UserAddReq true "type json string"
// @Success 200 {object} request.UserAddReq
// @router /add [post]
func (this *UserController) Add() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	var req request.UserAddReq
	err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}
	if req.RealName == "" {
		br.Msg = "请输入姓名"
		return
	}
	req.Phone = strings.TrimSpace(req.Phone)
	req.Email = strings.TrimSpace(req.Email)
	if req.Phone == "" && req.Email == "" {
		br.Msg = "至少输入一个手机号或邮箱"
		return
	}

	var userByPhone, userByEmail *models.User
	if req.Phone != "" {
		if req.AreaCode == "86" {
			if !utils.ValidateMobileFormatat(req.Phone) {
				br.Msg = "手机号格式有误, 请检查"
				return
			}
		}
		userByPhone, err = models.GetUserByPhone(req.Phone, req.AreaCode)
		if err != nil && err != orm.ErrNoRows {
			br.Msg = "添加用户失败"
			br.ErrMsg = "查询用户失败，系统错误，Err:" + err.Error()
			return
		}
		if userByPhone != nil {
			switch userByPhone.Status {
			case 0:
				userByPhone.Status = 2
			case 1:
				userByPhone.Status = 2
			default:
				br.Msg = "手机号已存在，请重新输入"
				return
			}
		}
	}
	if req.Email != "" {
		if !utils.ValidateEmailFormatat(req.Email) {
			br.Msg = "邮箱格式有误, 请检查"
			return
		}
		userByEmail, err = models.GetUserByEmail(req.Email)
		if err != nil && err != orm.ErrNoRows {
			br.Msg = "添加用户失败"
			br.ErrMsg = "查询用户失败，系统错误，Err:" + err.Error()
			return
		}
		if userByEmail != nil {
			switch userByEmail.Status {
			case 0:
				userByEmail.Status = 2
			case 1:
				userByEmail.Status = 2
			default:
				br.Msg = "邮箱已存在，请重新输入"
				return
			}
		}
	}
	if req.SellerId <= 0 {
		br.Msg = "请选择营业部/销售"
		return
	}
	if req.ValidStartTime == "" || req.ValidEndTime == "" {
		br.Msg = "请选择合理的有效期范围"
		return
	}

	validStartTime, err := time.Parse("2006-01-02", req.ValidStartTime)
	if err != nil {
		br.Msg = "错误的日期格式"
		return
	}
	validEndTime, err := time.Parse("2006-01-02", req.ValidEndTime)
	if err != nil {
		br.Msg = "错误的日期格式"
		return
	}
	if !validStartTime.Before(validEndTime) {
		br.Msg = "请选择合理的有效期范围"
		return
	}
	if req.Company == "" {
		br.Msg = "请输入所属公司"
		return
	}
	if userByPhone != nil && userByEmail != nil && userByPhone.UserId != userByEmail.UserId {
		br.Msg = "邮箱已存在，请重新输入"
		return
	}
	user := &models.User{}
	if userByPhone == nil && userByEmail == nil {
		user.CreateTime = time.Now()
		user.ModifyTime = time.Now()
	}

	validStartTime = validStartTime.Local().Add(-time.Hour * 8)
	validEndTime = validEndTime.Local().Add(-time.Hour*8 + time.Hour*24)
	curTime := time.Now()
	if curTime.Before(validEndTime) && curTime.After(validStartTime) {
		user.Status = 2
	} else {
		user.Status = 0
	}
	user.RealName = req.RealName
	user.AreaCode = req.AreaCode
	user.Phone = req.Phone
	user.Email = req.Email
	user.SellerId = req.SellerId
	user.ValidStartTime = validStartTime
	user.ValidEndTime = validEndTime
	user.Company = req.Company
	user.ModifyTime = time.Now()
	err = models.SaveUser(user, req.ChartPermission)
	if err != nil {
		br.Msg = "添加客户失败"
		br.ErrMsg = "添加客户失败，系统错误，Err:" + err.Error()
		return
	}

	br.Msg = "添加成功"
	br.Success = true
	br.Ret = 200

}

// edit
// @Title 编辑客户
// @Description 编辑客户
// @Param   request	body request.UserEidtReq true "type json string"
// @Success 200 {object} request.UserAddReq
// @router /edit [post]
func (this *UserController) Edit() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	var req request.UserEidtReq
	err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}
	if req.RealName == "" {
		br.Msg = "请输入姓名"
		return
	}
	req.Phone = strings.TrimSpace(req.Phone)
	req.Email = strings.TrimSpace(req.Email)
	if req.Phone == "" && req.Email == "" {
		br.Msg = "至少输入一个手机号或邮箱"
		return
	}
	if req.Phone != "" {
		if req.AreaCode == "86" {
			if !utils.ValidateMobileFormatat(req.Phone) {
				br.Msg = "手机号格式有误, 请检查"
				return
			}
		}
		user, err := models.GetUserByPhone(req.Phone, req.AreaCode)
		if err != nil && err != orm.ErrNoRows {
			br.Msg = "编辑用户失败"
			br.ErrMsg = "查询用户失败，系统错误，Err:" + err.Error()
			return
		}
		if user != nil && user.UserId != req.UserId {
			br.Msg = "手机号已存在，请重新输入"
			return
		}
	}
	if req.Email != "" {
		if !utils.ValidateEmailFormatat(req.Email) {
			br.Msg = "邮箱格式有误, 请检查"
			return
		}

		user, err := models.GetUserByEmail(req.Email)
		if err != nil && err != orm.ErrNoRows {
			br.Msg = "编辑用户失败"
			br.ErrMsg = "查询用户失败，系统错误，Err:" + err.Error()
			return
		}
		if user != nil && user.UserId != req.UserId {
			br.Msg = "邮箱已存在，请重新输入"
			return
		}
	}
	if req.SellerId <= 0 {
		br.Msg = "请选择营业部/销售"
		return
	}
	if req.ValidStartTime == "" || req.ValidEndTime == "" {
		br.Msg = "请选择合理的有效期范围"
		return
	}

	validStartTime, err := time.Parse("2006-01-02", req.ValidStartTime)
	if err != nil {
		br.Msg = "错误的日期格式"
		return
	}
	validStartTime = validStartTime.In(time.Local).Add(-time.Hour * 8)
	validEndTime, err := time.Parse("2006-01-02", req.ValidEndTime)
	if err != nil {
		br.Msg = "错误的日期格式"
		return
	}
	validEndTime = validEndTime.In(time.Local).Add(-time.Hour*8 + time.Hour*24)
	if !validStartTime.Before(validEndTime) {
		br.Msg = "请选择合理的有效期范围"
		return
	}
	if req.Company == "" {
		br.Msg = "请输入所属公司"
		return
	}

	user := &models.User{}
	user.UserId = req.UserId
	user.RealName = req.RealName
	user.AreaCode = req.AreaCode
	user.Phone = req.Phone
	user.Email = req.Email
	user.SellerId = req.SellerId
	user.ValidStartTime = validStartTime
	user.ValidEndTime = validEndTime
	user.Company = req.Company
	user.ModifyTime = time.Now()
	curTime := time.Now()
	if !req.IsEnabled {
		user.Status = 0
	} else if curTime.After(validStartTime) && curTime.Before(validEndTime) {
		user.Status = 2
	} else {
		user.Status = 1
	}
	err = models.SaveUser(user, req.ChartPermission)
	if err != nil {
		br.Msg = "添加客户失败"
		br.ErrMsg = "添加客户失败，系统错误，Err:" + err.Error()
		return
	}

	br.Msg = "编辑成功"
	br.Success = true
	br.Ret = 200
}

// Check
// @Title 编辑客户
// @Description 编辑客户
// @Param   request	body request.UserEidtReq true "type json string"
// @Success 200 {object} request.UserAddReq
// @router /check [post]
func (this *UserController) Check() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	var req request.UserCheckReq
	err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}

	req.Phone = strings.TrimSpace(req.Phone)
	req.Email = strings.TrimSpace(req.Email)
	if req.Phone == "" && req.Email == "" {
		br.Msg = "至少输入一个手机号或邮箱"
		return
	}
	userCheckResp := new(response.UserCheckResp)

	if req.Phone != "" {
		if req.AreaCode == "86" {
			if !utils.ValidateMobileFormatat(req.Phone) {
				br.Msg = "手机号格式有误, 请检查"
				return
			}
		}
		user, err := models.GetUserByPhone(req.Phone, req.AreaCode)
		if err != nil && err != orm.ErrNoRows {
			br.Msg = "编辑用户失败"
			br.ErrMsg = "查询用户失败，系统错误，Err:" + err.Error()
			return
		}
		if user != nil {
			userCheckResp.UserId = user.UserId
			if user.Status == 0 {
				br.Ret = 200
				br.Success = true
				br.Msg = "该用户已被禁用，确认启用并更新用户信息吗？"
				br.Data = userCheckResp
				return
			}
			if user.Status == 1 {
				br.Ret = 200
				br.Success = true
				br.Msg = "该用户已被禁用，确认启用并更新用户信息吗？"
				br.Data = userCheckResp
				return
			}
			if user.Status == 2 {
				br.Msg = "手机号已存在，请重新输入"
				return
			}
		}

	}
	if req.Email != "" {
		if !utils.ValidateEmailFormatat(req.Email) {
			br.Msg = "邮箱格式有误, 请检查"
			return
		}

		user, err := models.GetUserByEmail(req.Email)
		if err != nil && err != orm.ErrNoRows {
			br.Msg = "编辑用户失败"
			br.ErrMsg = "查询用户失败，系统错误，Err:" + err.Error()
			return
		}
		if user != nil {
			userCheckResp.UserId = user.UserId
			if user.Status == 0 {
				br.Ret = 200
				br.Success = true
				br.Msg = "该用户已被禁用，确认启用并更新用户信息吗？"
				br.Data = userCheckResp
				return
			}
			if user.Status == 1 {
				br.Ret = 200
				br.Success = true
				br.Msg = "该用户已被禁用，确认启用并更新用户信息吗？"
				br.Data = userCheckResp
				return
			}
			if user.Status == 2 {
				br.Msg = "邮箱已存在，请重新输入"
				return
			}
			return
		}
	}
	br.Ret = 200
	br.Success = true
	br.Msg = "邮箱或手机号合格"
}

// List
// @Title 用户列表
// @Description 用户列表
// @Param   PageSize   query   int  true       "每页数据条数"
// @Param   CurrentIndex   query   int  true       "当前页页码，从1开始"
// @Param   SellerId   query   int  true       "销售id"
// @Param   Status   query   int  true       "用户状态"
// @Param   KeyWord   query   string  true       "手机号/邮箱/姓名"
// @Param   IsRegistered   query   string  true       "是否注册"
// @Param   IsSubscribed   query   string  true       "是否关注"
// @Param   RegisterStartDate   query   string  true       "注册开始时间"
// @Param   RegisterEndDate   query   string  true       "注册结束时间"
// @Param   CreateStartDate   query   string  true       "创建开始时间"
// @Param   CreateEndDate   query   string  true       "创建结束时间"
// @Success 200 {object} models.LoginResp
// @router /list [get]
func (this *UserController) List() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	pageSize, _ := this.GetInt("PageSize")
	currentIndex, _ := this.GetInt("CurrentIndex")
	sellerId, _ := this.GetInt("SellerId")
	status := this.GetString("Status")
	keyWord := this.GetString("KeyWord")
	IsRegistered := this.GetString("IsRegisterd")
	IsSubscribed := this.GetString("IsSubscribed")
	registerStartDate := this.GetString("RegisterStartDate")
	registerEndDate := this.GetString("RegisterEndDate")
	createStartDate := this.GetString("CreateStartDate")
	createEndDate := this.GetString("CreateEndDate")

	var condition string
	var pars []interface{}

	if keyWord != "" {
		condition += ` AND (real_name LIKE ? OR phone LIKE ? OR email LIKE ?) `
		pars = utils.GetLikeKeywordPars(pars, keyWord, 3)
	}

	if pageSize <= 0 {
		pageSize = utils.PageSize20
	} else if pageSize > utils.PageSize100 {
		pageSize = utils.PageSize100
	}
	if currentIndex <= 0 {
		currentIndex = 1
	}

	if sellerId > 0 {
		condition += " AND seller_id=? "
		pars = append(pars, sellerId)
	}
	switch status {
	case "禁用":
		condition += " AND status=? "
		pars = append(pars, 0)
	case "潜在":
		condition += " AND status=? "
		pars = append(pars, 1)
	case "正式":
		condition += " AND status=? "
		pars = append(pars, 2)
	}
	switch IsRegistered {
	case "是":
		condition += " AND is_registered=? "
		pars = append(pars, true)
	case "否":
		condition += " AND is_registered=? "
		pars = append(pars, false)
	}
	switch IsSubscribed {
	case "是":
		condition += " AND is_subscribed=? "
		pars = append(pars, true)
	case "否":
		condition += " AND is_subscribed=? "
		pars = append(pars, false)
	}
	if registerStartDate != "" {
		registerStartTime, er := time.Parse("2006-01-02 15:04:05", registerStartDate)
		if er != nil {
			br.Msg = "日期格式有误"
			return
		}
		condition += " AND register_time>? "
		pars = append(pars, registerStartTime)
	}
	if registerEndDate != "" {
		registerEndTime, er := time.Parse("2006-01-02 15:04:05", registerEndDate)
		if er != nil {
			br.Msg = "日期格式有误"
			return
		}
		condition += " AND register_time<? "
		pars = append(pars, registerEndTime)
	}
	if createStartDate != "" {
		createStartTime, er := time.Parse("2006-01-02 15:04:05", createStartDate)
		if er != nil {
			br.Msg = "日期格式有误"
			return
		}
		condition += " AND create_time>? "
		pars = append(pars, createStartTime)
	}
	if createEndDate != "" {
		createEndTime, er := time.Parse("2006-01-02 15:04:05", createEndDate)
		if er != nil {
			br.Msg = "日期格式有误"
			return
		}
		condition += " AND create_time<? "
		pars = append(pars, createEndTime)
	}
	if pageSize <= 0 {
		pageSize = utils.PageSize20
	} else if pageSize > utils.PageSize100 {
		pageSize = utils.PageSize100
	}
	if currentIndex <= 0 {
		currentIndex = 1
	}
	startSize := utils.StartIndex(currentIndex, pageSize)

	total, err := models.GetUserCount(condition, pars)
	if err != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "获取失败，Err:" + err.Error()
		return
	}
	userList, err := models.GetUserList(condition, pars, startSize, pageSize)
	if err != nil {
		br.Msg = "查询用户失败"
		br.Msg = "查询用户失败，系统错误，Err:" + err.Error()
		return
	}
	page := paging.GetPaging(currentIndex, pageSize, total)
	resp := new(response.UserListResp)
	resp.Paging = page
	resp.List = userList

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

// Detail
// @Title 系统用户详情信息
// @Description 用户详情信息
// @Param   UserId   query   int  true       "系统用户id"
// @Success 200 {object} models.LoginResp
// @router /detail [get]
func (this *UserController) Detail() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	UserId, err := this.GetInt("UserId")
	if err != nil {
		br.Msg = "参数解析错误"
		return
	}
	if UserId <= 0 {
		br.Msg = "查询用户不存在"
		return
	}
	user, err := models.GetUserById(UserId)
	if err != nil {
		if err == orm.ErrNoRows {
			br.Msg = "用户不存在或已删除，请刷新页面"
			return
		}
		br.Msg = "查询用户失败"
		br.ErrMsg = "查询用户失败，系统错误，Err:" + err.Error()
		return
	}
	if user == nil {
		br.Msg = "用户不存在或已删除，请刷新页面"
		return
	}
	br.Msg = "查询成功"
	br.Ret = 200
	br.Success = true
	br.Data = user
}

// Delete
// @Title 系统用户详情信息
// @Description 用户详情信息
// @Param   request	body request.UserDeleteReq true "type json string"
// @Success 200 {object} models.LoginResp
// @router /delete [post]
func (this *UserController) Delete() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	var req request.UserDeleteReq
	err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}

	err = models.DeleteUserById(req.UserId)
	if err != nil {
		br.Msg = "删除失败"
		br.ErrMsg = "删除失败，系统错误，Err:" + err.Error()
		return
	}
	br.Msg = "删除成功"
	br.Success = true
	br.Ret = 200
}