util

type ArmRestartTracker struct {
	// Attempt is the current attempt count (1-based, 0 = not started)
	Attempt int `json:"attempt"`
	// MaxAttempts is set by caller (e.g., 2 for Secure Boot)
	MaxAttempts int `json:"maxAttempts"`
	// LastRestartTime is used for timeout/interval checks
	LastRestartTime time.Time `json:"lastRestartTime"`
	// InitialGeneration detects spec changes during flow
	InitialGeneration int64 `json:"initialGeneration"`
}

type BF4Artifact struct {
	UserData      []byte
	NetworkConfig []byte
}

type BlueFieldSpecs struct {
	// CPU is the number of cores
	CPU int
	// Mem is the memory size in GB.
	// Note: GB should be converted to "Gi" or "G" before comparison.
	Mem string
}

type CanProceedError struct {
	Reason  string
	Message string
}

type CapacityResult int

type ControllerContext struct {
	client.Client
	Scheme               *runtime.Scheme
	Options              DPUOptions
	Recorder             record.EventRecorder
	ClusterAllocator     allocator.Allocator
	JoinCommandGenerator NodeJoinCommandGenerator
	DPUArtifactGenerator DPUArtifactGenerator
	DPUInProvisioningMap *DPUInProvisioningMap
	DPFOperatorConfig    *operatorv1.DPFOperatorConfig
}

type DPUArtifactGenerator interface {
	GenerateBF3(ctx context.Context, req DPUArtifactRequest) ([]byte, error)
	GenerateBF4(ctx context.Context, req DPUArtifactRequest) (BF4Artifact, error)
}

type DPUArtifactRequest struct {
	ControllerContext *ControllerContext
	DPU               *provisioningv1.DPU
	Flavor            *provisioningv1.DPUFlavor
	BootstrapToken    string
}

type DPUID string

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

type DPUOptions struct {
	PrarprouterdImageWithTag    string
	ImagePullSecrets            []corev1.LocalObjectReference
	DPUInstallInterface         string
	DeploymentMode              string
	BFCFGTemplateFile           string
	BFBRegistry                 string
	BFBPVC                      string
	BFBRegistryLoadBalancer     string
	CustomCASecretName          string
	MaxDPUParallelInstallations int32
	// KubernetesAPIServerVIP is the Kubernetes API server VIP configured for the DMS/hostagent
	// Pod (from DPFOperatorConfig KubernetesAPIServerVIP, passed via --dms-pod-envs). It is added
	// to the bfb-registry server certificate SANs so the hostagent's VIP-based NodePort download
	// passes TLS verification. Empty when no VIP override is configured.
	KubernetesAPIServerVIP string
	// OSInstallTimeout is the maximum time allowed for OS installation.
	OSInstallTimeout time.Duration
	// OSInstallRetries is the maximum number of retryable OS installation attempts in
	// zero-trust mode before transitioning to Error. Defaults to DefaultOSInstallRetries.
	OSInstallRetries int32
	// FirmwareUpdateTimeout is the maximum time allowed for firmware update in zero-trust mode.
	FirmwareUpdateTimeout time.Duration
	// PreInstallAgentRegistrationTimeout is how long Initializing waits for preInstall.agentReported.
	// Set via --pre-install-agent-registration-timeout.
	PreInstallAgentRegistrationTimeout time.Duration
	// NodeEffectRemovalTimeout is the maximum time allowed for the Node Effect Removal phase.
	NodeEffectRemovalTimeout time.Duration
}

type IsoRootFile struct {
	Name string
	Data []byte
}

type KubeadmBootstrapTokenGenerator struct {
	client.Client
	TokenTTL time.Duration
}

type NodeJoinCommandGenerator interface {
	GenerateJoinCommand(ctx context.Context, dc *provisioningv1.DPUCluster, dpu *provisioningv1.DPU) (string, error)
}

type TaskWithRetry struct {
	Task       *future.Future
	RetryCount int
}