package controller

import (
	"encoding/base64"
	"gat1400Exchange/config"
	"gat1400Exchange/pkg"
	"gat1400Exchange/service"
	"net/http"
	"time"

	"gat1400Exchange/pkg/logger"
	"gat1400Exchange/repository"
	"gat1400Exchange/vo"

	"github.com/gin-gonic/gin"
)

type CaptureController struct {
	Repository repository.CaptureRepository
}

// 构造函数
func NewCaptureController() CaptureController {
	svr := repository.NewCaptureRepository()
	controller := CaptureController{Repository: svr}

	return controller
}

// 批量提交人脸
func (a CaptureController) Faces(c *gin.Context) {
	var req vo.RequestFaceList
	if err := c.BindJSON(&req); err != nil {
		c.AbortWithStatus(http.StatusBadRequest)
		return
	}

	if len(req.FaceListObject.FaceObject) == 0 {
		c.AbortWithStatus(http.StatusBadRequest)
		return
	}

	face := req.FaceListObject.FaceObject[0]
	logger.Debug("Receive new message, Id:%s Ip:%s faceId:%s, LeftTopX:%d, appearTime:%s", c.RemoteIP(), face.DeviceID, face.FaceID, face.LeftTopX, face.FaceAppearTime)

	// 如果开启了下级, 身份应该是消息代理, 不再转发到服务器
	if config.ClientConf.Enable && config.ServeConf.Role == "agent" {
		go a.Repository.VIIDMsgForward(&req)
	} else if config.ServeConf.Role == "cascade" {
		go service.AddFaceNotification(&face)
	}

	if config.ForwardConf.SyncServer != "" {
		go a.Repository.FaceForward(req.FaceListObject.FaceObject)
	}

	rspMsg := vo.ResponseStatus{
		RequestURL:   c.FullPath(),
		StatusCode:   vo.StatusSuccess,
		StatusString: vo.StatusString[vo.StatusSuccess],
		Id:           face.FaceID,
		LocalTime:    time.Now().Format("20060102150405"),
	}

	c.JSON(http.StatusOK, gin.H{"ResponseStatusObject": rspMsg})
}

func (a CaptureController) VideoLabels(c *gin.Context) {
	var req vo.RequestVideoLabelList
	if err := c.BindJSON(&req); err != nil {
		c.AbortWithStatus(http.StatusBadRequest)
		return
	}

	if len(req.VideoLabelListObject.VideoLabelObject) == 0 {
		c.AbortWithStatus(http.StatusBadRequest)
		return
	}

	videoLabel := req.VideoLabelListObject.VideoLabelObject[0]
	logger.Debug("Receive new message, Id:%s Ip:%s ", videoLabel.VideoLabelID, c.RemoteIP())

	// 转人脸消息
	var face vo.FaceObject
	face.FaceID = videoLabel.VideoLabelID
	face.InfoKind = 1
	face.SourceID = videoLabel.VideoImageID
	face.DeviceID = videoLabel.IVADeviceID
	face.LocationMarkTime = videoLabel.CreateTimeAbs
	face.FaceAppearTime = videoLabel.BehaviorAnalysisObject.BehaviorBeginTime
	face.FaceDisAppearTime = videoLabel.BehaviorAnalysisObject.BehaviorEndTime

	var hasTargetImage bool
	var bgImageWith, bgImageHeight int
	var bgImage *vo.SubImageInfoObject
	for idx, img := range videoLabel.SubImageList.SubImageInfoObject {
		videoLabel.SubImageList.SubImageInfoObject[idx].EventSort = 10

		// 判断是否有小图, 如果没有切一张
		if img.Type == "14" {
			bgImageWith, bgImageHeight = img.Width, img.Height
		} else {
			hasTargetImage = true
			bgImage = &videoLabel.SubImageList.SubImageInfoObject[idx]
		}

		face.SubImageList.SubImageInfoObject = append(
			face.SubImageList.SubImageInfoObject,
			videoLabel.SubImageList.SubImageInfoObject[idx],
		)
	}

	if !hasTargetImage && bgImage != nil {
		imgData, err := base64.StdEncoding.DecodeString(bgImage.Data)
		if err != nil {
			c.AbortWithStatus(http.StatusBadRequest)
			return
		}

		var subRect pkg.Rect
		subRect.Left = bgImageWith / 3
		subRect.Top = bgImageHeight / 3
		subRect.Right = subRect.Left * 2
		subRect.Bottom = subRect.Top * 2

		subImage, err := pkg.SubCutImage(imgData, &subRect, 0)
		if err != nil {
			c.AbortWithStatus(http.StatusBadRequest)
			return
		}

		var subImageInfo = vo.SubImageInfoObject{
			ImageID:     bgImage.ImageID + "1",
			EventSort:   10,
			DeviceID:    bgImage.DeviceID,
			StoragePath: "",
			Type:        "11",
			FileFormat:  "Jpeg",
			ShotTime:    bgImage.ShotTime,
			Width:       subRect.Right - subRect.Left,
			Height:      subRect.Top - subRect.Bottom,
			Data:        base64.StdEncoding.EncodeToString(subImage),
		}
		face.SubImageList.SubImageInfoObject = append(
			face.SubImageList.SubImageInfoObject,
			subImageInfo,
		)
	}

	// 如果开启了下级, 身份应该是消息代理, 不再转发到服务器
	if config.ClientConf.Enable && config.ServeConf.Role == "agent" {
		var faceObjList vo.RequestFaceList
		faceObjList.FaceListObject.FaceObject = append(faceObjList.FaceListObject.FaceObject, face)
		go a.Repository.VIIDMsgForward(&faceObjList)
	} else if config.ServeConf.Role == "cascade" {
		go service.AddFaceNotification(&face)
	}

	if config.ForwardConf.SyncServer != "" {
		go a.Repository.FaceForward([]vo.FaceObject{face})
	}

	rspMsg := vo.ResponseStatus{
		RequestURL:   c.FullPath(),
		StatusCode:   vo.StatusSuccess,
		StatusString: vo.StatusString[vo.StatusSuccess],
		Id:           videoLabel.VideoLabelID,
		LocalTime:    time.Now().Format("20060102150405"),
	}

	c.JSON(http.StatusOK, gin.H{"ResponseStatusObject": rspMsg})
}