#!/bin/bash

apt-get purge -y intel-basekit intel-aikit intel-hpckit intel-renderkit

# add keys
wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB \
| gpg --dearmor | tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null

echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | tee /etc/apt/sources.list.d/oneAPI.list

#basekit
apt-get install -y intel-basekit

basekit_packages=(
    "intel-basekit-env-2024.0"
    "intel-basekit-getting-started"
    "intel-basekit-getting-started-2024.0"
    "intel-oneapi-advisor"
    "intel-oneapi-ccl-devel"
    "intel-oneapi-common-licensing"
    "intel-oneapi-common-oneapi-vars"
    "intel-oneapi-common-vars"
    "intel-oneapi-compiler-dpcpp-cpp"
    "intel-oneapi-dal-devel"
    "intel-oneapi-dev-utilities"
    "intel-oneapi-diagnostics-utility"
    "intel-oneapi-dnnl-devel"
    "intel-oneapi-dpcpp-ct"
    "intel-oneapi-dpcpp-debugger"
    "intel-oneapi-ipp-devel"
    "intel-oneapi-ippcp-devel"
    "intel-oneapi-libdpstd-devel"
    "intel-oneapi-libdpstd-devel-2022.3"
    "intel-oneapi-mkl-devel"
    "intel-oneapi-tbb-devel"
    "intel-oneapi-tlt"
    "intel-oneapi-vtune"
    "intel-basekit"
)

for pkg in "${basekit_packages[@]}"; do
    apt-mark hold "$pkg"
done

#hpckit
apt-get install -y intel-hpckit

hpckit_packages=(
    "intel-basekit-env-2024.0"
    "intel-basekit-getting-started"
    "intel-basekit-getting-started-2024.0"
    "intel-oneapi-advisor"
    "intel-oneapi-ccl-devel"
    "intel-oneapi-common-licensing"
    "intel-oneapi-common-oneapi-vars"
    "intel-oneapi-common-vars"
    "intel-oneapi-compiler-dpcpp-cpp"
    "intel-oneapi-dal-devel"
    "intel-oneapi-dev-utilities"
    "intel-oneapi-diagnostics-utility"
    "intel-oneapi-dnnl-devel"
    "intel-oneapi-dpcpp-ct"
    "intel-oneapi-dpcpp-debugger"
    "intel-oneapi-ipp-devel"
    "intel-oneapi-ippcp-devel"
    "intel-oneapi-libdpstd-devel"
    "intel-oneapi-libdpstd-devel-2022.3"
    "intel-oneapi-mkl-devel"
    "intel-oneapi-tbb-devel"
    "intel-oneapi-tlt"
    "intel-oneapi-vtune"
    "intel-basekit"
    "intel-hpckit-env-2024.0"
    "intel-hpckit-getting-started-2024.0"
    "intel-oneapi-compiler-fortran-2024.0"
    "intel-oneapi-inspector"
    "intel-oneapi-itac"
    "intel-oneapi-mpi-devel"
)

for pkg in "${hpckit_packages[@]}"; do
    apt-mark hold "$pkg"
done

# aikit
apt-get install -y intel-aikit

aikit_packages=(
    "intel-aikit-getting-started"
    "intel-oneapi-common-licensing"
    "intel-oneapi-common-vars"
    "intel-oneapi-dev-utilities"
    "intel-oneapi-model-zoo"
    "intel-oneapi-modin"
    "intel-oneapi-neural-compressor"
    "intel-oneapi-python"
    "intel-oneapi-pytorch"
    "intel-oneapi-tensorflow"
)

for pkg in "${aikit_packages[@]}"; do
   apt-mark hold "$pkg"
done

# renderkit
apt-get install intel-renderkit

renderkit_packages=(
    "intel-oneapi-common-licensing"
    "intel-oneapi-common-vars"
    "intel-oneapi-embree"
    "intel-oneapi-ispc"
    "intel-oneapi-openpgl"
    "intel-oneapi-openvkl"
    "intel-oneapi-ospray-studio"
    "intel-oneapi-rkutil"
    "intel-oneapi-tlt"
    "intel-renderkit-env-2024.0"
    "intel-renderkit-getting-started"
    "intel-renderkit-getting-started-2024.0"
)

for pkg in "${renderkit_packages[@]}"; do
   apt-mark hold "$pkg"
done