#!/usr/bin/env bash

# Copyright (c) 2021-2024 tteck
# Author: MickLesk (CanbiZ)
# License: MIT
# https://github.com/tteck/Proxmox/raw/main/LICENSE

source /dev/stdin <<< "$FUNCTIONS_FILE_PATH"
color
verb_ip6
catch_errors
setting_up_container
network_check
update_os

msg_info "Installing Dependencies"
$STD apt-get install -y \
  gnupg2 \
  curl \
  sudo \
  mc \
  lsb-release 
msg_ok "Installed Dependencies"

msg_info "Setting up Adoptium Repository"
mkdir -p /etc/apt/keyrings
$STD wget -O - https://packages.adoptium.net/artifactory/api/gpg/key/public | tee /etc/apt/keyrings/adoptium.asc
$STD echo "deb [signed-by=/etc/apt/keyrings/adoptium.asc] https://packages.adoptium.net/artifactory/deb $(awk -F= '/^VERSION_CODENAME/{print$2}' /etc/os-release) main" | tee /etc/apt/sources.list.d/adoptium.list
$STD apt-get update
msg_ok "Set up Adoptium Repository"

read -r -p "Which Tomcat version would you like to install? (9, 10.1, 11): " version
case $version in
  9)
    TOMCAT_VERSION="9"
    echo "Which LTS Java version would you like to use? (8, 11, 17, 21): "
    read -r jdk_version
    case $jdk_version in
      8)
        msg_info "Installing Temurin JDK 8 (LTS) for Tomcat $TOMCAT_VERSION"
        export DEBIAN_FRONTEND=noninteractive apt-get install -y temurin-8-jdk
        msg_ok "Setup Temurin JDK 8 (LTS)"
        ;;
      11)
        msg_info "Installing Temurin JDK 11 (LTS) for Tomcat $TOMCAT_VERSION"
        $STD apt-get install -y temurin-11-jdk
        msg_ok "Setup Temurin JDK 11 (LTS)"
        ;;
      17)
        msg_info "Installing Temurin JDK 17 (LTS) for Tomcat $TOMCAT_VERSION"
        apt-get install -qqy temurin-17-jdk
        msg_ok "Setup Temurin JDK 17 (LTS)"
        ;;
      21)
        msg_info "Installing Temurin JDK 21 (LTS) for Tomcat $TOMCAT_VERSION"
        export DEBIAN_FRONTEND=noninteractive
        $STD apt-get install -y temurin-21-jdk
        msg_ok "Setup Temurin JDK 21 (LTS)"
        ;;
      *)
        msg_error "Invalid JDK version selected. Please enter 8, 11, 17 or 21."
        exit 1
        ;;
    esac
    ;;
  10|10.1)
    TOMCAT_VERSION="10"
    echo "Which LTS Java version would you like to use? (11, 17): "
    read -r jdk_version
    case $jdk_version in
      11)
        msg_info "Installing Temurin JDK 11 (LTS) for Tomcat $TOMCAT_VERSION"
        export DEBIAN_FRONTEND=noninteractive
        $STD apt-get install -y temurin-11-jdk
        msg_ok "Setup Temurin JDK 11"
        ;;
      17)
        msg_info "Installing Temurin JDK 17 (LTS) for Tomcat $TOMCAT_VERSION"
        export DEBIAN_FRONTEND=noninteractive
        $STD apt-get install -y temurin-17-jdk
        msg_ok "Setup Temurin JDK 17"
        ;;
      21)
        msg_info "Installing Temurin JDK 21 (LTS) for Tomcat $TOMCAT_VERSION"
        export DEBIAN_FRONTEND=noninteractive
        $STD apt-get install -y temurin-21-jdk
        msg_ok "Setup Temurin JDK 21 (LTS)"
        ;;
      *)
        msg_error "Invalid JDK version selected. Please enter 11 or 17."
        exit 1
        ;;
    esac
    ;;
  11)
    TOMCAT_VERSION="11"
    echo "Which LTS Java version would you like to use? (17, 21): "
    read -r jdk_version
    case $jdk_version in
      17)
        msg_info "Installing Temurin JDK 17 (LTS) for Tomcat $TOMCAT_VERSION"
        export DEBIAN_FRONTEND=noninteractive
        apt-get install -qqy temurin-17-jdk
        msg_ok "Setup Temurin JDK 17"
        ;;
      21)
        msg_info "Installing Temurin JDK 21 (LTS) for Tomcat $TOMCAT_VERSION"
        export DEBIAN_FRONTEND=noninteractive
        $STD apt-get install -y temurin-21-jdk
        msg_ok "Setup Temurin JDK 21 (LTS)"
        ;;
      *)
        msg_error "Invalid JDK version selected. Please enter 17 or 21."
        exit 1
        ;;
    esac
    ;;
  *)
    msg_error "Invalid Tomcat version selected. Please enter 9, 10.1 or 11."
    exit 1
    ;;
esac

msg_info "Installing Tomcat $TOMCAT_VERSION"
LATEST_VERSION=$(curl -s "https://dlcdn.apache.org/tomcat/tomcat-$TOMCAT_VERSION/" | grep -oP 'v[0-9]+\.[0-9]+\.[0-9]+(-M[0-9]+)?/' | sort -V | tail -n 1 | sed 's/\/$//; s/v//')
TOMCAT_URL="https://dlcdn.apache.org/tomcat/tomcat-$TOMCAT_VERSION/v$LATEST_VERSION/bin/apache-tomcat-$LATEST_VERSION.tar.gz"
wget -qO /tmp/tomcat.tar.gz "$TOMCAT_URL"
mkdir -p /opt/tomcat-$TOMCAT_VERSION
tar --strip-components=1 -xzf /tmp/tomcat.tar.gz -C /opt/tomcat-$TOMCAT_VERSION
chown -R root:root /opt/tomcat-$TOMCAT_VERSION

cat <<EOF > /etc/systemd/system/tomcat.service
[Unit]
Description=Apache Tomcat Web Application Container
After=network.target

[Service]
Type=forking
User=$(whoami)
Group=$(whoami)
Environment=JAVA_HOME=/usr/lib/jvm/temurin-${jdk_version}-jdk-amd64
Environment=CATALINA_HOME=/opt/tomcat-$TOMCAT_VERSION
Environment=CATALINA_BASE=/opt/tomcat-$TOMCAT_VERSION
Environment=CATALINA_PID=/opt/tomcat-$TOMCAT_VERSION/temp/tomcat.pid
ExecStart=/opt/tomcat-$TOMCAT_VERSION/bin/catalina.sh start
ExecStop=/opt/tomcat-$TOMCAT_VERSION/bin/catalina.sh stop
PIDFile=/opt/tomcat-$TOMCAT_VERSION/temp/tomcat.pid
SuccessExitStatus=143
Restart=on-abnormal

[Install]
WantedBy=multi-user.target
EOF
systemctl enable -q --now tomcat
msg_ok "Tomcat $LATEST_VERSION installed and started"

motd_ssh
customize

msg_info "Cleaning up"
rm -f /tmp/tomcat.tar.gz
$STD apt-get -y autoremove
$STD apt-get -y autoclean
msg_ok "Cleaned"