package mediaser

import (
	"91porn-server/app/appg"
	"91porn-server/app/service/mediatagser"
	"91porn-server/common"
	"91porn-server/common/constant/redisconst"
	"91porn-server/common/timeutil/timerange"
	"91porn-server/models/s/statrecordmod"
	"91porn-server/models/v/mediamod"
	"91porn-server/models/v/mediatagmod"
	"91porn-server/models/v/usermod"
	"strings"
	"sync"
	"time"

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

func buildDedupTagsAndDetails(tagIDs []primitive.ObjectID, tagMap map[primitive.ObjectID]*mediatagmod.MediaTag) ([]primitive.ObjectID, []mediamod.TagDetail) {
	if len(tagIDs) == 0 {
		return nil, nil
	}
	seen := make(map[string]struct{}, len(tagIDs))
	uniqueTagIDs := make([]primitive.ObjectID, 0, len(tagIDs))
	tagDetails := make([]mediamod.TagDetail, 0, len(tagIDs))
	for _, id := range tagIDs {
		tag, ok := tagMap[id]
		if !ok {
			continue
		}
		name := strings.TrimSpace(tag.Name)
		if _, exists := seen[name]; exists {
			continue
		}
		seen[name] = struct{}{}
		uniqueTagIDs = append(uniqueTagIDs, tag.ID)
		tagDetails = append(tagDetails, mediamod.TagDetail{
			Id:   tag.ID,
			Name: tag.Name,
		})
	}
	return uniqueTagIDs, tagDetails
}

func FillMedias(medias []*mediamod.Media, uid uint64, needDefaultContent bool, dramaCard ...bool) []*mediamod.AppMediaBase {
	if len(medias) == 0 {
		return nil
	}
	tagIDs := make([]primitive.ObjectID, 0)
	var (
		mIds         []primitive.ObjectID
		mStatus      map[primitive.ObjectID]mediamod.MediaStatus
		hasDrama     bool
		hasDramaCard bool
	)

	for i := range medias {
		if medias[i] == nil {
			continue
		}
		mIds = append(mIds, medias[i].ID)
		tagIDs = append(tagIDs, medias[i].Tags...)
		if medias[i].MediaType == mediamod.MediaTypeDrama {
			hasDrama = true
		}
	}
	var wg sync.WaitGroup
	wg.Add(1)
	common.Go(func() {
		defer wg.Done()
		mStatus = GetUserStatus(uid, mIds)
	})
	if hasDrama {
		if len(dramaCard) > 0 {
			hasDramaCard = dramaCard[0]
		} else {
			wg.Add(1)
			common.Go(func() {
				defer wg.Done()
				user, err := usermod.FindUserByUID(uid)
				hasDramaCard = err == nil && user != nil && user.DramaExpire.After(time.Now())
			})
		}
	}
	wg.Wait()
	var infos []*mediamod.AppMediaBase
	tagMap := mediatagser.GetTagsByIDs2Map(tagIDs)
	for i := range medias {
		if medias[i] == nil {
			continue
		}
		mid := ""
		if !medias[i].MID.IsZero() {
			mid = medias[i].MID.Hex()
		}
		temp := mediamod.AppMediaBase{
			ID:                 medias[i].ID,
			SortCode:           medias[i].SortCode,
			Status:             medias[i].Status,
			Title:              medias[i].Title,
			HorizontalCover:    medias[i].HorizontalCover,
			VerticalCover:      medias[i].VerticalCover,
			Tags:               make([]primitive.ObjectID, 0),
			TagDetails:         make([]mediamod.TagDetail, 0),
			Summary:            medias[i].Summary,
			TotalEpisode:       medias[i].TotalEpisode,
			CurrentEpisode:     medias[i].CurrentEpisode,
			FreeEpisode:        medias[i].FreeEpisode,
			MediaType:          medias[i].MediaType,
			MediaSubType:       medias[i].MediaSubType,
			Kind:               medias[i].Kind,
			Price:              medias[i].Price,
			ContentsPrice:      medias[i].ContentsPrice,
			Direction:          medias[i].Direction,
			UpdateStatus:       medias[i].UpdateStatus,
			Permission:         medias[i].Permission,
			PermissionIconHide: medias[i].PermissionIconHide,
			Style:              medias[i].Style,
			FreeTime:           medias[i].FreeTime,
			CountPurchases:     medias[i].CountPurchases,
			CountBrowse:        medias[i].CountBrowse,
			CountCollect:       medias[i].CountCollect,
			CountLike:          medias[i].CountLike,
			CountDisLike:       medias[i].CountDisLike,
			CountView:          medias[i].CountView,
			CountShare:         medias[i].CountShare,
			HasFollow:          medias[i].HasFollow,
			MediaStatus:        mStatus[medias[i].ID],
			CountComment:       medias[i].CountComment,
			Number:             medias[i].Number,
			CreatedAt:          medias[i].CreatedAt,
			UpdateTime:         medias[i].UpdateTime,
			LatestPublishedAt:  medias[i].LatestPublishedAt,
			MID:                mid,
			ModuleName:         medias[i].ModuleName,
			SectionName:        medias[i].SectionName,
			SID:                medias[i].SID,
			SectionSort:        medias[i].SectionSort,
			LsjId:              medias[i].LsjId,
		}
		if medias[i].MediaType == mediamod.MediaTypeDrama {
			temp.HasDramaCard = hasDramaCard
		}
		if temp.MediaStatus.HasCollected {
			temp.CountCollect = temp.CountCollect + 1 // 避免缓存详情的情况，收藏完成之后数字没有任何变化
		}
		if needDefaultContent {
			temp.DefaultContent = medias[i].DefaultContent
		}
		if temp.DefaultContent.Cover == "" && len(temp.DefaultContent.UrlSet) > 0 {
			temp.DefaultContent.Cover = temp.DefaultContent.UrlSet[0]
		}
		if strings.TrimSpace(temp.DefaultContent.Name) == "" {
			temp.DefaultContent.Name = "第一集"
		}
		temp.Tags, temp.TagDetails = buildDedupTagsAndDetails(medias[i].Tags, tagMap)

		infos = append(infos, &temp)
	}
	return infos
}

func FillNotStatusMedias(medias []*mediamod.Media, needDefaultContent bool) []*mediamod.AppMediaBase {
	if len(medias) == 0 {
		return []*mediamod.AppMediaBase{}
	}
	tagIDs := make([]primitive.ObjectID, 0)
	var (
		mIds []primitive.ObjectID
	)

	for i := range medias {
		mIds = append(mIds, medias[i].ID)
		if medias[i] != nil {
			tagIDs = append(tagIDs, medias[i].Tags...)
		}
	}

	infos := []*mediamod.AppMediaBase{}
	tagMap := mediatagser.GetTagsByIDs2Map(tagIDs)
	for i := range medias {
		if medias[i] == nil {
			continue
		}
		temp := mediamod.AppMediaBase{
			ID:                medias[i].ID,
			Title:             medias[i].Title,
			HorizontalCover:   medias[i].HorizontalCover,
			VerticalCover:     medias[i].VerticalCover,
			Tags:              make([]primitive.ObjectID, 0),
			TagDetails:        make([]mediamod.TagDetail, 0),
			Summary:           medias[i].Summary,
			TotalEpisode:      medias[i].TotalEpisode,
			CurrentEpisode:    medias[i].CurrentEpisode,
			FreeEpisode:       medias[i].FreeEpisode,
			MediaType:         medias[i].MediaType,
			MediaSubType:      medias[i].MediaSubType,
			Kind:              medias[i].Kind,
			Price:             medias[i].Price,
			ContentsPrice:     medias[i].ContentsPrice,
			Direction:         medias[i].Direction,
			UpdateStatus:      medias[i].UpdateStatus,
			Permission:        medias[i].Permission,
			CountPurchases:    medias[i].CountPurchases,
			CountBrowse:       medias[i].CountBrowse,
			CountCollect:      medias[i].CountCollect,
			CountLike:         medias[i].CountLike,
			HasFollow:         medias[i].HasFollow,
			CountComment:      medias[i].CountComment,
			Number:            medias[i].Number,
			CreatedAt:         medias[i].CreatedAt,
			UpdateTime:        medias[i].UpdateTime,
			LatestPublishedAt: medias[i].LatestPublishedAt,
		}
		if temp.MediaStatus.HasCollected {
			temp.CountCollect = temp.CountCollect + 1 // 避免缓存详情的情况，收藏完成之后数字没有任何变化
		}
		if needDefaultContent {
			temp.DefaultContent = medias[i].DefaultContent
		}
		if temp.DefaultContent.Cover == "" && len(temp.DefaultContent.UrlSet) > 0 {
			temp.DefaultContent.Cover = temp.DefaultContent.UrlSet[0]
		}
		if strings.TrimSpace(temp.DefaultContent.Name) == "" {
			temp.DefaultContent.Name = "第一集"
		}
		temp.Tags, temp.TagDetails = buildDedupTagsAndDetails(medias[i].Tags, tagMap)

		infos = append(infos, &temp)
	}
	return infos
}

// IncrCountBrowse 增加播放量
func IncrCountBrowse(mediaId primitive.ObjectID) {
	now := time.Now()
	recentMinute := timerange.RecentMinute(now, statrecordmod.FiveMinuteScale)
	key := redisconst.AcgBrowseCountListKey(recentMinute.Unix())
	if err := appg.Redis.RPush(key, mediaId.Hex()); err != nil {
		return
	}
	_, _ = appg.Redis.ExpireKey(key, time.Minute*20)
}