Function Maintenance

internal/cmd/plugin/maintenance/maintenance.go:38–94

Maintenance command implementation

(ctx context.Context,
	allNamespaces, reusePVC, confirmationRequired bool,
	clusterName string,
	setInProgressTo bool,
)

// Maintenance command implementation
func Maintenance(ctx context.Context,
	allNamespaces, reusePVC, confirmationRequired bool,
	clusterName string,
	setInProgressTo bool,
) error {
	clusters := tabby.New()
	clusters.AddLine("The following are the new values for the clusters")
	clusters.AddHeader(
		"Namespace",
		"Cluster Name",
		"Maintenance",
		"reusePVC")

	var clusterList apiv1.ClusterList
	var err error
	if allNamespaces || clusterName == "" {
		clusterList, err = getClusters(ctx, allNamespaces)
	} else {
		clusterList, err = getCluster(ctx, clusterName)
	}
	if err != nil {
		return err
	}

	if len(clusterList.Items) == 0 {
		if clusterName != "" {
			return fmt.Errorf("cluster '%v' couldn't be found", clusterName)
		}
		return fmt.Errorf("no cluster could be listed or no permission to list clusters")
	}

	for _, item := range clusterList.Items {
		clusters.AddLine(
			item.Namespace,
			item.Name,
			fmt.Sprintf("%v => %v", item.IsNodeMaintenanceWindowInProgress(), setInProgressTo),
			fmt.Sprintf("%v => %v", item.IsReusePVCEnabled(), reusePVC),
		)
	}

	clusters.Print()
	if confirmationRequired {
		proceed := askToProceed()
		if !proceed {
			return nil
		}
	}

	for _, item := range clusterList.Items {
		err := patchNodeMaintenanceWindow(ctx, item, setInProgressTo, reusePVC)
		if err != nil {
			return fmt.Errorf("unable to set progress to cluster %v in namespace %v: %w", item.Name, item.Namespace, err)
		}
	}

	return nil
}