From 332ee7ae8bd4858d6761146e84b722bdb979cbda Mon Sep 17 00:00:00 2001
From: neonmei <neonmei@pm.me>
Date: Mon, 23 Nov 2020 11:24:43 -0300
Subject: [PATCH] roles/agent: add task for determining which wazuh_managers to
 use through `register: yes` instead of just grabbing the first one on the
 list, otherwise fallback to first in the list

---
 roles/wazuh/ansible-wazuh-agent/defaults/main.yml | 1 +
 roles/wazuh/ansible-wazuh-agent/tasks/Linux.yml   | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/roles/wazuh/ansible-wazuh-agent/defaults/main.yml b/roles/wazuh/ansible-wazuh-agent/defaults/main.yml
index e59d4653..31c1ba85 100644
--- a/roles/wazuh/ansible-wazuh-agent/defaults/main.yml
+++ b/roles/wazuh/ansible-wazuh-agent/defaults/main.yml
@@ -92,6 +92,7 @@ wazuh_managers:
     api_user: wazuh
     max_retries: 5
     retry_interval: 5
+    register: yes
 
 ## Enrollment
 wazuh_agent_enrollment:
diff --git a/roles/wazuh/ansible-wazuh-agent/tasks/Linux.yml b/roles/wazuh/ansible-wazuh-agent/tasks/Linux.yml
index 034a3122..642e26a8 100644
--- a/roles/wazuh/ansible-wazuh-agent/tasks/Linux.yml
+++ b/roles/wazuh/ansible-wazuh-agent/tasks/Linux.yml
@@ -116,6 +116,13 @@
 - name: Linux | Agent registration via rest-API
   block:
 
+    - name: Establish target Wazuh Manager for registration task
+      set_fact:
+        target_manager: '{{ manager_primary | length | ternary(manager_primary, manager_fallback) | first }}'
+      vars:
+        manager_primary: "{{ wazuh_managers | selectattr('register','true') | list }}"
+        manager_fallback: "{{ wazuh_managers | list }}"
+
     - name: Linux | Create the agent key via rest-API
       uri:
         url: "{{ wazuh_managers.0.api_proto }}://{{ wazuh_agent_authd.registration_address }}:{{ wazuh_managers.0.api_port }}/agents/"