FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04

ENV DEBIAN_FRONTEND=noninteractive \
	TZ=Europe/Belgrade

# Remove any third-party apt sources to avoid issues with expiring keys.
# Install some basic utilities
RUN rm -f /etc/apt/sources.list.d/*.list && \
    apt-get update && apt-get install -y --no-install-recommends \
    curl \
    ca-certificates \
    sudo \
    git \
    wget \
    procps \
    git-lfs \
    zip \
    unzip \
    htop \
    vim \
    nano \
    bzip2 \
    libx11-6 \
    build-essential \
    libsndfile-dev \
    software-properties-common \
    supercollider \
    supercollider-server \
    jackd2 \
    libjack-jackd2-dev \
    ffmpeg \
    xvfb \
    x11-xserver-utils \
    telnet \
    iputils-ping \
    icecast2 \
    nginx \
 && rm -rf /var/lib/apt/lists/*

RUN add-apt-repository ppa:flexiondotorg/nvtop && \
    apt-get upgrade -y && \
    apt-get install -y --no-install-recommends nvtop

RUN curl -sL https://deb.nodesource.com/setup_21.x  | bash - && \
    apt-get install -y nodejs && \
    npm install -g configurable-http-proxy

# Create a working directory
WORKDIR /app

# Create a non-root user and switch to it
RUN adduser --disabled-password --gecos '' --shell /bin/bash user \
 && chown -R user:audio /app && usermod -aG audio user
RUN echo "user ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/90-user
RUN echo "@audio - rtprio 95" > /etc/security/limits.d/90-user
RUN echo "@audio - memlock unlimited" >> /etc/security/limits.d/90-user
RUN mkdir /data && chown user:audio /data

USER user

# All users can use /home/user as their home directory
ENV HOME=/home/user
ENV DISPLAY=127.0.0.1:99
RUN mkdir $HOME/.cache $HOME/.config \
 && chmod -R 777 $HOME

# Copy configuration files
COPY icecast.xml .
COPY nginx.conf .
COPY setup.sc .
COPY setup2.sc .
COPY start.sh .
COPY Consumer.py .
COPY test.html /data/
COPY Untitled1.ipynb /data/
COPY Player.ipynb /data/

# Standardize SC directories for the user (optional but good for debugging)
RUN mkdir -p /home/user/.local/share/SuperCollider/Extensions

# Set up the Conda environment
ENV CONDA_AUTO_UPDATE_CONDA=false \
    PATH=$HOME/miniconda/bin:$PATH
RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py39_4.10.3-Linux-x86_64.sh \
 && chmod +x ~/miniconda.sh \
 && ~/miniconda.sh -b -p ~/miniconda \
 && rm ~/miniconda.sh \
 && conda clean -ya

WORKDIR $HOME/app

#######################################
# Start root user section
#######################################

USER root

# User Debian packages
## Security warning : Potential user code executed as root (build time)
RUN --mount=target=/root/packages.txt,source=packages.txt \
    apt-get update && \
    xargs -r -a /root/packages.txt apt-get install -y --no-install-recommends \
    && rm -rf /var/lib/apt/lists/*

RUN --mount=target=/root/on_startup.sh,source=on_startup.sh,readwrite \
	bash /root/on_startup.sh

# Fix permissions
RUN chown -R user:user /app && chmod +x /app/start.sh && mkdir -p /app/logs && chown -R user:user /app/logs && \
    mkdir -p /var/log/nginx && chown -R user:user /var/log/nginx && \
    mkdir -p /var/lib/nginx && chown -R user:user /var/lib/nginx && chmod -R 777 /tmp && rm -fr /tmp/*

#######################################
# End root user section
#######################################

USER user

# Python packages
RUN --mount=target=requirements.txt,source=requirements.txt \
    pip install --no-cache-dir --upgrade -r requirements.txt

# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app

RUN chmod +x start_server.sh
RUN chmod +x start.sh

COPY --chown=user login.html /home/user/miniconda/lib/python3.9/site-packages/jupyter_server/templates/login.html

ENV PYTHONUNBUFFERED=1 \
	GRADIO_ALLOW_FLAGGING=never \
	GRADIO_NUM_PORTS=2 \
	GRADIO_SERVER_NAME=0.0.0.0 \
	GRADIO_THEME=huggingface \
	SYSTEM=spaces \
	SHELL=/bin/bash \
    DISPLAY=127.0.0.1:99 \
    XDG_RUNTIME_DIR=/data

# Install FoxDot Quark (needs to be done as the user)
# We run Xvfb momentarily just in case sclang needs a display (usually doesn't for CLI, but safe)
# 'xvfb-run' is a handy wrapper if we had it, but we can just run sclang directly for CLI text.
# RUN sclang setup.sc

EXPOSE 7860

CMD ["./start.sh"]