From b85ab63e2b860d6d920b2ccc17516eba6363a0e0 Mon Sep 17 00:00:00 2001
From: Miguel Caballer <micafer1@upv.es>
Date: Wed, 28 Feb 2024 16:16:17 +0100
Subject: [PATCH] Enable to open ports in slurm

---
 templates/slurm_cluster.yml | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/templates/slurm_cluster.yml b/templates/slurm_cluster.yml
index 28f1030..6ea5dd1 100644
--- a/templates/slurm_cluster.yml
+++ b/templates/slurm_cluster.yml
@@ -83,6 +83,18 @@ topology_template:
       constraints:
         - valid_values: [ '23.11.1', '20.11.9', '21.08.5', '21.08.8', '22.05.10' ]
 
+    fe_ports:
+      type: map
+      entry_schema:
+        type: PortSpec
+      description: |
+        List of ports to be Opened in FE node (eg. 22,80,443,2000:2100).
+        You can also include the remote CIDR (eg. 8.8.0.0/24).
+      default:
+        ssh_port:
+          source: 22
+          protocol: tcp
+
   node_templates:
 
 
@@ -101,10 +113,7 @@ topology_template:
           properties:
             dns_name: slurmserver
             network_name: PUBLIC
-            ports:
-              http_port:
-                protocol: tcp
-                source: 80
+            ports: { get_input: fe_ports }
         host:
           properties:
             num_cpus: { get_input: fe_cpus }