config

type Alert struct {
	Labels       LabelSet `json:"labels"`
	GeneratorURL string   `json:"generatorURL,omitempty"`
}

type AlertRule struct {
	Name                string  `yaml:"name"`
	Query               string  `yaml:"query"`
	Threshold           float64 `yaml:"threshold"`
	EvaluationInterval  string  `yaml:"evaluationInterval"`
	ConsecutiveCount    int     `yaml:"consecutiveCount"`
	Severity            string  `yaml:"severity"`
	Summary             string  `yaml:"summary"`
	Description         string  `yaml:"description"`
	RunBookURL          string  `yaml:"runbookURL"`
	AlertTargetInstance string  `yaml:"alertTargetInstance"`

	// when the rule is in test mode, it will not send alerts to alert manager
	TestMode bool

	FiringAlerts map[string]*FiringAlertCache
	// contains filtered or unexported fields
}

type AutoMigrationConfig struct {
	Enable bool                `yaml:"enable"`
	Scope  *AutoMigrationScope `yaml:"scope"`
}

type AutoMigrationRules struct {
	NamespaceNames    []string              `yaml:"namespaceNames"`
	NamespaceSelector *metav1.LabelSelector `yaml:"namespaceSelector"`
	PodSelector       *metav1.LabelSelector `yaml:"podSelector"`
}

type AutoMigrationScope struct {
	Includes *AutoMigrationRules `yaml:"includes"`
	Excludes *AutoMigrationRules `yaml:"excludes"`
}

type FiringAlertCache struct {
	Alert PostableAlert
	Count int
}

type GPUFitConfig struct {
	MaxWorkerPerNode int `json:"maxWorkerPerNode"`

	VramWeight   float64 `json:"vramWeight"`
	TflopsWeight float64 `json:"tflopsWeight"`
}

type GPUNetworkTopologyAwareConfig struct {
	TotalIntranetBandWidthGBps int64 `json:"totalIntranetBandWidthGBps"`
}

type GlobalConfig struct {
	MetricsTTL            string            `yaml:"metricsTTL"`
	MetricsFormat         string            `yaml:"metricsFormat"`
	MetricsExtraPodLabels map[string]string `yaml:"metricsExtraPodLabels"`

	AlertRules    []AlertRule          `yaml:"alertRules"`
	AutoMigration *AutoMigrationConfig `yaml:"autoMigration"`

	AutoScalingInterval  string `yaml:"autoScalingInterval"`
	GPUOperatorNamespace string `yaml:"gpuOperatorNamespace"`
}

type GpuInfo struct {
	Model         string            `json:"model"`
	Vendor        string            `json:"vendor"`
	CostPerHour   float64           `json:"costPerHour"`
	Fp16TFlops    resource.Quantity `json:"fp16TFlops"`
	FullModelName string            `json:"fullModelName"`

	// PartitionTemplates contains available partition templates for this GPU (e.g., MIG profiles)
	// Only applicable for GPUs that support hardware partitioning
	PartitionTemplates []PartitionTemplateInfo `json:"partitionTemplates,omitempty"`

	// MaxPartitions is the maximum number of partitions this GPU can support (e.g., 7 for MIG)
	MaxPartitions uint32 `json:"maxPartitions,omitempty"`

	// MaxPlacementSlots is the maximum number of placement slots this GPU can support (e.g., 8 for NVIDIA MIG)
	MaxPlacementSlots uint32 `json:"maxPlacementSlots,omitempty"`

	// MaxIsolationGroups is the maximum number of isolation groups (e.g., 4 for Ascend vGroups)
	// If not set, defaults to MaxPlacementSlots for backward compatibility
	MaxIsolationGroups uint32 `json:"maxIsolationGroups,omitempty"`

	// TotalExtendedResources defines the total capacity of extended resources for this GPU
	// For Ascend NPU: {"AICORE": 8, "AICPU": 7, "VPC": 12, "VENC": 3, "VDEC": 12, "JPEGD": 16, "JPEGE": 8}
	TotalExtendedResources map[string]uint32 `json:"totalExtendedResources,omitempty"`
}

type LabelSet map[string]string

type PartitionTemplateInfo struct {
	// TemplateID is the unique identifier for this partition template Profile `19` for 1g.10gb in A100
	TemplateID string `json:"templateId"`

	// Name is the unique identifier (e.g., "1g.24gb", "4g.94gb", "vir01", "vir04")
	Name string `json:"name"`

	// MemoryGigabytes is the memory allocated to this partition in gigabytes
	MemoryGigabytes uint64 `json:"memoryGigabytes"`

	// ComputePercent is the percent of sliced GPU (0-100)
	ComputePercent float64 `json:"computePercent"`

	// Description provides additional information about this template
	Description string `json:"description,omitempty"`

	// MaxPartition for this single template, eg. 1g.10gb+me can only be allocate once
	MaxPartition uint32 `json:"maxPartition"`

	// The placement limit for this template, use a bitmask to represent the placement limit
	// e.g. sudo nvidia-smi mig -i 0 -lgipp
	// GPU  0 Profile ID 19 Placements: {0,1,2,3,4,5,6}:1
	// GPU  0 Profile ID 20 Placements: {0,1,2,3,4,5,6}:1
	// GPU  0 Profile ID 15 Placements: {0,2,4,6}:2
	// GPU  0 Profile ID 14 Placements: {0,2,4}:2
	// GPU  0 Profile ID  9 Placements: {0,4}:4
	// GPU  0 Profile ID  5 Placement : {0}:4
	// GPU  0 Profile ID  0 Placement : {0}:8
	PlacementLimit  []uint32 `json:"placementLimit"`
	PlacementOffSet uint32   `json:"placementOffSet"`

	// ExtendedResources contains additional resource dimensions beyond compute and memory
	// For Ascend NPU: {"AICORE": 1, "AICPU": 1, "VPC": 1, "VENC": 0, "VDEC": 1, "JPEGD": 2, "JPEGE": 1, "PNGD": 0}
	// This enables multi-dimensional resource tracking for different accelerator types
	ExtendedResources map[string]uint32 `json:"extendedResources,omitempty"`

	// IsolationGroupSharing defines how isolation groups (vGroups for Ascend) are handled
	// "exclusive" - each partition requires its own isolation group (default for most templates)
	// "shared" - multiple partitions of this template can share an isolation group (time-sharing)
	// For Ascend vir01: shared (can have 2 vNPUs per vGroup with time-sharing)
	// For Ascend vir02/vir04: exclusive (each requires its own vGroup)
	IsolationGroupSharing string `json:"isolationGroupSharing,omitempty"`

	// MaxPartitionsPerIsolationGroup limits how many partitions of this template can share one group
	// Only applicable when IsolationGroupSharing is "shared"
	// For Ascend vir01: 2 (two vir01 can share one vGroup)
	MaxPartitionsPerIsolationGroup uint32 `json:"maxPartitionsPerIsolationGroup,omitempty"`

	// IsolationGroupSlots defines the minimum AI Core slots required by this template's isolation group
	// For Ascend: vir01 uses 1 AICORE but vGroup requires min 2 AICORE
	// vir02/vir04 use 2/4 AICORE respectively
	IsolationGroupSlots uint32 `json:"isolationGroupSlots,omitempty"`
}

type PostableAlert struct {
	Alert
	StartsAt    time.Time `json:"startsAt,omitempty"`
	EndsAt      time.Time `json:"endsAt,omitempty"`
	Annotations LabelSet  `json:"annotations,omitempty"`
}