Add labels instance.hetzner.cloud/provided-by,instance.hetzner.cloud/…

…is-root-server
identw · Nov 14, 2024 · b08bc2b · b08bc2b
1 parent 46327b2
commit b08bc2b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/hcloud/util.go b/hcloud/util.go
@@ -180,12 +180,15 @@ func syncLabels(k8sClient *kubernetes.Clientset, server *hcloud.Server) {
 func addTypeLabel(k8sClient *kubernetes.Clientset, name string, typeNode string) {
 	node, err := k8sClient.CoreV1().Nodes().Get(context.TODO(), name, metav1.GetOptions{})
 	if err == nil {
-		if _, ok := node.ObjectMeta.Labels[hcops.NameLabelType]; !ok {
-			node.ObjectMeta.Labels[hcops.NameLabelType] = typeNode
-		}
-		if node.ObjectMeta.Labels[hcops.NameLabelType] != typeNode {
-			node.ObjectMeta.Labels[hcops.NameLabelType] = typeNode
+		for k, v := range hcops.TypeLabels[typeNode] {
+			if _, ok := node.ObjectMeta.Labels[k]; !ok {
+				node.ObjectMeta.Labels[k] = v
+			}
+			if node.ObjectMeta.Labels[k] != v {
+				node.ObjectMeta.Labels[v] = v
+			}
 		}
+
 		k8sClient.CoreV1().Nodes().Update(context.TODO(), node, metav1.UpdateOptions{})
 	}
 }
diff --git a/internal/hcops/common.go b/internal/hcops/common.go
@@ -12,6 +12,20 @@ var (
 	NameLabelType = "node.hetzner.com/type"
 	NameCloudNode = "cloud"
 	NameDedicatedNode = "dedicated"
+
+	TypeLabels = map[string]map[string]string{
+		"cloud": map[string]string{
+			"node.hetzner.com/type": "cloud",
+			"instance.hetzner.cloud/provided-by": "cloud",
+			"instance.hetzner.cloud/is-root-server": "false",
+		},
+		"dedicated": map[string]string{
+			"node.hetzner.com/type": "dedicated",
+			"instance.hetzner.cloud/provided-by": "robot",
+			"instance.hetzner.cloud/is-root-server": "true",
+		},
+	}
+
 	ExcludeServer = &hcloud.Server{ 
 		ID: 999999,
 		ServerType: &hcloud.ServerType{Name: "exclude"},