pm/controllers/work/worker/user.go

package worker

import (
	"github.com/gin-gonic/gin"
	"zhiyuan/pkg/app"
	"zhiyuan/pkg/config"
	"zhiyuan/services/form"
	"zhiyuan/services/work/worker"
)

func WorkerRegister(c *gin.Context) {
	var addForm form.WorkerAdd
	if app.Bind(c, &addForm) != nil {
		return
	}
	id, err := worker.Register(addForm)
	if err != nil {
		app.Error(c, err.Error())
		return
	}
	app.Success(c, gin.H{"id": id})
}

func Login(c *gin.Context) {
	var form form.WorkLogin
	if app.Bind(c, &form) != nil {
		return
	}

	token, err := worker.Login(form)
	if err != nil {
		app.Error(c, err.Error())
		return
	}
	app.Success(c, map[string]string{
		"token": token,
	})
}

func Logout(c *gin.Context) {
	err := worker.Logout(c.GetInt("workerID"))
	if err != nil {
		app.ErrorMsg(c, err.Error(), nil)
		return
	}
	app.Success(c, nil)
}

func Info(c *gin.Context) {
	type SettingInfo struct {
		ImgHost string `json:"img_host"`
	}
	type UserInfo struct {
		ID         int         `json:"id"`
		Name       string      `json:"name"`
		TypeIds    string      `json:"type_ids"`
		NickName   string      `json:"nickname"`
		HeadImgUrl string      `json:"headimgurl"`
		Setting    SettingInfo `json:"setting"`
	}
	var userInfo UserInfo
	_, err := worker.GetInfoByID(c.GetInt("workerID"), nil, &userInfo)
	if err != nil {
		app.ErrorMsg(c, err.Error(), nil)
		return
	}
	userInfo.Setting = SettingInfo{
		ImgHost: config.Cfg.App.ImgHost,
	}
	app.Success(c, userInfo)
}

func UnBindWeixin(c *gin.Context) {
	if err := worker.UnBindWeixin(c.GetInt("workerID")); err != nil {
		app.ErrorMsg(c, err.Error(), nil)
		return
	}
	app.Success(c, nil)
}

func Password(c *gin.Context) {
	var form form.Password
	if app.Bind(c, &form) != nil {
		return
	}
	if err := worker.Password(c.GetInt("workerID"), form); err != nil {
		app.ErrorMsg(c, err.Error(), nil)
		return
	}
	app.Success(c, nil)
}