heartbeat

type AlertCallback func(h *Heartbeat, alertType string, details map[string]interface{})

type AlertState string

type CreateHeartbeatInput struct {
	Name            string `json:"name"`
	IntervalSeconds int    `json:"interval_seconds"`
	GraceSeconds    int    `json:"grace_seconds"`
}

type DefaultLicenseChecker struct {
	MaxHeartbeats int
}

type Deps struct {
	Store          HeartbeatStore // required
	Logger         *slog.Logger   // required
	LicenseChecker LicenseChecker // optional — defaults to community limits
	EventCallback  EventCallback  // optional — nil-safe
	AlertCallback  AlertCallback  // optional — nil-safe
	BaseURL        string         // optional
}

type EventCallback func(eventType string, data interface{})

type ExecutionOutcome string

type Heartbeat struct {
	ID                   string          `json:"id"`
	Name                 string          `json:"name"`
	Status               HeartbeatStatus `json:"status"`
	AlertState           AlertState      `json:"alert_state"`
	IntervalSeconds      int             `json:"interval_seconds"`
	GraceSeconds         int             `json:"grace_seconds"`
	LastPingAt           *time.Time      `json:"last_ping_at,omitempty"`
	NextDeadlineAt       *time.Time      `json:"next_deadline_at,omitempty"`
	CurrentRunStartedAt  *time.Time      `json:"current_run_started_at,omitempty"`
	LastExitCode         *int            `json:"last_exit_code,omitempty"`
	LastDurationMs       *int64          `json:"last_duration_ms,omitempty"`
	ConsecutiveFailures  int             `json:"consecutive_failures"`
	ConsecutiveSuccesses int             `json:"consecutive_successes"`
	Active               bool            `json:"active"`
	CreatedAt            time.Time       `json:"created_at"`
	UpdatedAt            time.Time       `json:"updated_at"`
	AgentID              string          `json:"agent_id"`
}

type HeartbeatExecution struct {
	ID          string           `json:"id"`
	HeartbeatID string           `json:"heartbeat_id"`
	StartedAt   *time.Time       `json:"started_at,omitempty"`
	CompletedAt *time.Time       `json:"completed_at,omitempty"`
	DurationMs  *int64           `json:"duration_ms,omitempty"`
	ExitCode    *int             `json:"exit_code,omitempty"`
	Outcome     ExecutionOutcome `json:"outcome"`
	Payload     *string          `json:"payload,omitempty"`
}

type HeartbeatPing struct {
	ID          string    `json:"id"`
	HeartbeatID string    `json:"heartbeat_id"`
	PingType    PingType  `json:"ping_type"`
	ExitCode    *int      `json:"exit_code,omitempty"`
	SourceIP    string    `json:"source_ip"`
	HTTPMethod  string    `json:"http_method"`
	Payload     *string   `json:"payload,omitempty"`
	Timestamp   time.Time `json:"timestamp"`
}

type HeartbeatStatus string

type HeartbeatStore interface {
	// Heartbeat CRUD
	CreateHeartbeat(ctx context.Context, h *Heartbeat) (string, error)
	GetHeartbeatByID(ctx context.Context, id string) (*Heartbeat, error)
	// GetHeartbeatByUUID looks up by the ping token (which is the id).
	GetHeartbeatByUUID(ctx context.Context, token string) (*Heartbeat, error)
	ListHeartbeats(ctx context.Context, opts ListHeartbeatsOpts) ([]*Heartbeat, error)
	UpdateHeartbeat(ctx context.Context, id string, input UpdateHeartbeatInput) error
	DeleteHeartbeat(ctx context.Context, id string) error

	// State updates
	UpdateHeartbeatState(ctx context.Context, id string, status HeartbeatStatus, alertState AlertState,
		lastPingAt *time.Time, nextDeadlineAt *time.Time, currentRunStartedAt *time.Time,
		lastExitCode *int, lastDurationMs *int64,
		consecutiveFailures, consecutiveSuccesses int) error
	PauseHeartbeat(ctx context.Context, id string) error
	ResumeHeartbeat(ctx context.Context, id string, nextDeadlineAt time.Time) error

	// Deadline scanning
	ListOverdueHeartbeats(ctx context.Context, now time.Time) ([]*Heartbeat, error)

	// License gating
	CountActiveHeartbeats(ctx context.Context) (int, error)

	// Pings
	InsertPing(ctx context.Context, p *HeartbeatPing) (string, error)
	ListPings(ctx context.Context, heartbeatID string, opts ListPingsOpts) ([]*HeartbeatPing, int, error)

	// Executions
	InsertExecution(ctx context.Context, e *HeartbeatExecution) (string, error)
	UpdateExecution(ctx context.Context, id string, completedAt *time.Time, durationMs *int64, exitCode *int, outcome ExecutionOutcome, payload *string) error
	GetCurrentExecution(ctx context.Context, heartbeatID string) (*HeartbeatExecution, error)
	ListExecutions(ctx context.Context, heartbeatID string, opts ListExecutionsOpts) ([]*HeartbeatExecution, int, error)

	// Retention
	DeletePingsBefore(ctx context.Context, before time.Time, batchSize int) (int64, error)
	DeleteExecutionsBefore(ctx context.Context, before time.Time, batchSize int) (int64, error)
}

type LicenseChecker interface {
	CanCreateHeartbeat(currentCount int) bool
	CanStorePayload() bool
}

type ListExecutionsOpts struct {
	Limit  int
	Offset int
}

type ListHeartbeatsOpts struct {
	Status          string
	IncludeInactive bool
	// AgentFilter filters by agent_id. Nil = no filter; "local" = the local
	// sentinel agent; UUID = specific agent.
	AgentFilter *string
}

type ListPingsOpts struct {
	Limit  int
	Offset int
}

type PingType string

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

type UpdateHeartbeatInput struct {
	Name            *string `json:"name,omitempty"`
	IntervalSeconds *int    `json:"interval_seconds,omitempty"`
	GraceSeconds    *int    `json:"grace_seconds,omitempty"`
}