package invite_r import ( "git.hilo.cn/hilo-common/domain" "git.hilo.cn/hilo-common/mycontext" "git.hilo.cn/hilo-common/utils" "github.com/gin-gonic/gin" "github.com/jinzhu/now" "hilo-user/cv/invite_cv" "hilo-user/domain/cache/user_c" "hilo-user/domain/model/invite_m" "hilo-user/domain/model/promotion_m" "hilo-user/domain/model/user_m" "hilo-user/myerr/bizerr" "hilo-user/req" "hilo-user/resp" "time" ) // @Tags 新人邀请 // @Summary 提交申请 // @Param newUserCode formData string true "被邀请人id" // @Param platform formData string true "平台" // @Param platformId formData string true "平台Id" // @Param recharge formData string true "充值金额" // @Param userCode formData string true "邀请人id" // @Param videoUrl formData string true "充值金额" // @Success 200 // @Router /v2/user/invite/apply [post] func InviteApply(c *gin.Context) (*mycontext.MyContext, error) { myCtx := mycontext.CreateMyContext(c.Keys) type paramStr struct { NewUserCode string `form:"newUserCode" binding:"required"` Platform string `form:"platform" binding:"required"` PlatformId string `form:"platformId" binding:"required"` Recharge string `form:"recharge" binding:"required"` UserCode string `form:"userCode" binding:"required"` VideoUrl string `form:"videoUrl" binding:"required"` } myUserId, err := req.GetUserId(c) if err != nil { return myCtx, err } var param paramStr if err := c.ShouldBind(¶m); err != nil { return myCtx, err } model := domain.CreateModelContext(myCtx) // 平台是否填写正确 platforms := promotion_m.GetPromotionPlatforms(model) var existsPlatform bool for _, v := range platforms { if v == param.Platform { existsPlatform = true } } if !existsPlatform { model.Log.Errorf("InviteApply param:%v", param) return myCtx, bizerr.InvalidParameter } // code 是否存在 newUser, err := user_m.GetUserByCode(model, param.NewUserCode) if err != nil { model.Log.Errorf("InviteApply param:%v", param) return myCtx, bizerr.InviteApplyNewCodeInvalid } user, err := user_m.GetUserByCode(model, param.UserCode) if err != nil { model.Log.Errorf("InviteApply param:%v", param) return myCtx, bizerr.InviteApplyCodeInvalid } if newUser.ID == 0 || user.ID == 0 { model.Log.Errorf("InviteApply param:%v", param) return myCtx, bizerr.InvalidParameter } // 邀请人是否有资格邀请 if !promotion_m.IsPromotionAgent(model, user.ID) { model.Log.Errorf("InviteApply 没有邀请资格 param:%v", param) return myCtx, bizerr.InviteApplyNoPermission } //if user.ID != myUserId && !promotion_m.IsMyPromotionManager(model, user.ID, myUserId) { // model.Log.Errorf("InviteApply 没有邀请资格 param:%v", param) // return myCtx, bizerr.InviteApplyNoPermission //} // 被邀请人是否已经被人提交过申请 isApply, err := invite_m.IsInInviteApply(model, newUser.ID) if err != nil { model.Log.Errorf("InviteApply param:%v, err:%v", param, err) return myCtx, err } if isApply { model.Log.Errorf("InviteApply 已经被别人邀请过了 param:%v", param) return myCtx, bizerr.InviteApplyAlreadyInvited } // 被邀请人是否符合条件 isInvite, err := promotion_m.IsPromotionInvitee(model, newUser.ID) if err != nil { model.Log.Errorf("InviteApply param:%v", param) return myCtx, err } if isInvite { model.Log.Errorf("InviteApply 已经被别人邀请了 param:%v", param) return myCtx, bizerr.InviteApplyAlreadyInvited } // 插入邀请表 err = invite_m.CreateInviteApply(model, user.ID, newUser.ID, myUserId, param.Platform, param.PlatformId, param.Recharge, param.VideoUrl) if err != nil { model.Log.Errorf("InviteApply param:%v", param) return myCtx, err } resp.ResponseOk(c, nil) return myCtx, nil } // @Tags 新人邀请 // @Summary 查询历史申请 // @Param pageIndex query int true "偏移值 默认:1" default(1) // @Param pageSize query int true "请求数量 默认:10" default(10) // @Param beginTime query string true "开始时间2006-01-02" // @Param endTime query string true "结束时间2006-01-02" // @Param type query int true "1.已申请2.待审核3.已通过4.已拒绝" // @Success 200 {object} invite_cv.InviteApplyRes // @Router /v2/user/invite/apply [get] func InviteApplyList(c *gin.Context) (*mycontext.MyContext, error) { myCtx := mycontext.CreateMyContext(c.Keys) type paramStr struct { PageIndex int `form:"pageIndex" binding:"required"` PageSize int `form:"pageSize" binding:"required"` BeginTime time.Time `form:"beginTime" binding:"required" time_format:"2006-01-02"` EndTime time.Time `form:"endTime" binding:"required" time_format:"2006-01-02"` Type int `form:"type"` } var param paramStr if err := c.ShouldBindQuery(¶m); err != nil { return myCtx, err } if param.Type < 0 || param.Type > 4 { return myCtx, bizerr.InvalidParameter } //beginTime, err := time.ParseInLocation(utils.DATE_FORMAT, param.BeginTime, time.Local) //if err != nil { // return nil, myerr.WrapErr(err) //} //endTime, err := time.ParseInLocation(utils.DATE_FORMAT, param.EndTime, time.Local) //if err != nil { // return nil, myerr.WrapErr(err) //} param.EndTime = utils.GetDayEndTime(param.EndTime) userId, err := req.GetUserId(c) if err != nil { return myCtx, err } var model = domain.CreateModelContext(myCtx) agentIds := []uint64{userId} if promotion_m.IsPromotionManager(model, userId) { agentIds, err = promotion_m.GetPromotionManagerAgentList(model, userId) if err != nil { return myCtx, err } } if param.Type == 0 { // 返回所有Type类型有多少条数 user, err := user_c.GetUserTinyById(model, userId) if err != nil { model.Log.Errorf("GetApplyList param:%v, err:%v", param, err) return myCtx, bizerr.InvalidParameter } numList := make([]*invite_cv.InviteApplyNumRes, 0, 4) for _, gType := range []int{1, 2, 3, 4} { // 1.已申请2.待审核3.已通过4.已拒绝 num, err := invite_m.GetInviteApplyNumByType(model, gType, param.BeginTime, param.EndTime, agentIds, userId) if err != nil { return myCtx, err } numList = append(numList, &invite_cv.InviteApplyNumRes{Type: gType, Num: num}) } resp.ResponsePageOk(c, &invite_cv.InviteApplyRes{NumList: numList, MyCode: user.Code}, 0, 0, 0) return myCtx, nil } list, total, err := invite_m.GetApplyList(model, userId, agentIds, param.PageIndex, param.PageSize, param.Type, param.BeginTime, param.EndTime) if err != nil { model.Log.Errorf("GetApplyList param:%v, err:%v", param, err) return myCtx, err } uids := make([]uint64, 0, len(list)+1) for _, v := range list { uids = append(uids, v.UserId, v.NewUserId) } users, err := user_c.GetUserTinyMap(model, uids, true) if err != nil { model.Log.Errorf("GetApplyList param:%v, err:%v", param, err) return myCtx, err } res := &invite_cv.InviteApplyRes{} res.List = make([]*invite_cv.InviteApply, 0, len(list)) for _, v := range list { res.List = append(res.List, &invite_cv.InviteApply{ NewUserCode: users[v.NewUserId].Code, Platform: v.Platform, Recharge: v.RechargeInfo, UserCode: users[v.UserId].Code, CreateUnix: v.CreatedTime.Unix(), Level: v.Level, Status: v.Status, Avatar: users[v.NewUserId].Avatar, ExternalId: users[v.NewUserId].ExternalId, Reason: v.Reason, Country: users[v.NewUserId].Country, PlatformId: v.PlatformId, }) } resp.ResponsePageOk(c, res, total, param.PageSize, param.PageIndex) return myCtx, nil } // @tags 新人邀请 // @Summary 平台列表 // @Success 200 {object} []string // @Router /v2/user/invite/platform [get] func PromotionPlatform(c *gin.Context) (*mycontext.MyContext, error) { myCtx := mycontext.CreateMyContext(c.Keys) resp.ResponseOk(c, []string{"Falla", "Yalla", "Whisper", "Ahlan", "Mashi", "YoYo", "Yoho", "Echo", "Hawa", "Yalla Ludo", "Hafla", "Imo", "Ola Party", "ShareChat", "Viya", "Hello Yo", "Bigo Live", "Hago"}) return myCtx, nil } // @Tags 新人邀请 // @Summary 查询周期 // @Success 200 {object} invite_cv.AgentPeriod // @Router /v2/user/invite/period [get] func AgentPeriod(c *gin.Context) (*mycontext.MyContext, error) { myCtx := mycontext.CreateMyContext(c.Keys) var response invite_cv.AgentPeriod week := now.BeginningOfWeek().AddDate(0, 0, 1) // 周一开始 for i := 0; i < 12; i++ { response.Week = append(response.Week, invite_cv.AgentPeriodWeek{ StartDate: week.Format("2006-01-02"), EndDate: week.AddDate(0, 0, 6).Format("2006-01-02"), }) week = week.AddDate(0, 0, -7) } resp.ResponseOk(c, response) return myCtx, nil }