package esmod

import (
	"fmt"
	"videoweb/app/api/proto"
	"videoweb/pub/common/log"
	"videoweb/pub/common/srvdef"
	"videoweb/pub/esmanager"
	"videoweb/pub/models/videoinformationmod"
)

// SearchVideo 根据关键字搜索媒体
func SearchVideo(content string, from int64, size int64, sort string) (int64, []*videoinformationmod.VideoInformation, *srvdef.SrvErr) {
	fields := []string{FN.ESVideo.Title, FN.ESVideo.CreatorName, FN.ESVideo.Keyworld, FN.ESVideo.Desc, FN.ESVideo.Remark}
	must := esmanager.M{
		"bool": esmanager.M{
			"must": esmanager.A{
				{"multi_match": esmanager.M{
					"query":  content,
					"type":   "best_fields",
					"fields": fields},
				},
				{"term": esmanager.M{FN.ESVideo.Status: videoinformationmod.InformationSatusVerifyOK}},
			},
		},
	}
	switch sort {
	case string(proto.SearchSortTypeHot):
		must = esmanager.M{
			"function_score": esmanager.M{
				"query": esmanager.M{
					"bool": esmanager.M{
						"must": esmanager.A{
							{"multi_match": esmanager.M{
								"query":  content,
								"type":   "best_fields",
								"fields": fields,
							}},
							{"term": esmanager.M{FN.ESVideo.Status: videoinformationmod.InformationSatusVerifyOK}},
						},
					},
				},
				"functions": esmanager.A{
					esmanager.M{
						"field_value_factor": esmanager.M{
							"field":    FN.ESVideo.WatchTimes,
							"factor":   1.0,
							"modifier": "none",
						},
					},
				},
				"score_mode": "sum",
				"boost_mode": "multiply",
			},
		}
	case string(proto.SearchSortTypeNew):
		must = esmanager.M{
			"bool": esmanager.M{
				"must": esmanager.A{
					{"multi_match": esmanager.M{
						"query":  content,
						"type":   "best_fields",
						"fields": fields},
					},
					{"term": esmanager.M{FN.ESVideo.Status: videoinformationmod.InformationSatusVerifyOK}},
				},
				"should": esmanager.A{
					{"range": esmanager.M{FN.ESVideo.AddedTime: esmanager.M{"gte": "now-30d/d", "boost": 2.0}}},  //7天内提升4倍权重
					{"range": esmanager.M{FN.ESVideo.AddedTime: esmanager.M{"gte": "now-180d/d", "boost": 1.5}}}, // 30天内提升3倍权重
					{"range": esmanager.M{FN.ESVideo.AddedTime: esmanager.M{"gte": "now-365d/d", "boost": 1.1}}}, // 90天内提升1.5倍权重
				},
				"minimum_should_match": 0, // 允许不匹配任何should条件
			},
		}
	}

	query := esmanager.M{
		"query": must,
		"sort": esmanager.A{
			esmanager.M{"_score": esmanager.M{"order": "desc"}},
		},
		"from": from,
		"size": size,
	}

	videos := &ESVideoHits{}
	err := esmanager.Instance.Search(videoIndex, videos, query)
	if err != nil {
		info := fmt.Sprintf(
			"(esmod:SearchVideo) Search failed! err:%+v:", err)
		log.Warn(info,
			log.Any("keywords", content),
			log.Any("from", from),
			log.Any("size", size))
		return 0, nil, srvdef.New(srvdef.ErrElsOperError, "", "")
	}
	total := int64(0)
	rslData := make([]*videoinformationmod.VideoInformation, 0)
	if videos != nil && videos.Hits != nil && len(videos.Hits) > 0 {
		total = videos.Total.Value
		for _, video := range videos.Hits {
			rslData = append(rslData, EsVideoInfoToData(video.Source))
		}
	}
	return total, rslData, nil
}

func EsVideoInfoToData(esInfo *ESVideo) *videoinformationmod.VideoInformation {
	info := &videoinformationmod.VideoInformation{
		Id:          esInfo.Id,
		Title:       esInfo.Title,
		MediaType:   esInfo.MediaType,
		AddedTime:   esInfo.AddedTime,
		CreatorId:   esInfo.CreatorId,
		CreatorName: esInfo.CreatorName,
		Keyworld:    esInfo.Keyworld,
		FirstTag:    esInfo.FirstTag,
		SecondTags:  esInfo.SecondTags,
		CoverImg:    esInfo.CoverImg,
		Desc:        esInfo.Desc,
		Remark:      esInfo.Remark,
		Nodes:       esInfo.Nodes,
		Status:      esInfo.Status,
		Video:       esInfo.Video,
		CategoryIds: esInfo.CategoryIds,
		Tags:        esInfo.Tags,
		IsHot:       esInfo.IsHot,
		Sort:        esInfo.Sort,
		Real:        esInfo.Real,
		VideoCount:  esInfo.VideoCount,
		UpdatedAt:   esInfo.UpdatedAt,
		CreatedAt:   esInfo.CreatedAt,
	}
	return info
}