From 13c2c4fa4c062e0b0fe7c6e1cf7ae27b07898af7 Mon Sep 17 00:00:00 2001
From: m0duspwnens <josh.patterson@securityonionsolutions.com>
Date: Tue, 26 May 2020 15:20:12 -0400
Subject: [PATCH] create and move files if they dont exist for addtotab.sh -
 https://github.com/Security-Onion-Solutions/securityonion-saltstack/issues/749

---
 pillar/data/addtotab.sh | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/pillar/data/addtotab.sh b/pillar/data/addtotab.sh
index b20bf22d9..1aa3d6780 100644
--- a/pillar/data/addtotab.sh
+++ b/pillar/data/addtotab.sh
@@ -1,6 +1,7 @@
 #!/usr/bin/env bash
 
 # This script adds sensors/nodes/etc to the nodes tab
+default_salt_dir=/opt/so/saltstack/default
 local_salt_dir=/opt/so/saltstack/local
 TYPE=$1
 NAME=$2
@@ -14,6 +15,14 @@ MONINT=$9
 #NODETYPE=$10
 #HOTNAME=$11
 
+if [ ! -d $local_salt_dir/pillar/data/ ]; then
+  mkdir -p $local_salt_dir/pillar/data/
+fi
+
+if [ ! -f $local_salt_dir/pillar/data/$TYPE.sls ]; then
+  cp $default_salt_dir/pillar/data/$TYPE.sls $local_salt_dir/pillar/data/$TYPE.sls
+fi
+
 echo "Seeing if this host is already in here. If so delete it"
 if grep -q $NAME "$local_salt_dir/pillar/data/$TYPE.sls"; then
   echo "Node Already Present - Let's re-add it"