# +=======================================+
# | PATCHES                               |
# | These should be merged into the image |
# +=======================================+

SNIPPET timezone-patch
RUN : && \
    . /etc/os-release && \
    if [ "$ID" = "ubuntu" ]; then \
        sudo apt-get update && \
        sudo -E apt-get install -y tzdata && \
        sudo rm -rf /var/cache/apt/archives /var/lib/apt/lists; \
    fi && \
    sudo ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && \
    :

# ===== END PATCHES ======
SNIPPET mitm-ca
COPY assets/mitm-ca.crt /usr/local/share/ca-certificates/mitm-ca.crt
RUN chmod 644 /usr/local/share/ca-certificates/mitm-ca.crt && \
    update-ca-certificates

# -------
SNIPPET dev-user
RUN useradd -ms /bin/bash dev && \
    usermod -a -G video dev && \
    printf "root ALL=(ALL:ALL) NOPASSWD: ALL\ndev ALL=(ALL:ALL) NOPASSWD: ALL\n" > /etc/sudoers && \
    true
WORKDIR /home/dev
ENV LANG="C.UTF-8"
USER dev

# -------
SNIPPET debian-addons
ENV DEBIAN_FRONTEND=noninteractive
RUN set -x && \
    sed -i 's/archive.ubuntu.com/mirrors.tencent.com/g' /etc/apt/sources.list && \
    sed -i 's/security.ubuntu.com/mirrors.tencent.com/g' /etc/apt/sources.list && \
    apt update && \
    apt upgrade -y && \
    apt install -y software-properties-common && \
    add-apt-repository -y ppa:ubuntu-toolchain-r/test && \
    add-apt-repository -y ppa:git-core/ppa && \
    apt-key adv --keyserver keyserver.ubuntu.com --recv-keys 15CF4D18AF4F7421 && \
    . /etc/lsb-release && \
    echo "deb http://apt.llvm.org/$DISTRIB_CODENAME/ llvm-toolchain-$DISTRIB_CODENAME-14 main" > /etc/apt/sources.list.d/llvm-14.list && \
    apt update && \
    apt install -y build-essential clang-14 clang-tidy-14 llvm-14 lld-14 gcc-11 g++-11 curl wget sudo python3-pip git unzip && \
    python3 -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip && \
    pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
    pip install -U cmake && \
    cd /usr && \
    curl -L https://github.com/rui314/mold/releases/download/v2.40.2/mold-2.40.2-x86_64-linux.tar.gz | tar -xz --strip-components=1 && \
    rm /usr/bin/ld && ln -sf /usr/bin/mold /usr/bin/ld && \
    rm -rf /var/cache/apt/archives /var/lib/apt/lists && \
    ln -sf /usr/bin/python3 /usr/bin/python && \
    ln -sf /usr/bin/clang-14 /usr/bin/clang && \
    ln -sf /usr/bin/clang++-14 /usr/bin/clang++ && \
    ln -sf /usr/bin/lld-14 /usr/bin/lld && \
    ln -sf /usr/bin/ld.lld-14 /usr/bin/ld.lld && \
    true

# mold version is not special, just the latest one at the time of writing
# lld is for AMDGPU backend, C++ code runs it.

# -------
SNIPPET debian-addons-test
ENV DEBIAN_FRONTEND=noninteractive
RUN set -x && \
    sed -i 's/archive.ubuntu.com/mirrors.tencent.com/g' /etc/apt/sources.list && \
    sed -i 's/security.ubuntu.com/mirrors.tencent.com/g' /etc/apt/sources.list && \
    apt update && apt upgrade -y && \
    apt install -y \
        curl wget sudo python3 python3-pip python3-distutils git unzip \
        libvulkan1 qtbase5-dev qtchooser qt5-qmake qtbase5-dev-tools libglfw3 ffmpeg \
        && \
    python3 -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip && \
    pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
    rm -rf /var/cache/apt/archives /var/lib/apt/lists && \
    true

# -------
FROM ubuntu:22.04 AS build-cpu
USE debian-addons
USE mitm-ca
RUN set -x && \
    apt update && apt install -y libtinfo-dev libx11-xcb-dev && \
    rm -rf /var/cache/apt/archives /var/lib/apt/lists && \
    true

USE dev-user
USE timezone-patch
BUILD build-cpu AS registry.botmaster.tgr/taichi-build-cpu:__TIME__

# -------
FROM ubuntu:22.04 AS test-cpu
USE debian-addons-test
USE mitm-ca
USE dev-user
USE timezone-patch
BUILD test-cpu AS registry.botmaster.tgr/taichi-test-cpu:__TIME__

# -------
SNIPPET gpu-build-image-deps
RUN set -x && \
    apt update && apt install -y \
        libtinfo-dev libx11-xcb-dev \
        ffmpeg libxrandr-dev \
        libxinerama-dev libxcursor-dev \
        libxi-dev libglu1-mesa-dev \
        freeglut3-dev mesa-common-dev \
        libssl-dev libglm-dev \
        libxcb-keysyms1-dev libxcb-dri3-dev \
        libxcb-randr0-dev libxcb-ewmh-dev \
        libpng-dev \
        libwayland-dev bison \
        liblz4-dev libzstd-dev \
        qtbase5-dev qtchooser qt5-qmake qtbase5-dev-tools \
        libglfw3 libglfw3-dev libjpeg-dev \
        && \
    rm -rf /var/cache/apt/archives /var/lib/apt/lists && \
    true

# -------
FROM rocm/dev-ubuntu-22.04:6.4.2 AS build-amdgpu
# The AMDGPU lists has bad distribution
RUN rm /etc/apt/sources.list.d/*.list
USE debian-addons
USE mitm-ca
USE gpu-build-image-deps
USE dev-user
USE timezone-patch
BUILD build-amdgpu AS registry.botmaster.tgr/taichi-build-amdgpu:__TIME__
BUILD build-amdgpu AS registry.botmaster.tgr/taichi-test-amdgpu:__TIME__

# -------
FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 AS build-cuda
ENV NVIDIA_DRIVER_CAPABILITIES compute,graphics,utility
USE debian-addons
USE mitm-ca
USE gpu-build-image-deps
# Remove mesa EGL driver, which interferes with the propritary NVIDIA drivers
RUN rm -f /usr/lib/x86_64-linux-gnu/libEGL_mesa*
USE dev-user
USE timezone-patch
BUILD build-cuda AS registry.botmaster.tgr/taichi-build-cuda:__TIME__

# -------
FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 AS test-cuda
ENV NVIDIA_DRIVER_CAPABILITIES compute,graphics,utility
USE debian-addons-test
USE mitm-ca
# Remove mesa EGL driver, which interferes with the propritary NVIDIA drivers
RUN rm -f /usr/lib/x86_64-linux-gnu/libEGL_mesa*
USE dev-user
USE timezone-patch
BUILD test-cuda AS registry.botmaster.tgr/taichi-test-cuda:__TIME__

# -------
FROM ubuntu:22.04 AS android-sdk
RUN set -x && \
    apt-get update && apt-get -y install wget unzip openjdk-11-jdk redis-tools && \
    mkdir /android-sdk && \
    wget -O clt.zip https://dl.google.com/android/repository/commandlinetools-linux-9477386_latest.zip && \
    unzip clt.zip && rm clt.zip && \
    mv cmdline-tools /android-sdk && \
    yes | /android-sdk/cmdline-tools/bin/sdkmanager --sdk_root=/android-sdk --install \
        'ndk-bundle' \
        'build-tools;30.0.3' \
        'build-tools;33.0.0' \
        'cmake;3.10.2.4988404' \
        'platforms;android-30' \
        'platforms;android-33' \
        'platform-tools' \
        emulator \
        && \
    chown -R 1000:1000 /android-sdk && \
    true

FROM build-cuda AS build-android
USER root
ENV ANDROID_SDK_ROOT=/android-sdk
RUN set -x && \
    rm /etc/apt/sources.list.d/* && apt-get update && apt-get -y install wget unzip openjdk-11-jdk redis-tools && \
    true

COPY --from=android-sdk /android-sdk /android-sdk
USER dev
COPY assets/dot-android.tgz /dot-android.tgz
RUN set -x && \
    cd /home/dev && mkdir .android && cd .android && \
    tar -xzf /dot-android.tgz && \
    chown -R dev:dev /home/dev/.android && \
    true

USE timezone-patch
BUILD build-android AS registry.botmaster.tgr/taichi-build-android:__TIME__
BUILD build-android AS registry.botmaster.tgr/taichi-test-android:__TIME__
# -------
# vim: ft=Dockerfile: