#!/bin/sh
# install.sh – One-command installer for o11pro-unpacked
#
# Usage:
#   sh -c "$(curl -fsSL https://raw.githubusercontent.com/Ap0dexMe0/o11pro-unpacked/main/install.sh)"
#   sh -c "$(wget -O-  https://raw.githubusercontent.com/Ap0dexMe0/o11pro-unpacked/main/install.sh)"
#   sh -c "$(fetch -o - https://raw.githubusercontent.com/Ap0dexMe0/o11pro-unpacked/main/install.sh)"
#
# Environment:
#   O11PRO_DIR    Install directory (default: /root/o11pro-unpacked, root required)
#   O11PRO_PORT   Server port (default: 1337)
#   O11PRO_VERBOSE Log level 0-5 (default: 2)
#   O11PRO_USER   Admin username
#   O11PRO_PASS   Admin password
#   O11PRO_YES    Skip confirmation prompts (set to 1)
#   O11PRO_BRANCH Git branch to clone (default: main)
#
# Options:
#   --yes, -y     Non-interactive mode (auto-confirm all prompts)
#   --help, -h    Show this help message

set -eu

REPO_URL="https://github.com/Ap0dexMe0/o11pro-unpacked.git"
REPO_RAW="https://raw.githubusercontent.com/Ap0dexMe0/o11pro-unpacked"
BRANCH="${O11PRO_BRANCH:-main}"

# ─── Parse flags ─────────────────────────────────────────────────────────
FORCE=0
for arg in "$@"; do
  case "$arg" in
    --yes|-y) FORCE=1 ;;
    --help|-h)
      sed -n '2,/^$/p' "$0" | sed 's/^# \?//'
      exit 0
      ;;
  esac
done
[ "${O11PRO_YES:-0}" = "1" ] && FORCE=1

# ─── Colors ──────────────────────────────────────────────────────────────
RST='\033[0m'
BLD='\033[1m'
DIM='\033[2m'
RED='\033[0;31m'
GRN='\033[0;32m'
YLW='\033[0;33m'
BLU='\033[0;34m'
CYN='\033[0;36m'

ok()    { printf "  ${GRN}\xE2\x9C\x94${RST} ${BLD}%s${RST} %s\n" "$1" "${2-}"; }
info()  { printf "  ${CYN}\xE2\x96\xB6${RST} ${DIM}%s${RST}\n" "$1"; }
warn()  { printf "  ${YLW}\xE2\x9A\xA0${RST} ${BLD}%s${RST} %s\n" "$1" "${2-}"; }
fail()  { printf "  ${RED}\xE2\x9C\x98${RST} ${RED}%s${RST}\n" "$1"; exit 1; }
sep()   { printf "  ${DIM}----------------------------------------${RST}\n"; }
header(){ printf "\n  ${BLD}%s${RST}\n  ${DIM}%s${RST}\n\n" "$1" "${2-}"; }

# ─── Check prerequisites ─────────────────────────────────────────────────
header "o11pro Installer" "One-command setup for o11pro-unpacked"

has_cmd() { command -v "$1" >/dev/null 2>&1; }

info "Checking prerequisites..."

if ! has_cmd git; then
  fail "git is required but not installed.\n       Install it with your package manager (apt install git, yum install git, etc.)"
fi
ok "git" "$(git --version 2>/dev/null)"

if ! has_cmd python3; then
  fail "python3 is required but not installed."
fi
ok "python3" "$(python3 --version 2>/dev/null)"

FETCH_CMD=""
DOWNLOAD_CMD=""
DOWNLOAD_PIPE=""
if has_cmd curl; then
  FETCH_CMD="curl -fsSL"
  DOWNLOAD_CMD="curl -fsSL -o"
  DOWNLOAD_PIPE="curl -fsSL"
  ok "curl" "available"
elif has_cmd wget; then
  FETCH_CMD="wget -O-"
  DOWNLOAD_CMD="wget -O"
  DOWNLOAD_PIPE="wget -O-"
  ok "wget" "available"
elif has_cmd fetch; then
  FETCH_CMD="fetch -o -"
  DOWNLOAD_CMD="fetch -o"
  DOWNLOAD_PIPE="fetch -o -"
  ok "fetch" "available"
else
  fail "curl, wget, or fetch is required to download files."
fi

# ─── Determine install directory ─────────────────────────────────────────
INSTALL_DIR="${O11PRO_DIR:-/root/o11pro-unpacked}"

# ─── Root check ──────────────────────────────────────────────────────────
if [ "$(id -u)" -ne 0 ]; then
  fail "This installer must be run as root.\n       Install directory ${INSTALL_DIR} requires root privileges.\n       Try: sudo sh -c \"\$(curl ...)\""
fi
ok "root" "privileges confirmed"

if [ "$FORCE" = "0" ] && [ -d "$INSTALL_DIR" ]; then
  printf "  Directory %s already exists.\n" "$INSTALL_DIR"
  printf "  Overwrite? [y/N] "; read -r REPLY
  case "$REPLY" in
    [yY]|[yY][eE][sS]) ;;
    *) fail "Installation cancelled." ;;
  esac
fi

# ─── Install ─────────────────────────────────────────────────────────────
sep

info "Installing to ${BLD}$INSTALL_DIR${RST}..."

# Ensure parent directory exists
PARENT_DIR=$(dirname "$INSTALL_DIR")
mkdir -p "$PARENT_DIR"

# Clone or update the repository
if [ -d "$INSTALL_DIR/.git" ]; then
  info "Updating existing installation..."
  cd "$INSTALL_DIR"
  git pull --ff-only origin "$BRANCH" 2>/dev/null || {
    warn "git pull failed, re-cloning..."
    cd /
    rm -rf "$INSTALL_DIR"
    git clone --depth 1 -b "$BRANCH" "$REPO_URL" "$INSTALL_DIR"
  }
else
  rm -rf "$INSTALL_DIR"
  git clone --depth 1 -b "$BRANCH" "$REPO_URL" "$INSTALL_DIR"
fi
ok "repository" "cloned to ${DIM}$INSTALL_DIR${RST}"

cd "$INSTALL_DIR"

# ─── Python virtual environment ──────────────────────────────────────────
VENV_DIR="$INSTALL_DIR/src/venv"
if [ ! -d "$VENV_DIR" ]; then
  info "Creating Python virtual environment..."
  python3 -m venv "$VENV_DIR"
fi
ok "venv" "ready"

# ─── Install Python dependencies ─────────────────────────────────────────
if [ -f "requirements.txt" ]; then
  info "Installing Python dependencies..."
  "$VENV_DIR/bin/pip" install --upgrade pip -q 2>/dev/null || true
  "$VENV_DIR/bin/pip" install -r requirements.txt -q 2>/dev/null || warn "Some pip packages failed to install"
  ok "dependencies" "installed"
fi

# ─── Make binary executable ──────────────────────────────────────────────
BINARY=""
for b in src/o11pro src/o11; do
  [ -f "$b" ] && BINARY="$b" && break
done

if [ -n "$BINARY" ]; then
  chmod +x "$BINARY"
  ok "binary" "ready (${DIM}$BINARY${RST})"
else
  warn "binary" "not found in src/ — check your download"
fi

# Make launcher executable
[ -f "src/RunMe.sh" ] && chmod +x "src/RunMe.sh"
ok "launcher" "ready"

# ─── Check for FFmpeg ────────────────────────────────────────────────────
if has_cmd ffmpeg; then
  ok "ffmpeg" "$(ffmpeg -version 2>&1 | head -1 | sed 's/ffmpeg version //' | sed 's/ Copyright.*//')"
else
  warn "ffmpeg not found" "install it for transcoding support"
fi

# ─── Create runtime directories ──────────────────────────────────────────
mkdir -p "$INSTALL_DIR/src/hls/live"
mkdir -p "$INSTALL_DIR/src/keys"
mkdir -p "$INSTALL_DIR/src/epg"
mkdir -p "$INSTALL_DIR/src/dl"
mkdir -p "$INSTALL_DIR/src/manifests"
mkdir -p "$INSTALL_DIR/src/offair"
mkdir -p "$INSTALL_DIR/src/overlay"
mkdir -p "$INSTALL_DIR/src/logos"
mkdir -p "$INSTALL_DIR/src/fonts"
mkdir -p "$INSTALL_DIR/src/rec"
mkdir -p "$INSTALL_DIR/src/scripts"
mkdir -p "$INSTALL_DIR/src/logs"
mkdir -p "$INSTALL_DIR/providers"
mkdir -p "$INSTALL_DIR/cache"
ok "directories" "created"

# ─── Done ────────────────────────────────────────────────────────────────
sep
echo
printf "  ${GRN}\xE2\x9C\x94${RST} ${BLD}o11pro installed successfully${RST}\n"
echo

printf "  ${DIM}Launch:${RST}\n"
printf "    cd ${INSTALL_DIR}/src && ./RunMe.sh ${O11PRO_PORT:-1337} ${O11PRO_VERBOSE:-2}\n"
echo