add crd list and check storagecluster ownership

Signed-off-by: Oded Viner <[email protected]>

Author: OdedViner

pkg/crd/crd.go

@@ -26,13 +26,15 @@ import (
 	"github.com/rook/kubectl-rook-ceph/pkg/k8sutil"
 	"github.com/rook/kubectl-rook-ceph/pkg/logging"
 	apierrors "k8s.io/apimachinery/pkg/api/errors"
+	"k8s.io/apimachinery/pkg/runtime/schema"
 	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
 	"k8s.io/apimachinery/pkg/types"
 )
 
 func RestoreCrdNew(ctx context.Context, k8sclientset *k8sutil.Clientsets, operatorNamespace, clusterNamespace,
 	groupName, versionResource string, args []string) {
+
 	crd := args[0]
 
 	var crName string
@@ -125,6 +127,86 @@ func RestoreCrdNew(ctx context.Context, k8sclientset *k8sutil.Clientsets, operat
 func removeOwnerRefOfUID(ctx context.Context, k8sclientset *k8sutil.Clientsets, operatorNamespace, clusterNamespace, targetUID string) {
 	logging.Info("Removing ownerreferences from resources with matching uid %s", targetUID)
 
+	customResources := []struct {
+		group    string
+		version  string
+		resource string
+	}{
+		// ceph.rook.io/v1
+		{"ceph.rook.io", "v1", "cephblockpoolradosnamespaces"},
+		{"ceph.rook.io", "v1", "cephblockpools"},
+		{"ceph.rook.io", "v1", "cephbucketnotifications"},
+		{"ceph.rook.io", "v1", "cephbuckettopics"},
+		{"ceph.rook.io", "v1", "cephclients"},
+		{"ceph.rook.io", "v1", "cephclusters"},
+		{"ceph.rook.io", "v1", "cephcosidrivers"},
+		{"ceph.rook.io", "v1", "cephfilesystemmirrors"},
+		{"ceph.rook.io", "v1", "cephfilesystems"},
+		{"ceph.rook.io", "v1", "cephfilesystemsubvolumegroups"},
+		{"ceph.rook.io", "v1", "cephnfses"},
+		{"ceph.rook.io", "v1", "cephobjectrealms"},
+		{"ceph.rook.io", "v1", "cephobjectstores"},
+		{"ceph.rook.io", "v1", "cephobjectstoreusers"},
+		{"ceph.rook.io", "v1", "cephobjectzonegroups"},
+		{"ceph.rook.io", "v1", "cephobjectzones"},
+		{"ceph.rook.io", "v1", "cephrbdmirrors"},
+	
+		// noobaa.io/v1alpha1
+		{"noobaa.io", "v1alpha1", "backingstores"},
+		{"noobaa.io", "v1alpha1", "bucketclasses"},
+		{"noobaa.io", "v1alpha1", "namespacestores"},
+		{"noobaa.io", "v1alpha1", "noobaaaccounts"},
+		{"noobaa.io", "v1alpha1", "noobaas"},
+	
+		// postgresql.cnpg.noobaa.io/v1alpha1
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "backups"},
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "clusterimagecatalogs"},
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "clusters"},
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "databases"},
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "imagecatalogs"},
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "poolers"},
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "publications"},
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "scheduledbackups"},
+		{"postgresql.cnpg.noobaa.io", "v1alpha1", "subscriptions"},
+	}
+	for _, crd := range customResources {
+		gvr := schema.GroupVersionResource{
+			Group:    crd.group,
+			Version:  crd.version,
+			Resource: crd.resource,
+		}
+	
+		crList, err := k8sclientset.ListResourcesDynamically(ctx, crd.group, crd.version, crd.resource, clusterNamespace)
+		if err != nil {
+			logging.Warning("Failed to list %s: %v", crd.resource, err)
+			continue
+		}
+	
+		for _, cr := range crList {
+			updated := false
+			owners := cr.GetOwnerReferences()
+			newOwners := []v1.OwnerReference{}
+	
+			for _, owner := range owners {
+				if string(owner.UID) != targetUID {
+					newOwners = append(newOwners, owner)
+				} else {
+					updated = true
+				}
+			}
+	
+			if updated {
+				logging.Info("Removing ownerReference for %s/%s", crd.resource, cr.GetName())
+				cr.SetOwnerReferences(newOwners)
+	
+				_, err := k8sclientset.Dynamic.Resource(gvr).Namespace(clusterNamespace).Update(ctx, &cr, v1.UpdateOptions{})
+				if err != nil {
+					logging.Fatal(fmt.Errorf("Failed to update ownerReferences for %s/%s: %v", crd.resource, cr.GetName(), err))
+				}
+			}
+		}
+	}
+
 	secrets, err := k8sclientset.Kube.CoreV1().Secrets(clusterNamespace).List(ctx, v1.ListOptions{})
 	if err != nil {
 		logging.Fatal(errors.Wrapf(err, "Failed to list secrets"))
@@ -234,4 +316,5 @@ func removeOwnerRefOfUID(ctx context.Context, k8sclientset *k8sutil.Clientsets,
 			}
 		}
 	}
+
 }