analytics

type AnalyticsStore interface {
	// SaveSnapshot stores a single savings snapshot.
	SaveSnapshot(ctx context.Context, snapshot *SavingsSnapshot) error

	// BulkInsertSnapshots inserts multiple snapshots efficiently.
	BulkInsertSnapshots(ctx context.Context, snapshots []SavingsSnapshot) error

	// QuerySavings retrieves savings snapshots based on query parameters.
	QuerySavings(ctx context.Context, req QueryRequest) ([]SavingsSnapshot, error)

	// Aggregated queries (using materialized views for performance). Scoping is
	// the dual-column model; pass empty filters only after enforcing scope.
	QueryMonthlyTotals(ctx context.Context, accountUUIDs []string, accountExternalIDsByProvider map[string][]string, months int) ([]MonthlySummary, error)
	QueryByProvider(ctx context.Context, accountUUIDs []string, accountExternalIDsByProvider map[string][]string, startDate, endDate time.Time) ([]ProviderBreakdown, error)
	QueryByService(ctx context.Context, accountUUIDs []string, accountExternalIDsByProvider map[string][]string, provider string, startDate, endDate time.Time) ([]ServiceBreakdown, error)

	// Partition management.
	CreatePartition(ctx context.Context, forMonth time.Time) error
	CreateFuturePartitions(ctx context.Context, monthsAhead int) error
	DropOldPartitions(ctx context.Context, retentionMonths int) error
	CreatePartitionsForRange(ctx context.Context, startDate, endDate time.Time) error

	// Materialized view management.
	RefreshMaterializedViews(ctx context.Context) error

	// Close cleans up resources.
	Close() error
}

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

type CollectorConfig struct {
	AnalyticsStore AnalyticsStore
}

type MonthlySummary struct {
	Month          time.Time `json:"month"`
	CloudAccountID *string   `json:"cloud_account_id,omitempty"`
	// AvgCoverage is the average reservation coverage for the month.
	// A nil value means coverage data is absent for this period and must
	// not be treated as zero coverage by consumers.
	AvgCoverage   *float64 `json:"avg_coverage,omitempty"`
	AccountID     string   `json:"account_id"`
	Provider      string   `json:"provider"`
	Service       string   `json:"service"`
	TotalSavings  float64  `json:"total_savings"`
	SnapshotCount int      `json:"snapshot_count"`
}

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

type ProviderBreakdown struct {
	AvgCoverage  *float64 `json:"avg_coverage,omitempty"`
	Provider     string   `json:"provider"`
	Service      string   `json:"service"`
	TotalSavings float64  `json:"total_savings"`
}

type QueryRequest struct {
	StartDate                    time.Time
	EndDate                      time.Time
	AccountExternalIDsByProvider map[string][]string
	Provider                     string
	Service                      string
	AccountUUIDs                 []string
	Limit                        int
}

type SavingsSnapshot struct {
	Timestamp          time.Time      `json:"timestamp"`
	TotalUsage         *float64       `json:"total_usage,omitempty"`
	CloudAccountID     *string        `json:"cloud_account_id,omitempty"`
	CoveragePercentage *float64       `json:"coverage_percentage,omitempty"`
	Metadata           map[string]any `json:"metadata,omitempty"`
	AccountID          string         `json:"account_id"`
	Provider           string         `json:"provider"`
	Service            string         `json:"service"`
	Region             string         `json:"region"`
	CommitmentType     string         `json:"commitment_type"`
	ID                 string         `json:"id"`
	TotalCommitment    *float64       `json:"total_commitment"`
	TotalSavings       float64        `json:"total_savings"`
}

type ServiceBreakdown struct {
	AvgCoverage  *float64 `json:"avg_coverage,omitempty"`
	Service      string   `json:"service"`
	Region       string   `json:"region"`
	TotalSavings float64  `json:"total_savings"`
}