#!/bin/sh -e

LOG_FILE="/var/log/unattended-upgrades-check.log"

# Colors
RED='\033[0;31m'
CYAN='\033[0;36m'
YELLOW='\033[0;33m'
LIGHT_GREEN='\033[0;92m'
BOLD='\033[1m'
NC='\033[0m' # No Color

echo_error() {
  printf "${BOLD}${RED}ERROR: ${NC}${RED}%s${NC}\n" "$1" >&2
}

echo_info() {
  printf "${BOLD}${CYAN}INFO: ${NC}${CYAN}%s${NC}\n" "$1"
}

echo_warning() {
  printf "${BOLD}${YELLOW}WARNING: ${NC}${YELLOW}%s${NC}\n" "$1"
}

echo_note() {
  printf "${BOLD}${LIGHT_GREEN}NOTE: ${NC}${LIGHT_GREEN}%s${NC}\n" "$1"
}

# ─< Check if the given command exists silently >─────────────────────────────────────────
command_exists() {
  command -v "$@" >/dev/null 2>&1
}

run_checks() {
  echo_info "Checking if unattended-upgrades is active..."

  if command_exists unattended-upgrades; then
    # if dpkg-query -W -f='${Status}' unattended-upgrades 2>/dev/null | grep -q "install ok installed"; then
    echo_note "unattended-upgrades is already installed."
  else
    echo_warning "unattended-upgrades is not installed. Attempting to install..."
    if command_exists apt-get; then
      if apt-get update && apt-get install --assume-yes unattended-upgrades; then
        echo_note "unattended-upgrades successfully installed."
      else
        echo_error "Failed to install unattended-upgrades. Exiting."
        exit 1
      fi
    else
      echo_error "apt is not available on this system. Exiting."
      exit 1
    fi
  fi

  # Enable unattended-upgrades
  UNATTENDED_UPGRADES_FILE="/etc/apt/apt.conf.d/50unattended-upgrades"
  if [ -f "$UNATTENDED_UPGRADES_FILE" ]; then
    echo_info "Configuring unattended upgrades in $UNATTENDED_UPGRADES_FILE"
    # Add or modify configurations as needed
    # Example: $_sudo sed -i 's/old_value/new_value/' "$UNATTENDED_UPGRADES_FILE"
  else
    echo_error "Unattended upgrades file not found!"
  fi
}

run_setup() {
  if command_exists unattended-upgrades; then
    systemctl enable --now unattended-upgrades || echo_error "Something went wrong! Could not setup the service/autostart"
  fi
}

# Main script
main() {
  echo_info "Starting unattended-upgrades check script..."

  if [ "$(id -u)" -ne 0 ]; then
    echo_error "This script must be run as root. Please run with sudo."
    exit 1
  fi

  run_checks

  run_setup && echo_note "Script completed successfully!"
}

main