# shellcheck shell=bash

# Atomic Bash Prompt for Bash-it
# By lfelipe base on the theme brainy of MunifTanjim

############
## Colors ##
############
IRed="\e[1;49;31m"
IGreen="\e[1;49;32m"
IYellow="\e[1;49;33m"
IWhite="\e[1;49;37m"
BIWhite="\e[1;49;37m"
BICyan="\e[1;49;36m"

#############
## Symbols ##
#############
Line="\342\224\200"
LineA="\342\224\214\342\224\200"
SX="\342\234\227"
LineB="\342\224\224\342\224\200\342\224\200"
Circle="\342\227\217"
Face="\342\230\273"

#############
## Parsers ##
#############

____atomic_top_left_parse() {
	ifs_old="${IFS}"
	IFS="|"
	read -r -a args <<< "$@"
	IFS="${ifs_old}"
	if [ -n "${args[3]}" ]; then
		_TOP_LEFT+="${args[2]}${args[3]}"
	fi
	_TOP_LEFT+="${args[0]}${args[1]}"
	if [ -n "${args[4]}" ]; then
		_TOP_LEFT+="${args[2]}${args[4]}"
	fi
	_TOP_LEFT+=""
}

____atomic_top_right_parse() {
	ifs_old="${IFS}"
	IFS="|"
	read -r -a args <<< "$@"
	IFS="${ifs_old}"
	_TOP_RIGHT+=" "
	if [ -n "${args[3]}" ]; then
		_TOP_RIGHT+="${args[2]}${args[3]}"
	fi
	_TOP_RIGHT+="${args[0]}${args[1]}"
	if [ -n "${args[4]}" ]; then
		_TOP_RIGHT+="${args[2]}${args[4]}"
	fi
	__TOP_RIGHT_LEN=$((__TOP_RIGHT_LEN + ${#args[1]} + ${#args[3]} + ${#args[4]} + 1))
	((__SEG_AT_RIGHT += 1))
}

____atomic_bottom_parse() {
	ifs_old="${IFS}"
	IFS="|"
	read -r -a args <<< "$@"
	IFS="${ifs_old}"
	_BOTTOM+="${args[0]}${args[1]}"
	[ ${#args[1]} -gt 0 ] && _BOTTOM+=" "
}

____atomic_top() {
	_TOP_LEFT=""
	_TOP_RIGHT=""
	__TOP_RIGHT_LEN=0
	__SEG_AT_RIGHT=0

	for seg in ${___ATOMIC_TOP_LEFT}; do
		info="$(___atomic_prompt_"${seg}")"
		[ -n "${info}" ] && ____atomic_top_left_parse "${info}"
	done

	___cursor_right="\e[500C"
	_TOP_LEFT+="${___cursor_right}"

	for seg in ${___ATOMIC_TOP_RIGHT}; do
		info="$(___atomic_prompt_"${seg}")"
		[ -n "${info}" ] && ____atomic_top_right_parse "${info}"
	done

	[ $__TOP_RIGHT_LEN -gt 0 ] && __TOP_RIGHT_LEN=$((__TOP_RIGHT_LEN - 0))
	___cursor_adjust="\e[${__TOP_RIGHT_LEN}D"
	_TOP_LEFT+="${___cursor_adjust}"

	printf "%s%s" "${_TOP_LEFT}" "${_TOP_RIGHT}"
}

____atomic_bottom() {
	_BOTTOM=""
	for seg in $___ATOMIC_BOTTOM; do
		info="$(___atomic_prompt_"${seg}")"
		[ -n "${info}" ] && ____atomic_bottom_parse "${info}"
	done
	printf "\n%s" "${_BOTTOM}"
}

##############
## Segments ##
##############

___atomic_prompt_user_info() {
	color=$white
	box="${normal}${LineA}\$([[ \$? != 0 ]] && echo \"${BIWhite}[${IRed}${SX}${BIWhite}]${normal}${Line}\")${Line}${BIWhite}[|${BIWhite}]${normal}${Line}"
	info="${IYellow}\u${IRed}@${IGreen}\h"

	printf "%s|%s|%s|%s" "${color}" "${info}" "${white}" "${box}"
}

___atomic_prompt_dir() {
	color=${IRed}
	box="[|]${normal}"
	info="\w"
	printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white}" "${box}"
}

___atomic_prompt_scm() {
	[ "${THEME_SHOW_SCM}" != "true" ] && return
	color=$bold_green
	box="${Line}[${IWhite}$(scm_char)] "
	info="$(scm_prompt_info)"
	printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white}" "${box}"
}

___atomic_prompt_python() {
	[ "${THEME_SHOW_PYTHON}" != "true" ] && return
	color=$bold_yellow
	box="[|]"
	info="$(python_version_prompt)"
	printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_blue}" "${box}"
}

___atomic_prompt_ruby() {
	[ "${THEME_SHOW_RUBY}" != "true" ] && return
	color=$bold_white
	box="[|]"
	info="rb-$(ruby_version_prompt)"
	printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_red}" "${box}"
}

___atomic_prompt_todo() {
	[ "${THEME_SHOW_TODO}" != "true" ] \
		|| [ -z "$(which todo.sh)" ] && return
	color=$bold_white
	box="[|]"
	info="t:$(todo.sh ls | grep -E "TODO: [0-9]+ of ([0-9]+)" | awk '{ print $4 }')"
	printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_green}" "${box}"
}

___atomic_prompt_clock() {
	[ "${THEME_SHOW_CLOCK}" != "true" ] && return
	color=$THEME_CLOCK_COLOR
	box="[|]"
	info="$(date +"${THEME_CLOCK_FORMAT}")"
	printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white}" "${box}"
}

___atomic_prompt_battery() {
	! _command_exists battery_percentage \
		|| [ "${THEME_SHOW_BATTERY}" != "true" ] \
		|| [ "$(battery_percentage)" = "no" ] && return

	batp=$(battery_percentage)
	if [ "$batp" -eq 50 ] || [ "$batp" -gt 50 ]; then
		color=$bold_green
	elif [ "$batp" -lt 50 ] && [ "$batp" -gt 25 ]; then
		color=$bold_yellow
	elif [ "$batp" -eq 25 ] || [ "$batp" -lt 25 ]; then
		color=$IRed
	fi
	box="[|]"
	ac_adapter_connected && info="+"
	ac_adapter_disconnected && info="-"
	info+=$batp
	[ "$batp" -eq 100 ] || [ "$batp" -gt 100 ] && info="AC"
	printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white}" "${box}"
}

___atomic_prompt_exitcode() {
	[ "${THEME_SHOW_EXITCODE}" != "true" ] && return
	color=$bold_purple
	[ "$exitcode" -ne 0 ] && printf "%s|%s" "${color}" "${exitcode}"
}

___atomic_prompt_char() {
	color=$white
	prompt_char="${__ATOMIC_PROMPT_CHAR_PS1}"
	if [ "${THEME_SHOW_SUDO}" == "true" ]; then
		if sudo -vn 1> /dev/null 2>&1; then
			prompt_char="${__ATOMIC_PROMPT_CHAR_PS1_SUDO}"
		fi
	fi
	printf "%s|%s" "${color}" "${prompt_char}"
}

#########
## cli ##
#########

__atomic_show() {
	typeset _seg=${1:-}
	shift
	export "THEME_SHOW_${_seg}"=true
}

__atomic_hide() {
	typeset _seg=${1:-}
	shift
	export "THEME_SHOW_${_seg}"=false
}

_atomic_completion() {
	local cur _action actions segments
	COMPREPLY=()
	cur="${COMP_WORDS[COMP_CWORD]}"
	_action="${COMP_WORDS[1]}"
	actions="show hide"
	segments="battery clock exitcode python ruby scm sudo todo"
	case "${_action}" in
		show | hide)
			# shellcheck disable=SC2207
			COMPREPLY=($(compgen -W "${segments}" -- "${cur}"))
			return 0
			;;
	esac

	# shellcheck disable=SC2207
	COMPREPLY=($(compgen -W "${actions}" -- "${cur}"))
	return 0
}

atomic() {
	typeset action=${1:-}
	shift
	typeset segs=${*:-}
	typeset func
	case $action in
		show)
			func=__atomic_show
			;;
		hide)
			func=__atomic_hide
			;;
	esac
	for seg in ${segs}; do
		seg=$(printf "%s" "${seg}" | tr '[:lower:]' '[:upper:]')
		$func "${seg}"
	done
}

complete -F _atomic_completion atomic

###############
## Variables ##
###############

export SCM_THEME_PROMPT_PREFIX=""
export SCM_THEME_PROMPT_SUFFIX=""

export RBENV_THEME_PROMPT_PREFIX=""
export RBENV_THEME_PROMPT_SUFFIX=""
export RBFU_THEME_PROMPT_PREFIX=""
export RBFU_THEME_PROMPT_SUFFIX=""
export RVM_THEME_PROMPT_PREFIX=""
export RVM_THEME_PROMPT_SUFFIX=""

export SCM_THEME_PROMPT_DIRTY=" ${bold_red}✗${normal}"
export SCM_THEME_PROMPT_CLEAN=" ${bold_green}✓${normal}"

THEME_SHOW_SUDO=${THEME_SHOW_SUDO:-"true"}
THEME_SHOW_SCM=${THEME_SHOW_SCM:-"true"}
THEME_SHOW_RUBY=${THEME_SHOW_RUBY:-"false"}
THEME_SHOW_PYTHON=${THEME_SHOW_PYTHON:-"false"}
THEME_SHOW_CLOCK=${THEME_SHOW_CLOCK:-"true"}
THEME_SHOW_TODO=${THEME_SHOW_TODO:-"false"}
THEME_SHOW_BATTERY=${THEME_SHOW_BATTERY:-"true"}
THEME_SHOW_EXITCODE=${THEME_SHOW_EXITCODE:-"false"}

THEME_CLOCK_COLOR=${THEME_CLOCK_COLOR:-"${BICyan}"}
THEME_CLOCK_FORMAT=${THEME_CLOCK_FORMAT:-"%a %b %d - %H:%M"}

__ATOMIC_PROMPT_CHAR_PS1=${THEME_PROMPT_CHAR_PS1:-"${normal}${LineB}${bold_white}${Circle}"}
__ATOMIC_PROMPT_CHAR_PS2=${THEME_PROMPT_CHAR_PS2:-"${normal}${LineB}${bold_white}${Circle}"}

__ATOMIC_PROMPT_CHAR_PS1_SUDO=${THEME_PROMPT_CHAR_PS1_SUDO:-"${normal}${LineB}${bold_red}${Face}"}
__ATOMIC_PROMPT_CHAR_PS2_SUDO=${THEME_PROMPT_CHAR_PS2_SUDO:-"${normal}${LineB}${bold_red}${Face}"}

___ATOMIC_TOP_LEFT=${___ATOMIC_TOP_LEFT:-"user_info dir scm"}
___ATOMIC_TOP_RIGHT=${___ATOMIC_TOP_RIGHT:-"exitcode python ruby todo clock battery"}
___ATOMIC_BOTTOM=${___ATOMIC_BOTTOM:-"char"}

############
## Prompt ##
############

__atomic_ps1() {
	printf "%s%s%s" "$(____atomic_top)" "$(____atomic_bottom)" "${normal}"
}

__atomic_ps2() {
	color=$bold_white
	printf "%s%s%s" "${color}" "${__ATOMIC_PROMPT_CHAR_PS2}  " "${normal}"
}

_atomic_prompt() {
	exitcode="$?"

	PS1="$(__atomic_ps1)"
	PS2="$(__atomic_ps2)"
}

safe_append_prompt_command _atomic_prompt