#!/usr/bin/env bash

APIkeyFile="${HOME}/.nagiosapikey"

APIKEY=""
XI_URL=""
curl="curl -k -s"
# curl -XGET "https://192.168.1.19/nagiosxi/api/v1/objects/service?apikey=<key>"

verbose="0"
doCreate=""
myHost=""
myService=""
myHG=""
mySG=""
myFields=""
myCommand=""
mySave=""
myFile=""
tmpJSON=""
moreJQ=""

myAPI="objects"
myAPIobject="servicestatus"

while [ -n "$1" ]; do
  case "$1" in
    --keyfile) APIkeyFile="$2"; shift 2;;
    -j|--jq) moreJQ="$2"; shift 2;;
    -v|--verbose) verbose=$(($verbose + 1)); shift 1;;
    --create) doCreate="true"; shift 1;;
    --key) APIKEY="$2"; shift 2;;
    --url) XI_URL="$2"; shift 2;;
    --save) mySave="$2"; shift 2;;
    --file) myFile="$2"; shift 2;;
    -f|--fields) myFields="$2"; shift 2;;
    -h|--host) myHost="$2"; shift 2;;
    -s|--service) myService="$2"; shift 2;;
    -c|--command) myCommand="$2"; shift 2;;
    -hg|--hostgroup) myHG="$2"; shift 2;;
    -sg|--servicegroup) mySG="$2"; shift 2;;
    --api) myAPI="$2"; shift 2;;
    --object) myAPIobject="$2"; shift 2;;
    *) shift 1;;
  esac
done

if [ -r "$APIkeyFile" ]; then
  while read url key; do
    if [ -z "$XI_URL" ]; then
      XI_URL="$url"
      APIKEY="$key"
      continue
    fi
    if [ "$url" = "$XI_URL" ]; then
      APIKEY="$key"
    fi
  done < "$APIkeyFile"
fi

do_debug() {
  [ "$verbose" -ge "$1" ] && echo "$2" >&2
}

do_api() {
  api_start="$1"
  api_command="$2"
  url="${XI_URL}/nagiosxi/api/v1/${api_start}/${api_command}?apikey=${APIKEY}&pretty=0"
  do_debug 2 "start=$1 command=$2"
  do_debug 1 "Executing: $url"
  $curl -XGET -k "$url"
}

# curl -XPOST "http://192.168.1.128/nagiosxi/api/v1/config/hostgroup?apikey=fsZZ4pXaKaVjSG7IYcjMRYhK8NqcqN2NGPck8gPhFoZMJGKj4YUjZCF8qSqsK7Ln&pretty=1" -d "hostgroup_name=testapihostgroup&alias=HostGroup&applyconfig=1"
do_api_post() {
  api_start="$1"
  api_command="$2"
  url="${XI_URL}/nagiosxi/api/v1/${api_start}/${api_command}?apikey=${APIKEY}&pretty=0"
  do_debug 2 "start=$1 command=$2"
  do_debug 1 "Executing: $url"
  $curl -XPOST -k "$url" -d "hostgroup_name=$myHG&alias=$myHG&applyconfig=0"
}

# At this time, all we can create ia a hostgroup
create_hostgroup() {
  do_debug 1 "about to do an API post call"
  do_api_post config hostgroup
}

do_create() {
  do_debug 1 "about to do a create_command"
  [ -n "$myHG" ] && create_hostgroup
  exit
}

# If we said "--create" then we want to make something
if [ -n "$doCreate" ]; then
  do_create
  exit
fi

# Grab a copy of the JSON data so we don't have to keep making calls over and over
# If we used an existing file, then just use that
if [ -z "$myFile" ]; then
  tmpJSON=`mktemp`
  do_debug 1 "tmp file is $tmpJSON"
  do_debug 2 "myAPI is $myAPI and myAPIobject is $myAPIobject"
  do_api "$myAPI" "$myAPIobject" > $tmpJSON
else
  do_debug 1 "myFile=$myFile"
  tmpJSON="$myFile"
fi

# if mySave is not empty, then we're just saving it into the file called $mySave
if [ -n "$mySave" ]; then
  mv $tmpJSON $mySave
  echo "JSON data saved to $mySave"
  exit
fi

# Otherwise, let's parse the JSON data here
# Parse our string
jqString=""
if [ "$myAPIobject" = "servicestatus" ]; then
  jqString=".servicestatus[]"
  [ -n "$myHost" ] && jqString="$jqString | select(.host_name | test(\"$myHost\"))"
  [ -n "$myService" ] && jqString="$jqString | select(.service_description | test(\"$myService\"))"
  [ -n "$myCommand" ] && jqString="$jqString | select(.check_command | test(\"$myCommand\"))"
  [ -n "$myHG" ] && jqString="$jqString | select(.hostgroup_name==\"$myHG\")"
  [ -n "$mySG" ] && jqString="$jqString | select(.servicegroup_name==\"$mySG\")"
elif [ "$myAPIobject" = "hostgroup" ]; then
  jqString=".[]"
  [ -n "$myHG" ] && jqString="$jqString | select(.hostgroup_name==\"$myHG\")"
elif [ "$myAPIobject" = "host" ]; then
  jqString=".[]"
  [ -n "$myHost" ] && jqString="$jqString | select(.host_name | test(\"$myHost\"))"
fi
[ -n "$myFields" ] && jqString="$jqString | $myFields"
jqString="$jqString $moreJQ"

cat $tmpJSON | jq -r "$jqString"
[ -z "$myFile" ] && rm $tmpJSON