From 418a6d1d9426bd98e9c4560f62feb45ce591684b Mon Sep 17 00:00:00 2001
From: Flawid DSouza <me@flawiddsouza.com>
Date: Thu, 14 Nov 2024 13:05:48 +0530
Subject: [PATCH] substitute vars in run name

---
 src/main.rs                             |  3 ++-
 test-ymls/use-vars-in-run-name.vars.yml |  2 ++
 test-ymls/use-vars-in-run-name.yml      |  7 +++++++
 test-ymls/use-vars-in-run-name.yml.out  |  4 ++++
 tests/integration_test.rs               | 10 ++++++++++
 5 files changed, 25 insertions(+), 1 deletion(-)
 create mode 100644 test-ymls/use-vars-in-run-name.vars.yml
 create mode 100644 test-ymls/use-vars-in-run-name.yml
 create mode 100644 test-ymls/use-vars-in-run-name.yml.out

diff --git a/src/main.rs b/src/main.rs
index 52e5eeb..d92e0f2 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -203,7 +203,8 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let deploy_file_dir = deploy_file_path.parent().unwrap_or(Path::new("."));
 
     for dep in deployments {
-        println!("{}", format!("Starting deployment: {}\n", dep.name).green());
+        let dep_name = utils::replace_placeholders(&dep.name, &vars_map);
+        println!("{}", format!("Starting deployment: {}\n", dep_name).green());
 
         let hosts: Vec<&str> = dep.hosts.split(',').map(|s| s.trim()).collect();
 
diff --git a/test-ymls/use-vars-in-run-name.vars.yml b/test-ymls/use-vars-in-run-name.vars.yml
new file mode 100644
index 0000000..0a802d8
--- /dev/null
+++ b/test-ymls/use-vars-in-run-name.vars.yml
@@ -0,0 +1,2 @@
+run_name: Test 1
+run_name_2: Test 2
diff --git a/test-ymls/use-vars-in-run-name.yml b/test-ymls/use-vars-in-run-name.yml
new file mode 100644
index 0000000..cbc839c
--- /dev/null
+++ b/test-ymls/use-vars-in-run-name.yml
@@ -0,0 +1,7 @@
+- name: Deployment {{ run_name }}
+  hosts: test
+  tasks:
+
+- name: Deployment {{ run_name_2 }}
+  hosts: test
+  tasks:
diff --git a/test-ymls/use-vars-in-run-name.yml.out b/test-ymls/use-vars-in-run-name.yml.out
new file mode 100644
index 0000000..2231fa9
--- /dev/null
+++ b/test-ymls/use-vars-in-run-name.yml.out
@@ -0,0 +1,4 @@
+Starting deployment: Deployment Test 1
+
+Starting deployment: Deployment Test 2
+
diff --git a/tests/integration_test.rs b/tests/integration_test.rs
index 02c86f7..7474f42 100644
--- a/tests/integration_test.rs
+++ b/tests/integration_test.rs
@@ -240,3 +240,13 @@ fn use_vars_in_task_name() {
     setup();
     run_tests_for_both_inventories("test-ymls/use-vars-in-task-name.yml", false, "");
 }
+
+#[test]
+fn use_vars_in_run_name() {
+    setup();
+    run_tests_for_both_inventories(
+        "test-ymls/use-vars-in-run-name.yml",
+        false,
+        "@test-ymls/use-vars-in-run-name.vars.yml",
+    );
+}