#!/usr/bin/env bash
set -Eeuo pipefail

pipe() {
  local code="99"
  msg="Failed to connect to $1, reason:"

  curl --disable --silent --max-time 15 --fail --location "${1}" || {
    code="$?"
  }

  case "${code,,}" in
    "6" ) error "$msg could not resolve host!" ;;
    "7" ) error "$msg no internet connection available!" ;;
    "28" ) error "$msg connection timed out!" ;;
    "99" ) return 0 ;;
    *) error "$msg $code" ;;
  esac

  return 1
}

getURL() {
  local id="${1/ /}"
  local ret="$2"
  local url=""
  local arm=""
  local name=""
  local body=""
  local version=""

  case "${id,,}" in
    "alma" | "almalinux" | "alma-linux" )
      version="9"
      name="AlmaLinux"
      if [[ "$ret" == "url" ]]; then
        url="https://repo.almalinux.org/almalinux/${version}/live/x86_64/AlmaLinux-${version}-latest-x86_64-Live-GNOME.iso"
        arm="https://repo.almalinux.org/almalinux/${version}/live/aarch64/AlmaLinux-${version}-latest-aarch64-Live-GNOME.iso"
      fi ;;
    "alpine" | "alpinelinux" | "alpine-linux" )
      name="Alpine Linux"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://dl-cdn.alpinelinux.org/alpine/latest-stable/releases/x86_64/latest-releases.yaml") || exit 65
        version=$(echo "$body" | awk '/"Xen"/{found=0} {if(found) print} /"Virtual"/{found=1}' | grep 'version:' | awk '{print $2}')
        url="https://dl-cdn.alpinelinux.org/alpine/latest-stable/releases/x86_64/alpine-virt-$version-x86_64.iso"
        arm="https://dl-cdn.alpinelinux.org/alpine/latest-stable/releases/aarch64/alpine-virt-$version-aarch64.iso"
      fi ;;
    "arch" | "archlinux" | "arch-linux" )
      name="Arch Linux"
      if [[ "$ret" == "url" ]]; then
        url="https://geo.mirror.pkgbuild.com/iso/latest/archlinux-x86_64.iso"
      fi ;;
    "cachy" | "cachyos" )
      name="CachyOS"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://cachyos.org/download/") || exit 65
        url=$(echo "$body" | tr '&' '\n' | grep "ISO/desktop" | grep -v 'iso.sha' | grep -v 'iso.sig' | cut -d';' -f2)
        arm=$(echo "$body" | tr '&' '\n' | grep "ISO/handheld" | grep -v 'iso.sha' | grep -v 'iso.sig' | cut -d';' -f2)
      fi ;;
    "centos" | "centosstream" | "centos-stream" )
      name="CentOS Stream"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://linuxsoft.cern.ch/centos-stream/") || exit 65
        version=$(echo "$body" | grep "\-stream" | cut -d'"' -f 6 | cut -d'-' -f 1 | head -n 1)
        url="https://mirrors.xtom.de/centos-stream/$version-stream/BaseOS/x86_64/iso/CentOS-Stream-$version-latest-x86_64-dvd1.iso"
        arm="https://mirrors.xtom.de/centos-stream/$version-stream/BaseOS/aarch64/iso/CentOS-Stream-$version-latest-aarch64-dvd1.iso"
      fi ;;
    "debian" )
      name="Debian"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://cdimage.debian.org/debian-cd/") || exit 65
        version=$(echo "$body" | grep '\.[0-9]/' | cut -d'>' -f 9 | cut -d'/' -f 1)
        url="https://cdimage.debian.org/debian-cd/current-live/amd64/iso-hybrid/debian-live-$version-amd64-standard.iso"
        arm="https://cdimage.debian.org/debian-cd/current/arm64/iso-dvd/debian-$version-arm64-DVD-1.iso"
      fi ;;
    "fedora" | "fedoralinux" | "fedora-linux" )
      name="Fedora Linux"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://getfedora.org/releases.json") || exit 65
        version=$(echo "$body" | jq -r 'map(.version) | unique | .[]' | sed 's/ /_/g' | sed '/_Beta/d' | sort -r | head -n 1)
        url=$(echo "$body" | jq -r "map(select(.arch==\"x86_64\" and .version==\"${version}\" and .variant==\"Workstation\" and .subvariant==\"Workstation\" )) | .[].link" | grep -m 1 .iso)
        arm=$(echo "$body" | jq -r "map(select(.arch==\"aarch64\" and .version==\"${version}\" and .variant==\"Workstation\" and .subvariant==\"Workstation\" )) | .[].link" | grep -m 1 .iso)
      fi ;;
    "gentoo" | "gentoolinux" | "gentoo-linux" )
      name="Gentoo Linux"
      if [[ "$ret" == "url" ]]; then
        if [[ "${PLATFORM,,}" != "arm64" ]]; then
          body=$(pipe "https://distfiles.gentoo.org/releases/amd64/autobuilds/latest-iso.txt") || exit 65
          version=$(echo "$body" | grep livegui | cut -d' ' -f1)
          url="https://distfiles.gentoo.org/releases/amd64/autobuilds/$version"
        else
          body=$(pipe "https://distfiles.gentoo.org/releases/arm64/autobuilds/latest-qcow2.txt")  || exit 65
          version=$(echo "$body" | grep cloudinit | cut -d' ' -f1)
          arm="https://distfiles.gentoo.org/releases/arm64/autobuilds/$version"
        fi
      fi ;;
    "kali" | "kalilinux" | "kali-linux" )
      name="Kali Linux"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://cdimage.kali.org/current/?C=M;O=D") || exit 65
        version=$(echo "$body" | grep -o ">kali-linux-.*-live-amd64.iso" | head -n 1 | cut -c 2-)
        url="https://cdimage.kali.org/current/$version"
        version=$(echo "$body" | grep -o ">kali-linux-.*-live-arm64.iso" | head -n 1 | cut -c 2-)
        arm="https://cdimage.kali.org/current/$version"
      fi ;;
    "kubuntu" )
      name="Kubuntu"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://api.launchpad.net/devel/ubuntu/series") || exit 65
        version=$(echo "$body" | jq -r '.entries | .[] | select(.status=="Current Stable Release").version')
        url="https://cdimage.ubuntu.com/kubuntu/releases/${version}/release/kubuntu-${version}-desktop-amd64.iso"
      fi ;;
    "lmde" )
      version="6"
      name="Linux Mint Debian Edition"
      if [[ "$ret" == "url" ]]; then
        url="https://pub.linuxmint.io/debian/lmde-${version}-cinnamon-64bit.iso"
      fi ;;
    "macos" | "osx" )
      name="macOS"
      error "To install $name use: https://github.com/dockur/macos" && return 1 ;;
    "mint" | "linuxmint" | "linux-mint" )
      version="22.2"
      name="Linux Mint"
      if [[ "$ret" == "url" ]]; then
        url="https://pub.linuxmint.io/stable/${version}/linuxmint-${version}-cinnamon-64bit.iso"
      fi ;;
    "manjaro" )
      name="Manjaro"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://gitlab.manjaro.org/web/iso-info/-/raw/master/file-info.json") || exit 65
        url=$(echo "$body" | jq -r .official.plasma.image)
      fi ;;
    "mx" | "mxlinux" | "mx-linux" )
      name="MX Linux"
      if [[ "$ret" == "url" ]]; then
        version=$(curl --disable -Ils "https://sourceforge.net/projects/mx-linux/files/latest/download" | grep -i 'location:' | cut -d? -f1 | cut -d_ -f1 | cut -d- -f3) || exit 65
        url="https://mirror.umd.edu/mxlinux-iso/MX/Final/Xfce/MX-${version}_x64.iso"
      fi ;;
    "nixos" )
      name="NixOS"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://nix-channels.s3.amazonaws.com/?delimiter=/") || exit 65
        version=$(echo "$body" | grep -o -E 'nixos-[[:digit:]]+\.[[:digit:]]+' | cut -d- -f2 | sort -nru | head -n 1)
        url="https://channels.nixos.org/nixos-$version/latest-nixos-gnome-x86_64-linux.iso"
        arm="https://channels.nixos.org/nixos-$version/latest-nixos-gnome-aarch64-linux.iso"
      fi ;;
    "opensuse" | "open-suse" | "suse" )
      name="OpenSUSE"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://download.opensuse.org/distribution/leap/") || exit 65
        version=$(echo "$body" | grep 'class="name"' | cut -d '/' -f2 | grep -v 42 | sort -r | head -n 1)
        url="https://download.opensuse.org/distribution/leap/$version/installer/iso/agama-installer.x86_64-Leap_${version}.iso"
        arm="https://download.opensuse.org/distribution/leap/$version/installer/iso/agama-installer.aarch64-Leap_${version}.iso"
      fi ;;
    "rocky" | "rockylinux" | "rocky-linux" )
      version="9"
      name="Rocky Linux"
      if [[ "$ret" == "url" ]]; then
        url="https://dl.rockylinux.org/pub/rocky/${version}/live/x86_64/Rocky-${version}-Workstation-x86_64-latest.iso"
        arm="https://dl.rockylinux.org/pub/rocky/${version}/live/aarch64/Rocky-${version}-Workstation-aarch64-latest.iso"
      fi ;;
    "slack" | "slackware" )
      name="Slackware"
      if [[ "$ret" == "url" ]]; then
        url="https://slackware.nl/slackware-live/slackware64-current-live/slackware64-live-current.iso"
      fi ;;
    "tails" )
      name="Tails"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://tails.net/install/v2/Tails/amd64/stable/latest.json") || exit 65
        url=$(echo "$body" | jq -r '.installations[0]."installation-paths"[]|select(.type=="iso")|."target-files"[0].url')
      fi ;;
    "ubuntu" | "ubuntu-desktop" )
      name="Ubuntu Desktop"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://api.launchpad.net/devel/ubuntu/series") || exit 65
        version=$(echo "$body" | jq -r '.entries | .[] | select(.status=="Current Stable Release").version')
        url="https://releases.ubuntu.com/${version}/ubuntu-${version}-desktop-amd64.iso"
        arm="https://cdimage.ubuntu.com/releases/${version}/release/ubuntu-${version}-desktop-arm64.iso"
      fi ;;
    "ubuntus" | "ubuntu-server")
      name="Ubuntu Server"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://api.launchpad.net/devel/ubuntu/series") || exit 65
        version=$(echo "$body" | jq -r '.entries | .[] | select(.status=="Current Stable Release").version')
        url="https://releases.ubuntu.com/${version}/ubuntu-${version}-live-server-amd64.iso"
        arm="https://cdimage.ubuntu.com/releases/${version}/release/ubuntu-${version}-live-server-arm64.iso"
      fi ;;
    "windows" )
      name="Windows"
      error "To install $name use: https://github.com/dockur/windows" && return 1 ;;
    "xubuntu" )
      name="Xubuntu"
      if [[ "$ret" == "url" ]]; then
        body=$(pipe "https://api.launchpad.net/devel/ubuntu/series") || exit 65
        version=$(echo "$body" | jq -r '.entries | .[] | select(.status=="Current Stable Release").version')
        url="https://cdimages.ubuntu.com/xubuntu/releases/${version}/release/xubuntu-${version}-desktop-amd64.iso"
      fi ;;
  esac

  case "${ret,,}" in
    "name" )
      echo "$name"
      ;;
    "url" )

      if [[ "${PLATFORM,,}" != "arm64" ]]; then
        if [ -n "$name" ] && [ -z "$url" ]; then
          error "No image for $name available!"
          return 1
        fi
      else
        if [ -n "$name" ] && [ -z "$arm" ]; then
          error "No image for $name is available for ARM64 yet! "
          return 1
        fi
      fi

      if [[ "${PLATFORM,,}" != "arm64" ]]; then
        echo "$url"
      else
        echo "$arm"
      fi ;;
  esac

  return 0
}

return 0