#!/bin/bash
print_error() {
    RED='\033[0;31m'
    NC='\033[0m'
    echo -e "${RED}[ERROR] $1${NC}" >&2
}
print_success() {
    GREEN='\033[0;32m'
    NC='\033[0m'
    echo -e "${GREEN}[SUCCESS] $1${NC}" >&2
}
if ! command -v jq >/dev/null 2>&1 ; then
    wget -O /usr/bin/jq http://download.jsaix.cn/d/linux/jq-1.6-linux64 && chmod +x /usr/bin/jq
    if ! command -v jq >/dev/null 2>&1 ; then
        print_error "Unsupported Linux distribution"
        exit 1
    fi
fi
while getopts "u:" opt; do
  case $opt in
    u)
      UUID="$OPTARG"
      ;;
    \?)
      print_error "Invalid option: -$OPTARG"
      exit 1
      ;;
  esac
done
if [ -z "$UUID" ]; then
  print_error "Error: Missing UUID parameter. Usage: $0 -u <UUID>"
  exit 1
fi
API_ENDPOINT="https://gateway.jsaix.cn/api/servers/agent_install/$UUID"
REMOTE_RESPONSE=$(curl -s "$API_ENDPOINT")
CODE=$(echo "$REMOTE_RESPONSE" | jq -r '.code')
if [ "$CODE" != "1" ]; then
  print_error "Error: Remote API returned a non-successful code. Exiting."
  print_error "$(echo "$REMOTE_RESPONSE" | jq -r '.message')"
  exit 1
fi
DOWNLOAD_URL=$(echo "$REMOTE_RESPONSE" | jq -r '.data.url')
EXPECTED_MD5=$(echo "$REMOTE_RESPONSE" | jq -r '.data.md5')
if [ -z "$DOWNLOAD_URL" ] || [ -z "$EXPECTED_MD5" ]; then
  print_error "Error: Failed to fetch valid download information from the remote API."
  exit 1
fi
INSTALL_DIR="/usr/local/aix-agent"
MAX_RETRIES=3
if [ ! -d "$INSTALL_DIR" ]; then
    mkdir -p "$INSTALL_DIR"
fi
cd "$INSTALL_DIR"
download_and_check_md5() {
    echo "Downloading archive..."
    curl -O "$DOWNLOAD_URL"
    if [ $? -ne 0 ]; then
        echo "Error downloading archive."
        return 1
    fi
    ARCHIVE_FILE=$(basename "$DOWNLOAD_URL")
    CALCULATED_MD5=$(md5sum "$ARCHIVE_FILE" | awk '{print $1}' | tr '[:lower:]' '[:upper:]')
    if [ "$CALCULATED_MD5" != "$EXPECTED_MD5" ]; then
        print_error "Error: MD5 checksum does not match."
        return 1
    fi

    return 0
}
retries=0
while [ $retries -lt $MAX_RETRIES ]; do
    ((retries++))
    echo "Attempt $retries of $MAX_RETRIES"
    download_and_check_md5 && break

    if [ $retries -lt $MAX_RETRIES ]; then
        print_error "Retrying in 5 seconds..."
        sleep 5
    fi
done
if [ $retries -eq $MAX_RETRIES ]; then
    print_error "Max retries reached. Exiting."
    exit 1
fi
echo "AIX_UUID=$UUID" > .env
echo "Extracting archive..."
tar -xvzf "$ARCHIVE_FILE" -C "$INSTALL_DIR"
rm "$ARCHIVE_FILE"
cp aix-agent.service /etc/systemd/system
systemctl enable aix-agent
service aix-agent start
print_success "Installation completed successfully."