#!/bin/sh

fatal() {
  echo "Error: $1"
  exit 1
}

error () {
  echo "Error: $1"
}

info() {
  echo "$1"
}

debug() {
  [ "$RODEO_DEBUG" = "1" ] && info "$1"
}

warning() {
  echo "Warning: $1"
}

remount () {
  overlay_config=/rodeo/configuration/included_overlays
  [ -n "$1" ] && overlay_config="$1"
  while IFS="" read -r dir || [ -n "$dir" ]
  do
    debug "Scanning for overlays to be mounted on /$dir"
    lowerdirs=""
      while IFS="" read -r overlay || [ -n "$overlay" ]
      do
        debug "Checking if $overlay has $dir"
        if [ -d "/rodeo/overlays/$overlay/$dir" ]; then
          debug "$overlay has $dir"
          if [ -z "$lowerdirs" ]; then
            lowerdirs="/rodeo/overlays/$overlay/$dir"
          else
            lowerdirs="$lowerdirs:/rodeo/overlays/$overlay/$dir"
          fi
        else
          debug "$overlay does not have $dir"
        fi
      done < "$overlay_config"
    umount "/$dir" || error "Unmounting of /$dir failed"
    if [ -z "$lowerdirs" ]; then
      warning "No overlay has $dir"
      mount --bind "/rodeo/root/$dir" "/$dir" || error "Bind-Mounting /$dir failed"
    else
      mount -t overlay overlay -o lowerdir="$lowerdirs",upperdir="/rodeo/root/$dir",workdir="/rodeo/tmp/$dir" "/$dir" || error "Overlay-Mounting /$dir failed"
    fi
  done < /rodeo/configuration/managed_directories
}

initial_mount () {
  overlay_config=/rodeo/configuration/included_overlays
  [ -n "$1" ] && overlay_config="$1"
  while IFS="" read -r dir || [ -n "$dir" ]
  do
    debug "Scanning for overlays to be mounted on /$dir"
    lowerdirs=""
      while IFS="" read -r overlay || [ -n "$overlay" ]
      do
        debug "Checking if $overlay has $dir"
        if [ -d "/rodeo/overlays/$overlay/$dir" ]; then
          debug "$overlay has $dir"
          if [ -z "$lowerdirs" ]; then
            lowerdirs="/rodeo/overlays/$overlay/$dir"
          else
            lowerdirs="$lowerdirs:/rodeo/overlays/$overlay/$dir"
          fi
        else
          debug "$overlay does not have $dir"
        fi
      done < "$overlay_config"
    if [ -z "$lowerdirs" ]; then
      warning "No overlay has $dir"
      mount --bind "/rodeo/root/$dir" "/$dir" || error "Bind-Mounting /$dir failed"
    else
      mount -t overlay overlay -o lowerdir="$lowerdirs",upperdir="/rodeo/root/$dir",workdir="/rodeo/tmp/$dir" "/$dir" || error "Overlay-Mounting /$dir failed"
    fi
  done < /rodeo/configuration/managed_directories
}

transfer_directory() {
  dir="$1"
  mv "/$dir" "/rodeo/root/$dir"
  mkdir -p "/$dir" "/rodeo/tmp/$dir"
  echo "$dir" >> /rodeo/configuration/managed_directories
  info "Added $dir to the list of managed directories"
  initial_mount
}

shell() {
  target="$1"
  overlay_file=/rodeo/configuration/included_overlays
  command="/bin/sh"
  [ -n "$2" ] && overlay_file="$2"
  [ -n "$3" ] && command="$3"
  [ ! -d "$target" ] && mkdir -p "$target"
  info "Mounting Base Filesystems"
  mount --bind / "$target"
  mount --bind /dev "$target/dev"
  mount --bind /sys "$target/sys"
  mount --bind /proc "$target/proc"
  info "Mounting Overlays"
  chroot "$target" /sbin/rodeo-overlay.sh initial-mount "$overlay_file"
  info "Executing $command"
  chroot "$target" "$command"
}

[ "$1" = "transfer-directory" ] && transfer_directory "$2"
[ "$1" = "remount" ] && remount "$2"
[ "$1" = "initial-mount" ] && initial_mount "$2"
[ "$1" = "shell" ] && shell "$2" "$3" "$4"