#!/bin/bash

################################################################################
#Copyright (C) 2022 Tony Grosinger. All rights reserved.
#
#Description:
# This script can be run in two modes depending on the file structure of the
# current working directory or the arguments passed to it.
#
# Mode 1: From within a cloned repository or downloaded snapshot with all files
#         already present and ready to be symlinked
#
# Mode 2: Standalone or through Curl in which case all required files will be
#         downloaded automatically to `~/.dotfiles` before creating symlinks.
#
################################################################################

DOTFILES_DIR="${HOME}/.dotfiles"
GIT_REPO_BASE="https://github.com/tgrosinger/dotfiles"
GIT_REPO="${GIT_REPO_BASE}.git"
REPO_TAR="${GIT_REPO_BASE}/archive/master.tar.gz"

################################################################################
# Define functions
################################################################################

# Remove a file if it exists then create a symlink to the one contained in ${DOTFILES_DIR}
function linkFile() {
    ln -fns ${DOTFILES_DIR}/$1 $1;
}

# Create a directory named by first parameter. Delete directory first if it already exists.
function createDirectory() {
    if [ -d $1 ]; then rm -rf $1; fi
    mkdir $1
}

# Perform the actual work of copying files and creating symlinks.
# Requires a home directory argument.
function performSetup() {
    if [ ${#} != 1 ]; then
        echo "Must pass home directory to ${FUNCNAME}"
        exit 1
    fi
    home=${1}

    echo "Moving to Home directory..."
    pushd ${home} > /dev/null

    echo "Linking shell configs..."
    linkFile ".bashrc"

    echo "Linking vim..."
    linkFile ".vim"
    linkFile ".vimrc"

    echo "Linking Git..."
    linkFile ".gitconfig"
    linkFile ".gitattributes"

    echo "Linking LazyGit..."
    mkdir -p ~/.config/lazygit
    ln -fns ${DOTFILES_DIR}/.lazygit ~/.config/lazygit/config.yml

    echo "Linking tmux..."
    linkFile ".tmux.conf"

    echo "Linking inputrc..."
    linkFile ".inputrc"

    echo "Installing tmux plugins"
    git clone https://github.com/tmux-plugins/tpm ~/.tmux/plugins/tpm

    echo "Linking gitmux..."
    linkFile ".gitmux.conf"

    echo "Linking neovim config"
    ln -fns ${DOTFILES_DIR}/nvim ~/.config/nvim

    if [[ "${OSTYPE}" == "darwin"* ]]; then
        brew install rsync rga pandoc poppler tesseract tmux ledger 
    else
        sudo apt install visidata ripgrep
    fi

    popd > /dev/null
}

################################################################################
# Initialize (Determine mode)
################################################################################

# Determine which mode we are running in. (Does the install script exist in the dotfiles directory?)
if [ -f ${DOTFILES_DIR}/install.sh ]; then
    echo "Running in Mode 1: Already cloned repository"
    performSetup ${HOME}
    echo "Done! Restart your shell to see changes"
else
    echo "Running in Mode 2: Direct from Curl"

    # Make sure git is installed. Exit if it isn't.
    if which git; then
        # Git is installed, clone the repository using read-only url
        if [ -d ${DOTFILES_DIR} ]; then
            if [ -d ${DOTFILES_DIR}/.git ]; then
                echo "Updating dotfiles repository in ${DOTFILES_DIR}"
                pushd ${DOTFILES_DIR} > /dev/null
                git pull > /dev/null
                popd > /dev/null
            else
                echo "Cloning dotfiles repository to ${DOTFILES_DIR}"
                rm -rf ${DOTFILES_DIR}
                git clone ${GIT_REPO} ${DOTFILES_DIR}
            fi
        else
            echo "Cloning dotfiles repository to ${DOTFILES_DIR}"
            git clone ${GIT_REPO} ${DOTFILES_DIR}
        fi
    else
        # Git is not installed, download the zip and extract
        if ! which wget; then
            echo "You must have either git or wget installed. Please install one before continuing."
            exit 1
        fi

        if [ -d ${DOTFILES_DIR} ]; then
            rm -rf ${DOTFILES_DIR}
        fi

        mkdir ${DOTFILES_DIR}
        curl -L ${REPO_TAR} | tar zx --strip 1 --directory ${DOTFILES_DIR}
    fi

    performSetup ${HOME}
    echo "Done! Restart your shell to see changes"
fi