From ebf537979e3281a96b2ebca7d692c9615c20c578 Mon Sep 17 00:00:00 2001
From: pb <pierre.bayle@irt-saintexupery.com>
Date: Tue, 3 Sep 2024 15:54:20 +0200
Subject: [PATCH] refactoring for better readability

---
 workflow_builder/argo_services.go | 148 ++++++++++++++++++++++++++++++
 1 file changed, 148 insertions(+)
 create mode 100644 workflow_builder/argo_services.go

diff --git a/workflow_builder/argo_services.go b/workflow_builder/argo_services.go
new file mode 100644
index 0000000..a990deb
--- /dev/null
+++ b/workflow_builder/argo_services.go
@@ -0,0 +1,148 @@
+package workflow_builder
+
+import (
+	"oc-monitord/models"
+	"strconv"
+	"strings"
+
+	"cloud.o-forge.io/core/oc-lib/models/resource_model"
+	"cloud.o-forge.io/core/oc-lib/models/resources/workflow/graph"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"gopkg.in/yaml.v3"
+)
+
+// TODO : refactor this method or the deserialization process in oc-lib to get rid of the mongo code
+func getExposeContract(expose resource_model.Model) map[string]map[string]string {
+	contract := make(map[string]map[string]string,0)
+	
+	mapped_info := bson.M{}
+	// var contract PortTranslation
+	_ , byt, _ := bson.MarshalValue(expose.Value)
+	
+	bson.Unmarshal(byt,&mapped_info)
+
+	for _,v := range mapped_info {
+		port := v.(primitive.M)["Key"].(string)
+		// exposed_port := map[string]interface{}{data["Key"] : ""}
+		port_translation := v.(primitive.M)["Value"]
+		contract[port] = map[string]string{}
+		for _,v2 := range port_translation.(primitive.A) {
+			if v2.(primitive.M)["Key"] == "reverse" {
+				contract[port]["reverse"] = v2.(primitive.M)["Value"].(string)
+			}
+			if v2.(primitive.M)["Key"] == "PAT" {
+				contract[port]["PAT"] = v2.(primitive.M)["Value"].(string)
+			}
+		}
+	}
+	return contract
+}
+
+
+func (b *ArgoBuilder) CreateService(processing graph.GraphItem) models.Service{
+	
+	// model {
+	// 	Type : "dict",
+	// 	Value : {
+	// 		"80" : {
+	// 			"reverse"	: "",
+	// 			"PAT"		: "34000"	
+	// 		},
+	// 		"344" : {
+	// 			"reverse"	: "",
+	// 			"PAT"		: "34400"	
+	// 		}
+	// 	} 
+	// }
+	
+
+	new_service := models.Service{APIVersion: "v1", 
+							Kind: "Service", 
+							Metadata: models.Metadata{
+								Name: "workflow-service" ,
+							}, 
+							Spec: models.ServiceSpec{
+								Selector: map[string]string{"app": "oc-service"},
+								Ports: []models.ServicePort{
+								},
+								Type: "NodePort",
+							},
+					}
+	
+	completeServicePorts(&new_service, processing)
+	yamlified, _ := yaml.Marshal(new_service)
+	x := string(yamlified)
+	_ = x
+	return new_service
+}
+
+func completeServicePorts(service *models.Service, processing graph.GraphItem) {
+	
+	contract := getExposeContract(processing.Processing.ResourceModel.Model["expose"])
+	
+
+	for str_port,translation_dict := range contract{
+		
+		port, err := strconv.ParseInt(str_port, 10, 64)
+		if err != nil {
+			logger.Error().Msg("Could not convert " + str_port + "to an int")
+			return
+		}
+
+		
+		if _, ok := translation_dict["PAT"]; ok{
+			port_translation, err := strconv.ParseInt(translation_dict["PAT"], 10, 64)
+			if err != nil {
+				logger.Error().Msg("Could not convert " + translation_dict["PAT"] + "to an int")
+				return
+			}
+			
+
+
+			new_port_translation := models.ServicePort{
+				Name: strings.ToLower(processing.Processing.Name) + processing.ID,
+				Port:  port_translation-30000,
+				TargetPort: port,
+				NodePort: port_translation,
+				Protocol: "TCP",
+			}
+			service.Spec.Ports = append(service.Spec.Ports, new_port_translation)
+		} 
+
+	}
+
+	return
+}
+
+func (b *ArgoBuilder) createService(service models.Service, processing_name string, processing_id string) {
+	if b.Services != nil{
+		b.Services.Spec.Ports = append(b.Services.Spec.Ports, service.Spec.Ports...)
+		}else {
+			b.Services = &service
+		}
+		
+	b.addLabel(processing_name,processing_id)
+
+}
+
+func (b *ArgoBuilder) addServiceToArgo() error {
+	service_manifest, err := yaml.Marshal(b.Services)
+	if err != nil {
+		logger.Error().Msg("Could not marshal service manifest")
+		return err
+	}
+	
+	service_template := models.Template{Name: "workflow-service-pod", 
+								Resource: models.ServiceResource{
+									Action: "create", 
+									SuccessCondition: "status.succeeded > 0",
+									FailureCondition: "status.failed > 3",
+									SetOwnerReference: true,
+									Manifest: string(service_manifest),
+									},
+								}
+	b.Workflow.Spec.Templates = append(b.Workflow.Spec.Templates, service_template)
+
+	return nil
+}
\ No newline at end of file