package service

import (
	"basic.com/pubsub/protomsg.git"
	"basic.com/valib/bhomeclient.git"
	"basic.com/valib/bhomedbapi.git"
	"encoding/json"
	"github.com/satori/go.uuid"
	"strings"
	"vamicro/config"
	"vamicro/stack-service/models"
	"vamicro/stack-service/vo"
)

type FileStackService struct {
	Bk bhomeclient.Broker
}

func (sv FileStackService) FindAll() []models.FileStack {
	var model models.FileStack
	list, _ := model.FindAll()
	if list == nil {
		list = []models.FileStack{}
	}
	return list
}

func (sv FileStackService) FindAllByPage(fType int, fileName string, page int, size int) (result vo.PageResult) {
	var crApi bhomedbapi.CameraRuleApi
	filterCamIds := ""
	if fType == 3 { //未配规则
		flag, allRules := crApi.FindAll()
		if flag && len(allRules) > 0 {
			for _,c := range allRules {
				if c.CameraInfo != nil && c.CameraInfo.Id != "" && len(c.Rules) > 0 {
					filterCamIds +="'"+c.CameraInfo.Id+"',"
				}
			}
		}
		if filterCamIds != "" {
			filterCamIds = strings.TrimRight(filterCamIds, ",")
		}
	}

	var model models.FileStack
	result.Total = model.Total(fileName, fType, &filterCamIds)


	if result.Total > 0 {
		arr, _ := model.FindAllByPage(fType, fileName, page, size, &filterCamIds)
		var pageList = make([]vo.FileStackWithWH, 0)
		if arr != nil {
			var resolutions []vo.Resolution
			if config.Server.Resolutions != nil && len(config.Server.Resolutions) > 0 {
				for _, rwh := range config.Server.Resolutions {
					resolutions = append(resolutions, vo.Resolution{
						Width:  rwh.Width,
						Height: rwh.Height,
					})
				}
			} else {
				dRe0 := vo.Resolution{
					Width: 0,
					Height: 0,
				}
				dRe1 := vo.Resolution{
					Width: 1080,
					Height: 720,
				}
				dRe2 := vo.Resolution{
					Width: 1920,
					Height: 1080,
				}
				dRe3 := vo.Resolution{
					Width: 2688,
					Height: 1520,
				}
				resolutions = []vo.Resolution{ dRe0, dRe1, dRe2, dRe3}
			}

			for _,sck := range arr {
				if !crApi.ExistRunningTask(sck.Id) {
					sck.Status = bhomeclient.Stack_Status_NoRule
				}
				v := vo.FileStackWithWH{}
				v.FileStack = sck
				v.Resolutions = resolutions
				pageList = append(pageList, v)
			}
		}
		result.DataList = pageList
	} else {
		result.DataList = []vo.FileStackWithWH{}
	}
	return result
}

func (sv FileStackService) Save(fsE *models.FileStack) bool {
	if fsE.Id == "" {
        fsE.Id = bhomeclient.Id_Stack_Pre + uuid.NewV4().String()
		if fsE.Add() {
			dbMsg := protomsg.DbChangeMessage{
				Id: fsE.Id,
				Table: protomsg.TableChanged_T_FileStack,
				Action: protomsg.DbAction_Insert,
			}
			pb,_ := json.Marshal(dbMsg)
			sv.Bk.Publish(ProcName, pb)
			return true
		}
	} else {
		if fsE.Update() {
			dbMsg := protomsg.DbChangeMessage{
				Id: fsE.Id,
				Table: protomsg.TableChanged_T_FileStack,
				Action: protomsg.DbAction_Update,
			}
			pb,_ := json.Marshal(dbMsg)
			sv.Bk.Publish(ProcName, pb)
			return true
		}
	}
	return false
}

func (sv FileStackService) UpdateStatus(ids []string, status int) bool {
	var sck models.FileStack
	ret := false
	if status == bhomeclient.Stack_Status_Doing {
		ret = sck.UpdateStatus(ids, status)
	} else if status == bhomeclient.Stack_Status_Done {
		for _,sckId := range ids {
			if sck.HavingWorkFiles(sckId) {
				sck.UpdateStatus([]string{ sckId }, bhomeclient.Stack_Status_Doing)
			} else {
				sck.UpdateStatus([]string{ sckId }, bhomeclient.Stack_Status_Done)
			}
		}
		ret = true
	}
	if ret {
		dbMsg := protomsg.DbChangeMessage{
			Id: strings.Join(ids, ","),
			Table: protomsg.TableChanged_T_FileStack,
			Action: protomsg.DbAction_Update,
		}
		pb,_ := json.Marshal(dbMsg)
		sv.Bk.Publish(ProcName, pb)
	}
	return ret
}

func (sv FileStackService) UpdateEnable(id string, enable bool) bool {
	var fs models.FileStack
	if fs.UpdateEnable(id, enable) {
		dbMsg := protomsg.DbChangeMessage{
			Table: protomsg.TableChanged_T_FileStack,
			Action: protomsg.DbAction_Update,
		}
		pb,_ := json.Marshal(dbMsg)
		sv.Bk.Publish(ProcName, pb)
		return true
	}
	return false
}

func (sv FileStackService) UpdateVideoEnable(enable bool) bool {
	var fset models.FileAnalysisSetting
	if fset.UpdateVideoEnable(enable) {
		dbMsg := protomsg.DbChangeMessage{
			Table: protomsg.TableChanged_T_FileSetting,
			Action: protomsg.DbAction_Update,
		}
		pb,_ := json.Marshal(dbMsg)
		sv.Bk.Publish(ProcName, pb)
		return true
	}
	return false
}

func (sv FileStackService) DeleteById(id string) bool {
	var fs models.FileStack
	if fs.DeleteById(id) {
		dbMsg := protomsg.DbChangeMessage{
			Table: protomsg.TableChanged_T_FileStack,
			Action: protomsg.DbAction_Delete,
		}
		pb,_ := json.Marshal(dbMsg)
		sv.Bk.Publish(ProcName, pb)
		return true
	}
	return false
}

func (sv FileStackService) UpdateChannelCount(count int) bool {
	var fas models.FileAnalysisSetting
	if fas.UpdateChannelCount(count)  {
		dbMsg := protomsg.DbChangeMessage{
			Table: protomsg.TableChanged_T_FileSetting,
			Action: protomsg.DbAction_Update,
		}
		pb,_ := json.Marshal(dbMsg)
		sv.Bk.Publish(ProcName, pb)
		return true
	}
	return false
}