observability

type AlertConfig struct {
	AlertType       string `json:"alert_type"`
	AlertTargetType string `json:"alert_target_type"`
}

type CheckResult struct {
	Healthy  bool          `json:"healthy"`
	Message  string        `json:"message,omitempty"`
	Duration time.Duration `json:"duration"`
	Error    string        `json:"error,omitempty"`
	Critical bool          `json:"critical"`
}

type CloudWatchLogsClient interface {
	CreateLogGroup(ctx context.Context, params *cloudwatchlogs.CreateLogGroupInput, optFns ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.CreateLogGroupOutput, error)
	CreateLogStream(ctx context.Context, params *cloudwatchlogs.CreateLogStreamInput, optFns ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.CreateLogStreamOutput, error)
	PutLogEvents(ctx context.Context, params *cloudwatchlogs.PutLogEventsInput, optFns ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.PutLogEventsOutput, error)
	DescribeLogGroups(ctx context.Context, params *cloudwatchlogs.DescribeLogGroupsInput, optFns ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.DescribeLogGroupsOutput, error)
	DescribeLogStreams(ctx context.Context, params *cloudwatchlogs.DescribeLogStreamsInput, optFns ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.DescribeLogStreamsOutput, error)
}

type HealthChecker interface {
	Check(ctx context.Context) error
	Name() string
	IsCritical() bool
}

type HealthStatus struct {
	Healthy   bool                   `json:"healthy"`
	Checks    map[string]CheckResult `json:"checks"`
	Timestamp time.Time              `json:"timestamp"`
	Version   string                 `json:"version,omitempty"`
}

type LogBuffer interface {
	Add(entry *LogEntry) error
	Flush(ctx context.Context) error
	Size() int
	IsFull() bool
	Clear()
	Close() error
}

type LogEntry struct {
	Timestamp time.Time      `json:"timestamp"`
	Level     string         `json:"level"`
	Message   string         `json:"message"`
	Fields    map[string]any `json:"fields,omitempty"`
	RequestID string         `json:"request_id,omitempty"`
	TenantID  string         `json:"tenant_id,omitempty"`
	UserID    string         `json:"user_id,omitempty"`
	TraceID   string         `json:"trace_id,omitempty"`
	SpanID    string         `json:"span_id,omitempty"`
}

type LogSink interface {
	Send(ctx context.Context, entries []*LogEntry) error
	Close() error
}

type LoggerConfig struct {
	// Basic configuration
	Level        string `json:"level"`
	Format       string `json:"format"` // "json" or "console"
	EnableCaller bool   `json:"enable_caller"`
	EnableStack  bool   `json:"enable_stack"`

	// CloudWatch specific
	LogGroup      string        `json:"log_group"`
	LogStream     string        `json:"log_stream"`
	BatchSize     int           `json:"batch_size"`
	FlushInterval time.Duration `json:"flush_interval"`
	BufferSize    int           `json:"buffer_size"`

	// Performance tuning
	AsyncLogging bool          `json:"async_logging"`
	MaxRetries   int           `json:"max_retries"`
	RetryDelay   time.Duration `json:"retry_delay"`

	// Multi-tenant context
	DefaultTenantID string `json:"default_tenant_id"`
	DefaultUserID   string `json:"default_user_id"`
}

type LoggerConfigOption func(*LoggerConfig)

type LoggerFactory interface {
	CreateConsoleLogger(config LoggerConfig) (StructuredLogger, error)
	CreateCloudWatchLogger(config LoggerConfig, client CloudWatchLogsClient) (StructuredLogger, error)
	CreateTestLogger() StructuredLogger
	CreateNoOpLogger() StructuredLogger
}

type LoggerStats struct {
	EntriesLogged    int64         `json:"entries_logged"`
	EntriesDropped   int64         `json:"entries_dropped"`
	FlushCount       int64         `json:"flush_count"`
	LastFlush        time.Time     `json:"last_flush"`
	BufferSize       int           `json:"buffer_size"`
	BufferCapacity   int           `json:"buffer_capacity"`
	AverageFlushTime time.Duration `json:"average_flush_time"`
	ErrorCount       int64         `json:"error_count"`
	LastError        string        `json:"last_error,omitempty"`
}

type MetricEntry struct {
	Name      string            `json:"name"`
	Value     float64           `json:"value"`
	Unit      string            `json:"unit"`
	Timestamp time.Time         `json:"timestamp"`
	Tags      map[string]string `json:"tags,omitempty"`
	Fields    map[string]any    `json:"fields,omitempty"`
}

type MetricsCollector interface {
	// Explicitly declare lift.MetricsCollector methods to avoid interface embedding issues
	Counter(name string, tags ...map[string]string) lift.Counter
	Histogram(name string, tags ...map[string]string) lift.Histogram
	Gauge(name string, tags ...map[string]string) lift.Gauge
	Flush() error

	// Additional methods for enhanced functionality
	RecordLatency(operation string, duration time.Duration)
	RecordError(operation string)
	RecordSuccess(operation string)

	// Context methods
	WithTags(tags map[string]string) MetricsCollector
	WithTag(key, value string) MetricsCollector

	// Batch operations
	RecordBatch(entries []*MetricEntry) error

	// Performance methods
	Close() error
	GetStats() MetricsStats
}

type MetricsStats struct {
	MetricsRecorded int64     `json:"metrics_recorded"`
	MetricsDropped  int64     `json:"metrics_dropped"`
	LastFlush       time.Time `json:"last_flush"`
	ErrorCount      int64     `json:"error_count"`
	LastError       string    `json:"last_error,omitempty"`
}

type ObservabilityProvider interface {
	Logger() StructuredLogger
	Metrics() MetricsCollector
	HealthChecker() HealthChecker
	Close() error
}

type SNSClient interface {
	Publish(ctx context.Context, params *sns.PublishInput, optFns ...func(*sns.Options)) (*sns.PublishOutput, error)
}

type SNSConfig struct {
	Client   SNSClient
	TopicARN string
}

type SNSNotificationMessage struct {
	AlertConfig AlertConfig `json:"alert_config"`
	LogTime     string      `json:"log_time"`
	Environment string      `json:"environment,omitempty"`
	Service     string      `json:"service,omitempty"`
	Partner     string      `json:"partner"`
	Stage       string      `json:"stage"`
	AWSRegion   string      `json:"aws_region"`
	AWSAccount  string      `json:"aws_account"`
	Function    string      `json:"function"`
	Subsystem   string      `json:"subsystem"`
	Severity    string      `json:"severity"`
	Message     string      `json:"message"`
}

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

type StructuredLogger interface {
	// Explicitly declare lift.Logger methods to avoid interface embedding issues
	Debug(message string, fields ...map[string]any)
	Info(message string, fields ...map[string]any)
	Warn(message string, fields ...map[string]any)
	Error(message string, fields ...map[string]any)
	WithField(key string, value any) lift.Logger
	WithFields(fields map[string]any) lift.Logger

	// Context methods for multi-tenant logging
	WithRequestID(requestID string) StructuredLogger
	WithTenantID(tenantID string) StructuredLogger
	WithUserID(userID string) StructuredLogger
	WithTraceID(traceID string) StructuredLogger
	WithSpanID(spanID string) StructuredLogger

	// Performance and health methods
	Flush(ctx context.Context) error
	Close() error
	IsHealthy() bool
	GetStats() LoggerStats
}

type TestObservabilityProvider interface {
	ObservabilityProvider

	// Test-specific methods
	GetLogEntries() []*LogEntry
	GetMetricEntries() []*MetricEntry
	ClearLogs()
	ClearMetrics()
	SimulateError(component string, err error)
	GetCallCount(method string) int
}