setup.sh

#!/bin/sh

function colour_echo() {
    COLOUR='\033[0;35m'
    ENDCOLOUR='\033[0m'
    if [[ $SHELL == "/bin/zsh" ]]; then
        echo -e "\n${COLOUR}$1${ENDCOLOUR}"
    else
        echo "\n$1"
    fi
}

function run_with_privileges() {
    type sudo &>/dev/null
    if [ $? -ne 0 ]; then
        sudo $1
    else
        $1
    fi
}

function shell_config_setup() {
    colour_echo "Backing up shell config"
    if [ -s ~/.zprofile ]; then
        mv ~/.zprofile ~/.dotfiles/archive/ &>/dev/null
    fi
    if [ -s ~/.bash_profile ]; then
        mv ~/.bash_profile ~/.dotfiles/archive/ &>/dev/null
    fi
    if [ -s ~/.zsh_prompt ]; then
        mv ~/.zsh_prompt ~/.dotfiles/archive/ &>/dev/null
    fi
    if [ -s ~/.bash_prompt ]; then
        mv ~/.bash_prompt ~/.dotfiles/archive/ &>/dev/null
    fi

    colour_echo "Configuring shell profile"
    case $SHELL in
    *'zsh'*)
        echo source ~/.dotfiles/shell/profile >> ~/.zshrc
        ;;
    *'bash'*)
        echo source ~/.dotfiles/shell/profile >> ~/.bashrc
        ;;
    esac
}

function create_custom_aliases() {
    colour_echo "Creating Custom Aliases file"
    custom_aliases=".custom_aliases"
    if [ -s ~/$custom_aliases ]; then
        touch ~/$custom_aliases
    fi
}

# function brew_setup() {
#     type brew &>/dev/null
#     if [ $? -ne 0 ]; then
#         colour_echo "Installing Brew"
#         /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
#         eval "$(/opt/homebrew/bin/brew shellenv)"
#         colour_echo "Finished installing Brew, installing Brew packages"
#         brew bundle --file ~/.dotfiles/brew/Brewfile
#     else
#         colour_echo "Skipping Brew, looks like it's already installed,
#         installing packages now"
#     fi
# }

# function git_setup() {
#     colour_echo "Configuring Git"
#     if [ -s ~/.gitconfig ]; then
#         mv ~/.gitconfig ~/.dotfiles/archive/ &>/dev/null
#     fi
#     ln -s ~/.dotfiles/git/gitconfig ~/.gitconfig
# }

# function work_git_setup() {
#     colour_echo "Copying gitconfig-extended into work folder"
#     mkdir -p ~/work/
#     if [ -s ~/work/.gitconfig-work ]; then
#         mv ~/work/.gitconfig-work ~/.dotfiles/archive/ &>/dev/null
#     fi
#     cp ~/.dotfiles/git/gitconfig-extended ~/work/.gitconfig-work
# }

# function personal_git_setup() {
#     colour_echo "Configuring personal gitconfig"
#     mkdir -p ~/personal/
#     if [ -s ~/personal/.gitconfig-personal ]; then
#         mv ~/personal/.gitconfig-personal ~/.dotfiles/archive/ &>/dev/null
#     fi
#     ln -s ~/.dotfiles/git/gitconfig-extended ~/personal/.gitconfig-personal

#     if [ ! -s ~/personal/dotfiles ]; then
#         ln -s ~/.dotfiles ~/personal/dotfiles
#     fi
# }

# function vim_setup() {
#     colour_echo "Configuring Vim"
#     if [ -s ~/.vimrc ]; then
#         mv ~/.vimrc ~/.dotfiles/archive/ &>/dev/null
#     fi
#     ln -s ~/.dotfiles/vim/vimrc ~/.vimrc
# }

# function vscode_setup() {
#     colour_echo "Configuring VS Code"
#     if [ -s ~/Library/Application\ Support/Code/User/settings.json ]; then
#         mv ~/Library/Application\ Support/Code/User/settings.json ~/.dotfiles/archive/ &>/dev/null
#     fi
#     ln -s ~/.dotfiles/vscode/settings.json ~/Library/Application\ Support/Code/User/settings.json

#     colour_echo "Installing VS Code extensions"
#     . ~/.dotfiles/vscode/extensions
# }

# function pyenv_setup() {
#     colour_echo "Configuring Pyenv"
#     run_with_privileges apt-get install make build-essential libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev -y
#     curl https://pyenv.run | bash
# }

# function npm_setup() {
#     colour_echo "Configuring NPM"
#     run_with_privileges apt-get install nodejs npm -y
#     colour_echo "Install Yarn"
#     run_with_privileges npm install -g yarn
#     colour_echo "Configuring NVM"
#     curl https://raw.githubusercontent.com/creationix/nvm/master/install.sh | bash
# }

# Create ssh and gpg folders
# function gpg_ssh_setup() {
#     colour_echo "Prepping GPG and SSH folders"
#     mkdir -p ~/.ssh
#     chmod 700 ~/.ssh
#     gpg --list-secret-keys --keyid-format LONG
#     mkdir -p ~/.gnupg
#     chmod 700 ~/.gnupg
#     echo "You may wantls to set a default"
#     echo "default-cache-ttl & max-cache-ttl"
#     echo "in ~/.gnupg/gpg-agent.conf"
#     echo "If not using any pin entry software"
# }

if [[ $(uname | tr '[:upper:]' '[:lower:]') = *linux* ]]; then
    mac=0
else
    mac=1
fi

colour_echo "Setting up dotfiles..."
if [[ $mac -eq 0 ]]; then
    colour_echo "Updating packages"
    apt-get &>/dev/null
    if [[ $? -eq 1 ]]; then
        run_with_privileges apt-get update -y
    else
        # Haven't tested this...
        run_with_privileges yum update -y
    fi
fi
shell_config_setup
if [[ $mac -eq 1 ]]; then
    brew_setup
fi
git_setup
personal_git_setup
vim_setup
if [[ $mac -eq 1 ]]; then
    vscode_setup
fi
if [[ $mac -eq 0 ]]; then
    pyenv_setup
    npm_setup
fi
gpg_ssh_setup
colour_echo "Setup Complete!"

if [[ "$1" == "work" ]]; then
    create_custom_aliases
    work_git_setup
fi

# Restart shell
exec $SHELL