queue

type JobMessage struct {
	JobID         string   `json:"job_id,omitempty"`
	RunID         string   `json:"run_id"`
	Repo          string   `json:"repo,omitempty"`
	InstanceType  string   `json:"instance_type"`
	Pool          string   `json:"pool,omitempty"`
	Private       bool     `json:"private"`
	Spot          bool     `json:"spot"`
	RunnerSpec    string   `json:"runner_spec"`
	OriginalLabel string   `json:"original_label,omitempty"`
	RetryCount    int      `json:"retry_count,omitempty"`
	ForceOnDemand bool     `json:"force_on_demand,omitempty"`
	Region        string   `json:"region,omitempty"`
	Environment   string   `json:"environment,omitempty"`
	OS            string   `json:"os,omitempty"`   // linux, windows
	Arch          string   `json:"arch,omitempty"` // amd64, arm64
	InstanceTypes []string `json:"instance_types,omitempty"`
	StorageGiB    int      `json:"storage_gib,omitempty"` // Disk storage in GiB
	TraceID       string   `json:"trace_id,omitempty"`
	SpanID        string   `json:"span_id,omitempty"`
	ParentID      string   `json:"parent_id,omitempty"`
}

type Message struct {
	// ID is the unique message identifier.
	ID string

	// Body contains the JSON-encoded JobMessage.
	Body string

	// Handle is used to delete/acknowledge the message.
	// For SQS: ReceiptHandle
	// For Valkey: stream message ID
	Handle string

	// Attributes contains message metadata (trace context, etc).
	Attributes map[string]string
}

type Queue interface {
	// SendMessage publishes a job message to the queue.
	// Returns error if the message cannot be sent.
	SendMessage(ctx context.Context, job *JobMessage) error

	// ReceiveMessages retrieves up to maxMessages with long polling.
	// waitTimeSeconds controls how long to wait for messages (0 = no wait).
	// Returns empty slice if no messages available.
	ReceiveMessages(ctx context.Context, maxMessages int32, waitTimeSeconds int32) ([]Message, error)

	// DeleteMessage acknowledges successful message processing.
	// handle is the receipt handle (SQS) or message ID (Valkey).
	DeleteMessage(ctx context.Context, handle string) error
}

type SQSAPI interface {
	SendMessage(ctx context.Context, params *sqs.SendMessageInput, optFns ...func(*sqs.Options)) (*sqs.SendMessageOutput, error)
	ReceiveMessage(ctx context.Context, params *sqs.ReceiveMessageInput, optFns ...func(*sqs.Options)) (*sqs.ReceiveMessageOutput, error)
	DeleteMessage(ctx context.Context, params *sqs.DeleteMessageInput, optFns ...func(*sqs.Options)) (*sqs.DeleteMessageOutput, error)
}

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

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

type ValkeyConfig struct {
	Addr       string
	Password   string
	DB         int
	Stream     string
	Group      string
	ConsumerID string
}