{%- if grains['role'] in ['so-node', 'so-searchnode', 'so-heavynode'] %}
  {%- set ELASTICSEARCH_HOST = salt['pillar.get']('node:mainip', '') -%}  
  {%- set ELASTICSEARCH_PORT = salt['pillar.get']('node:es_port', '') -%}
  {%- set LOG_SIZE_LIMIT = salt['pillar.get']('node:log_size_limit', '') -%}
{%- elif grains['role'] in ['so-eval', 'so-mastersearch', 'so-standalone'] %}
  {%- set ELASTICSEARCH_HOST = salt['pillar.get']('master:mainip', '') -%}
  {%- set ELASTICSEARCH_PORT = salt['pillar.get']('master:es_port', '') -%}
  {%- set LOG_SIZE_LIMIT = salt['pillar.get']('master:log_size_limit', '') -%}
{%- endif -%}

#!/bin/bash
#
# Copyright 2014,2015,2016,2017,2018 Security Onion Solutions, LLC
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

#. /usr/sbin/so-elastic-common
#. /etc/nsm/securityonion.conf

LOG="/opt/so/log/curator/so-curator-closed-delete.log"

# Check for 2 conditions:
# 1. Are Elasticsearch indices using more disk space than LOG_SIZE_LIMIT?
# 2. Are there any closed logstash- indices that we can delete?
# If both conditions are true, keep on looping until one of the conditions is false.
while [[ $(du -hs --block-size=1GB /nsm/elasticsearch/nodes | awk '{print $1}' ) -gt "{{LOG_SIZE_LIMIT}}" ]] &&
curl -s {{ELASTICSEARCH_HOST}}:{{ELASTICSEARCH_PORT}}/_cat/indices | grep "^      close logstash-" > /dev/null; do

	# We need to determine OLDEST_INDEX.
	# First, get the list of closed indices that are prefixed with "logstash-".
	# For example: logstash-ids-YYYY.MM.DD
	# Then, sort by date by telling sort to use hyphen as delimiter and then sort on the third field.
	# Finally, select the first entry in that sorted list.
	OLDEST_INDEX=$(curl -s {{ELASTICSEARCH_HOST}}:{{ELASTICSEARCH_PORT}}/_cat/indices | grep "^      close logstash-" | awk '{print $2}' | sort -t- -k3 | head -1)

	# Now that we've determined OLDEST_INDEX, ask Elasticsearch to delete it.
	curl -XDELETE {{ELASTICSEARCH_HOST}}:{{ELASTICSEARCH_PORT}}/${OLDEST_INDEX}

	# Finally, write a log entry that says we deleted it.
	echo "$(date) - Used disk space exceeds LOG_SIZE_LIMIT ({{LOG_SIZE_LIMIT}} GB) - Index ${OLDEST_INDEX} deleted ..." >> ${LOG}

done