Add drain pre and post hook and ProvisioningGeneration

PostDrainHooks and PreDrainHook fields are added to the DrainOptions.
The pre drain hook will run after cordon but before drain. The
post drain hook will run after the node has been updated (for example
upgrading rke2) but before the node is uncordoned.

Author: ibuildthecloud

pkg/apis/catalog.cattle.io/v1/zz_generated_deepcopy.go

@@ -29,6 +29,8 @@ type RKEClusterSpecCommon struct {
 	AdditionalManifest    string                 `json:"additionalManifest,omitempty"`
 	Registries            *Registry              `json:"registries,omitempty"`
 	ETCD                  *ETCD                  `json:"etcd,omitempty"`
+	// Increment to force all nodes to re-provision
+	ProvisionGeneration int `json:"provisionGeneration,omitempty"`
 }
 
 type LocalClusterAuthEndpoint struct {
@@ -80,6 +82,18 @@ type DrainOptions struct {
 	Timeout int `json:"timeout"`
 	// SkipWaitForDeleteTimeoutSeconds If pod DeletionTimestamp older than N seconds, skip waiting for the pod.  Seconds must be greater than 0 to skip.
 	SkipWaitForDeleteTimeoutSeconds int `json:"skipWaitForDeleteTimeoutSeconds,omitempty"`
+
+	// PreDrainHooks A list of hooks to run prior to draining a node
+	PreDrainHooks []DrainHook `json:"preDrainHooks,omitempty"`
+	// PostDrainHook A list of hooks to run after draining AND UPDATING a node
+	PostDrainHooks []DrainHook `json:"postDrainHooks,omitempty"`
+}
+
+type DrainHook struct {
+	// Annotation This annotation will need to be populated on the capi.Machine with the value from the annotation
+	// "rke.cattle.io/pre-drain" before the planner will continue with drain the specific node.  The annotation
+	// "rke.cattle.io/pre-drain" is used for pre-drain and "rke.cattle.io/post-drain" is used for post drain.
+	Annotation string `json:"annotation,omitempty"`
 }
 
 type Endpoint struct {

pkg/apis/cluster.cattle.io/v3/zz_generated_deepcopy.go

@@ -5,6 +5,7 @@ const (
 	CloudProviderFieldAWSCloudProvider       = "awsCloudProvider"
 	CloudProviderFieldAzureCloudProvider     = "azureCloudProvider"
 	CloudProviderFieldCustomCloudProvider    = "customCloudProvider"
+	CloudProviderFieldHarvesterCloudProvider = "harvesterCloudProvider"
 	CloudProviderFieldName                   = "name"
 	CloudProviderFieldOpenstackCloudProvider = "openstackCloudProvider"
 	CloudProviderFieldVsphereCloudProvider   = "vsphereCloudProvider"
@@ -14,6 +15,7 @@ type CloudProvider struct {
 	AWSCloudProvider       *AWSCloudProvider       `json:"awsCloudProvider,omitempty" yaml:"awsCloudProvider,omitempty"`
 	AzureCloudProvider     *AzureCloudProvider     `json:"azureCloudProvider,omitempty" yaml:"azureCloudProvider,omitempty"`
 	CustomCloudProvider    string                  `json:"customCloudProvider,omitempty" yaml:"customCloudProvider,omitempty"`
+	HarvesterCloudProvider *HarvesterCloudProvider `json:"harvesterCloudProvider,omitempty" yaml:"harvesterCloudProvider,omitempty"`
 	Name                   string                  `json:"name,omitempty" yaml:"name,omitempty"`
 	OpenstackCloudProvider *OpenstackCloudProvider `json:"openstackCloudProvider,omitempty" yaml:"openstackCloudProvider,omitempty"`
 	VsphereCloudProvider   *VsphereCloudProvider   `json:"vsphereCloudProvider,omitempty" yaml:"vsphereCloudProvider,omitempty"`

pkg/apis/management.cattle.io/v3/zz_generated_deepcopy.go

@@ -0,0 +1,10 @@
+package client
+
+const (
+	HarvesterCloudProviderType             = "harvesterCloudProvider"
+	HarvesterCloudProviderFieldCloudConfig = "cloudConfig"
+)
+
+type HarvesterCloudProvider struct {
+	CloudConfig string `json:"cloudConfig,omitempty" yaml:"cloudConfig,omitempty"`
+}