blob: ffa0bb4bfa345fa86937d5078a425b9c98eb73aa (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#!/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
|