cold-logistics/app/admin/controller/customer.go

package controller

import (
	"cold-logistics/app/admin/model"
	"cold-logistics/app/admin/service"
	"cold-logistics/app/admin/service/dto"
	"cold-logistics/common/actions"
	"cold-logistics/common/middleware/handler"
	"cold-logistics/conf"
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/gin-gonic/gin/binding"
	"go.uber.org/zap"
	"gogs.baozhida.cn/zoie/OAuth-core/api"
	"gogs.baozhida.cn/zoie/OAuth-core/pkg"
	"gogs.baozhida.cn/zoie/OAuth-core/pkg/jwtauth/user"
	_ "gogs.baozhida.cn/zoie/OAuth-core/pkg/response"
	"gogs.baozhida.cn/zoie/OAuth-core/pkg/sms"
	"golang.org/x/crypto/bcrypt"
	"net/http"
)

// Customer 定义客户控制器
type Customer struct {
	api.Api
}

// GetPage 获取客户列表
// @Summary 获取客户列表
// @Description 获取客户列表
// @Tags 客户
// @Param username query string false "登录用户名"
// @Param name query string false "姓名"
// @Param pageSize query int false "页条数"
// @Param page query int false "页码"
// @Success 200 {object} response.Response{data=response.Page{list=[]model.SysUser}}  "{"code": 200, "data": [...]}"
// @Router /api/customer [get]
// @Security Bearer
func (e Customer) GetPage(c *gin.Context) {
	s := service.Customer{}
	req := dto.CustomerGetPageReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.Query).
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	//数据权限检查
	p := actions.GetPermissionFromContext(c)

	list := make([]model.SysUser, 0)
	var count int64

	err = s.GetPage(&req, p, &list, &count)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}

	e.PageOK(list, int(count), req.GetPageIndex(), req.GetPageSize(), "查询成功")
}

// Get 通过id获取客户
// @Summary 通过id获取客户
// @Description 通过id获取客户
// @Tags 客户
// @Param id path int true "客户id"
// @Success 200 {object} response.Response{data=model.SysUser}  "{"code": 200, "data": [...]}"
// @Router /api/customer/{id} [get]
// @Security Bearer
func (e Customer) Get(c *gin.Context) {
	s := service.Customer{}
	req := dto.CustomerGetReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, nil).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	//数据权限检查
	//p := actions.GetPermissionFromContext(c)

	var object model.SysUser
	err = s.Get(&req, nil, &object)
	if err != nil {
		e.Error(http.StatusUnprocessableEntity, err, err.Error())
		return
	}
	e.OK(object, "查询成功")
}

// Insert 创建客户
// @Summary 创建客户
// @Description 创建客户
// @Tags 客户
// @Accept  application/json
// @Product application/json
// @Param data body dto.CustomerInsertReq true "body"
// @Success 200 {object} response.Response "{"code": 200, "data": [...]}"
// @Success 200 {string} string	"{"code": 200, "message": "添加成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "添加失败"}"
// @Router /api/customer [post]
// @Security Bearer
func (e Customer) Insert(c *gin.Context) {
	s := service.Customer{}
	req := dto.CustomerInsertReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	//数据权限检查
	p := actions.GetPermissionFromContext(c)
	if p.DeptId == 0 {
		err = errors.New("无权添加，请联系管理员！")
		e.Error(500, err, err.Error())
		return
	}
	// 设置创建人
	req.SetCreateBy(user.GetUserId(c))
	req.SetDeptId(p.DeptId)
	err = s.Insert(&req)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}

	e.OK(req.GetId(), "添加成功")
}

// Update 修改客户数据
// @Summary 修改客户数据
// @Description 修改客户数据
// @Tags 客户
// @Accept  application/json
// @Product application/json
// @Param data body dto.CustomerUpdateReq true "body"
// @Success 200 {object} response.Response "{"code": 200, "data": [...]}"
// @Success 200 {string} string	"{"code": 200, "message": "修改成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "修改失败"}"
// @Router /api/customer [put]
// @Security Bearer
func (e Customer) Update(c *gin.Context) {
	s := service.Customer{}
	req := dto.CustomerUpdateReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	req.SetUpdateBy(user.GetUserId(c))

	//数据权限检查
	//p := actions.GetPermissionFromContext(c)

	err = s.Update(&req, nil)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	e.OK(req.GetId(), "修改成功")
}

// Delete 通过id删除客户数据
// @Summary 通过id删除客户数据
// @Description 通过id删除客户数据
// @Tags 客户
// @Param data body dto.CustomerDeleteReq true "body"
// @Success 200 {object} response.Response "{"code": 200, "data": [...]}"
// @Success 200 {string} string	"{"code": 200, "message": "删除成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "删除失败"}"
// @Router /api/customer [delete]
// @Security Bearer
func (e Customer) Delete(c *gin.Context) {
	s := service.Customer{}
	req := dto.CustomerDeleteReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	userId := user.GetUserId(c)
	if userId == req.Id {
		err = errors.New("禁止删除自己")
		e.Error(500, err, err.Error())
		return
	}

	// 设置编辑人
	req.SetUpdateBy(userId)
	//数据权限检查
	//p := actions.GetPermissionFromContext(c)

	err = s.Remove(&req, nil)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	e.OK(req.GetId(), "删除成功")
}

// ResetPwd 重置客户密码
// @Summary 重置客户密码
// @Description 重置客户密码
// @Tags 客户
// @Accept  application/json
// @Product application/json
// @Param data body dto.ResetCustomerPwdReq true "body"
// @Success 200 {object} response.Response "{"code": 200, "data": [...]}"
// @Success 200 {string} string	"{"code": 200, "message": "修改成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "修改失败"}"
// @Router /api/user/pwd/reset [put]
// @Security Bearer
func (e Customer) ResetPwd(c *gin.Context) {
	s := service.Customer{}
	req := dto.ResetCustomerPwdReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	req.SetUpdateBy(user.GetUserId(c))

	//数据权限检查
	//p := actions.GetPermissionFromContext(c)

	err = s.ResetPwd(&req, nil)
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	e.OK(req.GetId(), "修改成功")

}

// UpdatePwd 修改密码
// @Summary 修改密码
// @Description 修改密码
// @Tags 个人中心
// @Accept  application/json
// @Product application/json
// @Param data body dto.CustomerPassWord true "body"
// @Success 200 {object} response.Response "{"code": 200, "data": [...]}"
// @Success 200 {string} string	"{"code": 200, "message": "密码修改成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "修改失败"}"
// @Router /api/customer/pwd/set [put]
// @Security Bearer
func (e Customer) UpdatePwd(c *gin.Context) {
	s := service.Customer{}
	req := dto.CustomerPassWord{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	var hash []byte
	if hash, err = bcrypt.GenerateFromPassword([]byte(req.NewPassword), bcrypt.DefaultCost); err != nil {
		req.NewPassword = string(hash)
	}

	err = s.UpdatePwd(user.GetUserId(c), req.OldPassword, req.NewPassword)
	if err != nil {
		e.Error(http.StatusForbidden, err, err.Error())
		return
	}

	e.OK(nil, "密码修改成功")

}

// GetProfile 获取个人中心客户
// @Summary 获取个人中心客户
// @Description 获取个人中心客户
// @Tags 个人中心
// @Success 200 {object} response.Response{user=model.SysUser,role=model.SysRole}  "{"code": 200, "data": {"user":[...],"role":[...]}}"
// @Router /api/user/profile [get]
// @Security Bearer
func (e Customer) GetProfile(c *gin.Context) {
	s := service.Customer{}
	req := dto.CustomerById{}
	err := e.MakeContext(c).
		MakeOrm().
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	req.Id = user.GetUserId(c)

	sysUser := model.SysUser{}
	err = s.GetProfile(&req, &sysUser)
	if err != nil {
		e.Logger.Errorf("get user profile error, %s", err.Error())
		e.Error(500, err, "获取客户信息失败")
		return
	}
	e.OK(gin.H{
		"user": sysUser,
	}, "查询成功")
}

// GetInfo 获取个人信息
// @Summary 获取个人信息
// @Description 获取个人信息
// @Tags 个人中心
// @Success 200 {object} response.Response "{"code": 200, "data": [...]}"
// @Router /api/user/info [get]
// @Security Bearer
func (e Customer) GetInfo(c *gin.Context) {
	req := dto.CustomerGetReq{}
	s := service.Customer{}
	r := service.SysRole{}
	err := e.MakeContext(c).
		MakeOrm().
		MakeService(&r.Service).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	var roles = make([]string, 1)
	roles[0] = user.GetRoleName(c)
	var permissions = make([]string, 1)
	permissions[0] = "*:*:*"
	var buttons = make([]string, 1)
	buttons[0] = "*:*:*"

	var mp = make(map[string]interface{})
	mp["roles"] = roles
	list, _ := r.GetById(user.GetRoleId(c))
	mp["permissions"] = list
	mp["buttons"] = list

	sysUser := model.SysUser{}
	req.Id = user.GetUserId(c)

	err = s.Get(&req, nil, &sysUser)
	if err != nil {
		e.Logger.Errorf("get user info error, %s", err.Error())
		e.Error(http.StatusUnauthorized, err, err.Error())
		return
	}

	mp["userName"] = sysUser.Username
	mp["userId"] = sysUser.Id
	mp["deptId"] = sysUser.DeptId
	mp["name"] = sysUser.NickName
	mp["code"] = 200
	e.OK(mp, "查询成功")
}

// VerifyCode 获取短信验证码
// @Summary 获取短信验证码
// @Description 获取短信验证码
// @Tags 登录
// @Accept  application/json
// @Product application/json
// @Success 200 {string} string "{"code": 200, "data": "18888888888"}"
// @Router /verify-code [get]
// @Security Bearer
func (e Customer) VerifyCode(c *gin.Context) {
	s := service.Customer{}
	req := dto.CustomerGetSMSVerifyCodeReq{}
	err := e.MakeContext(c).
		MakeService(&s.Service).
		Bind(&req, binding.Query).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	key := handler.GetVerifyCodeCacheKey(req.Phone)
	_, err = e.Cache.Get(key)
	// 验证吗缓存已存在
	if err == nil {
		e.Error(500, err, "验证吗已发送，请注意查收")
		return
	}

	code := pkg.GenerateRandomFigureKey6()

	ss := sms.NewSMS(conf.ExtConfig.SubMail.Appid, conf.ExtConfig.SubMail.Signature)
	content := fmt.Sprintf("【冷链运输平台】您的短信验证码：%s，请在10分钟内输入", code)
	res, err := ss.Send(req.Phone, content)
	if err != nil || res.Status != sms.SUCCESS {
		e.Logger.Error("发送短信验证码出现异常", zap.Any("res", res), zap.Error(err))
		e.Error(500, err, "验证吗发送失败，请重试")
		return
	}
	_ = e.Cache.Set(key, code, 600)
	e.OK(req.Phone, "发送成功")

}