#!/usr/bin/env bash
source <(curl -fsSL https://git.inbox.sk/proxmox/Ubuntu24_LXC/raw/branch/main/misc/build.func)
# Vlastný skript pre Ubuntu 24.04 LXC kontajner
# Zdroj: https://git.inbox.sk/proxmox/Ubuntu24_LXC

APP="Ubuntu"
var_tags="${var_tags:-os}"
var_cpu="${var_cpu:-4}"
var_ram="${var_ram:-4096}"
var_disk="${var_disk:-32}"
var_os="${var_os:-ubuntu}"
var_version="${var_version:-24.04}"
var_unprivileged="${var_unprivileged:-1}"

# URL pre stiahnutie users.json
USERS_JSON_URL="https://git.inbox.sk/proxmox/Ubuntu24_LXC/raw/branch/main/config/users.json"

header_info "$APP"
variables
color
catch_errors

# =============================================================================
# update_script() - aktualizácia existujúceho kontajnera
# =============================================================================
function update_script() {
  header_info
  check_container_storage
  check_container_resources
  if [[ ! -d /var ]]; then
    msg_error "No ${APP} Installation Found!"
    exit
  fi
  msg_info "Updating ${APP} LXC"
  $STD apt-get update
  $STD apt-get -y upgrade
  msg_ok "Updated ${APP} LXC"
  msg_ok "Updated successfully!"
  exit
}

# =============================================================================
# simple_install() - zjednodušená inštalácia s výberom užívateľov
# =============================================================================
simple_install() {
  pve_check
  shell_check
  root_check
  arch_check

  NEXTID=$(pvesh get /cluster/nextid)

  # Timezone
  if command -v timedatectl >/dev/null 2>&1; then
    timezone=$(timedatectl show --value --property=Timezone 2>/dev/null || echo "UTC")
  elif [ -f /etc/timezone ]; then
    timezone=$(cat /etc/timezone)
  else
    timezone="UTC"
  fi

  header_info

  # --- KROK 1: Hostname ---
  HN=$(whiptail --backtitle "Ubuntu LXC Setup" \
    --title "HOSTNAME" \
    --inputbox "\nZadaj hostname pre kontajner:" 10 58 "ubuntu" \
    3>&1 1>&2 2>&3) || exit_script
  HN=$(echo "${HN,,}" | tr -d ' ')
  [[ -z "$HN" ]] && HN="ubuntu"

  # --- KROK 2: IP adresa ---
  local ip_input
  ip_input=$(whiptail --backtitle "Ubuntu LXC Setup" \
    --title "IP ADRESA" \
    --inputbox "\nZadaj statickú IP adresu s maskou\n(napr. 192.168.1.100/24)\n\nAlebo nechaj prázdne pre DHCP:" 12 58 "" \
    3>&1 1>&2 2>&3) || exit_script

  if [[ -n "$ip_input" ]]; then
    NET="$ip_input"
    # Gateway
    local default_gw
    default_gw=$(echo "$ip_input" | cut -d'/' -f1 | sed 's/\.[0-9]*$/.1/')
    GATE=$(whiptail --backtitle "Ubuntu LXC Setup" \
      --title "GATEWAY" \
      --inputbox "\nZadaj gateway:" 10 58 "$default_gw" \
      3>&1 1>&2 2>&3) || exit_script
    [[ -z "$GATE" ]] && GATE="$default_gw"
  else
    NET="dhcp"
    GATE=""
  fi

  # --- KROK 3: Resources ---
  DISK_SIZE=$(whiptail --backtitle "Ubuntu LXC Setup" \
    --title "DISK" \
    --inputbox "\nVeľkosť disku v GB:" 10 58 "$var_disk" \
    3>&1 1>&2 2>&3) || exit_script
  [[ -z "$DISK_SIZE" ]] && DISK_SIZE="$var_disk"

  RAM_SIZE=$(whiptail --backtitle "Ubuntu LXC Setup" \
    --title "RAM" \
    --inputbox "\nVeľkosť RAM v MB:" 10 58 "$var_ram" \
    3>&1 1>&2 2>&3) || exit_script
  [[ -z "$RAM_SIZE" ]] && RAM_SIZE="$var_ram"

  CORE_COUNT=$(whiptail --backtitle "Ubuntu LXC Setup" \
    --title "CPU" \
    --inputbox "\nPočet CPU jadier:" 10 58 "$var_cpu" \
    3>&1 1>&2 2>&3) || exit_script
  [[ -z "$CORE_COUNT" ]] && CORE_COUNT="$var_cpu"

  # --- KROK 4: Výber užívateľov z users.json ---
  msg_info "Sťahujem zoznam užívateľov"
  local users_json
  users_json=$(curl -fsSL "$USERS_JSON_URL") || {
    msg_error "Nepodarilo sa stiahnuť users.json"
    exit 1
  }
  msg_ok "Zoznam užívateľov stiahnutý"

  # Parsovanie užívateľov bez jq — cez grep/sed
  local usernames=()
  local key_counts=()
  while IFS= read -r uname; do
    [[ -n "$uname" ]] && usernames+=("$uname")
  done < <(echo "$users_json" | grep '"username"' | sed 's/.*"username"[[:space:]]*:[[:space:]]*"\([^"]*\)".*/\1/')

  # Počet SSH kľúčov pre každého užívateľa
  local idx=0
  for uname in "${usernames[@]}"; do
    local kc
    kc=$(echo "$users_json" | sed -n "/$uname/,/\]/p" | grep 'ssh-' | wc -l)
    key_counts+=("$kc")
    idx=$((idx + 1))
  done

  local user_count=${#usernames[@]}

  if [[ "$user_count" -eq 0 ]]; then
    msg_warn "Žiadni užívatelia v users.json"
    SELECTED_USERS=""
  else
    local checklist_args=()
    for i in $(seq 0 $((user_count - 1))); do
      checklist_args+=("${usernames[$i]}" "${key_counts[$i]} SSH kľúč(ov)" "ON")
    done

    SELECTED_USERS=$(whiptail --backtitle "Ubuntu LXC Setup" \
      --title "UŽÍVATELIA" \
      --checklist "\nVyber užívateľov na vytvorenie:\n(SPACE = zaškrtni, ENTER = potvrď)" \
      $((user_count + 10)) 58 "$user_count" \
      "${checklist_args[@]}" \
      3>&1 1>&2 2>&3) || exit_script
  fi

  # Export pre install skript
  export SELECTED_USERS
  export USERS_JSON="$users_json"

  # --- Nastavenie premenných pre build_container ---
  CT_TYPE="$var_unprivileged"
  CT_ID="$NEXTID"
  BRG="${var_brg:-vmbr0}"
  MAC=""
  VLAN=""
  MTU=""
  SD=""
  NS=""
  IPV6_METHOD="auto"
  IPV6_ADDR=""
  IPV6_GATE=""
  SSH="no"
  SSH_AUTHORIZED_KEY=""
  APT_CACHER=""
  APT_CACHER_IP=""
  ENABLE_FUSE="no"
  ENABLE_TUN="no"
  ENABLE_GPU="no"
  ENABLE_NESTING="1"
  ENABLE_KEYCTL="0"
  ENABLE_MKNOD="0"
  PROTECT_CT="no"
  CT_TIMEZONE="$timezone"
  TAGS="community-script;${var_tags:-}"
  PW=""
  VERBOSE="no"
  METHOD="simple"
  DIAGNOSTICS="no"

  # Zobraz súhrn
  header_info
  echo -e "${DEFAULT}${BOLD}${BL}Ubuntu LXC - Inštalácia${CL}"
  echo -e "${TAB}${HOSTNAME}${YW} Hostname: ${GN}${HN}${CL}"
  echo -e "${TAB}${NETWORK}${YW} IP: ${GN}${NET}${CL}"
  if [[ -n "$GATE" ]]; then
    echo -e "${TAB}${GATEWAY}${YW} Gateway: ${GN}${GATE}${CL}"
  fi
  echo -e "${TAB}${DISKSIZE}${YW} Disk: ${GN}${DISK_SIZE}GB${CL}"
  echo -e "${TAB}${RAMSIZE}${YW} RAM: ${GN}${RAM_SIZE}MB${CL}"
  echo -e "${TAB}${CPUCORE}${YW} CPU: ${GN}${CORE_COUNT} jadier${CL}"
  if [[ -n "$SELECTED_USERS" ]]; then
    echo -e "${TAB}${ROOTSSH}${YW} Užívatelia: ${GN}${SELECTED_USERS}${CL}"
  fi
  echo ""

  # Potvrdenie
  if ! whiptail --backtitle "Ubuntu LXC Setup" \
    --title "POTVRDENIE" \
    --yesno "Pokračovať s inštaláciou?" 8 58; then
    exit_script
  fi
}

# Spustenie
simple_install
build_container
description

msg_ok "Completed successfully!\n"
echo -e "${CREATING}${GN}${APP} setup has been successfully initialized!${CL}"