package dbapi

import (
	"basic.com/pubsub/protomsg.git"
	"encoding/json"
	"fmt"
)

type TaskApi struct {}

//获取所有任务信息(包含算法定义)
func (api TaskApi) FindAll() (taskInfos []protomsg.TaskSdkInfo) {
	url := BASIC_URL + DATA_URL_PREFIX + "/task/findAll"
	client := NewClient()

	respBody, err := client.DoGetRequest(url, nil, nil)
	if err !=nil {
		return taskInfos
	}
	var res Result
	if err = json.Unmarshal(respBody, &res); err != nil {
		fmt.Println(err)
		return taskInfos
	}
	bytes, _ := json.Marshal(res.Data)
	err = json.Unmarshal(bytes, &taskInfos)
	return taskInfos
}

func (api TaskApi) FindAllTaskSdkRun() (taskSdks []protomsg.TaskSdkRun) {
	taskInfos := api.FindAll()
	if taskInfos !=nil {
		for _,ti :=range taskInfos{
			tsr := protomsg.TaskSdkRun{
				Task:ti.Task,
			}
			m := make(map[string]string,0)
			for _,s := range ti.Sdks{
				if _,ok :=m[s.IpcId];!ok{
					m[s.IpcId] = s.IpcId
					tsr.Sdks = append(tsr.Sdks,&protomsg.SdkRun{
						IpcId:s.IpcId,
						SdkType:s.SdkType,
					})
				}
			}

			taskSdks = append(taskSdks,tsr)
		}
	}
	return taskSdks
}

func (api TaskApi) Add(paramBody map[string]interface{}) (bool,interface{}) {
	url := BASIC_URL + DATA_URL_PREFIX + "/task/add"
	client := NewClient()
	respBody, err := client.DoPostRequest(url, CONTENT_TYPE_JSON, paramBody, nil, nil)
	if err != nil {
		return false,nil
	}
	var res Result
	if err = json.Unmarshal(respBody, &res); err != nil {
		fmt.Println(err)
		return false,nil
	}
	return res.Success,res.Data
}

func (api TaskApi) Delete(taskId string) (bool,interface{}){
	url := BASIC_URL + DATA_URL_PREFIX + "/task/delete"
	client := NewClient()
	paramQuery :=make(map[string]string,0)
	paramQuery["taskId"] = taskId
	respBody, err := client.DoGetRequest(url, paramQuery, nil)
	if err != nil {
		return false,nil
	}
	var res Result
	if err = json.Unmarshal(respBody, &res); err != nil {
		return false,nil
	}
	return res.Success,res.Data
}

func (api TaskApi) AddTaskSdk(taskId string,sdkId string) (bool,interface{}) {
	url := BASIC_URL + DATA_URL_PREFIX + "/task/addTaskSdk"
	client := NewClient()
	paramQuery :=make(map[string]string,0)
	paramQuery["taskId"] = taskId
	paramQuery["sdkId"] = sdkId
	respBody, err := client.DoGetRequest(url, paramQuery, nil)
	if err != nil {
		return false,nil
	}
	var res Result
	if err = json.Unmarshal(respBody, &res); err != nil {
		return false,nil
	}
	return res.Success,res.Data
}

func (api TaskApi) UpdateTaskStatus(paramBody map[string]string) (bool,interface{}) {
	url := BASIC_URL + DATA_URL_PREFIX + "/task/addTaskSdk"
	client := NewClient()
	respBody, err := client.DoPostRequest(url, CONTENT_TYPE_FORM, nil,paramBody, nil)
	if err != nil {
		return false,nil
	}
	var res Result
	if err = json.Unmarshal(respBody, &res); err != nil {
		return false,nil
	}
	return res.Success,res.Data
}

func (api TaskApi) DeleteTaskSdk(taskId string,sdkId string) (bool,interface{}) {
	url := BASIC_URL + DATA_URL_PREFIX + "/task/delTaskSdk"
	client := NewClient()
	paramQuery :=make(map[string]string,0)
	paramQuery["taskId"] = taskId
	paramQuery["sdkId"] = sdkId
	respBody, err := client.DoGetRequest(url, paramQuery, nil)
	if err != nil {
		return false,nil
	}
	var res Result
	if err = json.Unmarshal(respBody, &res); err != nil {
		return false,nil
	}
	return res.Success,res.Data
}