#!/usr/bin/env bash
# 
# Be careful what you are doing with this script. You may overwrite files.
# cd into the repository before executing the script!
set -e;

usage() {
	echo "Script to synchronize dotfiles-system skel files";
	echo "";
	echo "(1) /etc/skel --> ~/... ALL users (with given shell)";
	echo "";
	echo "(2) /etc/skel --> ~/specific user";
	echo "";  
	echo "(3) list line all users with given shell, e.g. zsh";
	echo "";
	echo "(0) or [CTRL+C] to exit";
}

menu() {
	usage;
	read -n 1 -e -p ">" menuInput
	echo "";

	if [ "$menuInput" = "1" ]; then
		# require root
		[[ $UID -eq 0 ]] || (>&2 echo "ERROR: need to be root!" && exit 1)	

		# determine shell
		echo "Which shell (shortname, e.g. zsh)?";
		read -e -p ">" shellInput
		echo "";

		# shell users
		users=$(cat /etc/passwd|grep $shellInput| awk -F':' '{ print $1 }');

		# stage
		tmpDir=/tmp/dotfiles-system-sync/;
		mkdir -p $tmpDir;
		srcDir=/etc/skel/.
		cp -R $srcDir $tmpDir;

		# copy   
		for user in $users; do
		  exists=$(id -u $user);
		  primaryGroup=$(id -gn $user);
		  chown -R $user:$primaryGroup $tmpDir;
		  trgDir=$(eval echo ~$user/);
		  echo "Copy from $tmpDir to $trgDir";
		  cp -Rp $tmpDir/. $trgDir/;
		done

		# remove stage
		rm -rf $tmpDir;
	elif [ "$menuInput" = "2" ]; then
		# determine shell
		echo "Which user?";
		read -e -p ">" user
		echo "";

		# stage
		tmpDir=/tmp/dotfiles-system-sync/;
		mkdir -p $tmpDir;
		srcDir=/etc/skel/.
		cp -R $srcDir $tmpDir;

		# copy   
		exists=$(id -u $user);

		if [ $UID -ne $exists ] && [ $UID -ne 0 ]; then
		  echo "ERROR: need to be $user or root!";
		  exit 1;
		fi

		primaryGroup=$(id -gn $user);
		chown -R $user:$primaryGroup $tmpDir;
		trgDir=$(eval echo ~$user/);
		echo "Copy from $tmpDir to $trgDir";
		cp -Rp $tmpDir/. $trgDir/;

		# remove stage
		rm -rf $tmpDir;

	elif [ "$menuInput" = "3" ]; then
	    echo "Which shell (shortname, e.g. zsh)?";
	    read -e -p ">" shellInput
	    echo "";

	    # shell users
	    users=$(cat /etc/passwd|grep $shellInput| awk -F':' '{ print $1 }');
	    echo "Users with shell $shellInput"

	    for u in $users; do
	      echo $u;
	    done
	elif [ "$menuInput" = "0" ]; then
		exit 0;
  fi

  menu;
}

menu;