package vidhelpser

import (
	"time"

	"91porn-server/models/v/tagmod"
	"91porn-server/models/v/usermod"
	"91porn-server/models/v/vidmod"

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

func getUsersBaseInfoFromMongo(uids []uint64) ([]*usermod.BaseInfoVip, error) {
	return usermod.GetUsersBaseInfoWithVip(uids)
}

func getTagsByIDsFromMongo(tids []primitive.ObjectID) ([]vidmod.TagInfo, error) {
	tInfo, err := tagmod.FindTagsByIDS(tids)
	if err != nil {
		return []vidmod.TagInfo{}, err
	}
	info := make([]vidmod.TagInfo, len(tInfo))
	for i, v := range tInfo {
		info[i] = vidmod.TagInfo{
			ID:          v.ID,
			Name:        v.TagName,
			CoverImg:    v.CoverImg,
			Description: v.Description,
		}
	}
	return info, nil
}

func getVideoListByIDsFromMongo(vids []primitive.ObjectID) ([]*vidmod.VideoModel, error) {
	return vidmod.GetVideoListByIDs(vids)
}

func getVideoListByIDsFromMongoBaseNoStatus(vids []primitive.ObjectID) ([]*vidmod.VideoModel, error) {
	return vidmod.GetVideoListByIDsNoStatus(vids)
}

func getNewestNewsFromMongo(page, size uint64, recentMinute time.Time) ([]*vidmod.VideoModel, bool, error) {
	return vidmod.GetNewestNews(page, size, recentMinute)
}

func getNewestNewsFromMongo_old(page, size uint64) ([]*vidmod.VideoModel, bool, error) {
	return vidmod.GetNewestNews_old(page, size)
}

func getNewestShortVideoFromMongo(page, size uint64) ([]*vidmod.VideoModel, bool, error) {
	return vidmod.GetNewestShortVideo(page, size)
}

func getNewsTop200FromMongo(recentMinute time.Time) []string {
	vModel, _, err := vidmod.GetNewestNews(1, MaxNewestCacheNum, recentMinute)
	if err != nil {
		return []string{}
	}
	ids := make([]string, len(vModel))
	for i, v := range vModel {
		ids[i] = v.ID.Hex()
	}
	return ids
}

func getNewsTop200FromMongo_old() []string {
	vModel, _, err := vidmod.GetNewestNews_old(1, MaxNewestCacheNum)
	if err != nil {
		return []string{}
	}
	ids := make([]string, len(vModel))
	for i, v := range vModel {
		ids[i] = v.ID.Hex()
	}
	return ids
}

func getNewsShortVideoTop200FromMongo() []string {
	vModel, _, err := vidmod.GetNewestShortVideo(1, MaxNewestCacheNum)
	if err != nil {
		return []string{}
	}
	ids := make([]string, len(vModel))
	for i, v := range vModel {
		ids[i] = v.ID.Hex()
	}
	return ids
}