package models

import (
	"basic.com/valib/bhomeclient.git"
	"strconv"
	"strings"
)

//本地文件处理
type FileAnalysis struct {
	Id 					string 		`gorm:"primary_key;column:id" json:"id"`
	Name 				string 		`gorm:"column:name" json:"name"`
	Path 				string 		`gorm:"column:path" json:"path"`
	Type 				int 		`gorm:"column:type;default:1" json:"type"`//类型：0：本地图片，1：本地视频，2：本地语音
	Sort 				int			`gorm:"column:sort;default:1" json:"sort"`//排序
	Alias     			string 	   	`gorm:"column:alias" json:"alias"` //摄像机的别名
	SnapshotUrl 		string     	`gorm:"column:snapshot_url" json:"snapshot_url"`//快照地址
	Duration 			string 		`gorm:"column:duration" json:"duration"`//时长
	IsRunning 			bool 		`gorm:"column:is_running" json:"is_running"`//是否正在解码
	Progress  			int 		`gorm:"column:progress;default:0" json:"progress"`//进度
	RuleType 			int 		`gorm:"column:ruleType;default:0" json:"ruleType"`//规则类型(0:独有规则，1：公共规则)
	Status      		int 		`gorm:"column:status;default:1" json:"status"`//状态(-1:已删除,0:暂停，1：等待 2：处理中  9:已完成)
	CreateTime 			string 		`gorm:"column:createTime" json:"createTime"`
	UpdateTime 			string 		`gorm:"column:updateTime" json:"updateTime"`
	Identifier			string 		`gorm:"column:identifier" json:"identifier"` 	//整个文件的唯一标识，md5
	StackId   			string   	`gorm:"column:stack_id" json:"stack_id"` //所属数据栈的id
	Size                int64       `gorm:"column:size" json:"size"`
}

func (FileAnalysis) TableName() string {
	return "file_analysis"
}

func IsStack(id string) bool {
	if strings.HasPrefix(id, bhomeclient.Id_Stack_Pre) {
		return true
	}
	return false
}

func (fa *FileAnalysis) ExistByPath(path string) (flag bool,file FileAnalysis) {
	foundResult := db.Table(fa.TableName()).Where("path=?", path).First(&file)
	if foundResult.Error !=nil {
		return false,file
	}
	if foundResult.RecordNotFound() {
		return false,file
	} else {
		return true,file
	}
}

func (fa *FileAnalysis) SelectById(id string) (rows int64, err error) {
	result := db.Table(fa.TableName()).Where("id = ?", id).First(&fa)
	if result.Error != nil || result.RowsAffected == 0 {
		return 0, err
	}
	return result.RowsAffected, nil
}

func (fa *FileAnalysis) FindByName(name string) (list []FileAnalysis, err error) {

	return list,err
}

func (fa *FileAnalysis) FindBySort(sort int) (rows int64, err error) {
	result := db.Table(fa.TableName()).Where("sort = ?", sort).First(&fa)
	if result.Error != nil || result.RowsAffected == 0 {
		return 0, err
	}
	return result.RowsAffected, nil
}

func (fa *FileAnalysis) GetMaxSortFile(stackId string) (rows int64, err error) {
	result := db.Table(fa.TableName()).Where("stack_id=?", stackId).Order("sort desc").First(&fa)
	if result.Error != nil || result.RowsAffected == 0 {
		return 0, err
	}
	return result.RowsAffected, nil
}

func (fa *FileAnalysis) Add() bool {

	result := db.Table(fa.TableName()).Create(&fa)
	if result.Error ==nil && result.RowsAffected >0 {
		return true
	}
	return false
}

func (fa *FileAnalysis) Total(fileName string, fType int, stackId string) int {
	var total int
	sql := "select count(1) from file_analysis f where f.status !=-1"
	if stackId != "" {
		sql += " and f.stack_id='"+stackId+"'"
	}
	if fileName !="" {
		sql += " and f.name like '%"+fileName+"%'"
	}
	if fType == 0 { //全部视频

	} else if fType == 1 { //等待处理
		sql += " and f.status=" + strconv.Itoa(bhomeclient.File_Status_Wait)
	} else if fType == 2 { //处理中
		sql += " and f.status=" + strconv.Itoa(bhomeclient.File_Status_Doing)
	} else if fType == 3 {
		sql += " and f.status=" + strconv.Itoa(bhomeclient.File_Status_Complete)
	} else if fType == 4 {
		sql += " and f.status=-2"
	}

	err := db.Raw(sql).Count(&total).Error
	if err !=nil {
		total = 0
	}
	return total
}

func (fa *FileAnalysis) FindAllByPage(fileName string, fType int, page int, size int, stackId string) (arr []FileAnalysis, err error) {
	sql := "select * from file_analysis f where f.status !=-1"
	if stackId != "" {
		sql += " and f.stack_id='"+stackId+"'"
	}
	if fileName !="" {
		sql += " and f.name like '%"+fileName+"%'"
	}
	if fType == 0 { //全部视频

	} else if fType == 1 { //处理完成
		sql += " and f.status=9"
	} else if fType == 2 { //处理中
		sql += " and f.status=3"
	} else if fType == 3 {
		sql += " and f.status=-2"
	}
	offset := (page-1)*size
	err = db.Raw(sql).Offset(offset).Limit(size).Order("sort asc").Find(&arr).Error
	return arr,err
}

func (fa *FileAnalysis) FindAll(stackId string, fileName string) (arr []FileAnalysis, err error) {
	sql := "select * from file_analysis where status !=-1"
	if fileName !="" {
		sql += " and name like '%"+fileName+"%'"
	}
	if stackId != "" {
		sql += " and stack_id='"+stackId+"'"
	}

	err = db.Raw(sql).Order("sort asc").Find(&arr).Error
	return arr,err
}

func (fa *FileAnalysis) FindByStackId(stackId string, fType int, name string, page int, size int) (arr []FileAnalysis, err error) {
	sql := "select * from file_analysis f where f.stack_id='"+stackId+"' and f.status !=-1"
	if name != "" {
		sql += " and f.name like '%"+name+"%'"
	}
	if fType == 0 { //全部视频

	} else if fType == 1 { //待处理
		sql += " and f.status="+strconv.Itoa(bhomeclient.File_Status_Wait)
	} else if fType == 2 { //处理中
		sql += " and f.status="+strconv.Itoa(bhomeclient.File_Status_Doing)
	} else if fType == 3 { //处理完成
		sql += " and f.status="+strconv.Itoa(bhomeclient.File_Status_Complete)
	} else if fType == 4 {//处理失败
		sql += " and f.status=-2"
	}
	offset := (page-1)*size
	err = db.Raw(sql).Offset(offset).Limit(size).Order("sort asc").Find(&arr).Error

	return arr,err
}

func (fa *FileAnalysis) FindAnalysisFiles() (arr []FileAnalysis, err error) {
	sql := "select * from file_analysis where status =1"
	err = db.Raw(sql).Order("sort asc").Find(&arr).Error
	return arr,err
}

func (fa *FileAnalysis) UpdateStatus(id string,status int) bool {
	result := db.Exec("update file_analysis set status=? where id=?",status,id)
	if result.Error ==nil && result.RowsAffected >0 {
		return true
	}
	return false
}

func (fa *FileAnalysis) UpdateIsRunning(id string, isRunning bool) bool {
	isRunningStr := "0"
	if isRunning {
		isRunningStr = "1"
	}
	result := db.Exec("update file_analysis set is_running="+isRunningStr+" where id='"+id+"'")
	if result.Error !=nil {
		return false
	}
	return result.RowsAffected>0
}

func (fa *FileAnalysis) UpdateProgress(ids []string,progress int) bool {
	idStr := ""
	for _,id := range ids {
		idStr += "'"+id+"',"
	}
	idStr = strings.Trim(idStr, ",")
	if idStr != "" {
		result := db.Exec("update file_analysis set status=? where id in ("+idStr+")",progress)
		if result.Error ==nil && result.RowsAffected >0 {
			return true
		}
	}

	return false
}

func (fa *FileAnalysis) Rename(id string, name string) bool {
	result := db.Exec("update file_analysis set name=? where id=?",name, id)
	if result.Error ==nil && result.RowsAffected >0 {
		return true
	}
	return false
}


func (fa *FileAnalysis) Move(id string, stackId string) bool {
	result := db.Exec("update file_analysis set stack_id=? where id=?",stackId, id)
	if result.Error ==nil && result.RowsAffected >0 {
		return true
	}
	return false
}