feat(): also handle deletes

internal/kube.go

@@ -65,7 +65,7 @@ func ApplyLabelToNode(client kubernetes.Interface, nodeName string, leaseName st
 	return nil
 }
 
-func RemoveLabelFromNode(nodeName string) error {
+func RemoveLabelFromNode(nodeName string, leaseName string, cfg *Config) error {
 	client, err := generateClient()
 	if err != nil {
 		return fmt.Errorf("Could not generate client: %s", err.Error())
@@ -75,22 +75,19 @@ func RemoveLabelFromNode(nodeName string) error {
 		"Caller": "LabelNode",
 	}).Info(fmt.Sprintf("Removing Label from node %s", nodeName))
 
-	labelKey := "metallb-speaker"
-
 	node, err := client.CoreV1().Nodes().Get(context.TODO(), nodeName, metav1.GetOptions{})
 	if err != nil {
 		panic(err.Error())
 	}
 
-	delete(node.Labels, labelKey)
+	delete(node.Labels, cfg.CiliumLabel)
+	delete(node.Labels, "cilium.uploadfilter24.eu/lease")
 
 	_, err = client.CoreV1().Nodes().Update(context.TODO(), node, metav1.UpdateOptions{})
 	if err != nil {
 		panic(err.Error())
 	}
 
-	fmt.Printf("Label %s removed from node %s\n", labelKey, nodeName)
-
 	return nil
 }