install

#!/usr/bin/env bash
set -euo pipefail

cd "$(dirname "${BASH_SOURCE[0]}")"
source ./install_utils.sh

task_sh() {
	[[ -z ${_TASK_SH+x} ]] || return 0
	[[ -n ${1+x} ]] || log_util 'TASK' 'sh'
	ensure_ln_s ./sh/inputrc ~/.inputrc
	ensure_ln_s ./sh/profile ~/.profile
	_TASK_SH=true
}

task_home() {
	log_util 'TASK' 'home'
	ensure_mkdir  ~/.config
	ensure_mkdir ~/desktop
	ensure_mkdir ~/doc
	ensure_mkdir ~/tmp
	ensure_mkdir ~/doc/templates
	ensure_mkdir ~/media/music
	ensure_mkdir ~/media/images
	ensure_mkdir ~/media/videos
	ensure_mkdir ~/public
	ensure_ln_s ./xdg/user-dirs.dirs ~/.config/user-dirs.dirs
	ensure_ln_s ./xdg/user-dirs.locale ~/.config/user-dirs.locale
}

task_vim() {
	log_util 'TASK' 'vim'
	ensure_mkdir ~/.vim
	ensure_mkdir ~/.vim/backups
	ensure_ln_s ./vim/vimrc ~/.vimrc

	if is_installed nvim; then
		local vim=nvim
	else
		local vim=vim
		ensure_pkg vim || { log_error "vim is required to install vim-plug"; return 1; }
	fi
	ensure_pkg git || { log_error "git is required to install vim-plug"; return 1; }
	ensure_pkg curl || { log_error "curl is required to install vim-plug"; return 1; }

	if [ ! -f ~/.vim/autoload/plug.vim ]; then
		log_info "Installing vim-plug"
		curl -sSfL -o ~/.vim/autoload/plug.vim --create-dirs \
			    https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim \
				|| { log_error "Failed to download plug.vim"; return 1; }
	fi

	log_info "Running PlugUpdate"
	$vim +PlugClean +PlugUpdate +qa
	ensure_ln_s ./vim/after ~/.vim/after
}

task_neovim() {
	log_util 'TASK' 'neovim'
	ensure_mkdir ~/.config/nvim
	ensure_ln_s ./vim/vimrc ~/.config/nvim/init.vim
	ensure_ln_s ./vim/lua ~/.config/nvim/lua

	ensure_pkg nvim neovim || { log_error "neovim is required to install vim-plug"; return 1; }
	ensure_pkg git || { log_error "git is required to install vim-plug"; return 1; }
	ensure_pkg curl || { log_error "curl is required to install vim-plug"; return 1; }

	ensure_mkdir ~/.local/share/nvim/site/autoload
	if [ ! -f ~/.local/share/nvim/site/autoload/plug.vim ]; then
		log_info "Installing vim-plug"
		curl -sSfL -o ~/.local/share/nvim/site/autoload/plug.vim --create-dirs \
				https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim \
				|| { log_error "Failed to download plug.vim"; return 1; }
	fi
	log_info "Running PlugUpdate"
	nvim +PlugClean +PlugUpdate +qa
	ensure_ln_s ./vim/after ~/.config/nvim/after
}

task_zsh() {
	log_util 'TASK' 'zsh'
	task_sh true
	ensure_ln_s ./bash/common.sh ~/.common.sh
	ensure_ln_s ./zsh/zlogin ~/.zlogin
	ensure_ln_s ./zsh/zshrc ~/.zshrc

	ensure_pkg zsh

	local shell_bin
	shell_bin="$(grep zsh /etc/shells | head -n 1)"
	if [ ! "$SHELL" = "$shell_bin" ]; then
		local chsh_cmd="chsh -s $shell_bin"
		if prompt_yn "Change login shell to zsh?"; then
			log_info "Changing login shell to zsh"
			eval "$chsh_cmd" || log_warn "${chsh_cmd} failed, please do this manually"
		else
			log_info "Shell can be changed later by running '${chsh_cmd}'"
		fi
	else
		log_info "$shell_bin is already the default shell"
	fi
}

task_bash() {
	log_util 'TASK' 'bash'
	task_sh true
	ensure_ln_s ./bash/common.sh ~/.common.sh
	ensure_ln_s ./bash/bash_login ~/.bash_login
	ensure_ln_s ./bash/bashrc ~/.bashrc
}

task_screen() {
	log_util 'TASK' 'screen'
	ensure_ln_s ./screen/screenrc ~/.screenrc
	ensure_pkg screen
}

task_git() {
	log_util 'TASK' 'git'
	ensure_ln_s ./git/gitconfig.defaults ~/.gitconfig.defaults
	ensure_ln_s ./git/gitignore.global ~/.gitignore.global
	ensure_ln_s ./git/gitattributes.global ~/.gitattributes.global
	#ensure_ln_s ./git/git-p4 /usr/lib/git-core/git-p4
	ensure_mkdir ~/bin
	ensure_pkg delta git-delta

	if [[ -L ~/.gitconfig ]]; then
		log_warn "~/.gitconfig appears to be a symlink, perhaps from an old install..."
		if prompt_yn "Remove ~/.gitconfig symlink and replace with a new one which includes ~/.gitconfig.defaults (the new symlink)?"; then
			log_info "Removing ~/.gitconfig"
			rm ~/.gitconfig
			if [ $? -ne 0 ]; then
				log_error "Failed to remove ~/.gitconfig"
				exit 1
			fi
			if [[ -f ~/.gitconfig.local ]] && prompt_yn "Use existing ~/.gitconfig.local as base for new ~/.gitconfig?"; then
				log_info "Moving ~/.gitconfig.local to ~/.gitconfig"
				mv ~/.gitconfig.local ~/.gitconfig
				if [ $? -ne 0 ]; then
					log_error "Failed to move ~/.gitconfig.local to ~/.gitconfig"
					return 1
				fi
			fi
		else
			log_error "Failed to install git"
			exit 1
		fi
	fi

	log_info "Ensuring that ~/.gitconfig includes ~/.gitconfig.defaults"
	if ! git config --get include.path '~/.gitconfig.defaults' > /dev/null; then
		[[ -s ~/.gitconfig ]] || echo > ~/.gitconfig
		sed -i '1s|^|[include]\n\tpath = ~/.gitconfig.defaults\n|' ~/.gitconfig
	fi

	ensure_pkg git
	ensure_pkg git-lfs
	git lfs install
}

task_icdiff() {
	log_util 'TASK' 'icdiff'
	ensure_pkg curl || { log_error "curl is required to install icdiff"; return 1; }
	ensure_mkdir ~/bin
	curl -sSf -o ~/bin/icdiff https://raw.githubusercontent.com/jeffkaufman/icdiff/release-1.9.5/icdiff
	curl -sSf -o ~/bin/git-icdiff https://raw.githubusercontent.com/jeffkaufman/icdiff/release-1.9.5/git-icdiff
	chmod ugo+rx ~/bin/icdiff
	chmod ugo+rx ~/bin/git-icdiff
}

task_tmux() {
	log_util 'TASK' 'tmux'
	ensure_mkdir ~/.tmux
	ensure_ln_s ./tmux/tmux.conf ~/.tmux.conf
	ensure_pkg tmux
}

task_gnupg() {
	log_util 'TASK' 'gnupg'
	ensure_mkdir ~/.gnupg
	ensure_ln_s ./gnupg/gpg.conf ~/.gnupg/gpg.conf
	ensure_pkg gpg gnupg
}

task_pgcli() {
	log_util 'TASK' 'pgcli'
	ensure_mkdir ~/.config/pgcli
	ensure_ln_s ./pgcli/config ~/.config/pgcli/config
	ensure_pkg pgcli
}

task_efmlangserver() {
	log_util 'TASK' 'efm-langserver'
	ensure_mkdir ~/.config/efm-langserver
	ensure_ln_s ./efm-langserver/config.yaml ~/.config/efm-langserver/config.yaml
	ensure_pkg efm-langserver
}

task_wezterm() {
	log_util 'TASK' 'wezterm'
	ensure_mkdir ~/.config/wezterm
	ensure_ln_s ./wezterm/wezterm.lua ~/.config/wezterm/wezterm.lua
	ensure_pkg wezterm
}

task_isobit_tools() {
	log_util 'TASK' 'isobit-tools'
	ensure_pkg curl || { log_error "curl is required to install isobit tools"; return 1; }
	ensure_pkg jq || { log_error "jq is required to install isobit tools"; return 1; }
	ensure_mkdir ~/bin
	ndog_latest="$(curl -sSfL -H 'accept: application/json' "https://github.com/isobit/ndog/releases/latest" | jq -r .tag_name)"
	pgt_latest="$(curl -sSfL -H 'accept: application/json' "https://github.com/isobit/pgt/releases/latest" | jq -r .tag_name)"
	tmpl_latest="$(curl -sSfL -H 'accept: application/json' "https://github.com/isobit/tmpl/releases/latest" | jq -r .tag_name)"
	ensure_download ~/bin/ndog "https://github.com/isobit/ndog/releases/download/${ndog_latest}/ndog-linux-amd64"
	ensure_download ~/bin/pgt "https://github.com/isobit/pgt/releases/download/${pgt_latest}/pgt-linux-amd64"
	ensure_download ~/bin/tmpl "https://github.com/isobit/tmpl/releases/download/${tmpl_latest}/tmpl-linux-amd64"
	chmod ugo+rx ~/bin/ndog
	chmod ugo+rx ~/bin/pgt
	chmod ugo+rx ~/bin/tmpl
}

DEFAULT_TASKS="home sh vim nvim bash zsh git screen gnupg isobit-tools"

while true; do
	case "${1:-}" in
		-h | --help)
			echo "Arguments should be a list of install tasks. If none are passed, the following defaults will be used:"
			echo "    $DEFAULT_TASKS"
			exit 1
			;;
		-y | --install-pkg-noconfirm)
			export INSTALL_PKG_NOCONFIRM=true
			shift
			;;
		--) # End of all options
			shift
			break
			;;
		-*)
			log_error "Unknown option: ${1:-}"
			break
			;;
		*)  # No more options
			break
			;;
	esac
done

if [[ $# -eq 0 ]]; then
	TASKS="$DEFAULT_TASKS"
	log_info "Installing default tasks"
else
	TASKS="$*"
fi

for task in $TASKS; do
	case $task in
		sh)
			task_sh
			;;
		home | xdg)
			task_home
			;;
		vim)
			task_vim
			;;
		neovim | nvim)
			task_neovim
			task_efmlangserver
			;;
		zsh)
			task_zsh
			;;
		bash)
			task_bash
			;;
		screen)
			task_screen
			;;
		git)
			task_git
			;;
		icdiff)
			task_icdiff
			;;
		tmux)
			task_tmux
			;;
		gnupg | gpg)
			task_gnupg
			;;
		pgcli)
			task_pgcli
			;;
		efm-langserver)
			task_efmlangserver
			;;
		wezterm)
			task_wezterm
			;;
		isobit-tools)
			task_isobit_tools
			;;
		*)
			log_error "Unknown task '$task'"
			;;
	esac
done