From e97a0ff7a35c9023602defba245abcbf2929ba93 Mon Sep 17 00:00:00 2001
From: Carlos Dominguez <43823505+cadoming@users.noreply.github.com>
Date: Mon, 14 Jan 2019 09:12:09 +0100
Subject: [PATCH] Labels configuration (#135)

---
 roles/wazuh/ansible-wazuh-agent/defaults/main.yml        | 5 +++++
 .../templates/var-ossec-etc-ossec-agent.conf.j2          | 9 +++++++++
 roles/wazuh/ansible-wazuh-manager/defaults/main.yml      | 5 +++++
 .../templates/var-ossec-etc-ossec-server.conf.j2         | 8 ++++++++
 4 files changed, 27 insertions(+)

diff --git a/roles/wazuh/ansible-wazuh-agent/defaults/main.yml b/roles/wazuh/ansible-wazuh-agent/defaults/main.yml
index c3e1f650..42e13eec 100644
--- a/roles/wazuh/ansible-wazuh-agent/defaults/main.yml
+++ b/roles/wazuh/ansible-wazuh-agent/defaults/main.yml
@@ -300,3 +300,8 @@ wazuh_agent_config:
         location: 'System'
       - format: 'syslog'
         location: 'active-response\active-responses.log'
+  labels:
+    enable: false
+    list:
+      - key: Env
+        value: Production
diff --git a/roles/wazuh/ansible-wazuh-agent/templates/var-ossec-etc-ossec-agent.conf.j2 b/roles/wazuh/ansible-wazuh-agent/templates/var-ossec-etc-ossec-agent.conf.j2
index 3cc6ba1b..6327441a 100644
--- a/roles/wazuh/ansible-wazuh-agent/templates/var-ossec-etc-ossec-agent.conf.j2
+++ b/roles/wazuh/ansible-wazuh-agent/templates/var-ossec-etc-ossec-agent.conf.j2
@@ -339,4 +339,13 @@
     </localfile>
   {% endfor %}
   {% endif %}
+
+{% if wazuh_agent_config.labels.enable == true %}
+  <labels>
+  {% for label in wazuh_agent_config.labels.list %}
+    <label key="{{ label.key }}"{% if label.hidden is defined %} hidden="{{ label.hidden }}"{% endif %}>{{ label.value }}</label>
+  {% endfor %}
+  </labels>
+{% endif %}
+
 </ossec_config>
diff --git a/roles/wazuh/ansible-wazuh-manager/defaults/main.yml b/roles/wazuh/ansible-wazuh-manager/defaults/main.yml
index 9f42df5e..b9817a3a 100644
--- a/roles/wazuh/ansible-wazuh-manager/defaults/main.yml
+++ b/roles/wazuh/ansible-wazuh-manager/defaults/main.yml
@@ -271,6 +271,11 @@ wazuh_manager_config:
     - server: null
       port: null
       format: null
+  labels:
+    enable: false
+    list:
+      - key: Env
+        value: Production
 
 wazuh_agent_configs:
   - type: os
diff --git a/roles/wazuh/ansible-wazuh-manager/templates/var-ossec-etc-ossec-server.conf.j2 b/roles/wazuh/ansible-wazuh-manager/templates/var-ossec-etc-ossec-server.conf.j2
index 7bc92ece..6f389a71 100644
--- a/roles/wazuh/ansible-wazuh-manager/templates/var-ossec-etc-ossec-server.conf.j2
+++ b/roles/wazuh/ansible-wazuh-manager/templates/var-ossec-etc-ossec-server.conf.j2
@@ -539,4 +539,12 @@
 {% endfor %}
 {% endif %}
 
+{% if wazuh_manager_config.labels.enable == true %}
+  <labels>
+  {% for label in wazuh_manager_config.labels.list %}
+    <label key="{{ label.key }}"{% if label.hidden is defined %} hidden="{{ label.hidden }}"{% endif %}>{{ label.value }}</label>
+  {% endfor %}
+  </labels>
+{% endif %}
+
 </ossec_config>