#! /usr/bin/env bash

set -o errexit

#container=$(buildah from nvidia/cuda:10.1-devel-ubuntu18.04)
container=$(buildah from nvidia/cuda:11.3.0-devel-ubuntu20.04)

buildah config --label maintainer="Suren A. Chilingaryan <csa@suren.me>" $container
buildah config --env LANG="C.UTF-8" --env LC_ALL="C.UTF-8" --env TERM="xterm" $container

buildah run $container ln -snf /usr/share/zoneinfo/Europe/London /etc/localtime
buildah run $container sh -c 'echo "Europe/London" > /etc/timezone'

buildah run $container sh -c 'echo "ssh:x:101:" >> /etc/group'
buildah run $container sh -c 'echo "sshd:x:101:101:sshd:/dev/null:/sbin/nologin" >> /etc/passwd'
buildah run $container sh -c 'echo "messagebus:x:102:" >> /etc/group'
buildah run $container sh -c 'echo "messagebus:x:102:102:messagebus:/dev/null:/sbin/nologin" >> /etc/passwd'
buildah run $container sh -c 'echo "rdma:x:103:" >> /etc/group'
buildah run $container sh -c 'echo "systemd-journal:x:104:" >> /etc/group'
buildah run $container sh -c 'echo "systemd-network:x:105:" >> /etc/group'
buildah run $container sh -c 'echo "systemd-network:x:105:105:systemd-network:/dev/null:/sbin/nologin" >> /etc/passwd'
buildah run $container sh -c 'echo "systemd-resolve:x:106:" >> /etc/group'
buildah run $container sh -c 'echo "systemd-resolve:x:106:106:systemd-resolve:/dev/null:/sbin/nologin" >> /etc/passwd'
buildah run $container sh -c 'echo "systemd-timesync:x:107:" >> /etc/group'
buildah run $container sh -c 'echo "systemd-timesync:x:107:107:systemd-timesync:/dev/null:/sbin/nologin" >> /etc/passwd'


buildah run $container sh -c 'apt-get update --fix-missing && apt-get install -y bash wget bzip2 mc ca-certificates git openssh-server'
buildah run $container sh -c 'apt-get install -y python3 python3-setuptools python3-pip cython3'
buildah run $container sh -c 'apt-get install -y cmake make autoconf automake libtool gcc g++ libboost-all-dev'
buildah run $container sh -c 'apt-get install -y ca-certificates pkg-config gobject-introspection liblapack-dev libjpeg-dev libtiff-dev libglib2.0-dev libjson-glib-dev libopenmpi-dev libhdf5-dev libclfft-dev libgsl-dev libgirepository1.0-dev fftw3-dev zlib1g-dev'
buildah run $container sh -c 'apt-get install -y python3-dev python3-numpy  python-gobject tifffile'
buildah run $container sh -c 'apt-get install -y ocl-icd-opencl-dev clinfo'
buildah run $container sh -c 'apt-get install -y jupyter-notebook'

buildah run $container sh -c 'update-alternatives --remove python /usr/bin/python2'
buildah run $container sh -c 'update-alternatives --install /usr/bin/python python /usr/bin/python3 10'

buildah run $container mkdir -p /etc/OpenCL/vendors
buildah run $container sh -c 'echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd'

buildah copy $container pre-setup /root/pre-setup
buildah run $container pip3 install -r /root/pre-setup/requirements.txt

buildah copy $container setup /root/setup
buildah commit --format docker $container pycuda:latest