#!/bin/bash

OPTIND=1         # Reset in case getopts has been used previously in the shell.

# Initialize our own variables:
count_search=""
name=""
password=""
query=""
url=""
value_search=""

script_name=$0

while getopts "c:n:p:q:u:v:" opt; do
  case "$opt" in
  c)  count_search=$OPTARG
    ;;
  n)  name=$OPTARG
    ;;
  p)  password=$OPTARG
    ;;
  q)  query=$OPTARG
    ;;
  u)  url=$OPTARG
    ;;
  v)  value_search=$OPTARG
    ;;
  esac
done

shift $((OPTIND-1))
[ "$1" = "--" ] && shift

#echo "`date` INFO:${script_name}: count_search=${count_search} ,name=${name} ,password=xxx query=${query} \
#             ,url=${url} ,value_search=${value_search} ,Leftovers: $@" >&2

if [[ ${url} = "" || ${name} = "" || ${password} = "" || ${query} = "" ]]; then
  echo "`date` ERROR:${script_name}: url, name, password and query are madatory fields" >&2
  echo  "{\"errorInfo\":\"missing parameter\"}"
  exit 1
fi
if [ `curl --write-out '%{http_code}' --silent --output /dev/null -u ${name}:${password} ${url} -d "<rpc><show><version/></show></rpc>"` != "200" ] ; then
  echo  "{\"errorInfo\":\"management host is not responding\"}"
  exit 1
fi

query_response=`curl -sS -u ${name}:${password} ${url} -d "${query}"`
# Validate first char of response is "<", otherwise no hope of being valid xml
if [[ ${query_response:0:1} != "<" ]] ; then
  echo  "{\"errorInfo\":\"no valid xml returned\"}"
  exit 1
fi
query_response_code=`echo $query_response | xmllint -xpath 'string(/rpc-reply/execute-result/@code)' -`

if [[ -z ${query_response_code} && ${query_response_code} != "ok" ]]; then
  echo  "{\"errorInfo\":\"query failed -${query_response_code}-\"}"
  exit 1
fi

#echo "`date` INFO:${script_name}: Query passed ${query_response_code}" >&2

if [[ ! -z $value_search ]]; then
  value_result=`echo $query_response | xmllint -xpath "string($value_search)" -`
  echo  "{\"errorInfo\":\"\",\"valueSearchResult\":\"${value_result}\"}"
  exit 0
fi

if [[ ! -z $count_search ]]; then
  count_line=`echo $query_response | xmllint -xpath "$count_search" -`
  count_string=`echo $count_search | cut -d '"' -f 2`
  count_result=`echo ${count_line} | tr "><" "\n" | grep -c ${count_string}`
  echo  "{\"errorInfo\":\"\",\"countSearchResult\":${count_result}}"
  exit 0
fi