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

	// WaitForNodesReboot waits for nodes to go offline (reboot started) then come back healthy.
	// Phase 1 polls the Talos version endpoint until all nodes are unreachable (offlineTimeout caps this phase).
	// Phase 2 polls until all nodes are healthy again within the remaining context deadline.
	WaitForNodesReboot(ctx context.Context, nodeAddresses []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
	WaitForNodesRebootFunc          func(ctx context.Context, nodeAddresses []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
	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
}