#!/bin/sh

# Copyright (c) 2017 rehashedsalt/vintagesalt
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
# 
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
# 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

# POSIX NOTICE
# This script should be fully POSIX-compliant
# If it is not, open a bug report at gitlab.com/rehashedsalt/home and I'll take
# care of it

# Environment variables
# Use Vim if we have it
export EDITOR="vi"
which vim > /dev/null 2>&1 && export EDITOR="vim"

# Patch PATH
desiredpath="/usr/local/bin:/usr/local/sbin:/usr/bin:/usr/sbin:/bin:/sbin:/usr/local/games:/usr/games:$HOME/.local/bin:$HOME/.local/bin/pip"
# Here, we eliminate the parts of the environment-supplied PATH that we already
# plan on using
IFS=":"
for path in $desiredpath; do
	PATH="${PATH#*$path}"
done
unset IFS
# We remove any prepending colons to make it appear neater
while [ "${PATH#:}" != "$PATH" ]; do
	PATH="${PATH#:}"
done
# Then finally set up our path
PATH="$desiredpath:$PATH"

# Grab bash_completion, if it exists
[ -f "/etc/profile.d/bash_completion.sh" ] && . "/etc/profile.d/bash_completion.sh"

# Grab pip completion, if it exists
if which pip > /dev/null 2>&1; then
	case "$(basename "$SHELL")" in
		*bash)
			if ! [ -f "$HOME/.pip-completion-bash" ]; then
				pip completion --bash > "$HOME/.pip-completion-bash"
				echo ".profile: Created pip completion for bash"
			fi
			source "$HOME/.pip-completion-bash"
			;;
		zsh)
			if ! [ -f "$HOME/.pip-completion-zsh" ]; then
				pip completion --zsh > "$HOME/.pip-completion-zsh"
				echo ".profile: Created pip completion for zsh"
			fi
			source "$HOME/.pip-completion-zsh"
			;;
		*)
			;;
	esac
	# Debian patches pip to assume the --user flag by default
	# This breaks using installation targets and prefixes
	# It's really dumb
	pipargs="--system"
	alias pip="pip $pipargs"
fi

# If emerge exists, add emerge aliases
if `which emerge > /dev/null 2>&1` ; then
	alias e-depclean='sudo emerge -a --depclean'
	alias e-inst='sudo emerge -a --jobs --tree --quiet-build y'
	alias e-upgrade='sudo emerge -DNUua --jobs --tree --quiet-build y --with-bdeps=y --keep-going --backtrack=1000 @world'
	alias e-newuse='sudo emerge -Uva --jobs --tree --quiet-build y @world'
	alias e-search='emerge -s'
	alias e-sync='sudo emerge --sync'
	if `which eclean > /dev/null 2>&1`; then
		alias e-cleanup='sudo eclean -d distfiles && sudo eclean -d packages'
	fi
fi

# Alias for the dotfile setup
if [ -d "$HOME/.dotfiles" ]; then
	dotcmd='git --git-dir=$HOME/.dotfiles/ --work-tree=$HOME'
	alias dot="$dotcmd"
	alias dotupdate="\
		printf 'Pulling...\n'; \
		$dotcmd pull
		printf 'Updating submodules...\n'; \
		$dotcmd submodule update --init; \
		printf 'Checking out masters...\n'; \
		$dotcmd submodule -q foreach --recursive 'git checkout -q master && git pull' && \
		$dotcmd status"
fi

# Aliases for common utilities
if [ "`uname`" = "Linux" ]; then
	# Assume we have GNU coreutils
	lsarguments='--color=auto --group-directories-first'
	alias l="ls -CF  --file-type $lsarguments"
	alias la="ls -AF  --file-type $lsarguments"
	alias ls="ls -F $lsarguments"
	alias ll="ls -AhlF  --file-type $lsarguments"
	unset lsarguments

	alias rm='rm -I'
else
	# Else only assume POSIX/BSD
	lsarguments='-F'
	alias l="ls -$lsarguments"
	alias la="ls -A $lsarguments"
	alias ls="ls $lsarguments"
	alias ll="ls -Ahl $lsarguments"
fi

alias cp='cp -i'

# Miscellaneous aliases
alias todo='$EDITOR "$HOME/Documents/todo"'
alias waitwhat='echo $?'

# Source ~/.functions, if it exists
[ -r "$HOME/.functions" ] && . "$HOME/.functions"

# Set up a default PS1
# This *should* work for all terminals. I know it works on ksh
function _ps1() {
	exitcode="$?"
	r="\e[0m"
	fg_blue="\e[34m"
	fg_cyan="\e[36m"
	fg_red="\e[31m"
	fg_green="\e[32m"
	fg_yellow="\e[33m"
	fg_grey="\e[37m"
	fg_bold="\e[1m"

	# Add hostname prefix in SSH sessions
	if [ "$SSH_CLIENT" ]; then
		prefix="${fg_bold}${fg_red}`hostname`${r}${fg_red}:"
	fi
	# Append a "restricted" prefix in rbash
	if [ "$0" == "rbash" ]; then
		prefix="${fg_bold}${fg_grey}rbash${r}:"
	fi
	# Change PWD color depending on the shell
	case $0 in
		*bash)
			prefix="${prefix}${fg_yellow}"
			;;
		ksh)
			prefix="${prefix}${fg_green}"
			;;
		sh)
			prefix="${prefix}${fg_grey}"
			;;
		*)
			;;
	esac
	# Show the tilde instead of $HOME
	cpwd="${PWD/#$HOME/\~}"
	# Alert us if the last command failed
	fail=""
	if ! [ "$exitcode" = "0" ]; then
		fail="${fg_bold}${fg_red}?"
	fi
	printf "[${prefix}${cpwd}${r}]${fail}${r}${fg_green}\$${r} "
}

if ! [ "$ZSH_NAME" ]; then
	# I've got a different ZSH theme
	export PS1='`_ps1`'
fi