Merge pull request #4 from StrongMonkey/add-windows-providers

Add rke.windows provider

Author: StrongMonkey

detector.go

@@ -23,6 +23,7 @@ func init() {
 	allProviders[providers.K3s] = providers.IsK3s
 	allProviders[providers.Minikube] = providers.IsMinikube
 	allProviders[providers.RKE] = providers.IsRKE
+	allProviders[providers.RKE_WINDOWS] = providers.IsRKEWindows
 	allProviders[providers.RKE2] = providers.IsRKE2
 }
 

providers/rke-windows.go

@@ -0,0 +1,33 @@
+package providers
+
+import (
+	"context"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes"
+)
+
+const RKE_WINDOWS = "rke.windows"
+
+func IsRKEWindows(ctx context.Context, k8sClient kubernetes.Interface) (bool, error) {
+	// if there are windows nodes
+	windowsNodes, err := k8sClient.CoreV1().Nodes().List(ctx, metav1.ListOptions{
+		Limit:         1,
+		LabelSelector: "kubernetes.io/os=windows",
+	})
+	if err != nil {
+		return false, err
+	}
+	if len(windowsNodes.Items) == 0 {
+		return false, nil
+	}
+
+	annos := windowsNodes.Items[0].Annotations
+	if _, ok := annos["rke.cattle.io/external-ip"]; ok {
+		return true, nil
+	}
+	if _, ok := annos["rke.cattle.io/internal-ip:"]; ok {
+		return true, nil
+	}
+	return false, nil
+}

providers/rke.go

@@ -10,6 +10,18 @@ import (
 const RKE = "rke"
 
 func IsRKE(ctx context.Context, k8sClient kubernetes.Interface) (bool, error) {
+	// if there are windows nodes then this should not be counted as rke.linux
+	windowsNodes, err := k8sClient.CoreV1().Nodes().List(ctx, metav1.ListOptions{
+		Limit:         1,
+		LabelSelector: "kubernetes.io/os=windows",
+	})
+	if err != nil {
+		return false, err
+	}
+	if len(windowsNodes.Items) != 0 {
+		return false, nil
+	}
+
 	// Any node created by RKE should have the annotation, so just grab 1
 	nodes, err := k8sClient.CoreV1().Nodes().List(ctx, metav1.ListOptions{Limit: 1})
 	if err != nil {