#!/bin/bash
set -e

REPO="fosrl/newt"
INSTALL_DIR="/opt/newt"
SERVICE_NAME="PVE-MGNT"
SYMLINK="${INSTALL_DIR}/newt_latest"
SERVICE_FILE="/etc/systemd/system/${SERVICE_NAME}.service"

info() { echo -e "\033[0;32m[INFO]\033[0m $1"; }
error() { echo -e "\033[0;31m[ERROR]\033[0m $1"; exit 1; }

get_latest_version() {
  curl -fsSL "https://api.github.com/repos/${REPO}/releases/latest" | grep -Po '"tag_name": *"\K[^"]+'
}

download_newt() {
  local version="$1"
  local arch=$(uname -m)
  local file=""

  case "$arch" in
    x86_64) file="newt_linux_amd64" ;;
    aarch64) file="newt_linux_arm64" ;;
    armv7l) file="newt_linux_arm32" ;;
    *) error "Nicht unterstützte Architektur: $arch" ;;
  esac

  local url="https://github.com/${REPO}/releases/download/${version}/${file}"
  local target="${INSTALL_DIR}/newt_${version}"
  mkdir -p "$INSTALL_DIR"

  info "⬇️ Lade $url herunter …"
  curl -fsSL "$url" -o "${target}.tmp" || error "Download fehlgeschlagen."
  chmod +x "${target}.tmp"
  mv "${target}.tmp" "$target"
  ln -sf "$target" "$SYMLINK"
  info "✅ newt ${version} installiert als ${target}"
}

setup_systemd_service() {
  echo "🆔 Bitte gib die Pangolin-ID ein: "
  read -r PANGOLIN_ID
  echo "🔑 Bitte gib das Secret ein: "
  read -r PANGOLIN_SECRET
  echo "🌐 Bitte gib den Endpoint (z. B. https://pangolin.domain.de) ein: "
  read -r PANGOLIN_ENDPOINT

  cat > "$SERVICE_FILE" <<EOF
[Unit]
Description=Newt Client - ${SERVICE_NAME}
After=network.target

[Service]
ExecStart=${SYMLINK} --id ${PANGOLIN_ID} --secret ${PANGOLIN_SECRET} --endpoint ${PANGOLIN_ENDPOINT}
Restart=always
RestartSec=5

[Install]
WantedBy=multi-user.target
EOF

  systemctl daemon-reload
  systemctl enable "$SERVICE_NAME"
  systemctl restart "$SERVICE_NAME"
  info "🛠️ systemd-Dienst ${SERVICE_NAME} eingerichtet und neu gestartet."
}

main() {
  local version
  version=$(get_latest_version)
  version=${version#v}
  info "📦 Neueste Version: $version"

  download_newt "$version"
  setup_systemd_service
  info "🚀 Reinstallation abgeschlossen!"
}

main "$@"