################################################################################
# Copyright 2022 Antonin Portelli 
# 
# This file is part of miria-cli.
#
# miria-cli is free software: you can redistribute it and/or modify it under 
# the terms of the GNU General Public License as published by the Free Software 
# Foundation, either version 3 of the License, or (at your option) any 
# later version.
#
# miria-cli is distributed in the hope that it will be useful, but WITHOUT ANY 
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS 
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with 
# miria-cli. If not, see <https://www.gnu.org/licenses/>.
################################################################################

# Functions used in miria-cli commands #########################################

#######################################
# Print error message
#
# Arguments:
#   Error message
#######################################
miria_error() {
  echo "[miria-cli] error: $1" 1>&2
}

#######################################
# Get Mira API URL
#
# Arguments:
#   None
# Output:
#   miria_url : Miria API URL
#######################################
miria_get_url() {
  if [ -e ~/.miria ]; then
    miria_url="$(grep -E ' *url +' ~/.miria | awk '{print $2}')"
  else
    miria_error 'file ~/.miria not found'
    exit 1
  fi
}

#######################################
# Get Mira agent
#
# Arguments:
#   None
# Output:
#   miria_agent : Miria agent name
#######################################
miria_get_agent() {
  if [ -e ~/.miria ]; then
    miria_agent="$(grep -E ' *agent +' ~/.miria | awk '{print $2}')"
  else
    miria_error 'file ~/.miria not found'
    exit 1
  fi
}

#######################################
# Execute Mira API command
#
# Arguments:
#   API command,
#   command argument 1 (key=value),
#   ...
#   command argument n (key=value)
# Outputs:
#   miria_last_xml    : last XML answer
#   miria_last_status : last return code
#   miria_last_error  : last error message
#######################################
miria_api_error_fatal='true'

miria_api_cmd() {
  local url_args=()
  local cmd="$1"

  miria_get_url
  shift
  for a in "$@"; do
    url_args+=('--data')
    url_args+=("${a}")
  done
  miria_last_xml="$(curl -sG "${url_args[@]}" "${miria_url}/${cmd}")"
  miria_last_status="$(miria_xml_get_attribute "${miria_last_xml}" '//ReturnCode' 'ADARetCode')"
  miria_last_error="$(miria_xml_get_string "${miria_last_xml}" '//ErrorMsg')"
  if [[ "${miria_last_status}" != '1' ]] && [[ "${miria_api_error_fatal}" == 'true' ]]; then
    miria_error "Miria API call returned status ${miria_last_status}"
    miria_error "${miria_last_error}"
    exit 1
  fi
}

#######################################
# Get XML string(s)
#
# Arguments:
#   XML data (in string form),
#   XPath of string
#######################################
miria_xml_get_string() {
  local xml="$1"
  local xpath="$2"

  echo "${xml}" | xmllint --xpath "string(${xpath})" -
}

#######################################
# Get XML string(s) from last answer
#
# Arguments:
#   XPath of string
#######################################
miria_last_get_string() {
  local xpath="$1"

  miria_xml_get_string "${miria_last_xml}" "//${xpath}"
}

#######################################
# Get XML attribute value(s)
#
# Arguments:
#   XML data (in string form),
#   XPath of attribute (without @attribute)
#   Attribute name
#######################################
miria_xml_get_attribute() {
  local xml="$1"
  local xpath="$2"
  local attr="$3"

  echo "${xml}" | xmllint --xpath "${xpath}/@${attr}" - \
                | sed -E "s/ *${attr}=\"(.+)\"/\1/g"
}