inventory

type Config struct {
	Version        semver.Version     `json:"version" yaml:"version"`
	ClusterStorage []string           `json:"cluster_storage" yaml:"cluster_storage"`
	Exclude        Exclude            `json:"exclude" yaml:"exclude"`
	Interconnect   ConfigInterconnect `json:"interconnect" yaml:"interconnect"`
	// contains filtered or unexported fields
}

type ConfigInterconnect struct {
	// ResourcePatterns is the list of glob patterns the inventory
	// operator matches kubelet's extended-resource names against to find
	// the interconnect HCA pool. A bare string with no glob metacharacter
	// is treated as a literal prefix match (preserves rc4 behavior).
	ResourcePatterns []string `json:"resource_patterns" yaml:"resource_patterns"`
}

type ConfigNodes struct {
	Exclude ExcludeRules `json:"exclude" yaml:"exclude"`
}

type ConfigStorage struct {
	Exclude ExcludeRules `json:"exclude" yaml:"exclude"`
}

type Exclude struct {
	Nodes       ExcludeRules        `json:"nodes" yaml:"nodes"`
	NodeStorage ExcludeNodesStorage `json:"node_storage" yaml:"node_storage"`
}

type ExcludeNodeStorage struct {
	NodeFilter *regexp.Regexp `json:"node_filter" yaml:"node_filter"`
	Classes    []string       `json:"classes" yaml:"classes"`
}

type ExcludeNodesStorage []ExcludeNodeStorage

type ExcludeRules []*regexp.Regexp

type IBDiscovery struct {
	// NCCLIBHCAPrefixes is every distinct HCA device-name family present
	// under /sys/class/infiniband — e.g. ["mlx5"] on a uniform Mellanox
	// host or ["mlx5","bnxt_re"] on a mixed-vendor node. Empty when no
	// devices are present. Surfaced into the provider as
	// NodeCapabilities.NCCLHCAPrefixes and joined with `,` by the
	// workload builder as NCCL_IB_HCA (NCCL accepts comma-separated
	// device prefixes natively).
	NCCLIBHCAPrefixes []string `json:"nccl_ib_hca_prefixes"`

	// Fabric is "infiniband" or "roce", read from the link_layer of port 1
	// on the first detected device. Empty when no devices are present.
	// Surfaced into the provider as NodeCapabilities.InterconnectFabric and used
	// by the inventory client to gate fabric-pinned bids.
	Fabric string `json:"fabric"`
}

type QuerierCluster interface {
	Query(ctx context.Context) (inventory.Cluster, error)
}

type QuerierNodes interface {
	Query(ctx context.Context) (inventory.Nodes, error)
}

type QuerierStorage interface{}

type RegistryGPUDevice struct {
	Name       string `json:"name"`
	Interface  string `json:"interface"`
	MemorySize string `json:"memory_size"`
}

type RegistryGPUDevices map[string]RegistryGPUDevice

type RegistryGPUVendor struct {
	Name    string             `json:"name"`
	Devices RegistryGPUDevices `json:"devices"`
}

type RegistryGPUVendors map[string]RegistryGPUVendor

type RemotePodCommandExecutor interface {
	ExecWithOptions(ctx context.Context, options rookexec.ExecOptions) (string, string, error)
	ExecCommandInContainerWithFullOutput(ctx context.Context, appLabel, containerName, namespace string, cmd ...string) (string, string, error)
	// ExecCommandInContainerWithFullOutputWithTimeout uses 15s hard-coded timeout
	ExecCommandInContainerWithFullOutputWithTimeout(ctx context.Context, appLabel, containerName, namespace string, cmd ...string) (string, string, error)
}

type Watcher interface {
	Watch(ctx context.Context, opts metav1.ListOptions) (watch.Interface, error)
}