#!/bin/bash

path=`pwd` 
if [ -d "$1" ]; then 
path="$1" 
fi

for i in dev proc run sys; do 
  if [ ! -d "$path/$i" ]; then
    echo "Error: Missing directory: $path/$i"; 
    echo "Path to chroot must have system directories: /dev, /proc, /run, /sys"
    exit 1;
  fi
done

sudo echo ""
if [ $? -ne 0 ]; then echo "Error"; exit 1; fi

sudo mount --rbind /dev  "$path/dev"
sudo mount --rbind /run  "$path/run"
sudo mount --rbind /sys  "$path/sys"
sudo mount -t proc proc  "$path/proc"

sudo cp /etc/resolv.conf "$path/etc/resolv.conf"

#xhost +local: 
tput reset

echo "====================================================="
echo "Entering chroot environment... "
echo "====================================================="
#echo "Network and display sharing is enabled. "
#echo "You can connect to the internet, and run GUI apps."
echo "Type 'exit' to quit the session"
echo ""

#sudo chroot "$path" /bin/bash -c "export DISPLAY=$DISPLAY"
sudo chroot "$path"

sync

sudo umount --lazy --force --recursive "$path/dev"
sudo umount --lazy --force --recursive "$path/run"
sudo umount --lazy --force --recursive "$path/sys"
sudo umount --lazy --force --recursive "$path/proc"

echo "==============================="
echo "Exited chroot environment"
echo "==============================="
echo ""
exit $?
