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
	HostUptimeChecker    reboot.HostUptimeChecker
	DPUInProvisioningMap *DPUInProvisioningMap
}

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
	// OSInstallTimeout is the maximum time allowed for OS installation in zero-trust mode.
	// Default: 45 minutes
	OSInstallTimeout time.Duration
	// FirmwareUpdateTimeout is the maximum time allowed for firmware update in zero-trust mode.
	// Default: 45 minutes
	FirmwareUpdateTimeout time.Duration
	// NodeEffectRemovalTimeout is the maximum time allowed for the Node Effect Removal phase.
	// Default: 30 minutes
	NodeEffectRemovalTimeout time.Duration
}

type IsoRootFile struct {
	Name string
	Data []byte
}

type KubeadmBootstrapTokenGenerator struct {
	client.Client
}

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

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