edgeapi

type BundleInfo struct {
	Name    string  `json:"name"`
	Version *string `json:"version"`
}

type Client struct {
	// The endpoint of the server conforming to this interface, with scheme,
	// https://api.deepmap.com for example. This can contain a path relative
	// to the server, such as https://api.deepmap.com/dev-test, and all the
	// paths in the swagger spec will be appended to the server.
	Server string

	*resty.Client

	RequestMiddleware []resty.RequestMiddleware
}

type ClientInterface interface {
	// Jobs deals with all the Jobs endpoints
	Jobs() JobsClient
	// Logs deals with all the Logs endpoints
	Logs() LogsClient
	// Worker deals with all the Worker endpoints
	Worker() WorkerClient
}

type ClientOption func(*Client) error

type EdgeJobFetched struct {
	// Command Execute the given Task.
	Command ExecuteTask `json:"command"`

	// ConcurrencySlots Number of concurrency slots the job requires.
	ConcurrencySlots int `json:"concurrency_slots"`

	// DagId Identifier of the DAG to which the task belongs.
	DagId string `json:"dag_id"`

	// MapIndex For dynamically mapped tasks the mapping number, -1 if the task is not mapped.
	MapIndex int `json:"map_index"`

	// RunId Run ID of the DAG execution.
	RunId string `json:"run_id"`

	// TaskId Task name in the DAG.
	TaskId string `json:"task_id"`

	// TryNumber The number of attempt to execute this task.
	TryNumber int `json:"try_number"`
}

type EdgeWorkerState string

type ExecuteTask struct {
	// BundleInfo Schema for telling task which bundle to run with.
	BundleInfo BundleInfo `json:"bundle_info"`
	DagRelPath string     `json:"dag_rel_path"`
	LogPath    *string    `json:"log_path"`

	// Ti Schema for TaskInstance with minimal required fields needed for Executors and Task SDK.
	Ti    TaskInstance `json:"ti"`
	Token string       `json:"token"`
	Type  *string      `json:"type,omitempty"`
}

type FetchJSONRequestBody = WorkerQueuesBody

type GeneralHTTPError struct {
	Response *resty.Response
	JSON     map[string]any
	Text     string
}

type HTTPExceptionResponse struct {
	Detail HTTPExceptionResponse_Detail `json:"detail"`
}

type HTTPExceptionResponseDetail0 = string

type HTTPExceptionResponseDetail1 map[string]interface{}

type HTTPExceptionResponse_Detail struct {
	// contains filtered or unexported fields
}

type HTTPValidationError struct {
	Detail *[]ValidationError `json:"detail,omitempty"`
}

type Job struct {
	// DagId Identifier of the DAG to which the task belongs.
	DagId string `json:"dag_id"`

	// EdgeWorker The worker processing the job during execution.
	EdgeWorker *string `json:"edge_worker"`

	// LastUpdate Last heartbeat of the job.
	LastUpdate *time.Time `json:"last_update"`

	// MapIndex For dynamically mapped tasks the mapping number, -1 if the task is not mapped.
	MapIndex int `json:"map_index"`

	// Queue Queue for which the task is scheduled/running.
	Queue string `json:"queue"`

	// QueuedDttm When the job was queued.
	QueuedDttm *time.Time `json:"queued_dttm"`

	// RunId Run ID of the DAG execution.
	RunId string `json:"run_id"`

	// State All possible states that a Task Instance can be in.
	//
	// Note that None is also allowed, so always use this in a type hint with Optional.
	State TaskInstanceState `json:"state"`

	// TaskId Task name in the DAG.
	TaskId string `json:"task_id"`

	// TryNumber The number of attempt to execute this task.
	TryNumber int `json:"try_number"`
}

type JobCollectionResponse struct {
	Jobs         []Job `json:"jobs"`
	TotalEntries int   `json:"total_entries"`
}

type JobsClient interface {
	// Fetch a job to execute on the edge worker.
	Fetch(ctx context.Context, workerName string, body *WorkerQueuesBody) (*struct {
		// Command Execute the given Task.
		Command ExecuteTask `json:"command"`

		// ConcurrencySlots Number of concurrency slots the job requires.
		ConcurrencySlots int `json:"concurrency_slots"`

		// DagId Identifier of the DAG to which the task belongs.
		DagId string `json:"dag_id"`

		// MapIndex For dynamically mapped tasks the mapping number, -1 if the task is not mapped.
		MapIndex int `json:"map_index"`

		// RunId Run ID of the DAG execution.
		RunId string `json:"run_id"`

		// TaskId Task name in the DAG.
		TaskId string `json:"task_id"`

		// TryNumber The number of attempt to execute this task.
		TryNumber int `json:"try_number"`
	}, error)
	// FetchResponse is a lower level version of [Fetch] and provides access to the raw [resty.Response]
	FetchResponse(ctx context.Context, workerName string, body *WorkerQueuesBody) (*resty.Response, error)

	// Update the state of a job running on the edge worker.
	State(ctx context.Context, dagId string, taskId string, runId string, tryNumber int, mapIndex int, state TaskInstanceState) (*interface{}, error)
	// StateResponse is a lower level version of [State] and provides access to the raw [resty.Response]
	StateResponse(ctx context.Context, dagId string, taskId string, runId string, tryNumber int, mapIndex int, state TaskInstanceState) (*resty.Response, error)
}

type LogsClient interface {

	// Push an incremental log chunk from Edge Worker to central site.
	Push(ctx context.Context, dagId string, taskId string, runId string, tryNumber int, mapIndex int, body *PushLogsBody) (*interface{}, error)
	// PushResponse is a lower level version of [Push] and provides access to the raw [resty.Response]
	PushResponse(ctx context.Context, dagId string, taskId string, runId string, tryNumber int, mapIndex int, body *PushLogsBody) (*resty.Response, error)
	// contains filtered or unexported methods
}

type MaintenanceRequest struct {
	// MaintenanceComment Comment describing the maintenance reason.
	MaintenanceComment string `json:"maintenance_comment"`
}

type PushLogsBody struct {
	// LogChunkData Log chunk data as incremental log text.
	LogChunkData string `json:"log_chunk_data"`

	// LogChunkTime Time of the log chunk at point of sending.
	LogChunkTime time.Time `json:"log_chunk_time"`
}

type PushLogsJSONRequestBody = PushLogsBody

type RegisterJSONRequestBody = WorkerStateBody

type SetStateJSONRequestBody = WorkerStateBody

type TaskInstance struct {
	ContextCarrier       *map[string]interface{} `json:"context_carrier"`
	DagId                string                  `json:"dag_id"`
	DagVersionId         openapi_types.UUID      `json:"dag_version_id"`
	Id                   openapi_types.UUID      `json:"id"`
	MapIndex             *int                    `json:"map_index,omitempty"`
	ParentContextCarrier *map[string]interface{} `json:"parent_context_carrier"`
	PoolSlots            int                     `json:"pool_slots"`
	PriorityWeight       int                     `json:"priority_weight"`
	Queue                string                  `json:"queue"`
	RunId                string                  `json:"run_id"`
	TaskId               string                  `json:"task_id"`
	TryNumber            int                     `json:"try_number"`
}

type TaskInstanceState string

type UpdateQueuesJSONRequestBody = WorkerQueueUpdateBody

type ValidationError struct {
	Loc  []ValidationError_Loc_Item `json:"loc"`
	Msg  string                     `json:"msg"`
	Type string                     `json:"type"`
}

type ValidationErrorLoc0 = string

type ValidationErrorLoc1 = int

type ValidationError_Loc_Item struct {
	// contains filtered or unexported fields
}

type Worker struct {
	// FirstOnline When the worker was first online.
	FirstOnline *time.Time `json:"first_online"`

	// JobsActive Number of active jobs the worker is running.
	JobsActive *int `json:"jobs_active,omitempty"`

	// LastHeartbeat When the worker last sent a heartbeat.
	LastHeartbeat *time.Time `json:"last_heartbeat"`

	// MaintenanceComments Comments about the maintenance state of the worker.
	MaintenanceComments *string `json:"maintenance_comments"`

	// Queues List of queues the worker is pulling jobs from. If not provided, worker pulls from all queues.
	Queues *[]string `json:"queues"`

	// State Status of a Edge Worker instance.
	State EdgeWorkerState `json:"state"`

	// Sysinfo System information of the worker.
	Sysinfo map[string]Worker_Sysinfo_AdditionalProperties `json:"sysinfo"`

	// WorkerName Name of the worker.
	WorkerName string `json:"worker_name"`
}

type WorkerClient interface {
	UpdateQueues(ctx context.Context, workerName string, body *WorkerQueueUpdateBody) (*interface{}, error)
	// UpdateQueuesResponse is a lower level version of [UpdateQueues] and provides access to the raw [resty.Response]
	UpdateQueuesResponse(ctx context.Context, workerName string, body *WorkerQueueUpdateBody) (*resty.Response, error)

	// Set state of worker and returns the current assigned queues.
	SetState(ctx context.Context, workerName string, body *WorkerStateBody) (*WorkerSetStateReturn, error)
	// SetStateResponse is a lower level version of [SetState] and provides access to the raw [resty.Response]
	SetStateResponse(ctx context.Context, workerName string, body *WorkerStateBody) (*resty.Response, error)

	// Register a new worker to the backend.
	Register(ctx context.Context, workerName string, body *WorkerStateBody) (*WorkerRegistrationReturn, error)
	// RegisterResponse is a lower level version of [Register] and provides access to the raw [resty.Response]
	RegisterResponse(ctx context.Context, workerName string, body *WorkerStateBody) (*resty.Response, error)
}

type WorkerCollectionResponse struct {
	TotalEntries int      `json:"total_entries"`
	Workers      []Worker `json:"workers"`
}

type WorkerQueueUpdateBody struct {
	// NewQueues Additional queues to be added to worker.
	NewQueues *[]string `json:"new_queues"`

	// RemoveQueues Queues to remove from worker.
	RemoveQueues *[]string `json:"remove_queues"`
}

type WorkerQueuesBody struct {
	// FreeConcurrency Number of free concurrency slots on the worker.
	FreeConcurrency int `json:"free_concurrency"`

	// Queues List of queues the worker is pulling jobs from. If not provided, worker pulls from all queues.
	Queues *[]string `json:"queues"`
}

type WorkerRegistrationReturn struct {
	// LastUpdate Time of the last update of the worker.
	LastUpdate time.Time `json:"last_update"`
}

type WorkerSetStateReturn struct {
	// MaintenanceComments Comments about the maintenance state of the worker.
	MaintenanceComments *string `json:"maintenance_comments"`

	// Queues List of queues the worker is pulling jobs from. If not provided, worker pulls from all queues.
	Queues *[]string `json:"queues"`

	// State Status of a Edge Worker instance.
	State EdgeWorkerState `json:"state"`
}

type WorkerStateBody struct {
	// JobsActive Number of active jobs the worker is running.
	JobsActive *int `json:"jobs_active,omitempty"`

	// MaintenanceComments Comments about the maintenance state of the worker.
	MaintenanceComments *string `json:"maintenance_comments"`

	// Queues List of queues the worker is pulling jobs from. If not provided, worker pulls from all queues.
	Queues *[]string `json:"queues"`

	// State Status of a Edge Worker instance.
	State EdgeWorkerState `json:"state"`

	// Sysinfo System information of the worker.
	Sysinfo map[string]WorkerStateBody_Sysinfo_AdditionalProperties `json:"sysinfo"`
}

type WorkerStateBodySysinfo0 = string

type WorkerStateBodySysinfo1 = int

type WorkerStateBody_Sysinfo_AdditionalProperties struct {
	// contains filtered or unexported fields
}

type WorkerSysinfo0 = string

type WorkerSysinfo1 = int

type Worker_Sysinfo_AdditionalProperties struct {
	// contains filtered or unexported fields
}