#!/bin/bash
# Installer for Datadog (www.datadoghq.com).
# Copyright 2016-present Datadog, Inc.
#
set -euo pipefail
umask 0

os=$(uname -s)
arch=$(uname -m)

if [[ "$os" != "Linux" || ("$arch" != "x86_64" && "$arch" != "aarch64") ]]; then
  echo "This installer only supports Linux running on amd64 or arm64." >&2
  exit 1
fi

if ! command -v sha256sum >/dev/null || ! (command -v curl >/dev/null || command -v wget >/dev/null); then
  echo "This installer requires sha256sum and either curl or wget to be installed." >&2
  exit 1
fi

flavor="default"
version="7.79.0-1"
export DD_INSTALLER_DEFAULT_PKG_VERSION_DATADOG_INSTALLER="$version"
case "$arch" in
x86_64)
  installer_sha256="8ac330999cc98097fabdab62761327a27c283054792ab68c34cc2c66c4028c8a"
  ;;
aarch64)
  installer_sha256="444ad78097b893ff816893090f9e9686ad1fbba9a0f6f9a073584c652f1fcbdc"
  ;;
esac
site=${DD_SITE:-datadoghq.com}
installer_domain=${DD_INSTALLER_REGISTRY_URL_INSTALLER_PACKAGE:-$([[ "$site" == "datad0g.com" ]] && echo "install.datad0g.com" || echo "install.datadoghq.com")}
installer_url="https://${installer_domain}/v2/agent-package/blobs/sha256:${installer_sha256}"

tmp_dir="/opt/datadog-packages/tmp"
tmp_bin="${tmp_dir}/installer"

if ((UID == 0)); then
  sudo_cmd=()
  # When running as root, no sudo needed
  sudo_env_cmd() {
    "$@"
  }
else
  sudo_cmd=(sudo)
  # sudo-rs (used in Ubuntu 25) doesn't support -E flag
  # Detect sudo-rs and use explicit env passing instead
  if sudo --version 2>&1 | sed -n '1p' | grep -q 'sudo-rs'; then
    sudo_env_cmd() {
      local env_args=()
      while IFS='=' read -r name value; do
        case "$name" in
          DD_*|HTTP_PROXY|HTTPS_PROXY|NO_PROXY|http_proxy|https_proxy|no_proxy)
            env_args+=("$name=$value")
            ;;
        esac
      done < <(env)
      sudo env "${env_args[@]}" "$@"
    }
  else
    # Traditional sudo supports -E flag
    sudo_env_cmd() {
      sudo -E "$@"
    }
  fi
fi

# This migrates legacy installs by removing the legacy deb / rpm installer package
if command -v dpkg >/dev/null && dpkg -s datadog-installer >/dev/null 2>&1; then
  "${sudo_cmd[@]+"${sudo_cmd[@]}"}" datadog-installer purge >/dev/null 2>&1 || true
  "${sudo_cmd[@]+"${sudo_cmd[@]}"}" dpkg --purge datadog-installer >/dev/null 2>&1 || true
  DATADOG_AGENT_OPTIONAL_REMOVE_DEB_CMD
elif command -v rpm >/dev/null && rpm -q datadog-installer >/dev/null 2>&1; then
  "${sudo_cmd[@]+"${sudo_cmd[@]}"}" datadog-installer purge >/dev/null 2>&1 || true
  "${sudo_cmd[@]+"${sudo_cmd[@]}"}" rpm -e datadog-installer >/dev/null 2>&1 || true
fi



"${sudo_cmd[@]+"${sudo_cmd[@]}"}" mkdir -p "$tmp_dir"

echo "Downloading the Datadog installer..."
if command -v curl >/dev/null; then
  if ! sudo_env_cmd curl -L --retry 3 "$installer_url" --output "$tmp_bin" >/dev/null; then
    echo "Error: Download failed with curl." >&2
    exit 1
  fi
else
  if ! sudo_env_cmd wget --tries=3 -O "$tmp_bin" "$installer_url" >/dev/null; then
    echo "Error: Download failed with wget." >&2
    exit 1
  fi
fi
"${sudo_cmd[@]+"${sudo_cmd[@]}"}" chmod +x "$tmp_bin"

echo "Verifying installer integrity..."
sha256sum -c <<<"$installer_sha256  $tmp_bin" >/dev/null
echo "Installer integrity verified."

echo "Starting the Datadog installer..."
sudo_env_cmd "$tmp_bin" setup --flavor "$flavor" "$@"

"${sudo_cmd[@]+"${sudo_cmd[@]}"}" rm -f "$tmp_bin"

exit 0