#!/usr/bin/env bash

#
# nagios-data-search(1) -- List all the hosts (or services) matching any of a
# set of strings.
#
#   $ nds abc-example-
#   $ nds abc- def- ghi-
#   $ nds -s WORDPRESS
#
# Add -d to include a unique list of the hosts downstream of any/all of the
# matched hosts. This is incompatible with the -s option.
#
# Author: Tom Ryder <tom@sanctum.geek.nz>
# Copyright: 2016
#

# Name self
self=nagios-data-search

# Usage printing function
usage() {
    printf 'USAGE: %s [-h | -d | -s] STRING\n' "$self"
}

# By default we search hosts, not services, and don't include downstreams
downstream=0
services=0

# Handle options, just -h help at the moment
OPTIND=1
while getopts 'dhs' opt ; do
    case "$opt" in
        h)
            usage
            exit 0
            ;;
        d)
            downstream=1
            ;;
        s)
            services=1
            ;;
        '?')
            usage >&2
            exit 2
            ;;
    esac
done
shift "$((OPTIND-1))"

# There is no such thing as a downstream service (well, not really), so bail if
# the user asked for them
if ((downstream && services)) ; then
    printf '%s: Cannot specify both -d and -s\n' "$self" >&2
    exit 2
fi

# We need exactly one argument after that
if ! (($#)) ; then
    usage >&2
    exit 2
fi

# Define the path to the Livestatus socket
socket=${MK_LIVESTATUS_SOCKET:-/usr/local/nagios/var/rw/live}

# Iterate through the search terms
for search ; do
    if ((services)) ; then
        unixcat "$socket" <<EOF
GET services
Separators: 10 47 44 124
Columns: host_name description
Filter: host_name ~ $search
Filter: description ~ $search
Or: 2
EOF
    else
        unixcat "$socket" <<EOF
GET hosts
Columns: host_name
Filter: host_name ~ $search
EOF
    fi
done |

# Pass output through a while-read loop that appends the output of ndl(1) for
# this host if downstream is set; otherwise, stdin just gets quietly duplicated
# by this block
while IFS= read -r host ; do
    printf '%s\n' "$host"
    if ((downstream)) ; then
        ndl -- "$host"
    fi
done |

# Sort the output version-wise (so that foo-bar-ap-1 comes before
# foo-bar-ap-10); note that -V is not a POSIX flag (supported by GNU sort)
sort -V |

# Ensure the output is unique
uniq