zsh_setup.sh

#!/bin/bash

echo "Starting command: $0"

if [ "$#" -ne 2 ] || [ ! -d "$1" ]; then
    echo "Error: $0 [home folder path] [username]";
	echo "Exiting command: $0"
else
	home_folder=$1
	#Add / at the end if not present
	[[ $home_folder != *\/ ]] && home_folder+="/"

	username=$2

	change_shell() {
		# If this user's login shell is not already "zsh", attempt to switch.
		TEST_CURRENT_SHELL=$(expr "$SHELL" : '.*/\(.*\)')
		if [ "$TEST_CURRENT_SHELL" != "zsh" ]; then
			# If this platform provides a "chsh" command (not Cygwin), do it, man!
			if hash chsh >/dev/null 2>&1; then
			  printf "${BLUE}Time to change your default shell to zsh!${NORMAL}\n"
			  chsh -s $(which zsh)
			  usermod -s $(which zsh) $username
			# Else, suggest the user do so manually.
			else
			  printf "I can't change your shell automatically because this system does not have chsh.\n"
			  printf "${BLUE}Please manually change your default shell to zsh!${NORMAL}\n"
			fi
		else
			printf "${YELLOW}Zsh is already your default shell. Exiting.${NORMAL}\n"
		fi
	}

	main_ohmyzsh() {
	  # Use colors, but only if connected to a terminal, and that terminal
	  # supports them.
	  if which tput >/dev/null 2>&1; then
		  ncolors=$(tput colors)
	  fi
	  if [ -t 1 ] && [ -n "$ncolors" ] && [ "$ncolors" -ge 8 ]; then
		RED="$(tput setaf 1)"
		GREEN="$(tput setaf 2)"
		YELLOW="$(tput setaf 3)"
		BLUE="$(tput setaf 4)"
		BOLD="$(tput bold)"
		NORMAL="$(tput sgr0)"
	  else
		RED=""
		GREEN=""
		YELLOW=""
		BLUE=""
		BOLD=""
		NORMAL=""
	  fi

	  # Only enable exit-on-error after the non-critical colorization stuff,
	  # which may fail on systems lacking tput or terminfo
	  set -e

	  CHECK_ZSH_INSTALLED=$(grep /zsh$ /etc/shells | wc -l)
	  if [ ! $CHECK_ZSH_INSTALLED -ge 1 ]; then
		printf "${YELLOW}Zsh is not installed!${NORMAL} Please install zsh first!\n"
		exit
	  fi
	  unset CHECK_ZSH_INSTALLED

	  if [ ! -n "$ZSH" ]; then
		ZSH=$home_folder".oh-my-zsh"
	  fi

	  if [ -d "$ZSH" ]; then
		printf "${YELLOW}You already have Oh My Zsh installed.${NORMAL}\n"
		printf "You'll need to remove $ZSH if you want to re-install.\n"

		change_shell

		exit
	  fi

	  # Prevent the cloned repository from having insecure permissions. Failing to do
	  # so causes compinit() calls to fail with "command not found: compdef" errors
	  # for users with insecure umasks (e.g., "002", allowing group writability). Note
	  # that this will be ignored under Cygwin by default, as Windows ACLs take
	  # precedence over umasks except for filesystems mounted with option "noacl".
	  umask g-w,o-w

	  printf "${BLUE}Cloning Oh My Zsh...${NORMAL}\n"
	  hash git >/dev/null 2>&1 || {
		echo "Error: git is not installed"
		exit 1
	  }
	  # The Windows (MSYS) Git is not compatible with normal use on cygwin
	  if [ "$OSTYPE" = cygwin ]; then
		if git --version | grep msysgit > /dev/null; then
		  echo "Error: Windows/MSYS Git is not supported on Cygwin"
		  echo "Error: Make sure the Cygwin git package is installed and is first on the path"
		  exit 1
		fi
	  fi
	  env git clone --depth=1 https://github.com/robbyrussell/oh-my-zsh.git $ZSH || {
		printf "Error: git clone of oh-my-zsh repo failed\n"
		exit 1
	  }

	  printf "${BLUE}Looking for an existing zsh config...${NORMAL}\n"
	  if [ -f $home_folder".zshrc" ] || [ -h $home_folder".zshrc" ]; then
		printf "${YELLOW}Found $home_folder .zshrc.${NORMAL}\n";
	  else
		printf "${RED}Error: $home_folder .zshrc not found.${NORMAL}\n";
	  fi

	  change_shell

	  printf "${GREEN}"
	  echo '         __                                     __   '
	  echo '  ____  / /_     ____ ___  __  __   ____  _____/ /_  '
	  echo ' / __ \/ __ \   / __ `__ \/ / / /  /_  / / ___/ __ \ '
	  echo '/ /_/ / / / /  / / / / / / /_/ /    / /_(__  ) / / / '
	  echo '\____/_/ /_/  /_/ /_/ /_/\__, /    /___/____/_/ /_/  '
	  echo '                        /____/                       ....is now installed!'
	  echo ''
	  echo ''
	  printf "${NORMAL}"
	  env zsh
	}

	## ZSH

	apt-get install -y zsh
	main_ohmyzsh
	git clone https://github.com/bhilburn/powerlevel9k.git $home_folder".oh-my-zsh/custom/themes/powerlevel9k"

	dos2unix $home_folder".zshrc"

	printf "${GREEN}Exiting command: $0 ${NORMAL}\n"

fi