feat: add instructions to patch system-upgrade-controller

Signed-off-by: PoAn Yang <[email protected]>

Author: FrankYang0529

pkg/plan/bootstrap.go

@@ -156,6 +156,28 @@ func (p *plan) addInstructions(cfg *config.Config, dataDir string) error {
 		}
 	}
 
+	// Above Rancher v2.10.x, we should patch system-upgrade-controller with meta.helm.sh/release-name:mcc-local-managed-system-upgrade-controller to continue cluster intialization.
+	if semver.Compare(cfg.RancherVersion, "v2.10.0") >= 0 {
+		if err := p.addInstruction(rancher.ToWaitMccLocalManagedSystemUpgradeControllerInstruction(k8sVersion)); err != nil {
+			return err
+		}
+		if err := p.addInstruction(rancher.ToPatchSUCServiceAccountInstruction(k8sVersion)); err != nil {
+			return err
+		}
+		if err := p.addInstruction(rancher.ToPatchSUCClusterRoleBindingInstruction(k8sVersion)); err != nil {
+			return err
+		}
+		if err := p.addInstruction(rancher.ToPatchSUCConfigMapInstruction(k8sVersion)); err != nil {
+			return err
+		}
+		if err := p.addInstruction(rancher.ToPatchSUCDeploymentInstruction(k8sVersion)); err != nil {
+			return err
+		}
+		if err := p.addInstruction(rancher.ToDeleteSUCInstruction(k8sVersion)); err != nil {
+			return err
+		}
+	}
+
 	if err := p.addInstruction(rancher.ToWaitSUCInstruction(cfg.RancherInstallerImage, cfg.SystemDefaultRegistry, k8sVersion)); err != nil {
 		return err
 	}

pkg/rancher/wait.go

@@ -52,6 +52,90 @@ func ToWaitSUCInstruction(_, _, k8sVersion string) (*applyinator.Instruction, er
 	}, nil
 }
 
+func ToWaitMccLocalManagedSystemUpgradeControllerInstruction(k8sVersion string) (*applyinator.Instruction, error) {
+	cmd, err := self.Self()
+	if err != nil {
+		return nil, fmt.Errorf("resolving location of %s: %w", os.Args[0], err)
+	}
+	return &applyinator.Instruction{
+		Name:       "wait-system-upgrade-controller",
+		SaveOutput: true,
+		Args:       []string{"retry", kubectl.Command(k8sVersion), "-n", "fleet-local", "get", "managedchart", "mcc-local-managed-system-upgrade-controller"},
+		Env:        kubectl.Env(k8sVersion),
+		Command:    cmd,
+	}, nil
+}
+
+func ToPatchSUCServiceAccountInstruction(k8sVersion string) (*applyinator.Instruction, error) {
+	cmd, err := self.Self()
+	if err != nil {
+		return nil, fmt.Errorf("resolving location of %s: %w", os.Args[0], err)
+	}
+	return &applyinator.Instruction{
+		Name:       "patch-system-upgrade-controller-serviceaccount",
+		SaveOutput: true,
+		Args:       []string{"retry", kubectl.Command(k8sVersion), "annotate", "--overwrite", "serviceaccount", "-n", "cattle-system", "system-upgrade-controller", "meta.helm.sh/release-name=mcc-local-managed-system-upgrade-controller"},
+		Env:        kubectl.Env(k8sVersion),
+		Command:    cmd,
+	}, nil
+}
+
+func ToPatchSUCConfigMapInstruction(k8sVersion string) (*applyinator.Instruction, error) {
+	cmd, err := self.Self()
+	if err != nil {
+		return nil, fmt.Errorf("resolving location of %s: %w", os.Args[0], err)
+	}
+	return &applyinator.Instruction{
+		Name:       "patch-system-upgrade-controller-configmap",
+		SaveOutput: true,
+		Args:       []string{"retry", kubectl.Command(k8sVersion), "annotate", "--overwrite", "configmap", "-n", "cattle-system", "system-upgrade-controller-config", "meta.helm.sh/release-name=mcc-local-managed-system-upgrade-controller"},
+		Env:        kubectl.Env(k8sVersion),
+		Command:    cmd,
+	}, nil
+}
+
+func ToPatchSUCClusterRoleBindingInstruction(k8sVersion string) (*applyinator.Instruction, error) {
+	cmd, err := self.Self()
+	if err != nil {
+		return nil, fmt.Errorf("resolving location of %s: %w", os.Args[0], err)
+	}
+	return &applyinator.Instruction{
+		Name:       "patch-system-upgrade-controller-clusterrolebinding",
+		SaveOutput: true,
+		Args:       []string{"retry", kubectl.Command(k8sVersion), "annotate", "--overwrite", "clusterrolebinding", "-n", "cattle-system", "system-upgrade-controller", "meta.helm.sh/release-name=mcc-local-managed-system-upgrade-controller"},
+		Env:        kubectl.Env(k8sVersion),
+		Command:    cmd,
+	}, nil
+}
+
+func ToPatchSUCDeploymentInstruction(k8sVersion string) (*applyinator.Instruction, error) {
+	cmd, err := self.Self()
+	if err != nil {
+		return nil, fmt.Errorf("resolving location of %s: %w", os.Args[0], err)
+	}
+	return &applyinator.Instruction{
+		Name:       "patch-system-upgrade-controller-deployment",
+		SaveOutput: true,
+		Args:       []string{"retry", kubectl.Command(k8sVersion), "annotate", "--overwrite", "deployment", "-n", "cattle-system", "system-upgrade-controller", "meta.helm.sh/release-name=mcc-local-managed-system-upgrade-controller"},
+		Env:        kubectl.Env(k8sVersion),
+		Command:    cmd,
+	}, nil
+}
+
+func ToDeleteSUCInstruction(k8sVersion string) (*applyinator.Instruction, error) {
+	cmd, err := self.Self()
+	if err != nil {
+		return nil, fmt.Errorf("resolving location of %s: %w", os.Args[0], err)
+	}
+	return &applyinator.Instruction{
+		Name:       "patch-system-upgrade-controller-deployment",
+		SaveOutput: true,
+		Args:       []string{"retry", kubectl.Command(k8sVersion), "delete", "bundle", "-n", "cattle-system", "mcc-local-managed-system-upgrade-controller"},
+		Env:        kubectl.Env(k8sVersion),
+		Command:    cmd,
+	}, nil
+}
+
 func ToWaitSUCPlanInstruction(_, _, k8sVersion string) (*applyinator.Instruction, error) {
 	cmd, err := self.Self()
 	if err != nil {