package controllers

import (
	"basic.com/dbapi.git"
	"webserver/extend/code"
	"webserver/extend/util"
	"github.com/gin-gonic/gin"
)

type SdkController struct {
}

type SdkVo struct {
	Id 		string `json:"id"`
	SdkType string `json:"sdk_type"`//人脸检测：FaceDetect,人脸提取：FaceExtract,人脸比对：FaceCompare,行为：Yolo
	SdkName string `json:"sdk_name"`    //算法名称
	Args    []SdkArgVo `json:"args"` //算法参数
	Icon    string `json:"icon"`       //算法图标
	Url     string `json:"url"`                       //算法下载地址
	CreateTime string `json:"create_time"`
	CreateBy string `json:"create_by"`
	UpdateTime string `json:"update_time"`
	Enable bool `json:"enable"`//是否启用
	DelFlag bool `json:"del_flag"`//逻辑删除
}

//算法参数定义
type SdkArgVo struct {
	Alias   string `json:"alias"`   //参数的别名
	Name  string `json:"name"`  //参数名称
	Type  string `json:"type"`  //参数类型(整数，字符串或数组)
	Must  bool   `json:"must"`  //是否必填
	Range string `json:"range"` //值的范围，eg：0,100表示从0到100
	Sort  int    `json:"sort"`  //参数顺序
}

// @Summary 算法保存
// @Description 算法保存
// @Produce json
// @Tags sdk
// @Param reqMap body controllers.SdkVo true "人脸检测"
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"",success:true}"
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
// @Router /data/api-v/sdk/save [POST]
func (sc SdkController) Save(c *gin.Context) {
	var sdk SdkVo
	var api dbapi.SdkApi

	if err := c.BindJSON(&sdk);err !=nil {
		util.ResponseFormat(c,code.RequestParamError,"参数错误")
		return
	}
	paramBody := util.Struct2Map(sdk)
	flag, data := api.Save(paramBody)
	if flag{
		util.ResponseFormat(c, code.Success, data)
	} else {
		util.ResponseFormat(c, code.ComError, data)
	}
}

// @Summary 查找所有算法
// @Description 查找所有算法
// @Produce json
// @Tags sdk
// @Param sdkName query string false "可选参数"
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"算法列表",success:true}"
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"[]",success:false}"
// @Router /data/api-v/sdk/findAllSdk [GET]
func (sc SdkController) FindAllSdk(c *gin.Context) {
	var api dbapi.SdkApi

	sdkName := c.Query("sdkName")
	sdks := api.FindAll(sdkName)
	util.ResponseFormat(c, code.Success, sdks)
}

// @Summary 根据id获取算法信息
// @Description 根据id获取算法信息
// @Produce json
// @Tags sdk
// @Param id query string true "算法id，必填"
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"算法信息",success:true}"
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
// @Router /data/api-v/sdk/getById [GET]
func (sc SdkController) GetById(c *gin.Context) {
	var api dbapi.SdkApi
	sdkId := c.Query("id")
	if sdkId == ""{
		util.ResponseFormat(c,code.RequestParamError,"sdkId不能为空")
		return
	}

	flag, sdk := api.GetById(sdkId)

	if flag {
		util.ResponseFormat(c,code.Success,sdk)
	} else {
		util.ResponseFormat(c,code.ComError,sdk)
	}
}

// @Summary 根据taskId获取算法信息
// @Description 根据taskId获取算法信息
// @Produce json
// @Tags sdk
// @Param taskId query string true "taskId，必填"
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"算法信息",success:true}"
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
// @Router /data/api-v/sdk/findByTaskId [GET]
func (sc SdkController) FindByTaskId(c *gin.Context) {
	var api dbapi.SdkApi
	taskId := c.Query("taskId")
	if taskId == "" {
		util.ResponseFormat(c,code.ComError,"任务id不能为空")
		return
	}
	flag,sdks := api.FindByTaskId(taskId)
	if flag {
		util.ResponseFormat(c,code.Success,sdks)
	} else {
		util.ResponseFormat(c,code.ComError,sdks)
	}
}