#!/usr/bin/env bash

set -e

##
# Global variables
##

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'
BOLD=$(tput bold)
NORM=$(tput sgr0)
UBUNTU_VER='Ubuntu 22.04'
BOOT_SPACE='51200'
SYSTEM_SPACE='358400'
OLD_NIC_NAMING=true
ZEN_VER='8.0'
ZEN_REPO_KEY="https://keys.zentyal.org/zentyal-${ZEN_VER}-packages-org.asc"
ZEN_REPO_URL="deb [signed-by=/etc/apt/trusted.gpg.d/zentyal-${ZEN_VER}-packages-org.asc] https://packages.zentyal.org/zentyal ${ZEN_VER} main extra"


##
# Functions
##

function check_ubuntu
{
  echo -e "\n${GREEN} - Checking Ubuntu version...${NC}"

  if ! lsb_release -d | egrep -q "${UBUNTU_VER}.?[0-9]? LTS$"
    then
      echo -e "${RED}  The version that you are using isn't valid. Zentyal requires ${UBUNTU_VER}.x LTS ${NC}"
      exit 1
  fi

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function check_broken_packages
{
  echo -e "${GREEN} - Checking for broken packages...${NC}"

  if [[ $(dpkg -l | egrep -v '^ii|rc' | awk '{if(NR>5)print}' | wc -l) -gt 0 ]]
    then
      echo -e "${RED}  You have broken packages, trying to repair.${NC}"

      for i in {1..10}; do DEBIAN_FRONTEND=noninteractive dpkg --configure -a; done

      if [[ $(dpkg -l | egrep -v '^ii|rc' | awk '{if(NR>5)print}' | wc -l) -gt 0 ]]
        then
          echo -e "${RED}  Couln't fix the broken packages.${NC}"
          exit 1
      fi

      echo -e "${GREEN} Broken packages fixed. ${NORM}";echo
  fi

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function check_available_packages
{
  echo -e "${GREEN} - Checking if the system is up-to-date...${NC}"

  apt -qq update
  if [[ $(apt list --upgradable 2> /dev/null | wc -l) -gt 1 ]]
    then
      echo -e "${RED}  Your server isn't up-to-date. You must run: sudo apt dist-upgrade${NC}"
      exit 1
  fi

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function check_disk_space
{
  echo -e "${GREEN} - Checking for available disk space...${NC}"

  if [ $(df /boot | tail -1 | awk '{print $4}') -lt ${BOOT_SPACE} ];
    then
      echo -e "${RED}  Upgrade cannot be performed due to low disk space (less than 50MB available on /boot)${NC}"
      exit 1
  fi

  for partition in / /var
    do
      if [ $(df ${partition} | tail -1 | awk '{print $4}') -lt ${SYSTEM_SPACE} ];
        then
          echo -e "${RED}  Upgrade cannot be performed due to low partition space (less than 350MB available on '${partition}')${NC}"
          exit 1
      fi
    done

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function check_connection
{
  local CHECK_DOMAIN='google.es'

  echo -e "${GREEN} - Checking the Internet connection...${NC}"

  if [[ ! -f '/usr/bin/ping' ]]
    then
      echo -e "${RED}  Ping command is not found, install it by running: "sudo apt install iputils-ping" .${NC}"
  fi

  if ! ping -4 -W 15 -q -c 5 ${CHECK_DOMAIN} > /dev/null
    then
      echo -e "${RED}  There are issues with the Internet resolution.${NC}"
      exit 1
  fi

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function check_webadmin_port
{
  echo -e "${GREEN} - Checking Webadmin 8443/tcp port...${NC}"

  if ss -tunpl | grep -q '8443'
    then
      echo -e "${RED}  The port 8443/tcp is already in use.${NC}"
      exit 1
  fi

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function check_nic_names
{
  echo -e "${GREEN} - Checking network interface naming...${NC}\n"

  interface_names=$(ip -o link show | awk -F': ' '{print $2}' | egrep 'eth' || true)

  if [ -z "${interface_names}" ] && [ "${OLD_NIC_NAMING}" == true ]; then
    echo -e "${RED}  The network interface naming is not using 'eth'. You can ommit this by changing to false the variable 'OLD_NIC_NAMING' or running the following commands:${NC}"
cat <<EOF
    sed -i 's/#GRUB_HIDDEN_TIMEOUT=0/GRUB_HIDDEN_TIMEOUT=0/' /etc/default/grub
    sed -i 's/\(GRUB_CMDLINE_LINUX_DEFAULT=".*\)"/\1 net.ifnames=0 biosdevname=0"/' /etc/default/grub
    update-grub
    reboot
EOF
    exit 1
  fi

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function check_requirements
{
  check_ubuntu
  check_broken_packages
  check_disk_space
  check_available_packages
  check_connection
  check_webadmin_port
  check_nic_names
}


function add_repositories
{
  echo -e "${GREEN} - Adding required repositories...${NC}\n"

  for url in packages.zentyal.com packages.zentyal.org archive.zentyal.org archive.zentyal.com
    do
      sed -i "/${url}/d" /etc/apt/sources.list

      for repo in $(find /etc/apt/sources.list.d/ -type f)
        do
          sed -i "/${url}/d" ${repo}
        done
    done

  # Set up Zentyal Development repository
  wget -q ${ZEN_REPO_KEY} -P /etc/apt/trusted.gpg.d/
  echo ${ZEN_REPO_URL} > /etc/apt/sources.list.d/zentyal.list

  ## Add Docker repository for zentyal-docker module
  curl -fsSL https://download.docker.com/linux/ubuntu/gpg | gpg --dearmor -o /etc/apt/trusted.gpg.d/docker.gpg
  chmod 0644 /etc/apt/trusted.gpg.d/docker.gpg
  echo "deb [arch=amd64 signed-by=/etc/apt/trusted.gpg.d/docker.gpg] https://download.docker.com/linux/ubuntu jammy stable" > /etc/apt/sources.list.d/docker.list

  ## Add Firefox repository for zenbuntu-desktop module
  wget -q https://packages.mozilla.org/apt/repo-signing-key.gpg -O- > /etc/apt/trusted.gpg.d/packages.mozilla.org.asc
  chmod 0644 /etc/apt/trusted.gpg.d/packages.mozilla.org.asc
  echo "deb [signed-by=/etc/apt/trusted.gpg.d/packages.mozilla.org.asc] https://packages.mozilla.org/apt mozilla main" > /etc/apt/sources.list.d/mozilla.list
cat <<EOF > /etc/apt/preferences.d/mozilla
Package: *
Pin: origin packages.mozilla.org
Pin-Priority: 1000
EOF

  apt -qq update

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function zentyal_gui
{
  echo -e "${GREEN} - Installing the graphical environment...${NC}\n"

  echo 'lxdm shared/default-x-display-manager select lxdm' | debconf-set-selections
  DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ${ZEN_GUI}

  if [[ ! -f /etc/X11/default-display-manager ]]
    then
      ## For Ubuntu Server
      echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
      continue
  fi

  CUR_GUI=$(cat /etc/X11/default-display-manager | xargs basename)

  case ${CUR_GUI} in
    gdm3) ## Ubuntu Desktop (Gnome)
      echo 'gdm3 shared/default-x-display-manager select lxdm' | debconf-set-selections
      DEBIAN_FRONTEND=noninteractive dpkg-reconfigure --force gdm3
      systemctl disable gdm3 lxdm
      which lxdm > /etc/X11/default-display-manager
    ;;
    sddm) ## Lubuntu and Kubuntu
      echo 'sddm shared/default-x-display-manager select lxdm' | debconf-set-selections
      DEBIAN_FRONTEND=noninteractive dpkg-reconfigure --force sddm
      systemctl disable sddm lxdm
      which lxdm > /etc/X11/default-display-manager
    ;;
    lightdm) ## Xubuntu
      echo 'lightdm shared/default-x-display-manager select lxdm' | debconf-set-selections
      DEBIAN_FRONTEND=noninteractive dpkg-reconfigure --force lightdm
      systemctl disable lightdm lxdm
      which lxdm > /etc/X11/default-display-manager
    ;;
  esac

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo

  echo -e "${GREEN} - Configuring the graphical environment...${NC}\n"

  /usr/share/zenbuntu-desktop/x11-setup >> /var/tmp/zentyal-installer.log 2>&1
  systemctl enable --now zentyal.lxdm

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo
}


function zentyal_installation
{
  echo -e "${GREEN} - Installing Zentyal...${NC}\n"

  DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends zentyal zenbuntu-core

  touch /var/lib/zentyal/.commercial-edition
  touch /var/lib/zentyal/.license

  echo -e "${GREEN}${BOLD}...OK${NC}${NORM}";echo

  echo -e "\n${GREEN}${BOLD}Installation complete, you can access the Zentyal Web Interface at:

    * https://<zentyal-ip-address>:8443/

  ${NC}${NORM}"

  if [[ -n ${ZEN_GUI} ]]
    then
      zentyal_gui
      sleep 10
      systemctl restart zentyal.lxdm
  fi
}


##
# Checks
##

if [[ ${EUID} -ne 0 ]]
  then
    echo -e "${RED}  The script must be run with 'sudo' rights.${NC}"
    exit 1
fi

echo -n "Do you want to install the Zentyal Graphical environment? (n|y) "
read ZEN_GUI

if [[ ${ZEN_GUI^} == 'Y' ]]
  then
    ZEN_GUI='zenbuntu-desktop'
elif [[ ${ZEN_GUI^} == 'N' ]]
  then
    ZEN_GUI=''
  else
    echo "Wrong answer. Please, type 'y' or 'n'."
    exit 1
fi


##
# Running the functions
##

check_requirements
add_repositories
zentyal_installation


###
# Final commands
###

# Disable cloud-init
touch /etc/cloud/cloud-init.disabled