package usermod

import (
	"time"

	"91porn-server/common/log"
	"91porn-server/models/l/payvidlgmod"
	"91porn-server/models/v/prdcthsomod"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo/options"
)

type UserListSelector struct {
	UID           *uint64 `form:"uid,omitempty" json:"uid,omitempty" bson:"uid,omitempty"`                      //用户id
	DevID         *string `form:"devID,omitempty" json:"devID,omitempty" bson:"devID,omitempty"`                //设备id
	DevType       *string `form:"devType,omitempty" json:"devType,omitempty" bson:"devType,omitempty"`          //设备类型
	DevToken      *string `form:"devToken,omitempty" json:"devToken,omitempty" bson:"devToken,omitempty"`       //设备toke`n ios 推送用
	RegisterIP    *string `form:"registerIP,omitempty" json:"registerIP,omitempty" bson:"registerIP,omitempty"` //注册IP
	Mobile        *string `form:"mobile,omitempty" json:"mobile,omitempty" bson:"mobile,omitempty"`             //手机号
	Gender        *string `form:"gender,omitempty" json:"gender,omitempty" bson:"gender,omitempty"`             //性别
	Channel       *string `form:"channel,omitempty" json:"channel,omitempty" bson:"channel,omitempty"`
	Name          *string `form:"name,omitempty" json:"name,omitempty" bson:"name,omitempty"`
	PromotionCode *string `form:"promotionCode,omitempty" json:"promotionCode,omitempty" bson:"promotionCode,omitempty"` //推广码
	Portrait      *string `form:"portrait,omitempty" json:"portrait,omitempty" bson:"portrait,omitempty"`                //头像
	Summary       *string `form:"summary,omitempty" json:"summary,omitempty" bson:"summary,omitempty"`                   //简介
	Region        *string `form:"region,omitempty" json:"region,omitempty" bson:"region,omitempty"`                      //地区
	Birthday      *string `form:"birthday,omitempty" json:"birthday,omitempty" bson:"birthday,omitempty"`                //生日
	VipLevel      *int    `form:"vipLevel,omitempty" json:"vipLevel,omitempty" bson:"vipLevel,omitempty"`                //vip等级
	HasLocked     *bool   `form:"hasLocked,omitempty" json:"hasLocked,omitempty" bson:"hasLocked,omitempty"`             //已禁止登陆
	HasBanned     *bool   `form:"hasBanned,omitempty" json:"hasBanned,omitempty" bson:"hasBanned,omitempty"`             //已禁言
	Token         *string `form:"token,omitempty" json:"token,omitempty" bson:"token,omitempty"`                         //用户令牌
	WatchCount    *uint64 `from:"watchCount,omitempty" json:"watchCount,omitempty" bson:"watchCount,omitempty"`          //观看次数
	BankActName   *string `json:"bankActName,omitempty" bson:"bankActName,omitempty"`                                    //银行卡绑定名字
	UpTag         *string `form:"upTag" json:"upTag,omitempty" bson:"upTag,omitempty"`                                   // 博主认证
}

// ExportUser 导出的用户信息
type ExportUser struct {
	UID uint64 `xlsx:"用户id"`
	//DevID          string     `xlsx:"设备id"`
	DevType    string `xlsx:"设备类型"`
	RegisterIP string `xlsx:"注册IP"`
	Mobile     string `xlsx:"手机号"`
	//Gender         string     `xlsx:"性别"`
	Channel string `xlsx:"渠道号"`
	//Name           string     `xlsx:"用户名"`
	PromotionCode string `xlsx:"推广码"`
	//Summary        string     `xlsx:"简介"`
	//Region         string     `xlsx:"地区"`
	//Birthday       string     `xlsx:"生日"`
	//VipLevel       int        `xlsx:"vip等级"`
	VipExpireDate time.Time `xlsx:"vip过期时间"`
	CreatedAt     time.Time `xlsx:"用户创建时间"`
	LastVisitAt   time.Time `xlsx:"最后活跃时间"`
	//MobileBindAt   *time.Time `xlsx:"手机绑定时间"`
	//HasLocked      bool       `xlsx:"已禁止登陆"`
	//HasBanned      bool       `xlsx:"已禁言"`
	WatchCount int `xlsx:"七日观看次数"` //七日观看次数，和数据库存的不同
	//VisitCount     int        `xlsx:"24小时内访问数"`
	//TotalVideoCnt  int        `xlsx:"视频总数"`
	//UndealVideoCnt int        `xlsx:"未审核视频数"`
}

// 用户VIP返回数据
type UserVipInfoRes struct {
	VideoDiscount           int                   `json:"videoDiscount"`           //视频折扣率
	VideoDiscountExpiration time.Time             `json:"videoDiscountExpiration"` // 视频折扣卡到期时间
	VipExpireDate           time.Time             `json:"vipExpireDate"`           //vip过期时间
	GoldVideoFreeExpire     time.Time             `json:"goldVideoFreeExpire"`     //金币视频免费日期
	PayVidDiscount          int                   `json:"payVidDiscount"`          //支付视频折扣
	VideoFreeExpiration     *time.Time            `json:"videoFreeExpiration"`     //视频免费过期时间
	GoldVideoCoupon         []UserGoldVideoCoupon `json:"goldVideoCoupon"`
}

func FindAgentByAccount(acc string) (user *User, err error) {
	if acc == "" {
		return
	}
	err = coll(nil).FindOne(&user, bson.M{"agentAccount": acc})
	return
}

func FindAllPayableAgents() ([]string, error) {
	var p []struct {
		AgentAcc string `bson:"agentAccount"`
	}
	if err := coll(nil).Find(&p, bson.M{
		"agentAccount": bson.M{"$gt": ""},
		"deposit":      bson.M{"$gt": 0},
		"payable":      true,
		"userType":     UserLoufengAgent,
	}, options.Find().SetProjection(bson.M{"agentAccount": 1})); err != nil {
		return []string{}, err
	}
	agents := make([]string, len(p))
	for i, agent := range p {
		agents[i] = agent.AgentAcc
	}
	return agents, nil
}

// FindMany 查询所有
func FindMany(filter bson.M, _ *options.FindOptions, skip int64, size int64) ([]*User, error) {
	sort := bson.D{{Key: "createdAt", Value: -1}}
	opts := options.FindOptions{}
	opts.SetSort(sort).SetSkip(skip).SetLimit(size)
	data := make([]*User, 0)
	if err := coll(nil).Find(&data, filter, &opts); err != nil {
		log.ZapLog.Warn("proxy FindMany Find fail", log.E(err))
		return data, err
	}
	return data, nil
}

// ExportFindMany 导出的时候查询所有
func ExportFindMany(filter bson.M, skip int64, size int64) ([]*User, error) {
	sort := bson.D{{Key: "createdAt", Value: 1}}
	filter["userType"] = bson.M{"$in": []UserType{UserUnkown, UserLoufengAgent}}
	opts := options.FindOptions{}
	opts.SetSort(sort).SetSkip(skip).SetLimit(size)
	data := make([]*User, 0)
	if err := coll(nil).Find(&data, filter, &opts); err != nil {
		log.ZapLog.Warn(" ExportFindMany Find fail", log.E(err))
		return data, err
	}
	return data, nil
}

// GetTotalCnt 查询数量
func GetTotalCnt(filter bson.M, _ *options.FindOptions) (total int64, err error) {
	total, err = coll(nil).Count(filter)
	if err != nil {
		log.ZapLog.Warn("GetTotalCnt fail", log.E(err))
	}
	return
}

// PayUIDCountByCreatedTime 消费包括 购买VIP、付费视屏
func PayUIDCountByCreatedTime(start time.Time, end time.Time) ([]uint64, error) {
	//购买VIP /分
	vUIDList, err := prdcthsomod.PayVipUIDSByCreatedTime(start, end)
	if err != nil {
		return nil, err
	}
	//付费视屏 /金币
	pUIDList, err := payvidlgmod.GetUIDListByPayTime(start, end)
	if err != nil {
		return nil, err
	}
	payUIDMap := make(map[uint64]byte, len(vUIDList)+len(pUIDList))
	//转载payUIDMap
	for _, uid := range vUIDList {
		payUIDMap[uid] = 1
	}
	for _, uid := range pUIDList {
		payUIDMap[uid] = 1
	}
	//取出payUID
	payUIDList := make([]uint64, len(payUIDMap))
	i := 0
	for k := range payUIDMap {
		payUIDList[i] = k
		i++
	}
	return payUIDList, nil
}

// NewAndPayUIDListByCreatedTime 消费包括 购买VIP、付费视屏
func NewAndPayUIDListByCreatedTime(start time.Time, end time.Time) ([]uint64, error) {
	uidList, err := FindUIDByCreateTime(start, end)
	if err != nil {
		return nil, err
	}
	vUIDList, err := prdcthsomod.PayVipUIDSByCreatedTime(start, end)
	if err != nil {
		return nil, err
	}
	pUIDList, err := payvidlgmod.GetUIDListByPayTime(start, end)
	if err != nil {
		return nil, err
	}
	payUIDMap := make(map[uint64]bool, len(uidList))
	//转载payUIDMap
	for _, uid := range pUIDList {
		payUIDMap[uid] = false
	}
	//标记消费的的UID
	for _, uid := range vUIDList {
		if _, ok := payUIDMap[uid]; ok {
			payUIDMap[uid] = true
		}
	}
	for _, uid := range pUIDList {
		if _, ok := payUIDMap[uid]; ok {
			payUIDMap[uid] = true
		}
	}
	//取出payUID
	payUIDList := make([]uint64, 0, len(payUIDMap))
	for k, v := range payUIDMap {
		if v {
			payUIDList = append(payUIDList, k)
		}
	}
	return payUIDList, nil
}

type UpdateFreeTimesCond struct {
	UID       uint64  `json:"uid" bson:"uid" binding:"required"` // 用户ID
	FreeTimes *int64  `json:"freeTimes" bson:"freeTimes"`        // 免费次数
	Mark      *string `json:"mark" bson:"mark"`                  // 备注
}

func (receiver *UpdateFreeTimesCond) Filter() bson.M {
	return bson.M{"uid": receiver.UID, "aiUndressFreeTimes": bson.M{"$gte": receiver.FreeTimes}}
}

func (receiver *UpdateFreeTimesCond) Update() bson.M {
	var update = bson.M{}
	if receiver.FreeTimes != nil {
		update["aiUndressFreeTimes"] = receiver.FreeTimes
	}
	return bson.M{"$inc": update}
}

type UpdateDownloadCond struct {
	UID      uint64  `json:"uid" bson:"uid" binding:"required"` // 用户ID
	Download *int64  `json:"download" bson:"download"`          // 下载次数
	Mark     *string `json:"mark" bson:"mark"`                  // 备注
}

func (receiver *UpdateDownloadCond) Filter() bson.M {
	return bson.M{"uid": receiver.UID}
}

func (receiver *UpdateDownloadCond) Update() bson.M {
	var update = bson.M{}
	if receiver.Download != nil {
		update["downloadCount"] = receiver.Download
	}
	return bson.M{"$inc": update}
}

type SetAdvertiseCond struct {
	UId    []uint64 `form:"uid" json:"uid" binding:"required"`       // 用户uid
	Enable *bool    `form:"enable" json:"enable" binding:"required"` // 是否开启
}

type UpdateLotteryTimesCond struct {
	UID          uint64  `json:"uid" bson:"uid" binding:"required"` // 用户ID
	LotteryTimes *int64  `json:"lotteryTimes" bson:"lotteryTimes"`  // 抽空次数
	Mark         *string `json:"mark" bson:"mark"`                  // 备注
}

func (receiver *UpdateLotteryTimesCond) Filter() bson.M {
	return bson.M{"uid": receiver.UID}
}

func (receiver *UpdateLotteryTimesCond) Update() bson.M {
	var update = bson.M{}
	if receiver.LotteryTimes != nil {
		update["lotteryTimes"] = receiver.LotteryTimes
	}
	return bson.M{"$inc": update}
}

type DeleteCommentCond struct {
	UID uint64 `json:"uid" bson:"uid" binding:"required"` // 用户ID
}

type QueryGameCodeCond struct {
	UID uint64 `form:"uid" json:"uid" bson:"uid,omitempty"` //用户ID
}

func (receiver *DeleteCommentCond) Filter() bson.M {
	return bson.M{"uid": receiver.UID}
}