#!/bin/bash
script_dir="$( cd "$( dirname "${BASH_SOURCE[0]}"  )" >/dev/null 2>&1 && pwd )"

report_error() {
    RED=$(tput setaf 1)
    NO_COLOR=$(tput sgr0)

    cat <<EOF
${RED}error:${NO_COLOR} input is invalid

build_image
Build an Occlum Docker image for a specific OS

USAGE:
    build_image.sh <OCCLUM_LABEL> <OS_NAME> <OCCLUM_BRANCH>

<OCCLUM_LABEL>:
    An arbitrary string chosen by the user to describe the version of Occlum preinstalled in the Docker image, e.g., "latest", "0.12.0", "prerelease", and etc.

<OS_NAME>:
    The name of the OS distribution that the Docker image is based on. Currently, <OS_NAME> must be one of the following values:
        ubuntu18.04         Use Ubuntu 18.04 as the base image
        ubuntu20.04         Use Ubuntu 20.04 as the base image
        centos8.2           Use CentOS 8.2 as the base image
        aliyunlinux3        Use AliyunLinux 3 as the base image
        anolis8.4           Use Anolis 8.4 as the base image

<OCCLUM_BRANCH>:
    The Occlum branch which the docker image is built on, e.g "0.24.0".
    It is optional, if not provided, "master" branch will be used.

The resulting Docker image will have "occlum/occlum:<OCCLUM_LABEL>-<OS_NAME>" as its label.
EOF
    exit 1
}

set -e

if [[ ( "$#" < 2 ) ]] ; then
    report_error
fi

occlum_label=$1
os_name=$2
occlum_branch=${3:-master}

function check_item_in_list() {
    item=$1
    list=$2
    [[ $list =~ (^|[[:space:]])$item($|[[:space:]]) ]]
}

check_item_in_list "$os_name" "ubuntu18.04 ubuntu20.04 centos8.2 aliyunlinux3 anolis8.4 anolis8.8" || report_error

cd "$script_dir/.."
docker build -f "$script_dir/Dockerfile.$os_name" -t "occlum/occlum:$occlum_label-$os_name" --build-arg OCCLUM_BRANCH=$occlum_branch .