package keywordanalyse

import (
	"main_program/common"
	"main_program/config"
	"main_program/entity"
	service "main_program/service"
	"strings"
	"time"
)

type keywordAnalyse struct{}

var KeywordAnalyseService keywordAnalyse

func (m *keywordAnalyse) Start() {
	config.Logger.Info("开始逐句解析...")
	var videoId string
	var err error
	//  获取需要解析的videoId
	if common.AnalyseRegion == "" {
		config.Logger.Info("直接获取VideoId")
		videoId, err = service.SrtFileService.QueryOneNotScanVideoId()
		if err != nil {
			config.Logger.Info("没有需要解析的SrtFile...")
			return
		}
	} else {
		config.Logger.Info("通过common.AnalyseRegion获取VideoId")
		videoId, err = service.SrtFileService.QueryOneNotScanVideoIdByRegion(common.AnalyseRegion)
		if err != nil {
			config.Logger.Info("没有需要解析的SrtFile...")
			return
		}
	}
	continueFlag := true
	for continueFlag {
		config.Logger.Infof("需要解析的VideoId:%s", videoId)

		// 根据videoId获取channel
		video, err := service.VideoService.QueryOneByVideoId(videoId)
		if err != nil {
			config.Logger.Infof("获取video失败,videoId:%s", videoId)
		}

		// 根据channelId获取channel
		channel, err := service.ChannelService.QueryOneByChannelId(video.ChannelId)
		if err != nil {
			config.Logger.Infof("获取Channel失败,channelId:%s", video.ChannelId)
		}

		// 根据region获取关键字
		keywords, err := service.KeywordService.QueryKeywordsByRegion(channel.Region)
		if err != nil {
			config.Logger.Infof("获取keywords失败,region:%s", channel.Region)
		}

		// 获取所有srtFile
		srtFiles, err := service.SrtFileService.QuerySrtFilesByVideoId(videoId)
		if err != nil {
			config.Logger.Infof("获取srtFiles失败,videoId:%s", videoId)
		}
		for i := 0; i < len(srtFiles); i++ {
			analyse(srtFiles[i], keywords)
		}

		// 5秒后循环获取
		config.Logger.Info("5秒后循环获取")
		time.Sleep(5 * time.Second)
		if common.AnalyseRegion == "" {
			videoId, err = service.SrtFileService.QueryOneNotScanVideoId()
			if err != nil {
				config.Logger.Info("没有需要解析的SrtFile...")
				continueFlag = false
			}
		} else {
			config.Logger.Info("通过common.AnalyseRegion获取VideoId")
			videoId, err = service.SrtFileService.QueryOneNotScanVideoIdByRegion(common.AnalyseRegion)
			if err != nil {
				config.Logger.Info("没有需要解析的SrtFile...")
				continueFlag = false
			}
		}
	}

}

func analyse(srtFile entity.Srtfile, keywords []entity.Keyword) {
	for i := 0; i < len(keywords); i++ {
		keyword := keywords[i]
		if strings.Contains(srtFile.SrtText, keyword.Word) {
			config.Logger.Infof("srtFileId: %d Ordinal: %d text: %s word: %s", srtFile.Id, srtFile.Ordinal, srtFile.SrtText, keyword.Word)
			// 匹配完成后，输入到result
			wordResultSet := entity.WorldResultSet{KeyWordId: int(keyword.Id), WordText: keyword.Word, SrtId: int(srtFile.Id),
				SrtOrdinal: srtFile.Ordinal, SrtText: srtFile.SrtText, VideoId: srtFile.VideoId}
			err := service.WordResultSetService.InsterOneByEntity(wordResultSet)
			if err != nil {
				config.Logger.Error("存储解析结果失败...")
				return
			}
		}
	}
	// 修改srtFile状态
	err := service.SrtFileService.UpdateIsScanById(int(srtFile.Id), 1)
	if err != nil {
		config.Logger.Error("更新srtFile失败")
	}
}