package controllers

import (
	"basic.com/valib/bhomeclient.git"
	"basic.com/valib/logger.git"
	"os"
	"path"
	"strconv"
	"strings"
	"vamicro/config"
	"vamicro/extend/util"
	"vamicro/stack-service/models"
	"vamicro/stack-service/service"
	"vamicro/stack-service/vo"
)

type FileStackController struct {

}


func (fsc FileStackController) FindAllDoingStacks(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
	//1.需要开启分析开关
	//2.需要配置有规则
	var model models.FileStack
	list,err := model.FindAll()
	if err ==nil {
		var resultList = make([]models.FileStack, 0)
		if list != nil {
			//var ctE models.CameraTask
			for _,fsE := range list {
				//if fsE.Enable && ctE.ExistRunningTask(fsE.Id) {
				if fsE.Enable {
					resultList = append(resultList, fsE)
				}
			}
		}
		return &bhomeclient.Reply{Success:true, Data:resultList}
	} else {
		return &bhomeclient.Reply{Msg:"查询失败"}
	}
}

// @Summary 获取数据栈列表
// @Description 获取数据栈列表
// @Produce json
// @Tags 数据栈
// @Param name query string false "根据数据栈名称查询"
// @Param type query int true "0:全部视频，1:处理完成，2:处理中，3:未配规则，4:未开启"
// @Param page query int true "当前页"
// @Param size query int true "每页数量"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileStack/findAllByPage [get]
func (fsc FileStackController) FindAllByPage(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	name := c.Query("name")
	page,_ := strconv.Atoi(c.Query("page"))
	size,_ := strconv.Atoi(c.Query("size"))
	fType,_ := strconv.Atoi(c.Query("type"))
	logger.Debug("FindAllFile fileName:",name,"page:",page,"size:",size,"fType:",fType)
	if page <= 0 {
		page = 1
	}
	if size <= 0 {
		size = 20
	}
	sv := service.FileStackService{Bk: h.Bk}
	result := sv.FindAllByPage(fType, name, page, size)

	return &bhomeclient.Reply{Success:true, Data:result}
}

// @Summary 获取所有数据栈列表
// @Description 获取所有数据栈列表
// @Produce json
// @Tags 数据栈
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileStack/findAll [get]
func (fsc FileStackController) FindAll(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{

	sv := service.FileStackService{Bk: h.Bk}
	result := sv.FindAll()

	return &bhomeclient.Reply{Success:true, Data:result}
}

// @Summary 新增或者更新
// @Description 新增或者更新，更新时id不能为空
// @Accept json
// @Produce json
// @Tags 数据栈
// @Param reqBody body models.FileStack true "数据栈结构体"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileStack/save [post]
func (fsc FileStackController) Save(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	var reqBody models.FileStack
	if err := c.BindJSON(&reqBody);err !=nil {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	sv := service.FileStackService{Bk: h.Bk}
	if sv.Save(&reqBody) {
		return &bhomeclient.Reply{Success:true, Msg:"保存成功"}
	} else {
		return &bhomeclient.Reply{Msg:"保存失败"}
	}
}

// @Summary 查看数据栈详情
// @Description 查看数据栈详情
// @Produce json
// @Tags 数据栈
// @Param id path string true "数据栈id"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileStack/show [get]
func (fsc FileStackController) Show(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	id := c.Query("id")
	if id == "" {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	var fsE models.FileStack
	rows, _ := fsE.SelectById(id)
	if rows > 0 {
		m := util.Struct2Map(fsE)

		if config.Server.Resolutions != nil && len(config.Server.Resolutions) > 0 {
			var arr []vo.Resolution
			for _,rwh := range config.Server.Resolutions {
				arr = append(arr, vo.Resolution{
					Width: rwh.Width,
					Height: rwh.Height,
				})
			}
			m["resolutions"] = arr
		} 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,
			}
			m["resolutions"] = []vo.Resolution{ dRe0, dRe1, dRe2, dRe3 }
		}
		return &bhomeclient.Reply{ Success:true, Data:m }
	} else {
		return &bhomeclient.Reply{Msg:"查询失败"}
	}
}

// @Summary 删除
// @Description 删除
// @Produce json
// @Tags 数据栈
// @Param id path string true "删除数据栈id"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileStack/delete [delete]
func (fsc FileStackController) Delete(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	id := c.Query("id")
	if id == "" {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	sv := service.FileStackService{Bk: h.Bk}
	var fa models.FileAnalysis
	allFiles, _ := fa.FindAll(id, "")
	if sv.DeleteById(id) {
		go deleteDiskFiles(allFiles)
		return &bhomeclient.Reply{Success:true, Msg:"删除成功"}
	} else {
		return &bhomeclient.Reply{Msg:"删除失败"}
	}
}

func deleteDiskFiles(files []models.FileAnalysis) {
	for _,f := range files {
		//如果是文件，则删除。是文件夹则跳过
		if util.DirExists(f.Path) {
			if util.Exists(f.Path) {
				os.Remove(f.Path)
			}
			ext := path.Ext(f.Path)
			md5Path := strings.Replace(f.Path, ext, "",-1)
			if util.Exists(md5Path) {
				os.RemoveAll(md5Path)
			}
		}
	}
}

// @Summary 切换enable
// @Description 切换enable
// @Accept json
// @Produce json
// @Tags 数据栈
// @Param reqBody body vo.EnableVo true "切换enable"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileStack/changeEnable [post]
func (fsc FileStackController) ChangeEnable(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	var reqBody vo.EnableVo
	c.BindJSON(&reqBody)
	if reqBody.Id == "" {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	sv := service.FileStackService{Bk: h.Bk}
	if sv.UpdateEnable(reqBody.Id, reqBody.Enable) {
		return &bhomeclient.Reply{Success:true, Msg:"更新成功"}
	} else {
		return &bhomeclient.Reply{Msg:"更新失败"}
	}
}

// @Summary 更新数据栈状态
// @Description 更新数据栈状态
// @Accept json
// @Produce json
// @Tags 数据栈
// @Param reqBody body controllers.FileStatusVo true "数据栈状态参数"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileStack/updateStatus [post]
func (fsc FileStackController) UpdateStatus(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	var reqBody FileStatusVo
	err := c.BindJSON(&reqBody)
	if err !=nil {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}

	sv := service.FileStackService{Bk: h.Bk}
	if sv.UpdateStatus(reqBody.Ids, reqBody.Status) {
		return &bhomeclient.Reply{Success:true, Msg:"更新成功"}
	} else {
		return &bhomeclient.Reply{Msg:"更新失败"}
	}
}