cluster

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

type ClusterClient interface {
	// WaitForNodesHealthy waits for nodes to be healthy and optionally match a specific version
	// Polls until all nodes are healthy (and correct version if specified) or timeout
	// skipServices is a list of service names to ignore during health checks
	WaitForNodesHealthy(ctx context.Context, nodeAddresses []string, expectedVersion string, skipServices []string) error

	// CaptureNodeBootIDs reads each node's current kernel boot ID, for later comparison
	// by WaitForNodesReboot. A node whose boot ID can't be read (e.g. permission denied,
	// unreachable) is simply omitted from the result; WaitForNodesReboot skips the
	// per-node check for any node missing from preActionBootIDs.
	CaptureNodeBootIDs(ctx context.Context, nodeAddresses []string) map[string]string

	// WaitForNodesReboot waits for nodes to reboot (confirmed by each node's kernel boot ID
	// changing from preActionBootIDs) then come back healthy. Phase 1 polls each node's
	// boot ID until it differs from before or offlineTimeout elapses; a node missing from
	// preActionBootIDs is treated as already satisfied. Phase 2 polls until all nodes are
	// healthy again within the remaining context deadline.
	WaitForNodesReboot(ctx context.Context, nodeAddresses []string, preActionBootIDs map[string]string, expectedVersion string, skipServices []string, offlineTimeout time.Duration) error

	// UpgradeNodes upgrades the specified nodes to the specified image. powercycle requests
	// a full ACPI reboot instead of the default kexec.
	UpgradeNodes(ctx context.Context, nodeAddresses []string, image string, powercycle bool) error

	// WaitForControlPlaneAPIReady waits for the kube-apiserver on a control-plane node
	// to accept TCP connections on port 6443. Returns nil immediately when the node is
	// not a control-plane (i.e. no etcd service present). outputFunc, when non-nil,
	// receives status messages during the polling loop. Returns an error if the node
	// role cannot be determined or the apiserver does not become reachable before the
	// context deadline.
	WaitForControlPlaneAPIReady(ctx context.Context, nodeAddress string, outputFunc func(string)) error

	// Close closes any open connections.
	Close()
}

type MockClusterClient struct {
	BaseClusterClient
	WaitForNodesHealthyFunc         func(ctx context.Context, nodeAddresses []string, expectedVersion string, skipServices []string) error
	CaptureNodeBootIDsFunc          func(ctx context.Context, nodeAddresses []string) map[string]string
	WaitForNodesRebootFunc          func(ctx context.Context, nodeAddresses []string, preActionBootIDs map[string]string, expectedVersion string, skipServices []string, offlineTimeout time.Duration) error
	UpgradeNodesFunc                func(ctx context.Context, nodeAddresses []string, image string, powercycle bool) error
	WaitForControlPlaneAPIReadyFunc func(ctx context.Context, nodeAddress string, outputFunc func(string)) error
	CloseFunc                       func()
}

type Shims struct {
	// Talos client operations
	TalosConfigOpen  func(configPath string) (*clientconfig.Config, error)
	TalosNewClient   func(ctx context.Context, opts ...client.OptionFunc) (*client.Client, error)
	TalosVersion     func(ctx context.Context, client *client.Client) (*machine.VersionResponse, error)
	TalosWithNodes   func(ctx context.Context, nodes ...string) context.Context
	TalosServiceList func(ctx context.Context, client *client.Client) (*machine.ServiceListResponse, error)
	TalosUpgrade     func(ctx context.Context, client *client.Client, image string, powercycle bool) error
	TalosRead        func(ctx context.Context, client *client.Client, path string) (io.ReadCloser, error)
	TalosClose       func(client *client.Client)

	// Network operations
	NetDialTimeout func(network, address string, timeout time.Duration) (net.Conn, error)
}

type TalosClusterClient struct {
	*BaseClusterClient
	// contains filtered or unexported fields
}