cd $(dirname $0)

MAX_SOURCE_OFFSET=3600
MAX_MASTER_OFFSET=300
MAX_SLAVE_OFFSET=300

adei_default_timeout=${default_timeout:-120}

. lib/adei/adei_version.sh
. lib/adei/adei_print.sh
. lib/adei/adei_error.sh
. lib/nagios/nagios.sh

shopt -s nocasematch;

function adei_curl {
    if [ -n "$adei_auth" ]; then
        curl -u "$adei_auth" "$@"
    else
        curl "$@"
    fi
}

# Auth with redirect (-L/--location-trusted) not working. Credentials are lost on hops.
function adei_query {
    local resp
    local timeout=${3:-$adei_default_timeout}

    local err=0
    if [ "$2" == "ecode" ]; then
        url="$1&mysql=master"
#	resp=$(adei_curl --proxy "" -f -m "$timeout" "$url" 2>&1 | grep "returned error")
	resp=$(adei_curl --proxy "" --location-trusted -sf -m "$timeout" -w "%{http_code}" -o /dev/null "$url" 2>&1);
	[ -z "$resp" -o "$resp" = "000" ] && resp="10" # error code 1
    elif [ "$2" == "emsg" ]; then
        url="$1&mysql=master"
	resp=$(adei_curl --proxy "" --location-trusted -f -m "$timeout" "$url" 2>&1 | grep -o "curl.*")
	[ -z "$resp" ] && resp="No response from"
    else
        if [ "$2" == "slave" ]; then
            url="$1&mysql=slave"
        else
            url="$1&mysql=master"
        fi

        resp=$(adei_curl --proxy "" --location-trusted -sf -m "$timeout" "$url"); err=$?
	if [ $err -ne 0 ]; then
	    resp=$(adei_query "$1" "ecode" "$timeout")
	    err=$(($(($resp / 10)) + $(($resp % 10))))

	    resp=$(adei_query "$1" "emsg" "$timeout")

	    [ -n "$debug" ] && echo "$(date) $timeout Failed $url" >> /tmp/adei.log
	else
	    [ -n "$debug" ] && echo "$(date) $timeout OK     $url" >> /tmp/adei.log
	fi
    fi

    echo -n $resp
    return $err
}

function adei_format_query {
    local with_auth="${1:-1}"
    local query="$2"
    local database=${3:-master}                         # ignoring here, it will be handled separately by adei_query
    local timeout=${4:-$adei_default_timeout}
    local source=${5:-"$adei_source"}
    local url="${6:-$adei_url}"

    auth=""
    [ $with_auth -gt 0 ] && auth="${adei_auth}@"
    if [[ "$query" =~ \? ]]; then
	echo "http://${auth}${url}/${query}${source}"
    else
	echo "http://${auth}${url}/${query}?xxxxx${source}"
    fi
}

function adei_simple_query {
    local req=$(adei_format_query 0 "$@")
    local database=${2:-master}
    adei_query "$req" $database "$timeout"; err=$?
    return $err
}

function adei_text_query {
    local out	# Local on the same string breaks error reporting
    out="$(adei_simple_query "$@")"; local err=$?

    if [ $err -gt 0 -o -z "$out" ]; then
#	echo "Error sending ADEI request: $(adei_format_query 0 "$@")"
	[ $err -eq 0 ] && local err=1
	[ -n "$out" ] && echo -n "$out"
	return $err
    elif [[ "$out" =~ "Error:" ]]; then
	echo -n "$out"
	return 7
    else
	echo -n "$out"
	return 0
    fi
}

function adei_xml_query {
    local out	# Local on the same string breaks error reporting
    out="$(adei_simple_query "$@")"; local err=$?

    if [ $err -ne 0 -o -z "$out" ]; then
#	echo "Error sending ADEI request: $(adei_format_query 0 '$@')"
	[ $err -eq 0 ] && err=1
	[ -n "$out" ] && echo "$out"
	return $err
    fi

    local xml="$(echo "$out" | xmllint --format - 2>/dev/null)"
    if [ $err -ne 0 -o -z "$out" ]; then
	echo "$out"
	if [[ "$out" =~ "Error:" ]]; then
	    return 7
	else
	    return 4
	fi
    fi

    error=$(echo "$xml" | grep "<Error>")
    if [ -n "$error" ]; then
	echo $error | sed -e "s|</\?Error>||g"
	return 7
    fi

    echo "$xml"
    return $err
}

function adei_value_query {
    local out	# Local on the same string breaks error reporting
    out="$(adei_xml_query "$@")"; local err=$?
    [ $err -ne 0 ] && { echo "$out"; return $err; }

    local values
    values="$(echo "$out" | grep "Value")"; err=$?
    [ $err -ne 0 ] && return 6

    echo "$values"
}


function adei_get_databases {
    local out
    out="$(adei_xml_query "list.php?target=databases" "$@")"
    [ $err -gt 0 ] && { echo -n "$out"; return $err; }
    echo "$out" | grep "Value" | sed -e "s/^.*db_name=\"\([^\"]*\)\".*$/\\1/" | sed -e "s/ /::space::/"
}

function adei_query_version {
    local version	# Local on the same string breaks error reporting

    version="$(adei_text_query "info.php?target=version&encoding=text" "$@")"; local err=$?
    [ $err -gt 0 ] && { echo "$version"; return $err; }

    adei_version="$version"

    adei_revision=$(echo $adei_version | cut -d '-' -f 1)
    if [ "$adei_revision" == "$adei_version" ]; then 
	adei_date=""
    else
	adei_date=$(echo $adei_version | cut -d '-' -f 2)
    fi
}

function adei_resolve_id {
    fn="$1"
    id="$2"
    host="$3"
    setup="$4"

    local var=$(cat "$fn"  | grep "^$id" | awk '{ print $2 }')
    [ -z "$var" -a -n "$host" ] && var=$(cat "$fn"  | grep "^$host"  | awk '{ print $2 }')
    [ -z "$var" -a -n "$setup" ] && var=$(cat "$fn"  | grep "^$setup" | awk '{ print $2 }')

    [ -n "$debug" ] && echo "$(date) resolved ($var) from $fn (id=$id, host=$host, setup=$setup) pwd ($(pwd))" >> /tmp/resolv.log
    echo "$var"
}

function adei_init_ {
    local id="$1" && shift		# Either URL or [setup]@[host]
#    local url="$1" && shift
    local server="$1" && shift
    local database="$1" && [[ ! "$server" =~ \&|= ]] && shift
    adei_args=( "$@" )

    [ -z "$id" ] && { echo "ADEI ID is not specified" && exit 8 ; }


    local url
    local host
    local setup
    if [[ $id =~ http.*:// ]]; then				# url
	url="$id"
	host="$(echo ${url#http*://} | cut -d '/' -f 1)"	# parse port, maybe

	unset $setup
    elif [[ $id =~ : ]]; then					# [setup@]host
	local seho
	IFS='@' read -ra seho <<< "$id" && shift
	host="${seho[1]:-${seho[0]}}"				# parse port, for sure
        setup="${seho[1]:+${seho[0]}}"

        unset $url
    else							# [setup]@[host] or both, no port
	local seho
	IFS='@' read -ra seho <<< "$id" && shift		# techincally we can try to resolve to decide which is which

	host="${seho[1]:-${seho[0]}}"				# This could be mixed up, but it is either not important, or correct, or misconfigured anyway
        setup="${seho[1]:+${seho[0]}}"				# Only error if only setup is provided. But we handle it in resolution code which agnostic and treats them as ids 

        url=$(adei_resolve_id "setup/adei.txt" "$id" "$host" "$setup")

        if [ -n "$url" ]; then					# Get URL, now we can determine host correctly and ignore the ids.
	    host="$(echo ${url#http*://} | cut -d '/' -f 1)"
	    unset $setup
	else							# only 'setup' is not allowed if URL is not configured, so it should be host (or error in configuration)
	    unset $url						# so either we have both (correctly) or only host (correctly)
	fi
    fi

    local hopo
    IFS=':' read -ra hopo <<< "$host" && shift
    host="${hopo[0]}"					# only non fqdn if also url not set
    port=":${hopo[1]:-80}"
    [ $port = ":80" ] && port=""

    # Preliminary for the sake of error reporting
    adei_host="$host"
    adei_port="$port"

    local fqdn
    if [ -z "$url" ]; then
        url=$(adei_resolve_id "setup/adei.txt" "$id" "$host" "$setup")
	if [ -n "$url" ]; then
	    fqdn="$(echo ${url#http*://} | cut -d '/' -f 1)"

	    IFS=':' read -ra hopo <<< "$fqdn" && shift
	    host="${hopo[0]}"				# again fqdn
	    port=":${hopo[1]:-80}"
	    [ $port = ":80" ] && port=""
	else
	    fqdn=$(resolve_fqdn "$host")		# this may be not fqdn
	    [ -n "$fqdn" ] && host="$fqdn"
	    url="${host}${port}/adei"
	fi
    fi

    # In some case we miss here "setup", but surely then the password is resolved host-based
    adei_auth=$(adei_resolve_id "security/adei.txt" "$id" "$host" "$setup")

    # Now build ADEI url
    adei_setup="$setup"
    adei_host="$host"
    adei_port="$port"
    adei_url="${url#http*://}/services"

    # Now check port
    adei_online=$(scripts/ping.pl "$host" "$port")
    [ $adei_online -ne 1 ] && return 2

    adei_query_version; local err=$?
    [ $err -gt 0 ] && return $err


    if [[ "$server" =~ \&|= ]]; then
	adei_source="&$server"
    else
	adei_source=""
	[ -n "$adei_setup" -a "$adei_setup" != "*" ] && adei_source+="&setup=$adei_setup"
        [ -n "$server" -a "$server" != "-" ] && adei_source+="&db_server=$server"

	if [ "$database" == "#1" ]; then
	    databases="$(adei_get_databases)"; err=$?
	    [ $err -gt 0 ] && { echo "Failed to query ADEI databases: $databases"; return $err; }
	    [ -z "$databases" ] && { echo "No databases reported by ADEI"; return 6; }
	    database=$(echo "$databases" | head -n 1)
	fi

	[ -n "$database" -a "$database" != "-" ] && adei_source+="&db_name=$database"
    fi

    return 0
}

function adei_init {
    adei_init_ "$@"; local code=$?

    adei_healthy=$(($code == 0))
    adei_process_error "$code" "" "*" "$(adei_print_status "$0" $adei_online $adei_healthy)"
}