From 0146c9d34faa959e9daca77c1c3353b2a41991de Mon Sep 17 00:00:00 2001
From: Edward Funnekotter <edward.funnekotter@solace.com>
Date: Tue, 24 Sep 2024 09:24:24 -0400
Subject: [PATCH] feat: AI-129: add ability to specify a default value for a an
 environment variable in a .yaml config file (#43)

---
 src/solace_ai_connector/main.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/src/solace_ai_connector/main.py b/src/solace_ai_connector/main.py
index 0adb4143..a62cf00a 100644
--- a/src/solace_ai_connector/main.py
+++ b/src/solace_ai_connector/main.py
@@ -1,5 +1,6 @@
 import os
 import sys
+import re
 import yaml
 from .solace_ai_connector import SolaceAiConnector
 
@@ -12,7 +13,7 @@ def load_config(file):
             yaml_str = f.read()
 
         # Substitute the environment variables using os.environ
-        yaml_str = os.path.expandvars(yaml_str)
+        yaml_str = expandvars_with_defaults(yaml_str)
 
         # Load the YAML string using yaml.safe_load
         return yaml.safe_load(yaml_str)
@@ -22,6 +23,19 @@ def load_config(file):
         sys.exit(1)
 
 
+def expandvars_with_defaults(text):
+    """Expand environment variables with support for default values.
+    Supported syntax: ${VAR_NAME} or ${VAR_NAME, default_value}"""
+    pattern = re.compile(r"\$\{([^}:\s]+)(?:\s*,\s*([^}]*))?\}")
+
+    def replacer(match):
+        var_name = match.group(1)
+        default_value = match.group(2) if match.group(2) is not None else ""
+        return os.environ.get(var_name, default_value)
+
+    return pattern.sub(replacer, text)
+
+
 def merge_config(dict1, dict2):
     """Merge a new configuration into an existing configuration."""
     merged = {}