Add additional helpers for rancher

Signed-off-by: Tamal Saha <tamal@appscode.com>

cluster/lib.go

@@ -132,7 +132,7 @@ func DetectClusterManager(kc client.Client) kmapi.ClusterManager {
 
 func IsDefault(kc client.Client, cm kmapi.ClusterManager, gvk schema.GroupVersionKind, key types.NamespacedName) (bool, error) {
 	if cm.ManagedByRancher() {
-		return IsRancherSystemResource(kc, key)
+		return IsInSystemProject(kc, key.Namespace)
 	}
 	return IsSingletonResource(kc, gvk, key)
 }

cluster/rancher.go

@@ -18,13 +18,12 @@ package cluster
 
 import (
 	"context"
-	"errors"
+	"sort"
 
 	core "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime/schema"
-	"k8s.io/apimachinery/pkg/types"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 )
 
@@ -40,17 +39,33 @@ func IsRancherManaged(mapper meta.RESTMapper) bool {
 	return false
 }
 
-func IsRancherSystemResource(kc client.Client, key types.NamespacedName) (bool, error) {
-	if !IsRancherManaged(kc.RESTMapper()) {
-		return false, errors.New("not a Rancher managed cluster")
+func IsInDefaultProject(kc client.Client, nsName string) (bool, error) {
+	return isInProject(kc, nsName, metav1.NamespaceDefault)
+}
+
+func IsInSystemProject(kc client.Client, nsName string) (bool, error) {
+	return isInProject(kc, nsName, metav1.NamespaceSystem)
+}
+
+func IsInUserProject(kc client.Client, nsName string) (bool, error) {
+	isDefault, err := IsInDefaultProject(kc, nsName)
+	if err != nil {
+		return false, err
 	}
+	isSys, err := IsInSystemProject(kc, nsName)
+	if err != nil {
+		return false, err
+	}
+	return !isDefault && !isSys, nil
+}
 
-	if key.Namespace == metav1.NamespaceSystem {
+func isInProject(kc client.Client, nsName, seedNS string) (bool, error) {
+	if nsName == seedNS {
 		return true, nil
 	}
 
 	var ns core.Namespace
-	err := kc.Get(context.TODO(), client.ObjectKey{Name: key.Namespace}, &ns)
+	err := kc.Get(context.TODO(), client.ObjectKey{Name: nsName}, &ns)
 	if err != nil {
 		return false, err
 	}
@@ -59,15 +74,74 @@ func IsRancherSystemResource(kc client.Client, key types.NamespacedName) (bool,
 		return false, nil
 	}
 
-	var sysNS core.Namespace
-	err = kc.Get(context.TODO(), client.ObjectKey{Name: metav1.NamespaceSystem}, &sysNS)
+	seedProjectId, _, err := GetProjectId(kc, seedNS)
 	if err != nil {
 		return false, err
 	}
+	return projectId == seedProjectId, nil
+}
 
-	sysProjectId, exists := ns.Labels[LabelKeyRancherProjectId]
-	if !exists {
-		return false, nil
+func GetDefaultProjectId(kc client.Client) (string, bool, error) {
+	return GetProjectId(kc, metav1.NamespaceDefault)
+}
+
+func GetSystemProjectId(kc client.Client) (string, bool, error) {
+	return GetProjectId(kc, metav1.NamespaceSystem)
+}
+
+func GetProjectId(kc client.Client, nsName string) (string, bool, error) {
+	var ns core.Namespace
+	err := kc.Get(context.TODO(), client.ObjectKey{Name: nsName}, &ns)
+	if err != nil {
+		return "", false, err
+	}
+	projectId, found := ns.Labels[LabelKeyRancherProjectId]
+	return projectId, found, nil
+}
+
+func ListSiblingNamespaces(kc client.Client, nsName string) ([]core.Namespace, error) {
+	projectId, found, err := GetProjectId(kc, nsName)
+	if err != nil || !found {
+		return nil, err
+	}
+	return ListProjectNamespaces(kc, projectId)
+}
+
+func AreSiblingNamespaces(kc client.Client, ns1, ns2 string) (bool, error) {
+	if ns1 == ns2 {
+		return true, nil
+	}
+
+	p1, found, err := GetProjectId(kc, ns1)
+	if err != nil || !found {
+		return false, err
+	}
+	p2, found, err := GetProjectId(kc, ns2)
+	if err != nil || !found {
+		return false, err
+	}
+	return p1 == p2, nil
+}
+
+func ListProjectNamespaces(kc client.Client, projectId string) ([]core.Namespace, error) {
+	var list core.NamespaceList
+	err := kc.List(context.TODO(), &list, client.MatchingLabels{
+		LabelKeyRancherProjectId: projectId,
+	})
+	if err != nil {
+		return nil, err
+	}
+	namespaces := list.Items
+	sort.Slice(namespaces, func(i, j int) bool {
+		return namespaces[i].Name < namespaces[j].Name
+	})
+	return namespaces, nil
+}
+
+func Names(in []core.Namespace) (ret []string) {
+	ret = make([]string, 0, len(in))
+	for _, ns := range in {
+		ret = append(ret, ns.Name)
 	}
-	return projectId == sysProjectId, nil
+	return
 }