rpc

type AgentInfo struct {
	Version      string            `json:"version"`
	Platform     string            `json:"platform"`
	Backend      string            `json:"backend"`
	Capacity     int               `json:"capacity"`
	CustomLabels map[string]string `json:"custom_labels"`
}

type Filter struct {
	Labels map[string]string `json:"labels"`
}

type LogEntry struct {
	StepUUID string `json:"step_uuid,omitempty"`
	Time     int64  `json:"time,omitempty"`
	Type     int    `json:"type,omitempty"`
	Line     int    `json:"line,omitempty"`
	Data     []byte `json:"data,omitempty"`
}

type Peer interface {
	// Version returns the server- & grpc-version
	Version(c context.Context) (*Version, error)

	// Next returns the next workflow in the queue
	Next(c context.Context, f Filter) (*Workflow, error)

	// Wait blocks until the workflow is complete
	Wait(c context.Context, workflowID string) error

	// Init signals the workflow is initialized
	Init(c context.Context, workflowID string, state WorkflowState) error

	// Done signals the workflow is complete
	Done(c context.Context, workflowID string, state WorkflowState) error

	// Extend extends the workflow deadline
	Extend(c context.Context, workflowID string) error

	// Update updates the step state
	Update(c context.Context, workflowID string, state StepState) error

	// EnqueueLog queues the step log entry for delayed sending
	EnqueueLog(logEntry *LogEntry)

	// RegisterAgent register our agent to the server
	RegisterAgent(ctx context.Context, info AgentInfo) (int64, error)

	// UnregisterAgent unregister our agent from the server
	UnregisterAgent(ctx context.Context) error

	// ReportHealth reports health status of the agent to the server
	ReportHealth(c context.Context) error
}

type StepState struct {
	StepUUID string `json:"step_uuid"`
	Started  int64  `json:"started"`
	Finished int64  `json:"finished"`
	Exited   bool   `json:"exited"`
	ExitCode int    `json:"exit_code"`
	Error    string `json:"error"`
}

type Version struct {
	GrpcVersion   int32  `json:"grpc_version,omitempty"`
	ServerVersion string `json:"server_version,omitempty"`
}

type Workflow struct {
	ID      string          `json:"id"`
	Config  *backend.Config `json:"config"`
	Timeout int64           `json:"timeout"`
}

type WorkflowState struct {
	Started  int64  `json:"started"`
	Finished int64  `json:"finished"`
	Error    string `json:"error"`
}