package service

import (
	"fmt"
	"service/api/consts"
	zonemomentproto "service/api/proto/zonemoment/proto"
	"service/bizcommon/util"
	"service/dbstruct"
	"strings"

	"github.com/gin-gonic/gin"
	goproto "google.golang.org/protobuf/proto"
)

var DefaultVideoModerationTaskResultHandler *VideoModerationTaskResultHandler

type VideoModerationTaskResultHandler struct {
	// 图像审核任务通过及回退方法生成器map
	videoModerationTaskUpdateFuncGeneratorMap map[string](func(ctx *gin.Context, task *dbstruct.VideoModerationTask, option int) func() error)
}

func NewVideoModerationTaskResultHandler() *VideoModerationTaskResultHandler {
	handler := &VideoModerationTaskResultHandler{}
	handler.initVideoModerationTaskUpdateFuncGeneratorMap()
	return handler
}

func (handler *VideoModerationTaskResultHandler) Handle(ctx *gin.Context, task *dbstruct.VideoModerationTask, option int) error {
	return handler.getVideoModerationTaskUpdateFunc(ctx, task, option)()
}

func (handler *VideoModerationTaskResultHandler) initVideoModerationTaskUpdateFuncGeneratorMap() {
	handler.videoModerationTaskUpdateFuncGeneratorMap = make(map[string]func(ctx *gin.Context, task *dbstruct.VideoModerationTask, option int) func() error)

	handler.generateZoneMomentMediaComponentUpdateFunc()
}

func (handler *VideoModerationTaskResultHandler) getVideoModerationTaskUpdateFunc(ctx *gin.Context, task *dbstruct.VideoModerationTask, option int) func() error {

	db := util.DerefString(task.AssociativeDatabase)
	table := util.DerefString(task.AssociativeTableName)
	column := util.DerefString(task.AssociativeTableColumn)

	key := fmt.Sprintf("%v|%v|%v", db, table, column)
	updateFuncGenerator := handler.videoModerationTaskUpdateFuncGeneratorMap[key]
	updateFunc := updateFuncGenerator(ctx, task, option)

	return updateFunc

}

// 私密动态表->图像内容，若不通过，则写入不通过的原因
func (handler *VideoModerationTaskResultHandler) generateZoneMomentMediaComponentUpdateFunc() {
	handler.videoModerationTaskUpdateFuncGeneratorMap["zone_moment|zone_moment|media_component"] = func(ctx *gin.Context, task *dbstruct.VideoModerationTask, option int) func() error {
		return func() error {
			if option == consts.VideoModerationTaskUpdate_Success {
				err := _DefaultZoneMoment.OpUpdate(ctx, &zonemomentproto.OpUpdateReq{
					ZoneMoment: &dbstruct.ZoneMoment{
						Id:               task.AssociativeTableId,
						ImageAuditStatus: goproto.Int64(consts.VideoModeration_Passed),
					},
				})
				if err != nil {
					return err
				}
				return _DefaultZoneMoment.TryToCompleteAudit(ctx, task.GetAssociativeTableId())
			}

			videoModerationOpinion := &strings.Builder{}
			if task.GetStatus() == consts.VideoModeration_ServiceFailed {
				videoModerationOpinion.WriteString("机审失败")
			} else {
				// 写入具体原因
				for i, pass := range task.AuditedMediaResults {
					if !pass {
						videoModerationOpinion.WriteString(fmt.Sprintf("视频%d：%s ", i+1, task.Description[i]))
					}
				}
			}

			err := _DefaultZoneMoment.OpUpdate(ctx, &zonemomentproto.OpUpdateReq{
				ZoneMoment: &dbstruct.ZoneMoment{
					Id:                task.AssociativeTableId,
					ImageAuditStatus:  task.Status,
					ImageAuditOpinion: goproto.String(videoModerationOpinion.String()),
				},
			})
			if err != nil {
				return err
			}
			return _DefaultZoneMoment.TryToCompleteAudit(ctx, task.GetAssociativeTableId())
		}
	}
}