From 174bb36dad4ccd218fa79f8cc9dde2561b6101f5 Mon Sep 17 00:00:00 2001
From: "dmitry.denisenko" <dmitry.denisenko@flant.com>
Date: Tue, 28 Jan 2025 19:42:12 +0400
Subject: [PATCH] ++

---
 internal/platform/cmd/module/enable/enable.go |  1 +
 .../cmd/module/operatemodule/switchmodule.go  | 25 ++++++++++++-------
 2 files changed, 17 insertions(+), 9 deletions(-)

diff --git a/internal/platform/cmd/module/enable/enable.go b/internal/platform/cmd/module/enable/enable.go
index c6ee7f30..76b09d0c 100644
--- a/internal/platform/cmd/module/enable/enable.go
+++ b/internal/platform/cmd/module/enable/enable.go
@@ -55,6 +55,7 @@ func enableModule(cmd *cobra.Command, moduleName []string) error {
 	if err != nil {
 		return fmt.Errorf("Failed to setup Kubernetes client: %w", err)
 	}
+
 	err = operatemodule.OperateModule(config, moduleName[0], true)
 	if err != nil {
 		return fmt.Errorf("Error enable module: %w", err)
diff --git a/internal/platform/cmd/module/operatemodule/switchmodule.go b/internal/platform/cmd/module/operatemodule/switchmodule.go
index 928e0ef6..24f876a5 100644
--- a/internal/platform/cmd/module/operatemodule/switchmodule.go
+++ b/internal/platform/cmd/module/operatemodule/switchmodule.go
@@ -15,7 +15,14 @@ import (
 	"k8s.io/utils/ptr"
 )
 
-func OperateModule(config *rest.Config, name string, enabled bool) error {
+type ModuleState bool
+
+const (
+	ModuleEnabled  ModuleState = true
+	ModuleDisabled ModuleState = false
+)
+
+func OperateModule(config *rest.Config, name string, moduleState ModuleState) error {
 	dynamicClient, err := dynamic.NewForConfig(config)
 	if err != nil {
 		return fmt.Errorf("Failed to create dynamic client: %v", err)
@@ -33,10 +40,10 @@ func OperateModule(config *rest.Config, name string, enabled bool) error {
 	if err != nil {
 		//fmt.Errorf("Error get options module '%s': %w", name, err)
 	}
-	patchSpec, err := patchSpec(enabled)
+	patchSpec, err := patchSpec(moduleState)
 	if customResource != nil {
-		if err = unstructured.SetNestedField(customResource.Object, enabled, "spec", "enabled"); err != nil {
-			return fmt.Errorf("failed to change spec.enabled to %v in the '%s' module config: %w", enabled, name, err)
+		if err = unstructured.SetNestedField(customResource.Object, moduleState, "spec", "enabled"); err != nil {
+			return fmt.Errorf("failed to change spec.enabled to %v in the '%s' module config: %w", moduleState, name, err)
 		}
 		if _, err = resourceClient.Patch(context.TODO(), name, types.MergePatchType, patchSpec, metav1.PatchOptions{}); err != nil {
 			return fmt.Errorf("failed to update the '%s' module config: %w", name, err)
@@ -44,7 +51,7 @@ func OperateModule(config *rest.Config, name string, enabled bool) error {
 		return nil
 	}
 
-	obj, err := createModuleConfig(name, enabled)
+	obj, err := createModuleConfig(name, moduleState)
 	if err != nil {
 		return fmt.Errorf("failed to convert the '%s' module config: %w", name, err)
 	}
@@ -54,7 +61,7 @@ func OperateModule(config *rest.Config, name string, enabled bool) error {
 	return err
 }
 
-func createModuleConfig(name string, enabled bool) (*unstructured.Unstructured, error) {
+func createModuleConfig(name string, moduleState ModuleState) (*unstructured.Unstructured, error) {
 	newCfg := &v1alpha1.ModuleConfigMeta{
 		TypeMeta: metav1.TypeMeta{
 			Kind:       "ModuleConfig",
@@ -64,17 +71,17 @@ func createModuleConfig(name string, enabled bool) (*unstructured.Unstructured,
 			Name: name,
 		},
 		Spec: v1alpha1.ModuleConfigSpec{
-			Enabled: ptr.To(enabled),
+			Enabled: (*bool)(ptr.To(moduleState)),
 		},
 	}
 	content, err := runtime.DefaultUnstructuredConverter.ToUnstructured(newCfg)
 	return &unstructured.Unstructured{Object: content}, err
 }
 
-func patchSpec(enabled bool) ([]byte, error) {
+func patchSpec(moduleState ModuleState) ([]byte, error) {
 	patchData := map[string]interface{}{
 		"spec": map[string]interface{}{
-			"enabled": enabled,
+			"enabled": moduleState,
 		},
 	}