From 52839e2a7d9c4f02d3cb50967cc2395dceb718ba Mon Sep 17 00:00:00 2001
From: Josh Patterson <josh.patterson@securityonionsolutions.com>
Date: Wed, 26 Feb 2025 15:22:36 -0500
Subject: [PATCH] implement regex for cpu and mem

---
 .../hypervisor/soc_hypervisor.yaml.jinja      | 38 ++++++++++++++++++-
 1 file changed, 36 insertions(+), 2 deletions(-)

diff --git a/salt/soc/dyanno/hypervisor/soc_hypervisor.yaml.jinja b/salt/soc/dyanno/hypervisor/soc_hypervisor.yaml.jinja
index ff654b4b8..cf272b221 100644
--- a/salt/soc/dyanno/hypervisor/soc_hypervisor.yaml.jinja
+++ b/salt/soc/dyanno/hypervisor/soc_hypervisor.yaml.jinja
@@ -116,9 +116,43 @@ No Virtual Machines Found
 {%-     for field in updated_template.uiElements -%}
 {%-       set updated_field = field.copy() -%}
 {%-       if field.field == 'cpu' -%}
-{%-         do updated_field.update({'label': field.label | replace('FREE', cpu_free | string) | replace('TOTAL', cpu_total | string)}) -%}
+{%-         if cpu_free < 10 -%}
+{%-           set cpu_regex = '^[1-' ~ cpu_free ~ ']$' -%}
+{%-         elif cpu_free < 100 -%}
+{%-           set tens_digit = cpu_free // 10 -%}
+{%-           set ones_digit = cpu_free % 10 -%}
+{%-           if ones_digit == 0 -%}
+{%-             set cpu_regex = '^([1-9]|[1-' ~ (tens_digit-1) ~ '][0-9]|' ~ tens_digit ~ '0)$' -%}
+{%-           else -%}
+{%-             set cpu_regex = '^([1-9]|[1-' ~ (tens_digit-1) ~ '][0-9]|' ~ tens_digit ~ '[0-' ~ ones_digit ~ '])$' -%}
+{%-           endif -%}
+{%-         else -%}
+{%-           set cpu_regex = '^([1-9]|[1-9][0-9]|100)$' -%}
+{%-         endif -%}
+{%-         do updated_field.update({
+            'label': field.label | replace('FREE', cpu_free | string) | replace('TOTAL', cpu_total | string),
+            'regex': cpu_regex,
+            'regexFailureMessage': 'Enter a value not exceeding ' ~ cpu_free | string ~ ' cores'
+          }) -%}
 {%-       elif field.field == 'memory' -%}
-{%-         do updated_field.update({'label': field.label | replace('FREE', mem_free | string) | replace('TOTAL', mem_total | string)}) -%}
+{%-         if mem_free < 10 -%}
+{%-           set mem_regex = '^[1-' ~ mem_free ~ ']$' -%}
+{%-         elif mem_free < 100 -%}
+{%-           set tens_digit = mem_free // 10 -%}
+{%-           set ones_digit = mem_free % 10 -%}
+{%-           if ones_digit == 0 -%}
+{%-             set mem_regex = '^([1-9]|[1-' ~ (tens_digit-1) ~ '][0-9]|' ~ tens_digit ~ '0)$' -%}
+{%-           else -%}
+{%-             set mem_regex = '^([1-9]|[1-' ~ (tens_digit-1) ~ '][0-9]|' ~ tens_digit ~ '[0-' ~ ones_digit ~ '])$' -%}
+{%-           endif -%}
+{%-         else -%}
+{%-           set mem_regex = '^([1-9]|[1-9][0-9]|[1-9][0-9][0-9])$' -%}
+{%-         endif -%}
+{%-         do updated_field.update({
+            'label': field.label | replace('FREE', mem_free | string) | replace('TOTAL', mem_total | string),
+            'regex': mem_regex,
+            'regexFailureMessage': 'Enter a value not exceeding ' ~ mem_free | string ~ ' GB'
+          }) -%}
 {%-       elif field.field == 'disk' -%}
 {%-         do updated_field.update({'label': field.label | replace('FREE', disk_free) | replace('TOTAL', disk_total | replace('\n', ','))}) -%}
 {%-       elif field.field == 'copper' -%}