psmapi

type AvailableFirmware struct {
	PMCMACAddress string
	Upgrades      []ComponentFirmwareUpgrades
}

type Chassis struct {
	SerialNumber string
	Model        string
	Manufacturer string
}

type Client interface {
	// GetPowershelves returns powershelf information for the specified PMC MAC addresses.
	// If pmcMacs is empty, returns all powershelves.
	GetPowershelves(ctx context.Context, pmcMacs []string) ([]PowerShelf, error)

	// RegisterPowershelves registers new powershelves with their PMC credentials.
	RegisterPowershelves(ctx context.Context, requests []RegisterPowershelfRequest) ([]RegisterPowershelfResponse, error)

	// PowerOn powers on the specified powershelves.
	PowerOn(ctx context.Context, pmcMacs []string) ([]PowerControlResult, error)

	// PowerOff powers off the specified powershelves.
	PowerOff(ctx context.Context, pmcMacs []string) ([]PowerControlResult, error)

	// UpdateFirmware performs firmware upgrades on the specified powershelves.
	UpdateFirmware(ctx context.Context, requests []UpdatePowershelfFirmwareRequest) ([]UpdatePowershelfFirmwareResponse, error)

	// GetFirmwareUpdateStatus returns the status of firmware updates for the specified PMC(s) and component(s).
	GetFirmwareUpdateStatus(ctx context.Context, queries []FirmwareUpdateQuery) ([]FirmwareUpdateStatus, error)

	// ListAvailableFirmware lists the firmware versions available for the specified powershelves.
	ListAvailableFirmware(ctx context.Context, pmcMacs []string) ([]AvailableFirmware, error)

	// SetDryRun configures whether the firmware manager is in Dry Run mode.
	SetDryRun(ctx context.Context, dryRun bool) error

	// Close closes the underlying gRPC connection.
	Close() error

	// The following are only valid in the mock environment and should only be called by unit tests.
	AddPowershelf(PowerShelf)
}

type ComponentFirmwareUpgrades struct {
	Component PowershelfComponent
	Upgrades  []FirmwareVersion
}

type Credentials struct {
	Username string
	Password string
}

type FirmwareUpdateQuery struct {
	PMCMACAddress string
	Component     PowershelfComponent
}

type FirmwareUpdateState int

type FirmwareUpdateStatus struct {
	PMCMACAddress string
	Component     PowershelfComponent
	State         FirmwareUpdateState
	Status        StatusCode
	Error         string
}

type FirmwareVersion struct {
	Version string
}

type PMCVendor int

type PowerControlResult struct {
	PMCMACAddress string
	Status        StatusCode
	Error         string
}

type PowerManagementController struct {
	MACAddress      string
	IPAddress       string
	Vendor          PMCVendor
	SerialNumber    string
	Model           string
	Manufacturer    string
	PartNumber      string
	FirmwareVersion string
	HardwareVersion string
}

type PowerShelf struct {
	PMC     PowerManagementController
	Chassis Chassis
	PSUs    []PowerSupplyUnit
}

type PowerSupplyUnit struct {
	ID              string
	Name            string
	Manufacturer    string
	Model           string
	SerialNumber    string
	CapacityWatts   string
	FirmwareVersion string
	HardwareVersion string
	PowerState      bool
	Sensors         []Sensor
}

type PowershelfComponent int

type RegisterPowershelfRequest struct {
	PMCMACAddress  string
	PMCIPAddress   string
	PMCVendor      PMCVendor
	PMCCredentials Credentials
}

type RegisterPowershelfResponse struct {
	PMCMACAddress string
	IsNew         bool
	Created       time.Time
	Status        StatusCode
	Error         string
}

type Sensor struct {
	ID              string
	Name            string
	Reading         float32
	ReadingRangeMax float64
	ReadingRangeMin float32
	ReadingType     string
	ReadingUnits    string
	Thresholds      SensorThresholds
}

type SensorThresholds struct {
	LowerCaution  float32
	LowerCritical float32
	UpperCaution  float32
	UpperCritical float32
}

type StatusCode int

type UpdateComponentFirmwareRequest struct {
	Component PowershelfComponent
	UpgradeTo FirmwareVersion
}

type UpdateComponentFirmwareResponse struct {
	Component PowershelfComponent
	Status    StatusCode
	Error     string
}

type UpdatePowershelfFirmwareRequest struct {
	PMCMACAddress string
	Components    []UpdateComponentFirmwareRequest
}

type UpdatePowershelfFirmwareResponse struct {
	PMCMACAddress string
	Components    []UpdateComponentFirmwareResponse
}