Setup zsh with oh-my-zsh and the spaceship prompt

setup-zsh.sh

#!/bin/bash

# Install oh-my-zsh and the spaceship theme.
# This script is to be run by a provisioner as root.

# Change this to the appropriate user.
USER_NAME="vagrant"

ZSH_CUSTOM="/home/$USER_NAME/.oh-my-zsh/custom"

# install oh-my-zsh
runuser -l $USER_NAME -c 'sh -c "$(wget --quiet https://raw.githubusercontent.com/robbyrussell/oh-my-zsh/master/tools/install.sh -O -)"'

# install spaceship prompt
git clone https://github.com/denysdovhan/spaceship-prompt.git "$ZSH_CUSTOM/themes/spaceship-prompt"
ln -s "$ZSH_CUSTOM/themes/spaceship-prompt/spaceship.zsh-theme" "$ZSH_CUSTOM/themes/spaceship.zsh-theme"

# set the spaceship prompt as the theme
echo 'ZSH_THEME="spaceship"' > ${ZSH_CUSTOM}/theme.zsh

echo -e '# run `ls -lah` when readline receives a single return character (and nothing else)
accept-line() {: "${BUFFER:="ls -lah"}"; zle ".$WIDGET"}
zle -N accept-line' > ${ZSH_CUSTOM}/lazy-list.zsh

# use zsh as the default shell for the user
chsh -s $(which zsh) $USER_NAME