services

type Agent struct {
	ID              uuid.UUID         `json:"id"`
	Name            string            `json:"name"`
	Labels          map[string]string `json:"labels"`
	Status          AgentStatus       `json:"status"`
	LastSeen        time.Time         `json:"last_seen"`
	GroupID         *string           `json:"group_id,omitempty"`
	GroupName       *string           `json:"group_name,omitempty"`
	Version         string            `json:"version"`
	Capabilities    []string          `json:"capabilities"`
	EffectiveConfig string            `json:"effective_config,omitempty"`
	CreatedAt       time.Time         `json:"created_at"`
	UpdatedAt       time.Time         `json:"updated_at"`
}

type AgentService interface {
	// Agent operations
	CreateAgent(ctx context.Context, agent *Agent) error
	GetAgent(ctx context.Context, id uuid.UUID) (*Agent, error)
	ListAgents(ctx context.Context) ([]*Agent, error)
	UpdateAgentStatus(ctx context.Context, id uuid.UUID, status AgentStatus) error
	UpdateAgentLastSeen(ctx context.Context, id uuid.UUID, lastSeen time.Time) error
	UpdateAgentEffectiveConfig(ctx context.Context, id uuid.UUID, effectiveConfig string) error
	DeleteAgent(ctx context.Context, id uuid.UUID) error

	// Group operations
	CreateGroup(ctx context.Context, group *Group) error
	GetGroup(ctx context.Context, id string) (*Group, error)
	GetGroupByName(ctx context.Context, name string) (*Group, error)
	ListGroups(ctx context.Context) ([]*Group, error)
	DeleteGroup(ctx context.Context, id string) error

	// Config operations
	CreateConfig(ctx context.Context, config *Config) error
	GetConfig(ctx context.Context, id string) (*Config, error)
	GetLatestConfigForAgent(ctx context.Context, agentID uuid.UUID) (*Config, error)
	GetLatestConfigForGroup(ctx context.Context, groupID string) (*Config, error)
	ListConfigs(ctx context.Context, filter ConfigFilter) ([]*Config, error)

	// StoreConfigForAgent validates and stores configuration for an agent
	// Returns the stored config or error if agent doesn't exist or doesn't support remote config
	StoreConfigForAgent(ctx context.Context, agentID uuid.UUID, content string) (*Config, error)
}

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

type AgentStatus string

type Config struct {
	ID         string     `json:"id"`
	Name       string     `json:"name"`
	AgentID    *uuid.UUID `json:"agent_id,omitempty"`
	GroupID    *string    `json:"group_id,omitempty"`
	ConfigHash string     `json:"config_hash"`
	Content    string     `json:"content"`
	Version    int        `json:"version"`
	CreatedAt  time.Time  `json:"created_at"`
}

type ConfigFilter struct {
	AgentID *uuid.UUID
	GroupID *string
	Limit   int
}

type Group struct {
	ID         string            `json:"id"`
	Name       string            `json:"name"`
	Labels     map[string]string `json:"labels"`
	AgentCount int               `json:"agent_count"`
	ConfigName string            `json:"config_name,omitempty"`
	CreatedAt  time.Time         `json:"created_at"`
	UpdatedAt  time.Time         `json:"updated_at"`
}

type Log struct {
	Timestamp      time.Time              `json:"timestamp"`
	AgentID        uuid.UUID              `json:"agent_id"`
	GroupID        *string                `json:"group_id,omitempty"`
	ServiceName    string                 `json:"service_name"`
	SeverityText   string                 `json:"severity_text"`
	SeverityNumber int                    `json:"severity_number"`
	Body           string                 `json:"body"`
	TraceID        *string                `json:"trace_id,omitempty"`
	SpanID         *string                `json:"span_id,omitempty"`
	LogAttributes  map[string]interface{} `json:"log_attributes"`
	ConfigHash     *string                `json:"config_hash,omitempty"`
	// Deprecated: use SeverityText instead
	Severity string `json:"severity,omitempty"`
	// Deprecated: use LogAttributes instead
	Attributes map[string]string `json:"attributes,omitempty"`
}

type LogQuery struct {
	AgentID   *uuid.UUID
	GroupID   *string
	Severity  *string
	Search    *string
	StartTime time.Time
	EndTime   time.Time
	Limit     int
}

type Metric struct {
	Timestamp        time.Time              `json:"timestamp"`
	AgentID          uuid.UUID              `json:"agent_id"`
	GroupID          *string                `json:"group_id,omitempty"`
	ServiceName      string                 `json:"service_name"`
	Name             string                 `json:"metric_name"`
	Value            float64                `json:"value"`
	MetricAttributes map[string]interface{} `json:"metric_attributes"`
	ConfigHash       *string                `json:"config_hash,omitempty"`
	Labels           map[string]string      `json:"labels,omitempty"`
	Type             MetricType             `json:"type,omitempty"`
}

type MetricQuery struct {
	AgentID    *uuid.UUID
	GroupID    *string
	MetricName *string
	StartTime  time.Time
	EndTime    time.Time
	Limit      int
}

type MetricType string

type Rollup struct {
	WindowStart time.Time      `json:"window_start"`
	AgentID     *uuid.UUID     `json:"agent_id,omitempty"`
	GroupID     *string        `json:"group_id,omitempty"`
	MetricName  string         `json:"metric_name"`
	Count       int64          `json:"count"`
	Sum         float64        `json:"sum"`
	Avg         float64        `json:"avg"`
	Min         float64        `json:"min"`
	Max         float64        `json:"max"`
	Interval    RollupInterval `json:"interval"`
}

type RollupInterval string

type RollupQuery struct {
	AgentID    *uuid.UUID
	GroupID    *string
	MetricName *string
	StartTime  time.Time
	EndTime    time.Time
	Interval   RollupInterval
}

type TelemetryOverview struct {
	TotalMetrics int64     `json:"totalMetrics"`
	TotalLogs    int64     `json:"totalLogs"`
	TotalTraces  int64     `json:"totalTraces"`
	ActiveAgents int       `json:"activeAgents"`
	Services     []string  `json:"services"`
	LastUpdated  time.Time `json:"lastUpdated"`
}

type TelemetryQueryService interface {
	// Query operations
	QueryMetrics(ctx context.Context, query MetricQuery) ([]Metric, error)
	QueryLogs(ctx context.Context, query LogQuery) ([]Log, error)
	QueryTraces(ctx context.Context, query TraceQuery) ([]Trace, error)
	QueryRaw(ctx context.Context, query string, args ...interface{}) ([]map[string]interface{}, error)

	// Rollup operations
	CreateRollups(ctx context.Context, window time.Time, interval RollupInterval) error
	QueryRollups(ctx context.Context, query RollupQuery) ([]Rollup, error)

	// Cleanup operations
	CleanupOldData(ctx context.Context, retention time.Duration) error

	// Overview operations
	GetTelemetryOverview(ctx context.Context) (*TelemetryOverview, error)
	GetServices(ctx context.Context) ([]string, error)
}

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

type Trace struct {
	Timestamp     time.Time         `json:"timestamp"`
	AgentID       uuid.UUID         `json:"agent_id"`
	ConfigHash    *string           `json:"config_hash,omitempty"`
	TraceID       string            `json:"trace_id"`
	SpanID        string            `json:"span_id"`
	ParentSpanID  *string           `json:"parent_span_id,omitempty"`
	Name          string            `json:"name"`
	Duration      int64             `json:"duration"`
	StatusCode    string            `json:"status_code"`
	StatusMessage string            `json:"status_message"`
	Attributes    map[string]string `json:"attributes"`
}

type TraceQuery struct {
	AgentID   *uuid.UUID
	GroupID   *string
	TraceID   *string
	StartTime time.Time
	EndTime   time.Time
	Limit     int
}