From 9b84a92ced276c4dc045e67cb501d89860c4a20f Mon Sep 17 00:00:00 2001
From: Mike Reeves <mike.reeves@securityonionsolutions.com>
Date: Wed, 31 Mar 2021 16:47:04 -0400
Subject: [PATCH] Manage the repo files

---
 salt/common/init.sls | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/salt/common/init.sls b/salt/common/init.sls
index d0dae49f5..1ee64bb5a 100644
--- a/salt/common/init.sls
+++ b/salt/common/init.sls
@@ -2,6 +2,7 @@
 {% if sls in allowed_states %}
 
 {% set role = grains.id.split('_') | last %}
+{% set managerupdates = salt['pillar.get']('global:managerupdate', '0') %}
 
 # Remove variables.txt from /tmp - This is temp
 rmvariablesfile:
@@ -84,6 +85,10 @@ crdebug:
   file.absent:
     - name: /etc/yum.repos.d/CentOS-Debuginfo.repo
 
+crdockerce:
+  file.absent: 
+    - name: /etc/yum.repos.d/docker-ce.repo
+    
 crfasttrack:
   file.absent:
     - name: /etc/yum.repos.d/CentOS-fasttrack.repo
@@ -120,6 +125,17 @@ crwazrepo:
   file.absent:
     - name: /etc/yum.repos.d/wazuh.repo
 
+crsecurityonionrepo:
+  file.managed:
+    {% if role in ['eval', 'standalone', 'import', 'manager' 'managersearch'] or managerupdates == 0 %}
+    - name: /etc/yum.repos.d/securityonion.repo
+    - source: salt://common/yum_repos.d/securityonion.repo
+    {% else %}
+    - name: /etc/yum.repos.d/securityonioncache.repo
+    - source: salt://commmon/yum_repos/securityonioncache.repo
+    {% endif %}
+    - mode: 644
+
 {% endif %}
 
 # Install common packages