v1payload

type CreateDatasetInput struct {
	ProjectID string `json:"project_id" valid:"required"`
}

type CreateDatasetOutput struct {
	DatasetSummary
	HeartbeatInterval time.Duration `json:"heartbeat_interval"`
}

type CreateTokenInput struct {
	ProjectID string `json:"project_id" valid:"required"`
}

type CreateTokenOutput struct {
	AWSRegion          string    `json:"aws_region"`
	AWSAccessKeyID     string    `json:"aws_access_key_id"`
	AWSExpiration      time.Time `json:"aws_expiration"`
	AWSSecretAccessKey string    `json:"aws_secret_access_key"`
	AWSSessionToken    string    `json:"aws_session_token"`
}

type CreateWorkloadInput struct {
	ProjectID      string            `json:"project_id" valid:"required"`
	Image          string            `json:"image" valid:"required"`
	NrOfWorkers    int               `json:"nr_of_workers" valid:"required"`
	InputDatasetID string            `json:"input_dataset_id"`
	UseCuteur      bool              `json:"use_cuteur"`
	Env            map[string]string `json:"env"`
}

type CreateWorkloadOutput struct {
	WorkloadSummary
}

type DatasetSummary struct {
	ProjectID    string `json:"project_id"`
	DatasetID    string `json:"dataset_id"`
	Bucket       string `json:"bucket"`
	DatasetRoot  string `json:"dataset_root"`
	ProjectRoot  string `json:"project_root"`
	UploadExpire int64  `json:"upload_expire"`
	UploadStatus string `json:"upload_status"`
	CreatedAt    int64  `json:"created_at"`
}

type DescribeDatasetInput struct {
	ProjectID string `json:"project_id" valid:"required"`
	DatasetID string `json:"dataset_id" valid:"required"`
}

type DescribeDatasetOutput struct {
	DatasetSummary
}

type DescribeTaskInput struct {
	ProjectID  string `json:"project_id" valid:"required"`
	WorkloadID string `json:"workload_id" valid:"required"`
	TaskID     int64  `json:"task_id" valid:"required"`
}

type DescribeTaskOutput struct {
	TaskSummary
	ExecutionARN   string `json:"execution_arn"`
	NumDispatches  int64  `json:"num_dispatches"`
	Result         string `json:"result,omitempty"`
	LastErrCode    string `json:"last_err_code,omitempty"`
	LastErrMessage string `json:"last_err_message,omitempty"`
}

type DescribeWorkloadInput struct {
	ProjectID  string `json:"project_id" valid:"required"`
	WorkloadID string `json:"workload_id"`
}

type DescribeWorkloadOutput struct {
	WorkloadSummary
	TaskCount map[TaskStatus]int `json:"task_count"`
	Env       map[string]string  `json:"env"`
}

type Error struct {

	//Retry indicates if the client can retry the request as is, this is mostly false on validation/encoding errors and true in other cases
	Retry bool `json:"retry"`

	//Message contains a overall message to the user, it should always be set to provide some feedback
	Message string `json:"message"`

	//Trace is set if the server is running in development mode, if it is empty it can be ignored
	Trace []string `json:"trace,omitempty"`

	//Fields can hold validation messages for individual fields, if empty the cause of the overal error is not due to specific field's input
	Fields map[string]string `json:"fields,omitempty"`
}

type ExpelProjectInput struct {
	ProjectID string `json:"project_id" valid:"required"`
}

type ExpelProjectOutput struct{}

type ListDatasetsInput struct {
	ProjectID string `json:"project_id" valid:"required"`
}

type ListDatasetsOutput struct {
	Datasets []*DatasetSummary
}

type ListTasksInput struct {
	ProjectID        string `json:"project_id" valid:"required"`
	WorkloadID       string `json:"workload_id" valid:"required"`
	OnlySuccessTasks bool   `json:"only_success_tasks" valid:"required"`
}

type ListTasksOutput struct {
	Tasks []*TaskSummary
}

type ListWorkloadsInput struct {
	ProjectID string `json:"project_id" valid:"required"`
}

type ListWorkloadsOutput struct {
	Workloads []*WorkloadSummary
}

type PlaceProjectInput struct {
	ProjectID string `json:"project_id" valid:"required"`
	Host      string `json:"host" valid:"required"`
	Token     string `json:"token"`
	CAPem     string `json:"ca_pem"`
	Password  string `json:"password"`
	Username  string `json:"username"`
	Insecure  bool   `json:"insecure"`
}

type PlaceProjectOutput struct {
}

type Run struct {
	ProjectID  string `json:"project_id"`
	WorkloadID string `json:"workload_id" valid:"required"`
	TaskID     int64  `json:"task_id"`
	Token      string `json:"token"`

	Cmd   []string          `json:"cmd"`
	Env   map[string]string `json:"env"`
	Stdin []byte            `json:"stdin"`
}

type SendRunFailureInput struct {
	ProjectID  string `json:"project_id" valid:"required"`
	WorkloadID string `json:"workload_id" valid:"required"`
	TaskID     int64  `json:"task_id" valid:"required"`
	RunToken   string `json:"run_token" valid:"required"`

	ErrorCode    string `json:"error_code"`
	ErrorMessage string `json:"error_message"`
}

type SendRunFailureOutput struct{}

type SendRunHeartbeatInput struct {
	ProjectID  string `json:"project_id" valid:"required"`
	WorkloadID string `json:"workload_id" valid:"required"`
	TaskID     int64  `json:"task_id" valid:"required"`
	RunToken   string `json:"run_token" valid:"required"`
}

type SendRunHeartbeatOutput struct {
	HasExpired bool `json:"has_expired"`
}

type SendRunSuccessInput struct {
	ProjectID  string `json:"project_id" valid:"required"`
	WorkloadID string `json:"workload_id" valid:"required"`
	TaskID     int64  `json:"task_id" valid:"required"`
	RunToken   string `json:"run_token" valid:"required"`

	Result          string `json:"result"`
	OutputDatasetID string `json:"output_dataset_id"`
}

type SendRunSuccessOutput struct{}

type SendUploadHeartbeatInput struct {
	ProjectID string `json:"project_id" valid:"required"`
	DatasetID string `json:"dataset_id" valid:"required"`
}

type SendUploadHeartbeatOutput struct {
	HasExpired bool `json:"has_expired"`
}

type SendUploadSuccessInput struct {
	ProjectID string `json:"project_id" valid:"required"`
	DatasetID string `json:"dataset_id" valid:"required"`
}

type SendUploadSuccessOutput struct{}

type StartTaskInput struct {
	ProjectID  string `json:"project_id" valid:"required"`
	WorkloadID string `json:"workload_id" valid:"required"`

	Cmd   []string          `json:"cmd"`
	Env   map[string]string `json:"env"`
	Stdin []byte            `json:"stdin"`
}

type StartTaskOutput struct {
	TaskSummary
}

type StopTaskInput struct {
	ProjectID  string `json:"project_id" valid:"required"`
	WorkloadID string `json:"workload_id" valid:"required"`
	TaskID     int64  `json:"task_id" valid:"required"`
}

type StopTaskOutput struct{}

type StopWorkloadInput struct {
	ProjectID  string `json:"project_id" valid:"required"`
	WorkloadID string `json:"workload_id" valid:"required"`
}

type StopWorkloadOutput struct{}

type TaskStatus string

type TaskSummary struct {
	TaskID          int64    `json:"task_id"`
	WorkloadID      string   `json:"workload_id"`
	Status          string   `json:"status,omitempty"`
	OutputDatasetID string   `json:"output_dataset_id"`
	Cmd             []string `json:"cmd"`
}

type WorkerCondition struct {
	ProbeTime time.Time `json:"probe_time"`
	Type      string    `json:"type"`
}

type WorkerLogsInput struct {
	ProjectID  string `json:"project_id"`
	WorkloadID string `json:"workload_id"`
	WorkerID   string `json:"worker_id"`
}

type WorkerLogsOutput struct {
	Data []byte `json:"data"`
}

type WorkerSummary struct {
	WorkerID   string             `json:"worker_id"`
	Status     string             `json:"status"`
	Conditions []*WorkerCondition `json:"conditions"`
}

type WorkloadSummary struct {
	ProjectID      string           `json:"project_id"`
	WorkloadID     string           `json:"workload_id"`
	QueueURL       string           `json:"queue_url"`
	Image          string           `json:"image"`
	NrOfWorkers    int              `json:"nr_of_workers"`
	InputDatasetID string           `json:"input_dataset_id"`
	CreatedAt      int64            `json:"created_at"`
	Workers        []*WorkerSummary `json:"workers"`
}