From 2ca5754481676e8b0e8a875f62f081683f8781c6 Mon Sep 17 00:00:00 2001
From: Darren Shepherd <darren@rancher.com>
Date: Thu, 28 May 2020 14:02:47 -0700
Subject: [PATCH] Move github.com/rancher/rancher/pkg/namespace to types

The namespace package contains constants that need to be used
globally so moving to types to other go modules can depend on it.
---
 namespace/namespace.go       |  18 ++++++
 namespace/namespace_utils.go | 118 +++++++++++++++++++++++++++++++++++
 2 files changed, 136 insertions(+)
 create mode 100644 namespace/namespace.go
 create mode 100644 namespace/namespace_utils.go

diff --git a/namespace/namespace.go b/namespace/namespace.go
new file mode 100644
index 00000000..263b0aca
--- /dev/null
+++ b/namespace/namespace.go
@@ -0,0 +1,18 @@
+package namespace
+
+import v1 "k8s.io/api/core/v1"
+
+var projectIDAnnotation = "field.cattle.io/projectId"
+
+func NsByProjectID(obj interface{}) ([]string, error) {
+	ns, ok := obj.(*v1.Namespace)
+	if !ok {
+		return []string{}, nil
+	}
+
+	if id, ok := ns.Annotations[projectIDAnnotation]; ok {
+		return []string{id}, nil
+	}
+
+	return []string{}, nil
+}
diff --git a/namespace/namespace_utils.go b/namespace/namespace_utils.go
new file mode 100644
index 00000000..02082d79
--- /dev/null
+++ b/namespace/namespace_utils.go
@@ -0,0 +1,118 @@
+package namespace
+
+import (
+	"encoding/json"
+	"time"
+
+	v1 "k8s.io/api/core/v1"
+)
+
+const (
+	statusAnn                   = "cattle.io/status"
+	GlobalNamespace             = "cattle-global-data"
+	NodeTemplateGlobalNamespace = "cattle-global-nt"
+)
+
+func SetNamespaceCondition(namespace *v1.Namespace, d time.Duration, conditionType string, conditionStatus bool, message string) error {
+	annotations := namespace.ObjectMeta.Annotations
+	if annotations == nil {
+		annotations = map[string]string{}
+	}
+	ann := annotations[statusAnn]
+	conditionStatusStr := "False"
+	if conditionStatus {
+		conditionStatusStr = "True"
+	}
+	bAnn, err := GenerateConditionAnnotation(ann, d, conditionType, conditionStatusStr, message)
+	if err != nil {
+		return err
+	}
+	annotations[statusAnn] = bAnn
+
+	namespace.ObjectMeta.Annotations = annotations
+
+	return nil
+}
+
+func GenerateConditionAnnotation(ann string, d time.Duration, conditionType string, conditionStatus string, message string) (string, error) {
+	status := &status{}
+	if ann != "" {
+		err := json.Unmarshal([]byte(ann), status)
+		if err != nil {
+			return "", err
+		}
+	}
+	if status.Conditions == nil {
+		status.Conditions = []condition{}
+	}
+
+	var idx int
+	found := false
+	for i, c := range status.Conditions {
+		if c.Type == conditionType {
+			idx = i
+			found = true
+			break
+		}
+	}
+
+	conditionMessage := ""
+	if conditionStatus != "True" {
+		conditionMessage = message
+	}
+
+	cond := condition{
+		Type:           conditionType,
+		Status:         conditionStatus,
+		Message:        conditionMessage,
+		LastUpdateTime: time.Now().Add(d).Format(time.RFC3339),
+	}
+
+	if found {
+		status.Conditions[idx] = cond
+	} else {
+		status.Conditions = append(status.Conditions, cond)
+	}
+
+	bAnn, err := json.Marshal(status)
+	if err != nil {
+		return "", err
+	}
+	return string(bAnn), nil
+}
+
+func IsNamespaceConditionSet(namespace *v1.Namespace, conditionType string, conditionStatus bool) (bool, error) {
+	if namespace.ObjectMeta.Annotations == nil {
+		return false, nil
+	}
+	ann := namespace.ObjectMeta.Annotations[statusAnn]
+	if ann == "" {
+		return false, nil
+	}
+	status := &status{}
+	err := json.Unmarshal([]byte(ann), status)
+	if err != nil {
+		return false, err
+	}
+	conditionStatusStr := "False"
+	if conditionStatus {
+		conditionStatusStr = "True"
+	}
+	for _, c := range status.Conditions {
+		if c.Type == conditionType && c.Status == conditionStatusStr {
+			return true, nil
+		}
+	}
+	return false, nil
+}
+
+type status struct {
+	Conditions []condition
+}
+
+type condition struct {
+	Type           string
+	Status         string
+	Message        string
+	LastUpdateTime string
+}