package client import ( "fmt" mpoauth2 "github.com/chanxuehong/wechat/mp/oauth2" "hudongzhuanjia/controllers" "hudongzhuanjia/libs/im" "hudongzhuanjia/libs/jwt" "hudongzhuanjia/libs/wechat" "hudongzhuanjia/models" activity_service "hudongzhuanjia/services/activity" "hudongzhuanjia/utils/code" "hudongzhuanjia/utils/define" "time" "github.com/ouxuanserver/osmanthuswine/src/core" ) //用户 type UserCtl struct { controllers.BaseCtl } // 录入登录 func (t *UserCtl) EntryLogin() { account := t.MustGet("account") password := t.MustGet("password") activityId := t.MustGetInt64("activity_id") // 需要获取 entryPeople := new(models.OrderEntryPerson) exist, err := core.GetXormAuto().Where("is_delete=0 and account = ? and password = ? and activity_id = ?", account, password, activityId).Get(entryPeople) t.CheckErr(err) t.Assert(exist, code.MSG_ENTRYPEOPLE_NOT_EXIST, "录入人员不存在") area := new(models.AreaStore) exist, err = models.GetById(area, entryPeople.AreaId) t.CheckErr(err) t.Assert(exist, code.MSG_AREASTORE_NOT_EXIST, "地区不存在") activity := new(models.Activity) exist, err = models.GetById(activity, activityId) t.CheckErr(err) t.Assert(exist, code.MSG_ACTIVITY_NOT_EXIST, "互动不存在") customer := new(models.Customer) exist, err = models.GetById(customer, entryPeople.Pid) t.CheckErr(err) t.Assert(exist, code.MSG_CUSTOMER_NOT_EXIST, "客户不存在") // 怎讲activity token, err := jwt.GenJwtToken("entry", entryPeople.Id, customer.Id, customer.Pid, entryPeople.AreaId, entryPeople.ActivityId) t.CheckErr(err) t.SetSession(define.TOKEN, token) entryPeople.Token = token entryPeople.AreaName = area.Name entryPeople.ActivityName = activity.Name t.JSON(map[string]interface{}{ "people": entryPeople, }) } func (t *UserCtl) WxLogin() { activityId := t.MustGetInt64("activity_id") wxcode := t.MustGet("code") _type, _ := t.Get("type") // mini 小程序登陆 base 静默授权 userinfo 或者为空 非静默授权 activity, exist, err := activity_service.GetActivityById(activityId) t.CheckErr(err) customerId := t.DefaultInt64("customer_id", activity.CustomerId) customer := new(models.Customer) exist, err = models.GetById(customer, customerId) t.CheckErr(err) t.Assert(exist, code.MSG_CUSTOMER_NOT_EXIST, "客户不存在") area := new(models.AreaStore) if customer.Pid == 0 { exist, err = area.GetMainAreaById(activityId) } else { exist, err = area.GetAreaStoreById(customer.AreaId) } t.CheckErr(err) t.Assert(exist, code.MSG_AREASTORE_NOT_EXIST, "地区不存在") user := new(models.User) user.ActivityId = activityId user.AreaId = area.Id user.AreaName = area.Name if _type == "mini" { session, err := wechat.MiniAuth(wxcode) t.CheckErr(err) user.Openid = session.OpenId exist, err = user.GetUserByOpenid(user.Openid) t.CheckErr(err) if exist { _, err = core.GetXormAuto().Insert(user) t.CheckErr(err) } } else if _type == "base" { token, err := wechat.GetToken(wxcode) t.CheckErr(err) user.Openid = token.OpenId exist, err = user.GetUserByOpenid(user.Openid) t.CheckErr(err) if exist { _, err = core.GetXormAuto().Insert(user) t.CheckErr(err) } } else if _type == "" { token, err := wechat.GetToken(wxcode) t.CheckErr(err) if token.AccessToken == "" || token.OpenId == "" { t.ERROR("code无效", code.MSG_ERR) } info, err := wechat.GetUserInfo(token) t.CheckErr(err) exist, err = user.GetUserByOpenid(info.OpenId) t.CheckErr(err) user.Nickname = info.Nickname user.Openid = info.OpenId user.Gender = func() string { if info.Sex == 1 { return "男" } return "女" }() user.Avatar = info.HeadImageURL user.Unionid = info.UnionId user.City = info.City user.Province = info.Province user.Country = info.Country user.CreatedAt = time.Now() user.UpdatedAt = time.Now() if !exist { _, err = core.GetXormAuto().Insert(user) } else { _, err = user.SaveUserInfo(user.Id) } t.CheckErr(err) } history := new(models.SignHistory) signExist, err := history.GetByUserId(activityId, user.Id, activity.RehearsalId, area.Id) t.CheckErr(err) var signIn = "未签到" if signExist { // 存在数据 signIn = "已签到" } sign := new(models.SignUp) exist, err = sign.GetByActivityId(activityId) t.CheckErr(err) t.Assert(exist, code.MSG_SIGN_UP_NOT_EXIST, "签到活动不存在") realSignExist, err := new(models.RealSignHistory).Check(user.Id, activity.Id, activity.RehearsalId) t.CheckErr(err) jwtToken, err := jwt.GenJwtToken(define.TYPE_H5USER, user.Id, customer.Id, customer.Pid, area.Id, activityId) t.CheckErr(err) t.SetSession(define.TOKEN, jwtToken) t.JSON(map[string]interface{}{ "user": user, "token": jwtToken, "activity": activity, "sign_in": signIn, "sign": signExist, "real_sign": realSignExist, "real_sign_form": sign.RealSignJsonForm, "tag": "activity", }) } func (t *UserCtl) Login() { wxcode := t.MustGet("code") _type, _ := t.Get("type") user := new(models.User) if _type == "mini" { session, err := wechat.MiniAuth(wxcode) t.CheckErr(err) user.Openid = session.OpenId exist, err := user.GetUserByOpenid(user.Openid) t.CheckErr(err) if exist { _, err = core.GetXormAuto().Insert(user) t.CheckErr(err) } } else if _type == "base" { token, err := wechat.GetToken(wxcode) t.CheckErr(err) user.Openid = token.OpenId exist, err := user.GetUserByOpenid(user.Openid) t.CheckErr(err) if exist { _, err = core.GetXormAuto().Insert(user) t.CheckErr(err) } } else if _type == "" || _type == "userinfo" { //token, err := wx.GetToken(wxcode) token, err := wechat.GetToken(wxcode) t.CheckErr(err) if token.AccessToken == "" || token.OpenId == "" { t.ERROR("code无效", code.MSG_ERR) } info, err := wechat.GetUserInfo(token) t.CheckErr(err) exist, err := user.GetUserByOpenid(info.OpenId) t.CheckErr(err) user.Nickname = info.Nickname user.Openid = info.OpenId user.Gender = func() string { if info.Sex == 1 { return "男" } return "女" }() user.Avatar = info.HeadImageURL user.Unionid = info.UnionId user.City = info.City user.Province = info.Province user.Country = info.Country user.CreatedAt = time.Now() user.UpdatedAt = time.Now() if !exist { _, err = core.GetXormAuto().Insert(user) } else { _, err = user.SaveUserInfo(user.Id) } t.CheckErr(err) } jwtToken, err := jwt.GenJwtToken(define.TYPE_LIVEUSER, user.Id, 0, 0, 0, 0) t.CheckErr(err) t.SetSession(define.TOKEN, jwtToken) sign, err := im.AccountImport(fmt.Sprintf("%d", user.Id), user.Nickname, user.Avatar) t.CheckErr(err) user.Sig = sign t.JSON(map[string]interface{}{ "user": user, "token": jwtToken, }) } // 模拟wx login func (t *UserCtl) DebugLogin() { uid := t.DefaultInt64("user_id", 1) customerId := t.DefaultInt64("customer_id", 1) activityId := t.DefaultInt64("activity_id", 5) user := new(models.User) exist, err := models.GetById(user, uid) t.CheckErr(err) t.Assert(exist, code.MSG_USER_NOT_EXIST, "用户不存在") customer := new(models.Customer) exist, err = models.GetById(customer, customerId) t.CheckErr(err) t.Assert(exist, code.MSG_CUSTOMER_NOT_EXIST, "客户不存在") area := new(models.AreaStore) if customer.Pid == 0 { exist, err = area.GetMainAreaById(activityId) } else { exist, err = area.GetAreaStoreById(customer.AreaId) } t.CheckErr(err) t.Assert(exist, code.MSG_AREASTORE_NOT_EXIST, "地区不存在") jwtToken, err := jwt.GenJwtToken("user", user.Id, customer.Id, customer.Pid, area.Id, activityId) t.CheckErr(err) t.SetSession(define.TOKEN, jwtToken) t.JSON(map[string]interface{}{ "user": user, "token": jwtToken, }) } type UserAuthCtl struct { controllers.AuthorCtl } func (t *UserAuthCtl) SaveUserInfo() { uid := t.MustGetUID() info := mpoauth2.SessionInfo{} t.RequestToStruct(&info) if info.Nickname == "" && info.AvatarUrl == "" { t.ERROR("参数错误", code.MSG_ERR_Param) return } user := models.User{} user.Nickname = info.Nickname user.Gender = func() string { if info.Gender == 1 { return "男" } return "女" }() user.Avatar = info.AvatarUrl user.Unionid = info.UnionId user.City = info.City user.Province = info.Province user.Country = info.Country _, err := user.SaveUserInfo(uid) t.CheckErr(err) // //exist, err := models.GetById(&user, uid) //t.CheckErr(err) //t.Assert(exist, code.MSG_USER_NOT_EXIST, "用户不存在") user.Id = uid sign, err := im.AccountImport(fmt.Sprintf("%d", user.Id), user.Avatar, user.Nickname) t.CheckErr(err) user.Sig = sign t.JSON(user) } //退出 func (t *UserAuthCtl) Logout() { if _, ok := t.Request.SESSION[define.TOKEN]; !ok { t.ERROR("未登录", code.MSG_ERR_Authority) } t.DeleteSession(define.TOKEN) t.SUCCESS("退出成功") }