# 基于最新的 CUDA 开发镜像构建
FROM nvidia/cuda:12.2.0-devel-ubuntu22.04 AS base

# 设置环境变量
ENV USERNAME=user \
    USER_UID=1000 \
    USER_GID=1000 \
    LANG=C.UTF-8 \
    LC_ALL=C.UTF-8 \
    NVIDIA_VISIBLE_DEVICES=all \
    NVIDIA_DRIVER_CAPABILITIES=all \
    EDITOR=code \
    VISUAL=code \
    GIT_EDITOR="code --wait" \
    OPENVSCODE_SERVER_ROOT=/home/.vscode \
    OPENVSCODE=/home/.vscode/bin/openvscode-server
    ENV access_token="EwCIA8l6BAAUbDba3x2OMJElkF7gJ4z/VbCPEz0AAQwSGkkb4/JzTr8+Ow72vC3fYBhCRS/AxwGCR1PqoElL6tQPzA7udYCTEbb5s7+YClJASBcFYQdsSPeV/pqJCNQoIcvGvDkHF4s6GcyOXAxvw98udailEOuKrygcIsrLkfUZiOECxrS1O3hZscpFdRE/8b2jYfwh6hNdary36kVCScxkCp0Ksn8VBhA7FnqP5x6yvVsh8nO9F0eJOxbwwTNqsXKmPHU7MCm+YSWofb06OdZC0dtQx4LvZVfft7vlg79k2iX/jRbTcUrBzUxpNRXqdUIOwrjuJb+/6EqLT5rxCEV+eykEv6gAVTwBOO6G97CmVzNyziIGThKOqYN8picQZgAAEHzYXyyWgN0GWonnjmJYADNQAsTodxomCQhGGRlv/MzWW8qy6svkhfBmLW8iMGB6lF/euD5nmrPd/4xLVDI1Wy6PR4oArT8aHLJsP3Z1Y36X8WzsEEdjSMToMFvGt7DuPXBSxK82sC2VWyVKpgzWFFi1k5NCvux9oOFM6XrSlzLQUTVlsh82uBxCHgETMA/3EEiIAioEklJJdOAcM3m7PRXHoz/eICSObjUuCSMKwJwF1hjbtkRRZflnllNZIgQolAoZBPX43XT42HCMtQoKI0OJM9JF8iSCzIvuoGKEZSS96dPgmLcb1I/39AJvqy93tpdLpD3vx2vKIV/tuSAert15TAp9GFNJ6WXAwb5gN4fSsVl5uxwVEO+UNpcNyFfzo/5lk3h6Z7AL8aTHwZKflj+nGE7rdtswzz+JW6qMHelAo9mm8TzS/wogYbmxHvJrcdsI4hCbutKregznoWzC352URypdeDwz0Q8YdmSwyudYnMXWiMFCEBUmLgy71SleszhgigGJVlIjYsfgDCD2Oa08dBCPALAmbB7ovbpmlK74MCqNhRUbLy7HdfwF7qOtBSoYs4yTMt2QhMz6k/mVTd812VhKAjPYj3yLYm3nkC+sEa6zVPKhbpjZlvZ0myqYABXw6p9wre972CfL6k5oEoF79zZZp57mb9nNzTUZlVaTFvMGd4R4S6xqy1Iho8qs/ySmHPyKXLopgF1Jvf64e/uKco0DLAmxirK6VJFg3sPD5b5EC52SFoPFtaG2ZWC6hkcWtKBWBDQcuEGTt/dyaLcJiGcHJEnvlGXvCuprT0auLb59Ag=="
    ENV drive_id="cf5e6afc6b196aac"

# 更新和安装必要的软件包，清理缓存以减少镜像大小
RUN apt-get update && apt-get install -y --no-install-recommends \
    curl \
    git \
    git-lfs \
    libatomic1 \
    locales \
    man \
    nano \
    net-tools \
    netcat \
    openssh-client \
    python3 \
    python3-pip \
    python3-venv \
    sudo \
    vim \
    wget \
    zsh \
    zip \
    unzip \
    ffmpeg \
    imagemagick \
    && git lfs install \
    && rm -rf /var/lib/apt/lists/*

# 修复 ImageMagick 策略
RUN sed -i '/<policy domain="path" rights="none" pattern="@\*"/d' /etc/ImageMagick-6/policy.xml

# 创建工作目录
WORKDIR /home/

# 安装应用
RUN curl -O https://downloads.rclone.org/rclone-current-linux-amd64.zip && \
    unzip rclone-current-linux-amd64.zip && cd rclone-*-linux-amd64 && cp rclone /usr/bin/ && \
    chown root:root /usr/bin/rclone && chmod 755 /usr/bin/rclone

# 安装 Node.js 和 npm
RUN curl -fsSL https://deb.nodesource.com/setup_20.x | sudo -E bash - && \
    apt-get install -y nodejs && \
    npm install -g npm@latest && \
    npm install -g yarn

# 下载并解压最新的 OpenVSCode Server 版本
RUN RELEASE_TAG=$(curl -sX GET "https://api.github.com/repos/gitpod-io/openvscode-server/releases/latest" | awk '/tag_name/{print $4;exit}' FS='[""]') && \
    arch=$(uname -m) && \
    if [ "${arch}" = "x86_64" ]; then \
        arch="x64"; \
    elif [ "${arch}" = "aarch64" ]; then \
        arch="arm64"; \
    elif [ "${arch}" = "armv7l" ]; then \
        arch="armhf"; \
    fi && \
    wget https://github.com/gitpod-io/openvscode-server/releases/download/${RELEASE_TAG}/${RELEASE_TAG}-linux-${arch}.tar.gz && \
    tar -xzf ${RELEASE_TAG}-linux-${arch}.tar.gz && \
    mv ${RELEASE_TAG}-linux-${arch} ${OPENVSCODE_SERVER_ROOT} && \
    cp ${OPENVSCODE_SERVER_ROOT}/bin/remote-cli/openvscode-server ${OPENVSCODE_SERVER_ROOT}/bin/remote-cli/code && \
    rm -f ${RELEASE_TAG}-linux-${arch}.tar.gz

# 安装 Firefox 和 GeckoDriver
RUN install -d -m 0755 /etc/apt/keyrings && \
    wget -q https://packages.mozilla.org/apt/repo-signing-key.gpg -O- | tee /etc/apt/keyrings/packages.mozilla.org.asc > /dev/null && \
    echo "deb [signed-by=/etc/apt/keyrings/packages.mozilla.org.asc] https://packages.mozilla.org/apt mozilla main" | tee /etc/apt/sources.list.d/mozilla.list > /dev/null && \
    apt-get update && \
    apt-get install -y firefox && \
    wget https://github.com/mozilla/geckodriver/releases/download/v0.34.0/geckodriver-v0.34.0-linux64.tar.gz && \
    tar -xvzf geckodriver-v0.34.0-linux64.tar.gz && \
    sudo mv geckodriver /usr/local/bin/ && \
    sudo chmod +x /usr/local/bin/geckodriver && \
    geckodriver --version && \
    rm geckodriver-v0.34.0-linux64.tar.gz

# 创建用户和用户组，设置权限
RUN groupadd --gid ${USER_GID} ${USERNAME} \
    && useradd --uid ${USER_UID} --gid ${USERNAME} -m -s /bin/bash ${USERNAME} \
    && echo ${USERNAME} ALL=\(root\) NOPASSWD:ALL > /etc/sudoers.d/${USERNAME} \
    && chmod 0440 /etc/sudoers.d/${USERNAME} \
    && chmod g+rw /home \
    && chown -R ${USERNAME}:${USERNAME} ${OPENVSCODE_SERVER_ROOT} \
    && chown -R ${USERNAME}:${USERNAME} /home/${USERNAME}

USER root

# 安装 oh-my-zsh
RUN yes | sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)"

# # 安装 VSCode 插件
RUN ${OPENVSCODE} --install-extension ms-python.python && \
     ${OPENVSCODE} --install-extension monokai.theme-monokai-pro-vscode && \
     ${OPENVSCODE} --install-extension zokugun.cron-tasks && \
     ${OPENVSCODE} --install-extension golf1052.code-sync && \
     ${OPENVSCODE} --install-extension kelvin.vscode-sshfs && \
     ${OPENVSCODE} --install-extension nonoroazoro.syncing && \
     ${OPENVSCODE} --install-extension zokugun.sync-settings

# 复制并安装 Python 依赖，修复权限问题
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# 复制 start.sh 脚本到容器中
COPY start.sh /start.sh

# 设置 start.sh 脚本的执行权限
RUN chmod +x /start.sh

# 将 ~/.local/bin 添加到 PATH
ENV PATH="/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/sbin:/usr/local/bin:/home/${USERNAME}/.local/bin:${PATH}"

# 启用连接令牌保护，启动 OpenVSCode Server
ENTRYPOINT ["/bin/sh", "-c", "/start.sh && exec $OPENVSCODE --host 0.0.0.0 --port 7860 --connection-token 'vscode'"]