# SPDX-License-Identifier: Apache-2.0

# Copyright 2025 Huawei Technologies Co., Ltd.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

FROM hub.oepkgs.net/openeuler/openeuler:22.03-lts-sp4

RUN set -ex && \
    echo "[OS]" > /etc/yum.repos.d/openEuler.repo && \
    echo "name=OS" >> /etc/yum.repos.d/openEuler.repo && \
    echo "baseurl=https://mirrors.huaweicloud.com/openeuler/openEuler-22.03-LTS-SP4/OS/\$basearch/" >> /etc/yum.repos.d/openEuler.repo && \
    echo "enabled=1" >> /etc/yum.repos.d/openEuler.repo && \
    echo "gpgcheck=1" >> /etc/yum.repos.d/openEuler.repo && \
    echo "gpgkey=https://mirrors.huaweicloud.com/openeuler/openEuler-22.03-LTS-SP4/OS/\$basearch/RPM-GPG-KEY-openEuler" >> /etc/yum.repos.d/openEuler.repo && \
    echo "" >> /etc/yum.repos.d/openEuler.repo && \
    echo "[everything]" >> /etc/yum.repos.d/openEuler.repo && \
    echo "name=everything" >> /etc/yum.repos.d/openEuler.repo && \
    echo "baseurl=https://mirrors.huaweicloud.com/openeuler/openEuler-22.03-LTS-SP4/everything/\$basearch/" >> /etc/yum.repos.d/openEuler.repo && \
    echo "enabled=1" >> /etc/yum.repos.d/openEuler.repo && \
    echo "gpgcheck=1" >> /etc/yum.repos.d/openEuler.repo && \
    echo "gpgkey=https://mirrors.huaweicloud.com/openeuler/openEuler-22.03-LTS-SP4/everything/\$basearch/RPM-GPG-KEY-openEuler" >> /etc/yum.repos.d/openEuler.repo && \
    echo "" >> /etc/yum.repos.d/openEuler.repo && \
    echo "[update]" >> /etc/yum.repos.d/openEuler.repo && \
    echo "name=update" >> /etc/yum.repos.d/openEuler.repo && \
    echo "baseurl=https://mirrors.huaweicloud.com/openeuler/openEuler-22.03-LTS-SP4/update/\$basearch/" >> /etc/yum.repos.d/openEuler.repo && \
    echo "enabled=1" >> /etc/yum.repos.d/openEuler.repo && \
    echo "gpgcheck=1" >> /etc/yum.repos.d/openEuler.repo && \
    echo "gpgkey=https://mirrors.huaweicloud.com/openeuler/openEuler-22.03-LTS-SP4/update/\$basearch/RPM-GPG-KEY-openEuler" >> /etc/yum.repos.d/openEuler.repo

RUN set -ex && \
    yum makecache && \
    yum install -y \
        bzip2 \
        bzip2-devel \
        cmake \
        curl \
        dpkg-devel \
        gcc \
        gdbm-devel \
        git \
        gperftools \
        gperftools-devel \
        kmod \
        libdb-devel \
        libffi-devel \
        llvm-toolset-17 \
        make \
        openssl-devel \
        pciutils \
        readline-devel \
        sqlite \
        sqlite-devel \
        sudo \
        tree \
        vim \
        wget \
        xz \
        xz-devel \
        zlib-devel \
        gzip \
        zip \
        coreutils \
    && yum clean all \
    && rm -rf /var/cache/yum

WORKDIR /root

RUN set -ex && \
    ARCH=$(uname -m) && \
    wget https://repo.anaconda.com/miniconda/Miniconda3-py311_25.1.1-2-Linux-${ARCH}.sh && \
    bash /root/Miniconda3-py311_25.1.1-2-Linux-${ARCH}.sh -b && \
    rm /root/Miniconda3-py311_25.1.1-2-Linux-${ARCH}.sh

ENV PATH="/root/miniconda3/bin:$PATH"
ENV PYTHONPATH="/root/miniconda3/lib/python3.11/site-packages"

RUN set -ex && \
    pip config set global.index-url 'https://mirrors.aliyun.com/pypi/simple/' && \
    pip config set global.trusted-host mirrors.aliyun.com

RUN set -ex && \
    echo "UserName=HwHiAiUser" >> /etc/ascend_install.info && \
    echo "UserGroup=HwHiAiUser" >> /etc/ascend_install.info && \
    echo "Firmware_Install_Type=full" >> /etc/ascend_install.info && \
    echo "Firmware_Install_Path_Param=/usr/local/Ascend" >> /etc/ascend_install.info && \
    echo "Driver_Install_Type=full" >> /etc/ascend_install.info && \
    echo "Driver_Install_Path_Param=/usr/local/Ascend" >> /etc/ascend_install.info && \
    echo "Driver_Install_For_All=no" >> /etc/ascend_install.info && \
    echo "Driver_Install_Mode=normal" >> /etc/ascend_install.info && \
    echo "Driver_Install_Status=complete" >> /etc/ascend_install.info

RUN set -ex && \
    echo "source /usr/local/Ascend/nnrt/set_env.sh" >> /root/.bashrc && \
    echo "source /usr/local/Ascend/ascend-toolkit/set_env.sh" >> /root/.bashrc

RUN set -ex && \
    pip install --no-cache-dir \
        cmake>=3.26 \
        decorator \
        ray==2.43.0 \
        protobuf==3.20.0 \
        ml_dtypes \
        wheel \
        setuptools \
        wrap \
        deprecated \
        packaging \
        ninja \
        setuptools-scm>=8 \
        numpy \
        numba \
        build

CMD ["bash"]