package smithy

import (
	"context"
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/pb"
	"go_dreamfactory/utils"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo/options"
	"google.golang.org/protobuf/proto"
)

//参数校验
func (this *apiComp) GetRandUserCheck(session comm.IUserSession, req *pb.SmithyGetRandUserReq) (code pb.ErrorCode) {
	if req.People <= 0 {
		code = pb.ErrorCode_ReqParameterError
		return
	}
	return
}

/// 获取一些玩家数据
func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.SmithyGetRandUserReq) (code pb.ErrorCode, data proto.Message) {
	var (
		szDbUser []*pb.DBUser
		mapUser  map[string]struct{}
	)
	mapUser = make(map[string]struct{}, req.People)
	code = this.GetRandUserCheck(session, req)
	if code != pb.ErrorCode_Success {
		return // 参数校验失败直接返回
	}
	// 获取在线玩家信息
	onlineList, err := this.module.ModuleUser.CrossUserOnlineList()
	if err != nil {
		code = pb.ErrorCode_DBError
		return
	}

	var szUid []string
	for _, v := range onlineList {
		if v.Uid == session.GetUserId() || v.Uid == "" { // 过滤自己信息
			continue
		}
		mapUser[v.Uid] = struct{}{}
		szUid = append(szUid, v.Uid)
	}
	// 随机在线玩家信息
	if len(szUid) > int(req.People) {
		randArr := utils.Numbers(0, len(szUid), int(req.People))
		for _, v := range randArr {
			if szUid[v] != "" {
				mapUser[szUid[v]] = struct{}{}
			}
		}
	} else { // 数量不足 则有多少给多少
		for _, v := range szUid {
			mapUser[v] = struct{}{}
		}
		left := int(req.People) - len(mapUser)
		if left > 0 { // 一个人也没有 那就从db 中随机取
			if _data, err1 := this.module.modelSmithy.DB.Find(core.SqlTable(comm.TableUser), bson.M{}, options.Find().SetSort(bson.M{"lv": -1}).SetLimit(int64(req.People))); err1 == nil {
				for _data.Next(context.TODO()) {
					temp := &pb.DBUser{}
					if err = _data.Decode(temp); err == nil {
						if len(mapUser) >= int(req.People) {
							break
						}
						if _, ok := mapUser[temp.Uid]; !ok {
							mapUser[temp.Uid] = struct{}{}
						}
					}
				}
			}
		}
	}
	for k := range mapUser {
		user, _ := this.module.ModuleUser.GetCrossUser(k)
		szDbUser = append(szDbUser, user) // 转成user对象
	}
	session.SendMsg(string(this.module.GetType()), SmithyGetRandUserResp, &pb.SmithyGetRandUserResp{User: szDbUser})
	return
}