#!/usr/bin/env bash
source <(curl -fsSL https://raw.githubusercontent.com/haris2887/Omada-LXC/main/misc/build.func)
# Copyright (c) 2021-2025 tteck
# Author: tteck (tteckster)
# License: MIT | https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE
# Source: https://www.tp-link.com/us/support/download/omada-software-controller/
# Modified to support non-AVX systems by forcing MongoDB 4.4

APP="Omada"
var_tags="${var_tags:-tp-link;controller}"
var_cpu="${var_cpu:-2}"
var_ram="${var_ram:-3072}"
var_disk="${var_disk:-8}"
var_os="${var_os:-debian}"
var_version="${var_version:-12}"
var_unprivileged="${var_unprivileged:-1}"

header_info "$APP"
variables
color
catch_errors

function update_script() {
  header_info
  check_container_storage
  check_container_resources
  if [[ ! -d /opt/tplink ]]; then
    msg_error "No ${APP} Installation Found!"
    exit
  fi

  msg_info "Updating MongoDB"
  # Force MongoDB 4.4 for systems without AVX support
  MONGODB_VERSION="4.4"
  
  if ! lscpu | grep -q 'avx'; then
    msg_warn "No AVX detected: Using MongoDB 4.4 (compatible with non-AVX systems)"
  else
    msg_info "AVX detected: Using MongoDB 4.4 for compatibility"
  fi

  # Remove any existing MongoDB repositories and packages
  $STD apt-get remove --purge -y mongodb-org mongodb-org-* mongodb 2>/dev/null || true
  $STD rm /etc/apt/sources.list.d/mongodb-org-*.list 2>/dev/null || true

  # Add MongoDB 4.4 repository using Ubuntu 20.04 (focal) repo for Debian 12 compatibility
  curl -fsSL "https://www.mongodb.org/static/pgp/server-${MONGODB_VERSION}.asc" | gpg --dearmor >/usr/share/keyrings/mongodb-server-${MONGODB_VERSION}.gpg
  
  # Use focal (Ubuntu 20.04) repo as it's more compatible with Debian 12 for MongoDB 4.4
  echo "deb [signed-by=/usr/share/keyrings/mongodb-server-${MONGODB_VERSION}.gpg] http://repo.mongodb.org/apt/ubuntu focal/mongodb-org/${MONGODB_VERSION} multiverse" >/etc/apt/sources.list.d/mongodb-org-${MONGODB_VERSION}.list
  
  $STD apt-get update
  
  # Install specific MongoDB 4.4 version with version pinning
  $STD apt-get install -y \
    mongodb-org=4.4.29 \
    mongodb-org-server=4.4.29 \
    mongodb-org-shell=4.4.29 \
    mongodb-org-mongos=4.4.29 \
    mongodb-org-tools=4.4.29

  # Hold packages to prevent unwanted upgrades
  $STD apt-mark hold mongodb-org mongodb-org-server mongodb-org-shell mongodb-org-mongos mongodb-org-tools

  # Enable and start MongoDB service
  $STD systemctl enable mongod
  $STD systemctl start mongod
  
  msg_ok "Updated MongoDB to $MONGODB_VERSION"

  msg_info "Checking if right Java is installed"
  # Check for OpenJDK 21 or install it
  if java -version 2>&1 | grep -q "openjdk version \"21\|temurin-21"; then
    msg_ok "OpenJDK 21 already installed"
  else
    # Install Eclipse Temurin OpenJDK 21
    $STD apt-get install -y wget apt-transport-https gnupg
    wget -O - https://packages.adoptium.net/artifactory/api/gpg/key/public | gpg --dearmor | tee /usr/share/keyrings/adoptium.gpg >/dev/null
    echo "deb [signed-by=/usr/share/keyrings/adoptium.gpg] 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
    $STD apt-get install -y temurin-21-jre
    msg_ok "Installed OpenJDK 21"
  fi

  msg_info "Updating Omada Controller"
  OMADA_URL=$(curl -fsSL "https://support.omadanetworks.com/en/download/software/omada-controller/" |
    grep -o 'https://static\.tp-link\.com/upload/software/[^"]*linux_x64[^"]*\.deb' |
    head -n1)
  OMADA_PKG=$(basename "$OMADA_URL")
  if [ -z "$OMADA_PKG" ]; then
    msg_error "Could not retrieve Omada package – server may be down."
    exit 1
  fi
  curl -fsSL "$OMADA_URL" -o "$OMADA_PKG"
  export DEBIAN_FRONTEND=noninteractive
  $STD dpkg -i "$OMADA_PKG"
  rm -f "$OMADA_PKG"
  msg_ok "Updated Omada Controller"
  exit 0
}

export FUNCTIONS_FILE_PATH="$(curl -fsSL https://raw.githubusercontent.com/haris2887/Omada-LXC/main/misc/install.func)"
export var_install="omada-install"

# Custom build_container function that uses our installation script
build_container() {
  if [ "$CT_TYPE" == "1" ]; then
    FEATURES="keyctl=1,nesting=1"
  else
    FEATURES="nesting=1"
  fi

  if [ "$ENABLE_FUSE" == "yes" ]; then
    FEATURES="$FEATURES,fuse=1"
  fi

  if [[ $DIAGNOSTICS == "yes" ]]; then
    post_to_api
  fi

  # Build network string - this was missing!
  NET_STRING="-net0 name=eth0,bridge=$BRG$MAC,ip=$NET$GATE$VLAN$MTU"
  case "$IPV6_METHOD" in
  auto) NET_STRING="$NET_STRING,ip6=auto" ;;
  dhcp) NET_STRING="$NET_STRING,ip6=dhcp" ;;
  static)
    NET_STRING="$NET_STRING,ip6=$IPV6_ADDR"
    [ -n "$IPV6_GATE" ] && NET_STRING="$NET_STRING,gw6=$IPV6_GATE"
    ;;
  none) ;;
  esac

  TEMP_DIR=$(mktemp -d)
  pushd "$TEMP_DIR" >/dev/null
  export FUNCTIONS_FILE_PATH="$(curl -fsSL https://raw.githubusercontent.com/haris2887/Omada-LXC/main/misc/install.func)"
  export CACHER="$APT_CACHER"
  export CACHER_IP="$APT_CACHER_IP"
  export tz="$timezone"
  export APPLICATION="$APP"
  export app="$NSAPP"
  export PASSWORD="$PW"
  export VERBOSE="$VERBOSE"
  export SSH_ROOT="${SSH}"
  export SSH_AUTHORIZED_KEY
  export CTID="$CT_ID"
  export CTTYPE="$CT_TYPE"
  export ENABLE_FUSE="$ENABLE_FUSE"
  export ENABLE_TUN="$ENABLE_TUN"
  export PCT_OSTYPE="$var_os"
  export PCT_OSVERSION="$var_version"
  export PCT_DISK_SIZE="$DISK_SIZE"
  export PCT_OPTIONS="
    -features $FEATURES
    -hostname $HN
    -tags $TAGS
    $SD
    $NS
    $NET_STRING
    -onboot 1
    -cores $CORE_COUNT
    -memory $RAM_SIZE
    -unprivileged $CT_TYPE
    $PW
  "
  
  # Create the container using the original script
  bash -c "$(curl -fsSL https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/misc/create_lxc.sh)" $?

  # Wait for container to be fully started and ready
  msg_info "Starting LXC Container"
  pct start "$CTID"
  
  # Wait for container to reach running state
  for i in {1..30}; do
    if pct status "$CTID" | grep -q "status: running"; then
      msg_ok "LXC Container is running"
      break
    fi
    if [ $i -eq 30 ]; then
      msg_error "LXC Container failed to start properly"
      exit 1
    fi
    sleep 2
  done

  # Wait for container init process to be ready
  msg_info "Waiting for container init process"
  for i in {1..20}; do
    if pct exec "$CTID" -- test -f /sbin/init >/dev/null 2>&1; then
      msg_ok "Container init process ready"
      break
    fi
    if [ $i -eq 20 ]; then
      msg_error "Container init process not ready"
      exit 1
    fi
    sleep 3
  done

  # Additional wait for network and services
  msg_info "Waiting for container network"
  for i in {1..15}; do
    if pct exec "$CTID" -- ping -c1 -W2 8.8.8.8 >/dev/null 2>&1; then
      msg_ok "Container network ready"
      break
    fi
    if [ $i -eq 15 ]; then
      msg_warn "Container network may not be ready, continuing anyway"
    fi
    sleep 5
  done

  # Now install using our custom script
  msg_info "Installing Omada Controller (No-AVX Compatible)"
  pct exec "$CTID" -- bash -c "$(curl -fsSL https://raw.githubusercontent.com/haris2887/Omada-LXC/main/install/omada-install.sh)"
  msg_ok "Omada Installation Completed"
}

start
build_container
description

msg_ok "Completed Successfully!\n"
echo -e "${CREATING}${GN}${APP} setup has been successfully initialized!${CL}"
echo -e "${INFO}${YW} Access it using the following URL:${CL}"
echo -e "${TAB}${GATEWAY}${BGN}https://${IP}:8043${CL}"