package hdl import ( "applet/app/db" "applet/app/e" "applet/app/es/hdl" md2 "applet/app/es/md" "applet/app/lib/auth" "applet/app/md" "applet/app/svc" "applet/app/utils" "applet/app/utils/cache" "applet/app/utils/qrcode" "code.fnuoos.com/EggPlanet/egg_models.git/src/implement" "code.fnuoos.com/EggPlanet/egg_models.git/src/model" "code.fnuoos.com/EggPlanet/egg_system_rules.git/aliyun" "code.fnuoos.com/EggPlanet/egg_system_rules.git/baidu" md3 "code.fnuoos.com/EggPlanet/egg_system_rules.git/md" md4 "code.fnuoos.com/EggPlanet/egg_system_rules.git/rule/egg_energy/md" es2 "code.fnuoos.com/EggPlanet/egg_system_rules.git/utils/es" "code.fnuoos.com/go_rely_warehouse/zyos_go_es.git/es" "code.fnuoos.com/go_rely_warehouse/zyos_go_mq.git/rabbit" "context" "encoding/json" "fmt" "github.com/gin-gonic/gin" "github.com/olivere/elastic/v7" "github.com/syyongx/php2go" "github.com/tidwall/gjson" "strings" "time" ) // UserInfo // @Summary 用户信息 // @Tags 用户信息 // @Description 用户信息 // @Accept json // @Produce json // @param Authorization header string true "验证参数Bearer和token空格拼接" // @Success 200 {object} md.UserInfoResp "具体数据" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/userInfo [get] func UserInfo(c *gin.Context) { user := svc.GetUser(c) res := md.UserInfoResp{ Id: utils.Int64ToStr(user.Id), Phone: user.Phone, Nickname: user.Phone, InviteCode: user.SystemInviteCode, IsBindExtend: "0", } if user.Avatar == "" { user.Avatar = svc.GetSysCfgStr("default_avatar") } if user.Avatar != "" { res.HeadImg = svc.GetOssUrl(user.Avatar) } if user.CustomInviteCode != "" { res.InviteCode = user.CustomInviteCode } if user.ParentUid > 0 { res.IsBindExtend = "1" } e.OutSuc(c, res, nil) return } // InviteCodeUserInfo // @Summary 邀请码获取用户信息 // @Tags 登录注册 // @Description 邀请码获取用户信息 // @Accept json // @Produce json // @Param req body md.InviteCodeUserInfoReq true "注册参数" // @Success 200 {object} md.InviteCodeUserInfoResp "具体数据" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/inviteCode/userInfo [post] func InviteCodeUserInfo(c *gin.Context) { var req md.InviteCodeUserInfoReq err := c.ShouldBindJSON(&req) if err != nil { err = svc.HandleValidateErr(err) err1 := err.(e.E) e.OutErr(c, err1.Code, err1.Error()) return } if req.InviteCode == "" { e.OutErr(c, 400, e.NewErr(400, "邀请码不能为空")) return } userDb := implement.NewUserDb(db.Db) user, err := userDb.UserGetOneByParams(map[string]interface{}{ "key": "system_invite_code", "value": req.InviteCode, }) if user == nil { user, _ = userDb.UserGetOneByParams(map[string]interface{}{ "key": "custom_invite_code", "value": req.InviteCode, }) if user == nil { e.OutErr(c, 400, e.NewErr(400, "用户不存在")) return } } nickname := user.Nickname if nickname == "" { nickname = user.Phone } if php2go.IsNumeric(nickname) { nickname = utils.HideTrueName(user.Nickname) } user = svc.UserImg(user) res := md.InviteCodeUserInfoResp{ Nickname: nickname, HeadImg: user.Avatar, } e.OutSuc(c, res, nil) return } // UserBindParent // @Summary 绑定上级-要登陆的 // @Tags 登录注册 // @Description 绑定上级 // @Accept json // @Produce json // @Param req body md.InviteCodeUserInfoReq true "注册参数" // @Success 200 {string} "具体数据" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/memberCenter/bindParent [post] func UserBindParent(c *gin.Context) { var req md.InviteCodeUserInfoReq err := c.ShouldBindJSON(&req) if err != nil { err = svc.HandleValidateErr(err) err1 := err.(e.E) e.OutErr(c, err1.Code, err1.Error()) return } if req.InviteCode == "" { e.OutErr(c, 400, e.NewErr(400, "邀请码不能为空")) return } userDb := implement.NewUserDb(db.Db) user, err := userDb.UserGetOneByParams(map[string]interface{}{ "key": "system_invite_code", "value": req.InviteCode, }) if user == nil { user, _ = userDb.UserGetOneByParams(map[string]interface{}{ "key": "custom_invite_code", "value": req.InviteCode, }) if user == nil { e.OutErr(c, 400, e.NewErr(400, "用户不存在")) return } } ownUser := svc.GetUser(c) if ownUser.ParentUid > 0 { e.OutErr(c, 400, e.NewErr(400, "已有导师")) return } if user.Id == ownUser.Id { e.OutErr(c, 400, e.NewErr(400, "不能绑定自己")) return } ownUser.ParentUid = user.Id _, err = db.Db.Where("id=?", ownUser.Id).Cols("parent_uid").Update(ownUser) if err != nil { e.OutErr(c, 400, e.NewErr(400, "绑定失败,请重试")) return } initLV := 1 ur := new(model.UserRelate) ur.ParentUid = user.Id ur.Uid = ownUser.Id ur.Level = initLV ur.InviteTime = ownUser.CreateAt userRelateDb := implement.NewUserRelateDb(db.Db) _, err = userRelateDb.UserRelateInsert(ur) if err != nil { e.OutErr(c, e.ERR_DB_ORM, err) return } // 插入多级关联 go svc.RoutineMultiRelate(ur.ParentUid, ur.Uid, initLV) //TODO 绑定成功后 加群之类的怎么处理 if ownUser.ParentUid > 0 { //TODO::推入mq异步处理 ch, err := rabbit.Cfg.Pool.GetChannel() if err != nil { e.OutErr(c, e.ERR_INIT_RABBITMQ, err.Error()) return } defer ch.Release() ch.Publish(md4.EggAppExchange, md4.AddPublicPlatoonUserRelationCommissionReq{RecommendUid: utils.Int64ToStr(ownUser.ParentUid), Uid: utils.Int64ToStr(ownUser.Id)}, md4.EggRoutKeyForAddPublicPlatoonUserRelationCommission) } e.OutSuc(c, "success", nil) return } // BindUserInfo // @Summary 绑定用户信息 // @Tags 会员中心 // @Description 绑定用户信息 // @Accept json // @Produce json // @Param req body md.WechatAccountUserInfoReq true "注册参数" // @Success 200 {string} "具体数据" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/memberCenter/bindUserInfo [post] func BindUserInfo(c *gin.Context) { var req md.WechatAccountUserInfoReq err := c.ShouldBindJSON(&req) if err != nil { err = svc.HandleValidateErr(err) err1 := err.(e.E) e.OutErr(c, err1.Code, err1.Error()) return } if req.WechatAccount == "" { e.OutErr(c, 400, e.NewErr(400, "微信号不能为空")) return } ownUser := svc.GetUser(c) cols := "" if req.Nickname != "" { ownUser.Nickname = req.Nickname cols += ",nickname" } if req.HeadImg != "" { ownUser.Avatar = req.HeadImg cols += ",avatar" } if req.WechatAccount != "" { ownUser.WechatAccount = req.WechatAccount cols += ",wechat_account" } if cols == "" { e.OutErr(c, 400, e.NewErr(400, "修改失败")) return } _, err = db.Db.Where("id=?", ownUser.Id).Cols(cols[1:]).Update(ownUser) if err != nil { e.OutErr(c, 400, e.NewErr(400, "修改失败")) return } e.OutSuc(c, "success", nil) return } // UpdatePassword // @Summary 修改密码-不要原密码 换成验证码 // @Tags 账号与安全 // @Description 修改密码 // @Accept json // @Produce json // @Param req body md.UpdatePasswordReq true "注册参数" // @Success 200 string "登录成功返回" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/memberCenter/updatePassword [post] func UpdatePassword(c *gin.Context) { var req md.UpdatePasswordReq err := c.ShouldBindJSON(&req) if err != nil { err = svc.HandleValidateErr(err) err1 := err.(e.E) e.OutErr(c, err1.Code, err1.Error()) return } user := svc.GetUser(c) data := svc.AliyunSmsBase(c, "") //校验短信 err = aliyun.AliyunCheckSms(data["aliyun_sms_id"], data["aliyun_sms_secret"], user.Phone, req.Code) if err != nil { e.OutErr(c, 400, e.NewErr(400, "验证码错误,请重试")) return } user.Password = utils.Md5(req.Password) db.Db.Where("id=?", user.Id).Cols("password").Update(user) e.OutSuc(c, "success", nil) return } // UpdatePasscode // @Summary 修改支付宝密码 // @Tags 账号与安全 // @Description 修改支付宝密码 // @Accept json // @Produce json // @Param req body md.UpdatePasscodeReq true "注册参数" // @Success 200 string "登录成功返回" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/memberCenter/updatePasscode [post] func UpdatePasscode(c *gin.Context) { var req md.UpdatePasscodeReq err := c.ShouldBindJSON(&req) if err != nil { err = svc.HandleValidateErr(err) err1 := err.(e.E) e.OutErr(c, err1.Code, err1.Error()) return } user := svc.GetUser(c) data := svc.AliyunSmsBase(c, "") //校验短信 err = aliyun.AliyunCheckSms(data["aliyun_sms_id"], data["aliyun_sms_secret"], user.Phone, req.Code) if err != nil { e.OutErr(c, 400, e.NewErr(400, "验证码错误,请重试")) return } user.Passcode = utils.Md5(req.PassCode) db.Db.Where("id=?", user.Id).Cols("passcode").Update(user) e.OutSuc(c, "success", nil) return } // InviteUrl // @Summary 邀请链接 // @Tags 邀请海报 // @Description 邀请链接 // @Accept json // @Produce json // @Success 200 {object} md.InviteUrl "登录成功返回" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/memberCenter/inviteUrl [get] func InviteUrl(c *gin.Context) { user := svc.GetUser(c) link := svc.GetSysCfgStr("kuaizhan_url") res := md.InviteUrl{ Link: "", InviteCode: user.SystemInviteCode, } if user.CustomInviteCode != "" { res.InviteCode = user.CustomInviteCode } link += "?inviteCode=" + res.InviteCode EggUserShortLink := md2.EggUserShortLink boolQueryToItem := elastic.NewBoolQuery() // 创建bool查询 aggsMatch := elastic.NewMatchQuery("link", link) //设置查询条件 boolQueryToItem.Must(aggsMatch) result, _ := hdl.EsSelectOne(context.Background(), EggUserShortLink, boolQueryToItem, false) isHas := 0 if result != nil && len(result.Hits.Hits) > 0 { for _, hit := range result.Hits.Hits { if hit == nil { continue } jsonByte, _ := hit.Source.MarshalJSON() if gjson.Get(string(jsonByte), "short_link").String() != "" && gjson.Get(string(jsonByte), "date").Int() > time.Now().Unix() { isHas = 1 link = gjson.Get(string(jsonByte), "short_link").String() } } } if isHas == 0 { url, _ := baidu.BaiduShortenUrl(svc.GetSysCfgStr("baidu_token"), link) if url != "" { var uniqueId = php2go.Md5(link) oldLink := link link = url tmp := map[string]interface{}{ "uid": user.Id, "link": oldLink, "short_link": link, "date": time.Now().Unix() + 365*86400, } doc, _ := es.FirstDoc(EggUserShortLink, uniqueId) if doc == nil { es.CreateDoc(EggUserShortLink, uniqueId, tmp) } else { es.UpdateDoc(EggUserShortLink, uniqueId, tmp) } } } res.Link = link QRcode := qrcode.GetPNGBase64(link) QRcode = strings.ReplaceAll(QRcode, "\u0000", "") res.Qrcode = QRcode e.OutSuc(c, res, nil) return } // ParentInfo // @Summary 导师信息 // @Tags 会员中心 // @Description 导师信息 // @Accept json // @Produce json // @Success 200 {object} md.ParentInfo "登录成功返回" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/memberCenter/parentInfo [get] func ParentInfo(c *gin.Context) { ownUser := svc.GetUser(c) if ownUser.ParentUid == 0 { e.OutSuc(c, md.ParentInfo{}, nil) return } NewUserDb := implement.NewUserDb(db.Db) user, _ := NewUserDb.GetUser(ownUser.ParentUid) // 1. 获取会员等级名称 userLevelDb := implement.NewUserLevelDb(db.Db) level, err := userLevelDb.UserLevelByID(user.Level) if err != nil { e.OutErr(c, e.ERR_DB_ORM, nil) return } code := user.SystemInviteCode if user.CustomInviteCode != "" { code = user.CustomInviteCode } res := md.ParentInfo{ Nickname: user.Nickname, LevelName: level.LevelName, InviteCode: code, HeadImg: svc.GetOssUrl(user.Avatar), Id: utils.Int64ToStr(user.Id), Phone: user.Phone, WechatAccount: user.WechatAccount, } e.OutSuc(c, res, nil) return } // Delete // @Summary 注销账号操作 // @Tags 账号与安全 // @Description 注销账号操作 // @Accept json // @Produce json // @Param req body md.DeleteUserReq true "注册参数" // @Success 200 string "登录成功返回" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/memberCenter/delete [post] func Delete(c *gin.Context) { var req md.DeleteUserReq err := c.ShouldBindJSON(&req) if err != nil { err = svc.HandleValidateErr(err) err1 := err.(e.E) e.OutErr(c, err1.Code, err1.Error()) return } user := svc.GetUser(c) data := svc.AliyunSmsBase(c, "") //校验短信 err = aliyun.AliyunCheckSms(data["aliyun_sms_id"], data["aliyun_sms_secret"], user.Phone, req.Code) if err != nil { e.OutErr(c, 400, e.NewErr(400, "验证码错误,请重试")) return } user.State = 3 db.Db.Where("id=?", user.Id).Cols("state").Update(user) tmp := model.UserDeleteInfo{ Uid: int(user.Id), Phone: user.Phone, CreateAt: time.Now(), } db.Db.Insert(&tmp) ch, err := rabbit.Cfg.Pool.GetChannel() if err == nil { defer ch.Release() err = ch.PublishV2(md.EggUserExchange, md.CommUserId{ Uid: utils.Int64ToStr(user.Id), ParentUid: utils.Int64ToStr(user.ParentUid), }, md.EggUserDelete) if err != nil { ch.PublishV2(md.EggUserExchange, md.CommUserId{ Uid: utils.Int64ToStr(user.Id), ParentUid: utils.Int64ToStr(user.ParentUid), }, md.EggUserDelete) } } // 清掉token cacheKey := fmt.Sprintf(auth.TokenKey, user.Id) _, err = cache.SetEx(cacheKey, "", 1) e.OutSuc(c, "success", nil) return } // DeleteInfo // @Summary 注销账号信息 // @Tags 账号与安全 // @Description 注销账号信息 // @Accept json // @Produce json // @Success 200 {object} md.UserDeleteInfo "登录成功返回" // @Failure 400 {object} md.Response "具体错误" // @Router /api/v1/memberCenter/delete/info [get] func DeleteInfo(c *gin.Context) { user := svc.GetUser(c) resp := md.UserDeleteInfo{ Url: fmt.Sprintf("%s%s?id=%s&is_hide=1", svc.GetSysCfgStr("wap_host"), "/#/pages/course-detail-page/course-detail-page", "115"), } extendUserCount, _ := db.Db.Where("parent_uid=?", user.Id).Count(&model.User{}) NewUserWalletDb := implement.NewUserWalletDb(db.Db) walletAmount := "0" wallet, _ := NewUserWalletDb.GetUserVirtualWallet(user.Id) if wallet != nil { walletAmount = wallet.Amount } NewUserVirtualAmountDb := implement.NewUserVirtualAmountDb(db.Db) virtualWallet, _ := NewUserVirtualAmountDb.GetUserVirtualAllWallets(user.Id) virtualAmount1 := "0" virtualAmount2 := "0" virtualAmount3 := "0" settingDb := implement.NewEggEnergyBasicSettingDb(db.Db) setting, _ := settingDb.EggEnergyBasicSettingGetOne() if virtualWallet != nil { for _, v := range *virtualWallet { if v.CoinId == setting.PersonEggPointsCoinId { virtualAmount3 = utils.Float64ToStrPrec8(utils.StrToFloat64(v.Amount) + utils.StrToFloat64(virtualAmount3)) } if v.CoinId == setting.TeamEggPointsCoinId { virtualAmount3 = utils.Float64ToStrPrec8(utils.StrToFloat64(v.Amount) + utils.StrToFloat64(virtualAmount3)) } if v.CoinId == setting.PersonEggEnergyCoinId { virtualAmount2 = utils.Float64ToStrPrec8(utils.StrToFloat64(v.Amount) + utils.StrToFloat64(virtualAmount2)) } if v.CoinId == setting.TeamEggEnergyCoinId { virtualAmount2 = utils.Float64ToStrPrec8(utils.StrToFloat64(v.Amount) + utils.StrToFloat64(virtualAmount2)) } if v.CoinId == setting.ContributionCoinId { virtualAmount1 = utils.Float64ToStrPrec8(utils.StrToFloat64(v.Amount) + utils.StrToFloat64(virtualAmount1)) } } } score := 60.00 now := time.Now() esIndex := es2.GetLatestEffectiveIndexFromAlias(now) esIndexName := md3.EggEnergyUserEggScoreEsAlias + "_" + esIndex results, _ := es.FirstDoc(esIndexName, esIndex+"_"+utils.Int64ToStr(user.Id)) if results != nil { var doc md.UserEggFlowReqRespList json.Unmarshal(results.Source, &doc) score = doc.ScoreValue } uid := user.Id sql := fmt.Sprintf("SELECT COUNT(*)AS total FROM `public_platoon_user_relation` WHERE father_uid1 = %d OR father_uid2= %d OR father_uid3= %d OR father_uid4= %d OR father_uid5= %d OR father_uid6= %d OR father_uid7= %d OR father_uid8= %d OR father_uid9= %d", uid, uid, uid, uid, uid, uid, uid, uid, uid) nativeString1, _ := db.QueryNativeString(db.Db, sql) hasUserCount := "0" if len(nativeString1) > 0 { hasUserCount = nativeString1[0]["total"] } info := []md.UserDeleteInfoList{ {Title: "贡献值", Content: "贡献值 " + virtualAmount1}, {Title: "能量值", Content: "能量值 " + virtualAmount2}, {Title: "活跃值", Content: "活跃值 " + virtualAmount3}, {Title: "蛋蛋分", Content: "蛋蛋分 " + utils.Float64ToStr(score)}, {Title: "余额", Content: "余额 " + walletAmount}, {Title: "直推好友", Content: "直推好友 " + utils.Int64ToStr(extendUserCount) + " 人"}, {Title: "团队", Content: "团队 " + hasUserCount + " 人"}, } resp.Info = info e.OutSuc(c, resp, nil) return }