Docker.sh

#!/bin/bash

echo "Sart time=$(date +"%T")"
IMG_NAME="voxfusion"
IMG_TAG="latest"
CTNR_NAME="${IMG_NAME}_ctnr"
WORKDIR="/app"

VOLUME=""
while getopts "f:b:ei:t:v:r:" opt
do
  case $opt in
    f)
        DOCKERFILE="$OPTARG"
        ;;
    b) 
        if [ "$OPTARG" == "n" ]; then
            CACHE="--no-cache"
        else
            CACHE=""
        fi

        START="$(TZ=UTC0 printf '%(%s)T\n' '-1')" # `-1`  is the current time
        
        cd ./dockerfiles
        docker rmi $IMG_NAME
        if [ -z ${DOCKERFILE+x} ]; then
            docker build $CACHE -t $IMG_NAME .
        else
            docker build $CACHE -t $IMG_NAME -f ${DOCKERFILE} .
        fi

        # Pring elapsed time
        ELAPSED=$(( $(TZ=UTC0 printf '%(%s)T\n' '-1') - START ))
        TZ=UTC0 printf 'Build duration=%(%H:%M:%S)T\n' "$ELAPSED"
        ;;
    e)
        docker exec -it --user=root $CTNR_NAME  bash
        ;;
    i)
        IMG_NAME="$OPTARG"
        ;;
    t)
        IMG_TAG="$OPTARG"
        ;;
    v)
        VOLUME="-v $OPTARG"
        ;;
    r)
        RM=""
        GPU=""
        DISPLAY_ENV="DISPLAY=$DISPLAY"
        DISPLAY_VOLUME="--volume=/tmp/.X11-unix:/tmp/.X11-unix:rw"
        if [[ $OPTARG == *"m"* ]]; then
            RM="--rm"
        fi
        if [[ $OPTARG == *"g"* ]]; then
            GPU="--gpus all"
        fi
        if [[ $OPTARG == *"d"* ]]; then
            DISPLAY_ENV="DISPLAY=host.docker.internal:0"
            DISPLAY_VOLUME=""
        fi
        # Enable tracing
        set -x
        # --mount type=volume,src="",dst="" \
        # --mount type=bind,src="",dst="" \
        # --user="$(id -u):$(id -g)" \

        sudo xhost + local:docker &&
            docker run -it $RM $GPU $DISPLAY_VOLUME $VOLUME \
                -e QT_X11_NO_MITSHM=1 \
                -e XDG_RUNTIME_DIR=$XDG_RUNTIME_DIR \
                -p 8087:8087 \
                -v /dev:/dev:ro \
                --net=host \
                --privileged \
                --init \
                --ipc=host \
                --env="$DISPLAY_ENV" \
                --mount type=volume,src="vscode-extensions",dst="/root/.vscode-server/extensions" \
                --volume="$PWD:$WORKDIR" \
                --workdir $WORKDIR \
                --name $CTNR_NAME \
                "$IMG_NAME:$IMG_TAG"

        # Disable tracing
        set +x
        ;;
    \?) 
        echo "Invalid option -$OPTARG" >&2
        exit 1
        ;;
    :)
        echo "Option -$OPTARG requires an argument." >&2
        exit 1
        ;;
    *)
        echo "*"
        ;;
  esac
done