#!/usr/bin/env bash
set -Eeuo pipefail

readonly LOG_PREFIX="[wine-desktop]"
DISPLAY="${DISPLAY:-:0}"
export DISPLAY
DISPLAY_WIDTH="${DISPLAY_WIDTH:-1280}"
DISPLAY_HEIGHT="${DISPLAY_HEIGHT:-720}"
DISPLAY_DEPTH="${DISPLAY_DEPTH:-24}"
VNC_PORT="${VNC_PORT:-5900}"
NOVNC_PORT="${NOVNC_PORT:-6080}"
WINEPREFIX="${WINEPREFIX:-/home/container/.wine}"
export DISPLAY DISPLAY_WIDTH DISPLAY_HEIGHT DISPLAY_DEPTH VNC_PORT NOVNC_PORT WINEPREFIX

children=()
cleanup() {
    trap - EXIT INT TERM
    for pid in "${children[@]:-}"; do
        kill "$pid" 2>/dev/null || true
    done
}
trap cleanup EXIT INT TERM

log() { echo "${LOG_PREFIX} $*"; }

if ! xdpyinfo -display "$DISPLAY" >/dev/null 2>&1; then
    log "Starting Xvfb on ${DISPLAY} (${DISPLAY_WIDTH}x${DISPLAY_HEIGHT}x${DISPLAY_DEPTH})"
    Xvfb "$DISPLAY" -screen 0 "${DISPLAY_WIDTH}x${DISPLAY_HEIGHT}x${DISPLAY_DEPTH}" -ac +extension GLX +render -noreset &
    children+=("$!")
    for _ in {1..50}; do
        xdpyinfo -display "$DISPLAY" >/dev/null 2>&1 && break
        sleep 0.1
    done
fi

if ! xdpyinfo -display "$DISPLAY" >/dev/null 2>&1; then
    log "X server did not become ready"
    exit 1
fi

export XDG_RUNTIME_DIR="${XDG_RUNTIME_DIR:-/tmp/runtime-${USER:-container}}"
mkdir -p "$XDG_RUNTIME_DIR"
chmod 0700 "$XDG_RUNTIME_DIR"

if command -v dbus-launch >/dev/null 2>&1; then
    eval "$(dbus-launch --sh-syntax)"
    children+=("${DBUS_SESSION_BUS_PID:-}")
fi

if command -v pulseaudio >/dev/null 2>&1 && ! pgrep -x pulseaudio >/dev/null 2>&1; then
    log "Starting PulseAudio"
    pulseaudio --system --daemonize=yes --disallow-exit --exit-idle-time=-1 >/tmp/wine-desktop-pulseaudio.log 2>&1 || \
        log "PulseAudio could not start; applications may run without audio"
fi

if ! pgrep -x xfce4-session >/dev/null 2>&1; then
    log "Starting XFCE desktop"
    startxfce4 >/tmp/wine-desktop-xfce.log 2>&1 &
    children+=("$!")
fi

vnc_args=(-display "$DISPLAY" -rfbport "$VNC_PORT" -forever -shared -nopw)
if [[ -n "${VNC_PASSWORD:-}" ]]; then
    password_file="${WINEPREFIX}/.vncpasswd"
    mkdir -p "$WINEPREFIX"
    x11vnc -storepasswd "$VNC_PASSWORD" "$password_file" >/dev/null
    chmod 0600 "$password_file"
    vnc_args=(-display "$DISPLAY" -rfbport "$VNC_PORT" -forever -shared -rfbauth "$password_file")
fi
log "Starting x11vnc on TCP ${VNC_PORT}"
x11vnc "${vnc_args[@]}" >/tmp/wine-desktop-x11vnc.log 2>&1 &
children+=("$!")

novnc_web="/usr/share/novnc"
[[ -d "$novnc_web" ]] || novnc_web="/usr/share/novnc/utils"
log "Starting noVNC on TCP ${NOVNC_PORT}"
websockify --web="$novnc_web" "${NOVNC_PORT}" "localhost:${VNC_PORT}" >/tmp/wine-desktop-novnc.log 2>&1 &
children+=("$!")

mkdir -p "$WINEPREFIX"
if [[ ! -f "$WINEPREFIX/system.reg" ]]; then
    log "Initializing Wine prefix at ${WINEPREFIX}"
    wineboot --init >/tmp/wine-desktop-wineboot.log 2>&1 || {
        log "Wine prefix initialization failed; see /tmp/wine-desktop-wineboot.log"
        exit 1
    }
fi

# The inherited command remains the official yolk command, normally:
# /bin/bash /entrypoint.sh. This preserves its update, winetricks, startup,
# and signal behavior while tini remains PID 1.
log "Starting inherited Wine command: $*"
exec "$@"