feat: got first provider working

Author: jtschelling

distros/kubernetes/nvsentinel/charts/janitor-provider-kwok/templates/networkpolicy.yaml

@@ -0,0 +1,30 @@
+# Copyright (c) 2025, NVIDIA CORPORATION.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+apiVersion: networking.k8s.io/v1
+kind: NetworkPolicy
+metadata:
+  name: {{ include "provider.fullname" . }}-allow-service-traffic
+  labels:
+    {{- include "provider.labels" . | nindent 4 }}
+spec:
+  podSelector:
+    matchLabels:
+      {{- include "provider.selectorLabels" . | nindent 6 }}
+  policyTypes:
+    - Ingress
+  ingress:
+    - ports:
+        - port: {{ .Values.service.port | default 50051 }}
+          protocol: TCP
+

providers/kwok/main.go

@@ -33,7 +33,21 @@ type kwokProviderServer struct {
 
 func (s *kwokProviderServer) SendRebootSignal(ctx context.Context, req *cspv1alpha1.SendRebootSignalRequest) (*cspv1alpha1.SendRebootSignalResponse, error) {
 	slog.Info("Sending reboot signal", "node", req.NodeName)
-	return nil, status.Errorf(codes.Unimplemented, "method SendRebootSignal not implemented")
+	return &cspv1alpha1.SendRebootSignalResponse{
+		RequestId: "1234567890",
+	}, nil
+}
+
+func (s *kwokProviderServer) IsNodeReady(ctx context.Context, req *cspv1alpha1.IsNodeReadyRequest) (*cspv1alpha1.IsNodeReadyResponse, error) {
+	slog.Info("Checking if node is ready", "node", req.NodeName)
+	return &cspv1alpha1.IsNodeReadyResponse{
+		IsReady: true,
+	}, nil
+}
+
+func (s *kwokProviderServer) SendTerminateSignal(ctx context.Context, req *cspv1alpha1.SendTerminateSignalRequest) (*cspv1alpha1.SendTerminateSignalResponse, error) {
+	slog.Info("Sending terminate signal", "node", req.NodeName)
+	return nil, status.Errorf(codes.Unimplemented, "method SendTerminateSignal not implemented")
 }
 
 func main() {