package controllers

import (
	"basic.com/valib/bhomeclient.git"
	"basic.com/valib/logger.git"
	"encoding/json"
	"github.com/satori/go.uuid"
	"strconv"
	"time"
	"vamicro/stack-service/models"
	"vamicro/stack-service/service"
	"vamicro/stack-service/vo"
)

type FileAnalysisController struct {

}

// @Summary 获取本地文件列表
// @Description 获取本地文件列表
// @Produce json
// @Tags 本地文件
// @Param stackId query string true "数据栈id"
// @Param fileName 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/fileAnalysis/findAllFile [get]
func (fac FileAnalysisController) FindAllFile(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	fileName := c.Query("fileName")
	stackId := c.Query("stackId")
	page,_ := strconv.Atoi(c.Query("page"))
	size,_ := strconv.Atoi(c.Query("size"))
	fType,_ := strconv.Atoi(c.Query("type"))
	logger.Debug("FindAllFile fileName:",fileName,"page:",page,"size:",size,"fType:",fType,"stackId:",stackId)
	if page < 0 {
		page = 1
	}
	if size <= 0 {
		size = 20
	}
	sv := service.FileAnalysisService{Bk: h.Bk}
	result := sv.FindAllByPage(fileName, fType, page, size, stackId)

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

// @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/fileAnalysis/show [get]
func (fac FileAnalysisController) Show(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	id := c.Query("id")
	if id == "" {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	var file models.FileAnalysis
	rows, _ := file.SelectById(id)
	if rows >0 {
		return &bhomeclient.Reply{Success:true, Data:file}
	} else {
		return &bhomeclient.Reply{Msg:"记录不存在"}
	}
}

// @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/fileAnalysis/getAnalysisFiles [get]
func (fac FileAnalysisController) GetAnalysisFiles(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	sv := service.FileAnalysisService{Bk: h.Bk}
	arr, _ := sv.GetAnalysisFiles()
	if arr !=nil && len(arr) >0 {
		return &bhomeclient.Reply{Success:true, Data:arr}
	} else {
		return &bhomeclient.Reply{Success:true, Data:[]models.FileAnalysis{}}
	}
}

type FileProgressVo struct {
	Ids []string `json:"ids" binding:"required"`
	Progress int `json:"progress" binding:"required"`
}

// @Summary 更新文件分析进度
// @Description 更新文件分析进度
// @Accept json
// @Produce json
// @Tags 本地文件
// @Param reqBody body controllers.FileProgressVo true "更新进度参数,progress为1-100"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileAnalysis/updateProgress [post]
func (fac FileAnalysisController) UpdateProgress(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	var reqBody FileProgressVo
	err := c.BindJSON(&reqBody)
	logger.Debug("updateProgress reqBody:",reqBody,"bindJson err:",err)
	if err !=nil {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}

	sv := service.FileAnalysisService{Bk: h.Bk}
	if sv.UpdateProgress(reqBody.Ids, reqBody.Progress) {
		if reqBody.Progress == bhomeclient.File_Status_Complete {
			type donePub struct {
				Topic string
				Ids   []string
			}
			param := donePub{
				Topic: "FileDecodeDone",
				Ids: reqBody.Ids,
			}
			data,_ := json.Marshal(param)
			h.Bk.Publish(service.ProcName, data)
		}

		return &bhomeclient.Reply{Success:true, Msg:"更新成功"}
	} else {
		return &bhomeclient.Reply{Msg:"更新失败"}
	}
}

type FileStatusVo struct {
	Ids []string `json:"ids" binding:"required"`
	Status int `json:"status"`
}

// @Summary 开启或暂停文件分析
// @Description 开启或暂停文件分析
// @Accept json
// @Produce json
// @Tags 本地文件
// @Param reqBody body controllers.FileStatusVo true "开启暂停参数,暂停status=0，开启status=1"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileAnalysis/updateStatus [post]
func (fac FileAnalysisController) 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.FileAnalysisService{Bk: h.Bk}
	if sv.UpdateStatus(reqBody.Ids, reqBody.Status) {
		return &bhomeclient.Reply{Success:true, Msg:"更新成功"}
	} else {
		return &bhomeclient.Reply{Msg:"更新失败"}
	}
}


// @Summary 删除本地文件
// @Description 删除本地文件
// @Accept json
// @Produce json
// @Tags 本地文件
// @Param reqBody body vo.IdArrVo 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/fileAnalysis/delete [post]
func (fac FileAnalysisController) Delete(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	var reqBody vo.IdArrVo
	err := c.BindJSON(&reqBody)
	if err !=nil {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}

	sv := service.FileAnalysisService{Bk: h.Bk}
	if sv.Delete(&reqBody) {
		return &bhomeclient.Reply{Success:true, Msg:"删除成功"}
	} else {
		return &bhomeclient.Reply{Msg:"删除失败"}
	}
}

// @Summary 本地文件排序
// @Description 本地文件排序
// @Accept json
// @Produce json
// @Tags 本地文件
// @Param reqBody body vo.SortVo true "排序参数,向上direct=1，向下direct=2"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileAnalysis/sortFile [post]
func (fac FileAnalysisController) SortFile(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	var reqBody vo.SortVo
	err := c.BindJSON(&reqBody)
	if err !=nil || (reqBody.Direct !=service.Sort_Up && reqBody.Direct !=service.Sort_Down) {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}

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

// @Summary 新增文件
// @Description 新增文件
// @Accept json
// @Produce json
// @Tags 本地文件
// @Param reqBody body models.FileAnalysis true "文件新增json"
// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
// @Router /data/api-v/fileAnalysis/save [post]
func (fac FileAnalysisController) Save(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	var reqBody models.FileAnalysis
	err := c.BindJSON(&reqBody)
	if err !=nil || reqBody.StackId == "" {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	logger.Debug("file add reqBody:", reqBody)
    if reqBody.Id == "" {
        reqBody.Id = uuid.NewV4().String()
    }

	reqBody.CreateTime = time.Now().Format("2006-01-02 15:04:05")
	var tmpMax models.FileAnalysis
	rows, err := tmpMax.GetMaxSortFile(reqBody.StackId)
	if err == nil && rows >0 {
		reqBody.Sort = tmpMax.Sort +1
	}
	sv := service.FileAnalysisService{Bk: h.Bk}
	if sv.Add(reqBody) {
		return &bhomeclient.Reply{Success:true, Msg:"保存成功"}
	} else {
		return &bhomeclient.Reply{Msg:"保存失败"}
	}
}

// @Param stackId query string true "stackId"
// @Param type query int true "type"
// @Param name query string false "搜索条件"
// @Param page query int true "当前页"
// @Param size query int true "每页数量"
func (fac FileAnalysisController) FindByStackId(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	stackId := c.Query("stackId")
	if stackId == "" {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	typ,_ := strconv.Atoi(c.Query("type"))
	page,_ := strconv.Atoi(c.Query("page"))
	size,_ := strconv.Atoi(c.Query("size"))
	name := c.Query("name")
	if page <= 0 {
		page = 1
	}
	if size <= 0 {
		size = 20
	}
	sv := service.FileAnalysisService{Bk: h.Bk}
	result := sv.FindByStackId(stackId, typ, name, page, size)
	return &bhomeclient.Reply{Success:true, Data: result}
}

// @Summary 重命名
// @Description 重命名
// @Accept json
// @Produce json
// @Tags 本地文件
// @Param id formData string true "文件id"
// @Param name formData string 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/fileAnalysis/rename [post]
func (fac FileAnalysisController) Rename(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	id := c.PostForm("id")
	name := c.PostForm("name")
	if id == "" || name == "" {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	sv := service.FileAnalysisService{Bk: h.Bk}
	if sv.Rename(id, name) {
		return &bhomeclient.Reply{Success: true, Msg:"修改成功"}
	} else {
		return &bhomeclient.Reply{Msg:"修改失败"}
	}
}

// @Summary 移动
// @Description 移动
// @Accept json
// @Produce json
// @Tags 本地文件
// @Param id formData string true "文件id"
// @Param stackId formData 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/fileAnalysis/move [post]
func (fac FileAnalysisController) Move(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	id := c.PostForm("id")
	stackId := c.PostForm("stackId")
	if id == "" || stackId == "" {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}
	var tmp models.FileAnalysis
	rows, err := tmp.SelectById(id)
	if rows > 0 && err == nil {
		sv := service.FileAnalysisService{Bk: h.Bk}
		if sv.Move(id, stackId) {
			return &bhomeclient.Reply{Success:true, Msg:"移动成功"}
		}
	}
	return &bhomeclient.Reply{Msg:"移动失败"}
}

// @Summary 复制
// @Description 复制
// @Accept json
// @Produce json
// @Tags 本地文件
// @Param reqBody body vo.FileMoveVo 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/fileAnalysis/copy [post]
func (fac FileAnalysisController) Copy(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply{
	var reqBody vo.FileMoveVo
	err := c.BindJSON(&reqBody)
	if err != nil {
		return &bhomeclient.Reply{Msg:"参数有误"}
	}

	var tmp models.FileAnalysis
	rows, err := tmp.SelectById(reqBody.Id)
	if rows > 0 && err == nil {
		sv := service.FileAnalysisService{Bk: h.Bk}
		for _,sckId := range reqBody.StackIds {
			if sckId != tmp.StackId {
				tmp.Id = uuid.NewV4().String()
				tmp.StackId = sckId
				sv.Add(tmp)
			}
		}

		return &bhomeclient.Reply{Success: true, Msg:"拷贝成功"}
	}
	return &bhomeclient.Reply{Msg:"拷贝失败"}
}