update_base.sh

#! /bin/bash

set -eu
set -o pipefail

usage() {
  echo "Usage:"
  echo "./update_base.sh <BASE_IMAGENAME> <BASE_CONTAINERFILE> <BASE_CONTAINERNAME> [<extra_args>]"
}

# check that we have exactly 3 params
if [ $# -lt 3 ]; then
  usage
  exit 1
fi

DEVC_REPO_DIR=/home/shank/code/misc/dev-container

# params - base
BASE_IMAGENAME=$1
BASE_CONTAINERFILE=$2
BASE_CONTAINERNAME=$3

echo "BASE_IMAGENAME:${BASE_IMAGENAME}"
echo "BASE_CONTAINERFILE:${BASE_CONTAINERFILE}"
echo "BASE_CONTAINERNAME:${BASE_CONTAINERNAME}"

# default mount volumes
DEFAULT_MOUNTS=(
  "--volume /tmp/.X11-unix:/tmp/.X11-unix"
  "--volume $HOME:/host_home"
  "--volume $HOME/code:/home/shank/code"
  "--volume $HOME/.ssh:/home/shank/.ssh"
  "--volume $SSH_AUTH_SOCK:/tmp/ssh-agent.socket"
  # "--volume $XDG_RUNTIME_DIR/podman/podman.sock:/run/podman/podman.sock"
)

if [ -d $HOME/.rustup ]; then
  DEFAULT_MOUNTS+=("--volume $HOME/.rustup:/home/shank/.rustup")
elif [ -d /evaldisk/shank/.rustup ]; then
  DEFAULT_MOUNTS+=("--volume /evaldisk/shank/.rustup:/home/shank/.rustup")
fi

if [ -d $HOME/.cargo ]; then
  DEFAULT_MOUNTS+=("--volume $HOME/.cargo:/home/shank/.cargo")
elif [ -d /evaldisk/shank/.cargo ]; then
  DEFAULT_MOUNTS+=("--volume /evaldisk/shank/.cargo:/home/shank/.cargo")
fi

# if /workdisk exists, then add it to the list of mounts as well
if [ -d /workdisk ]; then
  DEFAULT_MOUNTS+=("--volume /workdisk:/workdisk")
fi

# if /evaldisk exists, then add it to the list of mounts as well
if [ -d /evaldisk ]; then
  DEFAULT_MOUNTS+=("--volume /evaldisk:/evaldisk")
fi

# if both /home/common and /evaldisk/shank exist, then as user which to mount at /home/common
if [ -d /home/common ] && [ -d /evaldisk/shank ]; then
  echo "Mount which to /home/common? (enter 1/2/3)"
  echo "1. /home/common"
  echo "2. /evaldisk/shank"
  echo "3. None"
  read selection
  # ensure that the user enters a valid selection
  if [[ $selection != "1" && $selection != "2" && $selection != "3" ]]; then
    echo "Invalid selection"
    exit 1
  fi
  if [[ $selection == "1" ]]; then
    DEFAULT_MOUNTS+=("--volume /home/common:/home/common")
  elif [[ $selection == "2" ]]; then
    DEFAULT_MOUNTS+=("--volume /evaldisk/shank:/home/common")
  fi

elif [ -d /home/common ]; then
  # if /home/common exists, then add it to the list of mounts as well
  read -p "Mount /home/common to /home/common? (y/n) : " mount_home_common
  if [[ $mount_home_common == "y" || $mount_home_common == "Y" ]]; then
    DEFAULT_MOUNTS+=("--volume /home/common:/home/common")
  fi

elif [ -d /evaldisk/shank ]; then
  # if /evaldisk exists, then add /evaldisk/shank to the list of mounts as well
  read -p "Mount /evaldisk/shank to /home/common? (y/n) : " mount_evaldisk_shank_to_common
  if [[ $mount_evaldisk_shank_to_common == "y" || $mount_evaldisk_shank_to_common == "Y" ]]; then
    DEFAULT_MOUNTS+=("--volume /evaldisk/shank:/home/common")
  fi
fi

# default published ports
DEFAULT_PORTS=(
  # "--publish 3306:3306"
)

# stop running devc container
echo ">> [0] stopping (and removing) existing container if any..."
! docker container stop $BASE_CONTAINERNAME
! docker container rm --force $BASE_CONTAINERNAME

# update base image
echo ">> [1] updating base image..."

# Separate build args from run args
BUILD_ARGS=()
RUN_ARGS=()

i=4 # Start from 4th argument (after the 3 required params)
while [ $i -le $# ]; do
  arg="${!i}"
  case "$arg" in
    -v | --volume | --mount)
      # Volume args go to run command (this arg + next arg)
      RUN_ARGS+=("$arg")
      i=$((i + 1))
      if [ $i -le $# ]; then
        RUN_ARGS+=("${!i}")
      fi
      ;;
    -v=* | --volume=* | --mount=*)
      # Volume args with = go to run command
      RUN_ARGS+=("$arg")
      ;;
    *)
      # Everything else goes to build
      BUILD_ARGS+=("$arg")
      ;;
  esac
  i=$((i + 1))
done

docker image build \
  "${BUILD_ARGS[@]}" \
  --build-arg UID=$(id -u) \
  --build-arg GID=$(id -g) \
  --tag ${BASE_IMAGENAME} \
  --file "$DEVC_REPO_DIR/$BASE_CONTAINERFILE" .

# build container
echo ">> [2] creating and running the proj container..."
docker container run \
  -it \
  --label="no-prune" \
  --name ${BASE_CONTAINERNAME} \
  --hostname "${BASE_CONTAINERNAME}-$(cat /etc/hostname)" \
  --security-opt label=disable \
  --security-opt seccomp=unconfined \
  --cap-add=SYS_PTRACE \
  --cap-add=NET_RAW \
  --cap-add=MKNOD \
  --cap-add=AUDIT_WRITE \
  --env DISPLAY=${DISPLAY:-:0} \
  ${DEFAULT_MOUNTS[@]} \
  ${DEFAULT_PORTS[@]} \
  "${RUN_ARGS[@]}" \
  ${BASE_IMAGENAME}
# --userns=keep-id \
# --env CONTAINER_HOST=unix:///run/podman/podman.sock \
# --env DISPLAY=$DISPLAY \
# --userns=keep-id \