grpc

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

type ClientOptions struct {
	// Address is the gRPC server address (e.g., "grpc.spooled.cloud:443")
	Address string
	// APIKey is the API key for authentication
	APIKey string
	// UseTLS forces TLS on or off. When nil, TLS is used for port 443 (and for an
	// address with no port, which gRPC dials on 443) and plaintext is used for
	// every other port, including localhost:50051. Set it to dial a TLS
	// endpoint on another port, such as ":8443".
	UseTLS *bool
	// TLSConfig is custom TLS configuration (optional)
	TLSConfig *tls.Config
	// DialOptions are additional gRPC dial options
	DialOptions []grpc.DialOption
	// Timeout is the connection timeout
	Timeout time.Duration
}

type CompleteRequest struct {
	JobID    string
	WorkerID string
	Result   map[string]any
	// LeaseID is the lease fencing token from the dequeued Job (Job.LeaseID).
	// It is required: the server answers a Complete without it with
	// InvalidArgument, and one that does not match the job's current lease is
	// rejected.
	LeaseID string
}

type DequeueRequest struct {
	QueueName        string
	WorkerID         string
	BatchSize        int32
	LeaseDurationSec int32
}

type DequeueResponse struct {
	Jobs []*Job
}

type EnqueueRequest struct {
	QueueName      string
	Payload        map[string]any
	Priority       int32
	MaxRetries     *int32
	TimeoutSeconds *int32
	ScheduledAt    *time.Time
	IdempotencyKey string
}

type EnqueueResponse struct {
	JobID   string
	Created bool
}

type FailRequest struct {
	JobID    string
	WorkerID string
	Error    string
	// Retry has no effect any more: a failed job is retried, up to the job's
	// max_retries, unless NoRetry is set.
	//
	// Deprecated: the zero value used to mean "dead-letter now" because proto3
	// cannot tell an omitted bool from false, so a Fail without Retry skipped
	// every retry. Retrying is now the default; set NoRetry to opt out.
	Retry bool
	// NoRetry dead-letters the job immediately instead of scheduling a retry.
	NoRetry bool
	// LeaseID is the lease fencing token from the dequeued Job (Job.LeaseID).
	// It is required: the server answers a Fail without it with
	// InvalidArgument, and one that does not match the job's current lease is
	// rejected.
	LeaseID string
}

type FailResponse struct {
	Success bool
	// WillRetry is true when the job went back to pending for another attempt,
	// false when it was dead-lettered (NoRetry, or max_retries used up).
	WillRetry bool
	// NextRetryDelaySecs is the backoff before the retry; 0 when the job was
	// dead-lettered.
	NextRetryDelaySecs int32
}

type Job struct {
	ID             string
	QueueName      string
	Payload        map[string]any
	Priority       int32
	RetryCount     int32
	MaxRetries     int32
	TimeoutSeconds int32
	LeaseExpiresAt *time.Time
	// LeaseID is the lease fencing token issued by the server for this
	// dequeue. Echo it back on Complete, Fail and RenewLease: the server
	// requires it there (a settlement without it fails with InvalidArgument
	// "lease_id is required for job settlement") and applies the call only to
	// the lease this worker actually holds.
	LeaseID string
}

type QueueStats struct {
	QueueName  string
	Pending    int64
	Scheduled  int64
	Processing int64
	Completed  int64
	Failed     int64
	Deadletter int64
	Total      int64
	MaxAgeMs   int64
}

type RegisterWorkerRequest struct {
	QueueName      string
	Hostname       string
	MaxConcurrency int32
	Version        string
	Metadata       map[string]string
	// WorkerID is an optional stable worker id: 1-128 characters from
	// [A-Za-z0-9._-]. Send the same id after a restart and the server updates
	// that worker instead of adding one, so a restarting worker does not count
	// twice against the plan's worker limit. Empty: the server assigns a new
	// id. An id another organization already uses fails with AlreadyExists.
	WorkerID string
}

type RegisterWorkerResponse struct {
	WorkerID             string
	HeartbeatIntervalSec int32
	LeaseDurationSec     int32
}

type RenewLeaseRequest struct {
	JobID         string
	WorkerID      string
	ExtensionSecs int32
	// LeaseID is the lease fencing token from the dequeued Job (Job.LeaseID).
	// It is required: the server answers a RenewLease without it with
	// InvalidArgument, and one that does not match the job's current lease is
	// rejected.
	LeaseID string
}

type RenewLeaseResponse struct {
	Success      bool
	NewExpiresAt *time.Time
}

type WorkerHeartbeatRequest struct {
	WorkerID    string
	CurrentJobs int32
	Status      string
}