From a2ee4c6568b08099a248a6012cee36c88d5748e5 Mon Sep 17 00:00:00 2001
From: "Wile E. Coyote" <rhizomatica@rhizomatica.org>
Date: Mon, 21 Nov 2016 12:08:51 +0000
Subject: [PATCH] Add some BTS names

---
 modules/rhizo_base/manifests/init.pp     |  3 +++
 modules/rhizo_base/templates/vars.sh.erb | 11 +++++++++++
 2 files changed, 14 insertions(+)

diff --git a/modules/rhizo_base/manifests/init.pp b/modules/rhizo_base/manifests/init.pp
index b0694d1..183b977 100644
--- a/modules/rhizo_base/manifests/init.pp
+++ b/modules/rhizo_base/manifests/init.pp
@@ -40,14 +40,17 @@ class rhizo_base {
   $bts1_ip_address = hiera('rhizo::bts1_ip_address')
   $arfcn_A         = hiera('rhizo::arfcn_A')
   $arfcn_B         = hiera('rhizo::arfcn_B', false)
+  $bts1_name       = hiera('rhizo::bts1_name', "${site_name}_1")
 
   $bts2_ip_address = hiera('rhizo::bts2_ip_address', false)
   $arfcn_C         = hiera('rhizo::arfcn_C', false)
   $arfcn_D         = hiera('rhizo::arfcn_D', false)
+  $bts2_name       = hiera('rhizo::bts2_name', "${site_name}_2")
 
   $bts3_ip_address = hiera('rhizo::bts3_ip_address', false)
   $arfcn_E         = hiera('rhizo::arfcn_E', false)
   $arfcn_F         = hiera('rhizo::arfcn_F', false)
+  $bts3_name       = hiera('rhizo::bts3_name', "${site_name}_3")
 
   $max_power_red   = hiera('rhizo::max_power_red')
 
diff --git a/modules/rhizo_base/templates/vars.sh.erb b/modules/rhizo_base/templates/vars.sh.erb
index 198936a..47ddcca 100644
--- a/modules/rhizo_base/templates/vars.sh.erb
+++ b/modules/rhizo_base/templates/vars.sh.erb
@@ -19,3 +19,14 @@ BTS3=<%= @bts3_ip_address %>
 <% end -%>
 
 RECIPIENTS="<%= @mail_admins -%>"
+
+function bname {
+    case $1 in
+        0) echo -n "<%= @bts1_name %> ONE" ;;
+        1) echo -n "<%= @bts1_name %> TWO" ;;
+        2) echo -n "<%= @bts2_name %> ONE" ;;
+        3) echo -n "<%= @bts2_name %> TWO" ;;
+        4) echo -n "<%= @bts3_name %> ONE" ;;
+        5) echo -n "<%= @bts3_name %> TWO" ;;
+    esac
+}