feat(core): mvp

Signed-off-by: Pavel Tishkov <[email protected]>

Author: fl64

images/hooks/cmd/discovery-workload-nodes/main.go renamed to images/hooks/cmd/discovery-virthandler-nodes/main.go

@@ -0,0 +1,85 @@
+/*
+Copyright 2025 Flant JSC
+
+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.
+*/
+
+package main
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/deckhouse/module-sdk/pkg"
+	"github.com/deckhouse/module-sdk/pkg/app"
+	"github.com/deckhouse/module-sdk/pkg/registry"
+	"k8s.io/utils/ptr"
+
+	"hooks/pkg/common"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+const (
+	discoveryNodesSnapshot               = "discovery-virthandler-nodes"
+	virthandlerSchedulableNodeLabel      = "kubevirt.internal.virtualization.deckhouse.io/schedulable"
+	virthandlerSchedulableNodeLabelValue = "true"
+	kvmEnabledLabel                      = "virtualization.deckhouse.io/kvm-enabled"
+	kvmEnabledLabelValue                 = "true"
+)
+
+var _ = registry.RegisterFunc(configDiscoveryService, handleDiscoveryNodes)
+
+var configDiscoveryService = &pkg.HookConfig{
+	OnBeforeHelm: &pkg.OrderedConfig{Order: 1},
+	Kubernetes: []pkg.KubernetesConfig{
+		{
+			Name:       discoveryNodesSnapshot,
+			APIVersion: "v1",
+			Kind:       "Node",
+			JqFilter:   ".metadata",
+			LabelSelector: &metav1.LabelSelector{
+				MatchLabels: map[string]string{
+					virthandlerSchedulableNodeLabel: virthandlerSchedulableNodeLabelValue,
+				},
+			},
+			ExecuteHookOnSynchronization: ptr.To(false),
+		},
+	},
+
+	Queue: fmt.Sprintf("modules/%s", common.MODULE_NAME),
+}
+
+func handleDiscoveryNodes(_ context.Context, input *pkg.HookInput) error {
+	nodes := input.Snapshots.Get(discoveryNodesSnapshot)
+	if len(nodes) == 0 {
+		return nil
+	}
+
+	for _, node := range nodes {
+		metadata := &metav1.ObjectMeta{}
+		if err := node.UnmarshalTo(metadata); err != nil {
+			input.Logger.Error(fmt.Sprintf("Failed to unmarshal node metadata %v", err))
+		}
+		if _, ok := metadata.Labels[kvmEnabledLabel]; !ok {
+			metadata.Labels[kvmEnabledLabel] = kvmEnabledLabelValue
+		}
+		input.PatchCollector.PatchWithMerge(metadata, "v1", "Node", "", metadata.)
+	}
+
+	return nil
+}
+
+func main() {
+	app.Run()
+}

images/hooks/cmd/migrate-virthandler-kvm-node-labels/main.go

@@ -34,4 +34,5 @@ shell:
   - go build -ldflags="-s -w" -o /hooks/prevent-default-vmclasses-deletion ./cmd/prevent-default-vmclasses-deletion
   - go build -ldflags="-s -w" -o /hooks/generate-secret-for-dvcr ./cmd/generate-secret-for-dvcr
   - go build -ldflags="-s -w" -o /hooks/discovery-clusterip-service-for-dvcr ./cmd/discovery-clusterip-service-for-dvcr
-  - go build -ldflags="-s -w" -o /hooks/discovery-workload-nodes ./cmd/discovery-workload-nodes
+  - go build -ldflags="-s -w" -o /hooks/discovery-virthandler-nodes ./cmd/discovery-virthandler-nodes
+  - go build -ldflags="-s -w" -o /hooks/migrate-virthandler-kvm-node-labels ./cmd/migrate-virthandler-kvm-node-labels

images/hooks/werf.inc.yaml

@@ -27,7 +27,7 @@ spec:
   template:
     spec:
       nodeSelector:
-        {{- $defaultLabels | toYaml | nindent 8 }}
+{{ $defaultLabels | toYaml | nindent 8 }}
 {{- end -}}
 
 {{- define "kubevirt.virthandler_nodeseletor_strategic_patch_json" -}}

templates/kubevirt/_kubevirt_helpers.tpl

@@ -92,10 +92,10 @@ spec:
       patch: {{ include "kubevirt.virthandler_nodeaffinity_strategic_patch_json" $dvpNestingLevel }}
       type: strategic
     # Run virt-handler pods on required nodes
-    - resourceType: DaemonSet
-      resourceName: virt-handler
-      patch: {{ include "kubevirt.virthandler_nodeseletor_strategic_patch_json" . }}
-      type: strategic
+    # - resourceType: DaemonSet
+    #   resourceName: virt-handler
+    #   patch: {{ include "kubevirt.virthandler_nodeseletor_strategic_patch_json" . }}
+    #   type: strategic
     {{- if (include "helm_lib_ha_enabled" .) }}
     # HA settings for deploy/virt-api.
     - resourceType: Deployment

templates/kubevirt/kubevirt.yaml

@@ -22,7 +22,7 @@ spec:
     # limitations under the License.
 
     kvm_enabled_label='virtualization.deckhouse.io/kvm-enabled=true'
-    kvm_disabled_label='virtualization.deckhouse.io/kvm-enabled=false'
+    kvm_disabled_label='virtualization.deckhouse.io/kvm-enabled-'
 
     label=$kvm_enabled_label
 
@@ -41,15 +41,15 @@ spec:
       exit 0
     fi
 
-    node=$(bb-d8-node-name)
+    # node=$(bb-d8-node-name)
 
-    max_attempts=3
-    until bb-kubectl --kubeconfig $kubeconfig label --overwrite=true node "$node" "$label"; do
-      attempt=$(( attempt + 1 ))
-      if [ "$attempt" -gt "$max_attempts" ]; then
-        bb-log-error "failed to label node $node with label $label after $max_attempts attempts"
-        exit 1
-      fi
-      echo "Waiting for label node $node with label $label (attempt $attempt of $max_attempts)..."
-      sleep 5
-    done
+    # max_attempts=3
+    # until bb-kubectl --kubeconfig $kubeconfig label --overwrite=true node "$node" "$label"; do
+    #   attempt=$(( attempt + 1 ))
+    #   if [ "$attempt" -gt "$max_attempts" ]; then
+    #     bb-log-error "failed to label node $node with label $label after $max_attempts attempts"
+    #     exit 1
+    #   fi
+    #   echo "Waiting for label node $node with label $label (attempt $attempt of $max_attempts)..."
+    #   sleep 5
+    # done