aws

type Capabilities struct {
	InstanceType         string   `json:"instance_type"`
	Found                bool     `json:"found"`
	Architectures        []string `json:"architectures,omitempty"` // e.g. ["x86_64"], ["arm64"]
	ClusterPlacement     bool     `json:"cluster_placement"`       // supports the "cluster" placement strategy (MPI)
	EFA                  bool     `json:"efa"`                     // supports Elastic Fabric Adapter
	Hibernation          bool     `json:"hibernation"`             // supports On-Demand hibernation
	NestedVirtualization bool     `json:"nested_virtualization"`   // can run KVM/Hyper-V in-instance
	GPUs                 int32    `json:"gpus,omitempty"`
	BareMetal            bool     `json:"bare_metal"`
}

type CapacityBlockOfferingOptions struct {
	InstanceType          string // required
	InstanceCount         int32  // required by us; defaults to 1
	CapacityDurationHours int32  // required by AWS
	StartAfter            string // ISO-8601; earliest block start → StartDateRange
	EndBy                 string // ISO-8601; latest block end → EndDateRange
	Verbose               bool
}

type CapacityBlockOfferingResult struct {
	OfferingID       string `json:"offering_id" yaml:"offering_id"`
	InstanceType     string `json:"instance_type" yaml:"instance_type"`
	InstanceCount    int32  `json:"instance_count" yaml:"instance_count"`
	AvailabilityZone string `json:"availability_zone" yaml:"availability_zone"`
	Region           string `json:"region" yaml:"region"`
	StartDate        string `json:"start_date" yaml:"start_date"`
	EndDate          string `json:"end_date" yaml:"end_date"`
	DurationHours    int32  `json:"duration_hours" yaml:"duration_hours"`
	UpfrontFee       string `json:"upfront_fee" yaml:"upfront_fee"` // total up-front price (string per AWS)
	CurrencyCode     string `json:"currency_code" yaml:"currency_code"`
	Tenancy          string `json:"tenancy,omitempty" yaml:"tenancy,omitempty"`
}

type CapacityBlockOptions struct {
	InstanceTypes []string
	MinDuration   int32  // Minimum duration in hours
	MaxDuration   int32  // Maximum duration in hours
	StartAfter    string // ISO format datetime
	StartBefore   string // ISO format datetime
	OnlyActive    bool
	Verbose       bool
}

type CapacityBlockResult struct {
	CapacityBlockID       string   `json:"capacity_block_id" yaml:"capacity_block_id"`
	InstanceType          string   `json:"instance_type" yaml:"instance_type"`
	InstanceCount         int32    `json:"instance_count" yaml:"instance_count"`
	AvailabilityZone      string   `json:"availability_zone" yaml:"availability_zone"`
	StartDate             string   `json:"start_date" yaml:"start_date"`
	EndDate               string   `json:"end_date" yaml:"end_date"`
	DurationHours         int32    `json:"duration_hours" yaml:"duration_hours"`
	State                 string   `json:"state" yaml:"state"` // scheduled, active, completed, cancelled
	ReservationFee        float64  `json:"reservation_fee" yaml:"reservation_fee"`
	UltraClusterPlacement bool     `json:"ultra_cluster_placement" yaml:"ultra_cluster_placement"`
	Tags                  []string `json:"tags,omitempty" yaml:"tags,omitempty"`
}

type CapacityReservationOptions struct {
	InstanceTypes  []string
	OnlyAvailable  bool  // Only show reservations with available capacity
	OnlyActive     bool  // Only show active reservations
	IncludeExpired bool  // Include expired reservations
	MinCapacity    int32 // Minimum available capacity
	Verbose        bool
}

type CapacityReservationResult struct {
	ReservationID     string   `json:"reservation_id" yaml:"reservation_id"`
	InstanceType      string   `json:"instance_type" yaml:"instance_type"`
	Region            string   `json:"region" yaml:"region"`
	AvailabilityZone  string   `json:"availability_zone" yaml:"availability_zone"`
	TotalCapacity     int32    `json:"total_capacity" yaml:"total_capacity"`
	AvailableCapacity int32    `json:"available_capacity" yaml:"available_capacity"`
	UsedCapacity      int32    `json:"used_capacity" yaml:"used_capacity"`
	State             string   `json:"state" yaml:"state"`
	Tenancy           string   `json:"tenancy" yaml:"tenancy"`
	EBSOptimized      bool     `json:"ebs_optimized" yaml:"ebs_optimized"`
	EndDate           string   `json:"end_date,omitempty" yaml:"end_date,omitempty"`
	Platform          string   `json:"platform" yaml:"platform"`
	Tags              []string `json:"tags,omitempty" yaml:"tags,omitempty"`
}

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

type FilterOptions struct {
	IncludeAZs       bool    // If true, populate InstanceTypeResult.AvailableAZs (one extra API call per type)
	Architecture     string  // Filter to "x86_64" or "arm64"; empty matches both
	MinVCPUs         int     // Minimum vCPU count; 0 disables this filter
	MinMemory        float64 // Minimum memory in GiB; 0 disables this filter
	MinPhysicalCores int     // Minimum physical core count; 0 disables this filter
	ExactVCPUs       bool    // If true, match exact vCPU count instead of minimum
	ExactMemory      bool    // If true, match exact memory instead of minimum
	ExactCores       bool    // If true, match exact physical core count instead of minimum
	InstanceFamily   string  // Restrict to a family prefix, e.g. "m6i"; empty matches all
	NestedVirt       bool    // If true, only types supporting nested virtualization (KVM/Hyper-V in-instance)
	Verbose          bool    // If true, log per-region progress to stderr
}

type Finder interface {
	// GetEnabledRegions returns regions enabled for the caller's account.
	GetEnabledRegions(ctx context.Context) ([]string, error)

	// GetAllRegions returns all AWS regions (enabled or not).
	GetAllRegions(ctx context.Context) ([]string, error)

	// GetInstanceTypes returns all instance type names available in a region.
	GetInstanceTypes(ctx context.Context, region string) ([]string, error)

	// SearchInstanceTypes finds instance types matching a pattern across regions.
	SearchInstanceTypes(ctx context.Context, regions []string, matcher *regexp.Regexp, opts FilterOptions) ([]InstanceTypeResult, error)

	// GetSpotPricing retrieves current Spot pricing for instance types.
	GetSpotPricing(ctx context.Context, instances []InstanceTypeResult, opts SpotOptions) ([]SpotPriceResult, error)

	// GetCapacityReservations retrieves ODCRs across regions.
	GetCapacityReservations(ctx context.Context, regions []string, opts CapacityReservationOptions) ([]CapacityReservationResult, error)

	// GetCapacityBlocks retrieves Capacity Blocks for ML across regions.
	GetCapacityBlocks(ctx context.Context, regions []string, opts CapacityBlockOptions) ([]CapacityBlockResult, error)

	// GetCapacityBlockOfferings discovers purchasable Capacity Block offerings across regions.
	GetCapacityBlockOfferings(ctx context.Context, regions []string, opts CapacityBlockOfferingOptions) ([]CapacityBlockOfferingResult, error)

	// OnDemandPrice returns the on-demand $/hr for an instance type in a region.
	OnDemandPrice(ctx context.Context, instanceType, region string) (float64, error)

	// HourlyRate returns $/hr under a given purchase model ("on-demand" or "spot").
	HourlyRate(ctx context.Context, instanceType, region, model string) (float64, error)
}

type InstanceTypeResult struct {
	InstanceType    string   `json:"instance_type" yaml:"instance_type"`                                     // EC2 instance type, e.g. "m6i.2xlarge"
	Region          string   `json:"region" yaml:"region"`                                                   // AWS region where this type is available
	AvailableAZs    []string `json:"availability_zones,omitempty" yaml:"availability_zones,omitempty"`       // AZs with capacity; populated when FilterOptions.IncludeAZs is true
	VCPUs           int32    `json:"vcpus,omitempty" yaml:"vcpus,omitempty"`                                 // Default vCPU count
	PhysicalCores   int32    `json:"physical_cores,omitempty" yaml:"physical_cores,omitempty"`               // Physical CPU cores (vCPUs / threads-per-core)
	ThreadsPerCore  int32    `json:"threads_per_core,omitempty" yaml:"threads_per_core,omitempty"`           // Threads per physical core (1 for Graviton, 2 for most x86)
	MemoryMiB       int64    `json:"memory_mib,omitempty" yaml:"memory_mib,omitempty"`                       // Memory in MiB
	Architecture    string   `json:"architecture,omitempty" yaml:"architecture,omitempty"`                   // CPU architecture: "x86_64" or "arm64"
	InstanceFamily  string   `json:"instance_family,omitempty" yaml:"instance_family,omitempty"`             // Family prefix, e.g. "m6i"
	GPUs            int32    `json:"gpus,omitempty" yaml:"gpus,omitempty"`                                   // Number of GPUs; 0 for non-GPU instances
	GPUMemoryMiB    int64    `json:"gpu_memory_mib,omitempty" yaml:"gpu_memory_mib,omitempty"`               // Total GPU memory in MiB across all GPUs
	GPUModel        string   `json:"gpu_model,omitempty" yaml:"gpu_model,omitempty"`                         // GPU model name, e.g. "A100"
	GPUManufacturer string   `json:"gpu_manufacturer,omitempty" yaml:"gpu_manufacturer,omitempty"`           // GPU vendor, e.g. "nvidia"
	OnDemandPrice   float64  `json:"on_demand_price,omitempty" yaml:"on_demand_price,omitempty"`             // On-demand $/hr; 0 if not yet fetched
	SpawnSupported  bool     `json:"spawn_supported,omitempty" yaml:"spawn_supported,omitempty"`             // True if spawn can launch instances in this region
	NestedVirt      bool     `json:"nested_virtualization,omitempty" yaml:"nested_virtualization,omitempty"` // True if the type supports nested virtualization (KVM/Hyper-V in-instance)
	Service         string   `json:"service,omitempty" yaml:"service,omitempty"`                             // Offering namespace: "" / "ec2" (default) or "sagemaker" for ml.* types

	// SageMaker-only fields (populated when Service == "sagemaker"):
	ManagedSpotEligible bool     `json:"managed_spot_eligible,omitempty" yaml:"managed_spot_eligible,omitempty"` // Type can be used with managed spot training (has a "spot training job usage" quota). Managed spot is a billed-time discount (up to 90%), not a spot market — there is no per-type spot price.
	TrainingJobQuota    *float64 `json:"training_job_quota,omitempty" yaml:"training_job_quota,omitempty"`       // Account limit for "training job usage" of this type; nil when the region exposes no such quota. 0 means an increase must be requested before launching.
}

type OnDemandPricer interface {
	// OnDemandPrice returns the On-Demand $/hr for instanceType in region.
	// It returns (0, error) when no price could be determined.
	OnDemandPrice(ctx context.Context, instanceType, region string) (float64, error)
}

type SageMakerPricer interface {
	// SageMakerPrice returns the On-Demand $/hr for an ml.* instanceType in
	// region. It returns (0, error) when no price could be determined.
	SageMakerPrice(ctx context.Context, instanceType, region string) (float64, error)
}

type SageMakerTypeLister interface {
	// OfferedTypes returns the ml.*-prefixed types offered in region, with quota detail.
	OfferedTypes(ctx context.Context, region string) ([]quotas.SageMakerTypeQuota, error)
}

type SpotOptions struct {
	MaxPrice      float64 // Exclude results above this $/hr threshold; 0 disables
	ShowSavings   bool    // Populate SpotPriceResult.SavingsPercent by comparing with on-demand price
	LookbackHours int     // Hours of price history to query; 0 uses the AWS default (typically 1 hour)
	OnlyActive    bool    // Exclude price history entries with no current offering
	Verbose       bool    // If true, log per-region progress to stderr
}

type SpotPriceResult struct {
	InstanceType     string  `json:"instance_type" yaml:"instance_type"`                         // EC2 instance type
	Region           string  `json:"region" yaml:"region"`                                       // AWS region
	AvailabilityZone string  `json:"availability_zone" yaml:"availability_zone"`                 // AZ where this price applies
	SpotPrice        float64 `json:"spot_price" yaml:"spot_price"`                               // Current Spot price in $/hr
	OnDemandPrice    float64 `json:"on_demand_price,omitempty" yaml:"on_demand_price,omitempty"` // On-demand $/hr for savings calculation; 0 if unavailable
	SavingsPercent   float64 `json:"savings_percent,omitempty" yaml:"savings_percent,omitempty"` // Discount vs on-demand: 100*(1-spot/ondemand); set when SpotOptions.ShowSavings is true
	Timestamp        string  `json:"timestamp" yaml:"timestamp"`                                 // RFC3339 timestamp of the price observation
	ProductType      string  `json:"product_type,omitempty" yaml:"product_type,omitempty"`       // OS description, e.g. "Linux/UNIX"
}