escli.bash

#!/usr/bin/env bash

#------------------------------------------------
# Detect how we were called es or kb and [l|p|c] for env
#------------------------------------------------
calledCmd="$(basename $0)"
[[ $calledCmd == "esl" ]] && env="lab1"
[[ $calledCmd == "esp" ]] && env="rdu1"
[[ $calledCmd == "esc" ]] && env="aws1"
[[ $calledCmd == "kbl" ]] && env="lab1" && kbHost="logs-lab"
[[ $calledCmd == "kbp" ]] && env="rdu1" && kbHost="logs"
[[ $calledCmd == "kbc" ]] && env="aws1" && kbHost="XXXXX"

#------------------------------------------------
# source escli.conf variables
#------------------------------------------------
# g* tools via brew install coreutils
[ $(uname) == "Darwin" ] && readlink=greadlink || readlink=readlink

# source default escli.conf
. $(dirname $($readlink -f $0))/escli.conf
# source secondary escli_c.conf if called w/ esc
[[ "$env" == "aws1" ]] && . $(dirname $($readlink -f $0))/escli_c.conf

usage () {
    cat <<-EOF

    USAGE: $0 [HEAD|GET|PUT|POST] '...ES/KB REST CALL...'

    EXAMPLES:

        $0 GET  '_cat/shards?pretty'
        $0 GET  '_cat/indices?pretty&v&human'
        $0 GET  '_cat'
        $0 GET  ''
        $0 PUT  '_all/_settings'   -d "\$DATA"
        $0 POST '_cluster/reroute' -d "\$DATA"


	EOF
    exit 1
}

[ "$1" == "" ] && usage

# Target either Elastic or Kibana URL + content type
baseUrl=""
conType=""
if [[ $calledCmd == "esl" || $calledCmd == "esp" || $calledCmd == "esc" ]]; then
    baseUrl=$esBaseUrl
    conType=$esContType
else
    baseUrl=$kbBaseUrl
    conType=$kbContType
fi

#------------------------------------------------
# retrieves user/pass, replace them....
#   - XXXXXXXX (username)
#   - YYYYYYYY (password)
#------------------------------------------------
[ "$4" != "" ] && tmpArg4="$4"
arg4=$(echo "$tmpArg4" | sed -e "s|XXXXXXXX|$(${usernameCmd})|" -e "s|YYYYYYYY|$(${passwordCmd})|")

#------------------------------------------------
# ...ways to call curl.....
#------------------------------------------------
if [ "${1}" == "HEAD" ]; then
    curl -I -skK \
        <(cat <<<"user = \"$( ${usernameCmd} ):$( ${passwordCmd} )\"") \
        "${baseUrl}/$2"
elif [ "${1}" == "PUT" ]; then
    curl -skK \
        <(cat <<<"user = \"$( ${usernameCmd} ):$( ${passwordCmd} )\"") \
        -X$1 -H "${conType}" "${baseUrl}/$2" "$3" "$4"
elif [ "${1}" == "POST" ]; then
    curl -skK \
        <(cat <<<"user = \"$( ${usernameCmd} ):$( ${passwordCmd} )\"") \
        -X$1 -H "${conType}" "${baseUrl}/$2" "$3" "$arg4"
#elif [ "${1}" == "KIBANA" ]; then
#    curl -skK \
#        <(cat <<<"user = \"$( ${usernameCmd} ):$( ${passwordCmd} )\"") \
#        -XPOST -H "${conType}" -H "kbn-xsrf: reporting" "$2"
else
    if [[ $3 || $4 || $5 ]]; then
        argvars="$3 $4 $5"
        curl -skK \
            <(cat <<<"user = \"$( ${usernameCmd} ):$( ${passwordCmd} )\"") \
            -X$1 -H "${conType}" "${baseUrl}/$2" "$argvars"
    else
        curl -skK \
            <(cat <<<"user = \"$( ${usernameCmd} ):$( ${passwordCmd} )\"") \
            -X$1 -H "${conType}" "${baseUrl}/$2" #"$3" "$4" "$5"
    fi
fi

##### TODO #####
# the username logic above which uses `sed` to swap out XXXX and YYYY w/ username + password is a tricky bit. We've tested it with passwords like this:
#
#    echo 'CH\#(8jCCUr;d*aD{}m2dPZY`_9tLRC.3o5K>;~YW%,[Zw"2+D(FK-jPbdsake2^?#a7;N-Y)+4uZ8)(Qe"b#r!,$!*k#]Xr+4Nv'
#
# Bear in mind that `sed` makes use of `|` and so a password that includes this character will likely break things. At some point we should improve the 
# username + password handling, but for today it can at least deal w/ passwords such as the one above.