ARG CUDA_VER=12.8.0
ARG UBUNTU_VER=24.04

FROM nvidia/cuda:${CUDA_VER}-devel-ubuntu${UBUNTU_VER} AS builder
SHELL ["/bin/bash", "-o", "pipefail", "-c"]

RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential autoconf pkg-config git git-lfs curl wget jq ca-certificates \
    libpng-dev libfreetype6-dev libfontconfig1-dev libxml2-dev ghostscript webp \
    libgtk-3-dev libglib2.0-dev \
  && rm -rf /var/lib/apt/lists/*

ARG GO_VERSION=1.24.1
RUN curl -fsSL "https://go.dev/dl/go${GO_VERSION}.linux-amd64.tar.gz" -o /tmp/go.tgz \
 && rm -rf /usr/local/go \
 && tar -C /usr/local -xzf /tmp/go.tgz \
 && rm /tmp/go.tgz
ENV PATH=/usr/local/go/bin:$PATH

# Clone danser-go and build danser-cli inside the repo directory
RUN git clone --depth 1 --branch fix-screenshot-slider-ticks \
      https://github.com/Arlind-dev/danser-go.git /src/danser \
 && cd /src/danser \
 && go build -ldflags="-s -w" -o danser-cli . \
 && strip danser-cli || true

# Fetch ffmpeg prebuilt binaries
RUN set -eux; \
  mkdir -p /out/ffmpeg; \
  URL="$(curl -s https://api.github.com/repos/BtbN/FFmpeg-Builds/releases/latest \
      | jq -r '.assets[].browser_download_url' \
      | grep -E 'linux64-gpl(\-shared)?\.tar\.xz$' \
      | head -n 1)"; \
  wget -qO /tmp/ffmpeg.tar.xz "$URL"; \
  tar -xf /tmp/ffmpeg.tar.xz -C /tmp; \
  FF_DIR="$(find /tmp -maxdepth 1 -type d -name 'ffmpeg-*' | head -n 1)"; \
  cp "$FF_DIR/bin/ffmpeg" /out/ffmpeg/; \
  cp "$FF_DIR/bin/ffprobe" /out/ffmpeg/ || true; \
  cp "$FF_DIR/bin/ffplay"  /out/ffmpeg/ || true; \
  if [ -d "$FF_DIR/lib" ]; then mkdir -p /out/ffmpeg/lib && cp -a "$FF_DIR/lib/." /out/ffmpeg/lib/; fi; \
  rm -rf /tmp/ffmpeg* /tmp/ffmpeg.tar.xz

# Build and stage ImageMagick into /tmp/im/usr/local
RUN set -eux; \
  IMAGICK_URL="$(curl -s https://api.github.com/repos/ImageMagick/ImageMagick/releases/latest | jq -r '.tarball_url')"; \
  mkdir -p /tmp/im; \
  wget -qO- "$IMAGICK_URL" | tar xz -C /tmp; \
  cd /tmp/ImageMagick-*; \
  ./configure --prefix=/usr/local \
    --disable-dependency-tracking \
    --disable-docs \
    --without-perl \
    --with-fontconfig=yes \
    --with-freetype=yes \
    --with-gslib=yes \
    --with-jpeg=yes \
    --with-png=yes \
    --with-tiff=yes \
    --with-xml=yes; \
  make -j"$(nproc)"; \
  make install DESTDIR=/tmp/im; \
  rm -rf /tmp/ImageMagick-*

FROM nvidia/cuda:${CUDA_VER}-runtime-ubuntu${UBUNTU_VER} AS runtime
SHELL ["/bin/bash", "-o", "pipefail", "-c"]

RUN apt-get update && apt-get install -y --no-install-recommends \
    git curl ca-certificates \
    xz-utils \
    libglvnd0 libgl1 libegl1 libgles2 \
    libx11-6 libxext6 \
    xvfb \
    libgtk-3-0 libglib2.0-0 \
    libpng16-16 libfreetype6 libfontconfig1 libxml2 ghostscript webp \
    fonts-urw-base35 fonts-noto-color-emoji fonts-dejavu-core \
    jq \
  && rm -rf /var/lib/apt/lists/* \
  && fc-cache -f -v

ARG NODE_VERSION=20.11.1
RUN curl -fsSL "https://nodejs.org/dist/v${NODE_VERSION}/node-v${NODE_VERSION}-linux-x64.tar.xz" -o /tmp/node.tar.xz \
 && tar -C /usr/local --strip-components=1 -xf /tmp/node.tar.xz \
 && rm /tmp/node.tar.xz \
 && node --version \
 && npm --version

RUN install -d -m 755 /etc/glvnd/egl_vendor.d \
 && cat >/etc/glvnd/egl_vendor.d/10_nvidia.json <<'EOF'
{
  "file_format_version": "1.0.0",
  "ICD": {
    "library_path": "libEGL_nvidia.so.0"
  }
}
EOF

RUN groupadd -g 1000 appuser 2>/dev/null || true \
 && id -u 1000 >/dev/null 2>&1 || useradd -u 1000 -g 1000 -m -s /bin/bash appuser \
 && mkdir -p /app

COPY --from=builder /src/danser        /app/danser
COPY --from=builder /out/ffmpeg/       /app/danser/ffmpeg/
COPY --from=builder /tmp/im/usr/local/ /usr/local/

ENV LD_LIBRARY_PATH=/app/danser:/app/danser/ffmpeg/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}

RUN ldconfig \
 && mkdir -p /app/danser/{songs,settings,videos,screenshots,custom-replays,skins/default-skin,ffmpeg} \
 && chown -R 1000:1000 /app/danser

COPY --chown=1000:1000 src/maps/         /app/danser/songs/
COPY --chown=1000:1000 src/replays/      /app/danser/custom-replays/
COPY --chown=1000:1000 src/settings/     /app/danser/settings/
COPY --chown=1000:1000 src/default-skin/ /app/danser/skins/default-skin/

USER 1000:1000
WORKDIR /app/danser