From 657ddc42a85d3f804c9a467be870abfc5f374926 Mon Sep 17 00:00:00 2001
From: Mike Reeves <mike.reeves@securityonionsolutions.com>
Date: Fri, 27 Sep 2019 10:40:28 -0400
Subject: [PATCH] Playbook - Add flag for runtime

---
 salt/top.sls        | 10 +++++++++-
 so-setup-network.sh |  1 +
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/salt/top.sls b/salt/top.sls
index f3f8c0a73..cf5d47699 100644
--- a/salt/top.sls
+++ b/salt/top.sls
@@ -3,6 +3,7 @@
 {%- set WAZUH = salt['pillar.get']('master:wazuh', '0') -%}
 {%- set GRAFANA = salt['pillar.get']('master:grafana', '0') -%}
 {%- set THEHIVE = salt['pillar.get']('master:thehive', '0') -%}
+{%- set PLAYBOOK = salt['pillar.get']('master:playbook', '0') -%}
 base:
   'G@role:so-sensor':
     - ca
@@ -55,6 +56,10 @@ base:
     {%- if THEHIVE != 0 %}
     - hive
     {%- endif %}
+    {%- if PLAYBOOK != 0 %}
+    - playbook
+    {%- endif %}
+
 
 
   'G@role:so-master':
@@ -87,7 +92,10 @@ base:
     {%- if THEHIVE != 0 %}
     - hive
     {%- endif %}
-
+    {%- if PLAYBOOK != 0 %}
+    - playbook
+    {%- endif %}
+    
 
   # Storage node logic
 
diff --git a/so-setup-network.sh b/so-setup-network.sh
index c0159e71d..69ee7839e 100644
--- a/so-setup-network.sh
+++ b/so-setup-network.sh
@@ -539,6 +539,7 @@ master_pillar() {
   echo "  osquery: $OSQUERY" >> /opt/so/saltstack/pillar/masters/$HOSTNAME.sls
   echo "  wazuh: $WAZUH" >> /opt/so/saltstack/pillar/masters/$HOSTNAME.sls
   echo "  thehive: $THEHIVE" >> /opt/so/saltstack/pillar/masters/$HOSTNAME.sls
+  echo "  playbook: 0" >> /opt/so/saltstack/pillar/masters/$HOSTNAME.sls
   }
 
 master_static() {