###############################################################################
#
# Prompt theme name:
#   budspencer
#
# Description:
#   a sophisticated airline/powerline theme
#
# Author:
#   Joseph Tannhuber <sepp.tannhuber@yahoo.de>
#
# Sections:
#   -> Key bindings
#   -> Files
#   -> Functions
#     -> Pre execute
#     -> Directory history
#     -> Command history
#     -> Bookmarks
#     -> Sessions
#     -> Commandline editing with $EDITOR
#     -> Git segment
#     -> Bind-mode segment
#     -> Symbols segment
#   -> Prompt initialization
#   -> Left prompt
#
###############################################################################

###############################################################################
# => Key bindings
###############################################################################

if [ $fish_key_bindings = 'fish_vi_key_bindings' ]
  bind '#' __budspencer_toggle_symbols
  bind -M visual '#' __budspencer_toggle_symbols
  bind ' ' __budspencer_toggle_pwd
  bind -M visual ' ' __budspencer_toggle_pwd
  bind L __budspencer_cd_next
  bind H __budspencer_cd_prev
  bind m mark
  bind M unmark
  bind . __budspencer_edit_commandline
  bind \cv __budspencer_edit_commandline
  bind -M insert \cv __budspencer_edit_commandline
  bind -M insert \n __budspencer_preexec
  bind \n __budspencer_preexec
end

###############################################################################
# => Files
###############################################################################

# Config file
set -g budspencer_config "$HOME/.config/fish/budspencer_config.fish"

# Temporary files
set -g budspencer_tmpfile '/tmp/'(echo %self)'_budspencer_edit.fish'

###############################################################################
# => Functions
###############################################################################

################
# => Pre execute
################
function __budspencer_preexec -d 'Execute after hitting <Enter> before doing anything else'
  set -l cmd (commandline | sed 's|\s\+|\x1e|g')
  if [ $_ = 'fish' ]
    if [ -z $cmd[1] ]
      set -e cmd[1]
  end
  if [ -z $cmd[1] ]
    return
  end
  set -e budspencer_prompt_error[1]
  if not type -q $cmd[1]
    if [ -d $cmd[1] ]
      set budspencer_prompt_error (cd $cmd[1] ^&1)
      and commandline ''
      commandline -f repaint
      return
      end
  end
  switch $cmd[1]
  case 'c'
    if begin
      [ (count $cmd) -gt 1 ]
      and [ $cmd[2] -gt 0 ]
      and [ $cmd[2] -lt $pcount ]
    end
    commandline $prompt_hist[$cmd[2]]
    echo $prompt_hist[$cmd[2]]  | xsel
    commandline -f repaint
    return
    end
  case 'cd'
    if [ (count $cmd) -le 2 ]
      set budspencer_prompt_error (eval $cmd ^&1)
      and commandline ''
      if [ (count $budspencer_prompt_error) -gt 1 ]
        set budspencer_prompt_error $budspencer_prompt_error[1]
      end
      commandline -f repaint
      return
      end
  case 'day' 'night'
    if [ (count $cmd) -eq 1 ]
      eval $cmd
      commandline ''
      commandline -f repaint
      return
      end
  end
end
commandline -f execute
end

#####################
# => Fish termination
#####################
function __budspencer_on_termination -s HUP -s INT -s QUIT -s TERM --on-process %self -d 'Execute when shell terminates'
  set -l item (contains -i %self $budspencer_sessions_active_pid ^ /dev/null)
  __budspencer_detach_session $item
end

######################
# => Directory history
######################
function __budspencer_create_dir_hist -v PWD -d 'Create directory history without duplicates'
  if [ "$pwd_hist_lock" = false ]
    if contains $PWD $$dir_hist
      set -e $dir_hist[1][(contains -i $PWD $$dir_hist)]
  end
  set $dir_hist $$dir_hist $PWD
  set -g dir_hist_val (count $$dir_hist)
end
end

function __budspencer_cd_prev -d 'Change to previous directory, press H in NORMAL mode.'
  if [ $dir_hist_val -gt 1 ]
    set dir_hist_val (expr $dir_hist_val - 1)
    set pwd_hist_lock true
    cd $$dir_hist[1][$dir_hist_val]
    commandline -f repaint
end
end

function __budspencer_cd_next -d 'Change to next directory, press L in NORMAL mode.'
  if [ $dir_hist_val -lt (count $$dir_hist) ]
    set dir_hist_val (expr $dir_hist_val + 1)
    set pwd_hist_lock true
    cd $$dir_hist[1][$dir_hist_val]
    commandline -f repaint
end
end

function d -d 'List directory history, jump to directory in list with d <number>'
  set -l num_items (expr (count $$dir_hist) - 1)
  if [ $num_items -eq 0 ]
    set_color $fish_color_error[1]
    echo 'Directory history is empty. '(set_color normal)'It will be created automatically'
    return
end
for i in (seq $num_items)
  if [ (expr \( $num_items - $i \) \% 2) -eq 0 ]
    set_color normal
  else
    set_color $budspencer_colors[4]
  end
  echo (expr $num_items - $i)\t$$dir_hist[1][$i]  | sed "s|$HOME|~|"
end
if [ $num_items -eq 1 ]
  set last_item ''
else
  set last_item '-'(expr $num_items - 1)
end
echo -en $budspencer_cursors[2]
set input_length (expr length (expr $num_items - 1))
read -p 'echo -n (set_color -b $budspencer_colors[2] $budspencer_colors[5])" ♻ Goto [e|0"$last_item"] "(set_color -b normal $budspencer_colors[2])" "(set_color $budspencer_colors[5])' -n $input_length -l dir_num
switch $dir_num
case (seq 0 (expr $num_items - 1))
  cd $$dir_hist[1][(expr $num_items - $dir_num)]
  return 0
case 'e'
  read -p 'echo -n (set_color -b $budspencer_colors[2] $budspencer_colors[5])" ♻ Erase [0"$last_item"] "(set_color -b normal $budspencer_colors[2])" "(set_color $budspencer_colors[5])' -n $input_length -l dir_num
  set -e $dir_hist[1][(expr $num_items - $dir_num)] ^ /dev/null
  set dir_hist_val (count $$dir_hist)
  return 0
end
return 1
end

####################
# => Command history
####################
function __budspencer_create_cmd_hist -e fish_prompt -d 'Create command history without duplicates'
  if [ $_ = 'fish' ]
    set pcount (expr $pcount + 1)
    set -l IFS ''
    set -l cmd (echo $history[1] | fish_indent | expand -t 4)
    # Create prompt history
    set prompt_hist $prompt_hist $cmd
    # Create command history
    if not begin
      expr $cmd : '[cdms] ' >  /dev/null
      or contains $cmd $budspencer_nocmdhist
      end
      if contains $cmd $$cmd_hist
        set -e $cmd_hist[1][(contains -i $cmd $$cmd_hist)]
      end
      set $cmd_hist $$cmd_hist $cmd
  end
end
set fish_bind_mode insert
echo -n \a
end

function c -d 'List command history, load command from prompt with c <prompt number>'
  set -l num_items (count $$cmd_hist)
  if [ $num_items -eq 0 ]
    set_color $fish_color_error[1]
    echo 'Command history is empty. '(set_color normal)'It will be created automatically.'
    return
end
for i in (seq $num_items)
  if [ (expr \( $num_items - $i \) \% 2) -eq 0 ]
    set_color normal
  else
    set_color $budspencer_colors[4]
  end
  echo -n (expr $num_items - $i)
  set -l item (echo $$cmd_hist[1][$i])
  echo -n \t$item\n
end
if [ $num_items -eq 1 ]
  set last_item ''
else
  set last_item '-'(expr $num_items - 1)
end
echo -en $budspencer_cursors[4]
set input_length (expr length (expr $num_items - 1))
read -p 'echo -n (set_color -b $budspencer_colors[2] $budspencer_colors[9])" ↩ Exec [e|0"$last_item"] "(set_color -b normal $budspencer_colors[2])" "(set_color $budspencer_colors[9])' -n $input_length -l cmd_num
switch $cmd_num
case (seq 0 (expr $num_items - 1))
  commandline $$cmd_hist[1][(expr $num_items - $cmd_num)]
  echo $$cmd_hist[1][(expr $num_items - $cmd_num)]  | xsel
  return 0
case 'e'
  read -p 'echo -n (set_color -b $budspencer_colors[2] $budspencer_colors[9])" ↩ Erase [0"$last_item"] "(set_color -b normal $budspencer_colors[2])" "(set_color $budspencer_colors[9])' -n $input_length -l cmd_num
  set -e $cmd_hist[1][(expr $num_items - $cmd_num)] ^ /dev/null
  return 0
end
return 1
end

##############
# => Bookmarks
##############
function mark -d 'Create bookmark for present working directory.'
  if not contains $PWD $bookmarks
    set -U bookmarks $PWD $bookmarks
    set pwd_hist_lock true
    commandline -f repaint
end
end

function unmark -d 'Remove bookmark for present working directory.'
  if contains $PWD $bookmarks
    set -e bookmarks[(contains -i $PWD $bookmarks)]
    set pwd_hist_lock true
    commandline -f repaint
end
end

function m -d 'List bookmarks, jump to directory in list with m <number>'
  set -l num_items (count $bookmarks)
  if [ $num_items -eq 0 ]
    set_color $fish_color_error[1]
    echo 'Bookmark list is empty. '(set_color normal)'Enter '(set_color $fish_color_command[1])'mark '(set_color normal)'in INSERT mode or '(set_color $fish_color_command[1])'m '(set_color normal)'in NORMAL mode, if you want to add the current directory to your bookmark list.'
    return
end
for i in (seq $num_items)
  if [ $PWD = $bookmarks[$i] ]
    set_color $budspencer_colors[10]
  else
    if [ (expr \( $num_items - $i \) \% 2) -eq 0 ]
      set_color normal
      else
        set_color $budspencer_colors[4]
      end
  end
  echo (expr $num_items - $i)\t$bookmarks[$i]  | sed "s|$HOME|~|"
end
if [ $num_items -eq 1 ]
  set last_item ''
else
  set last_item '-'(expr $num_items - 1)
end
echo -en $budspencer_cursors[1]
set input_length (expr length (expr $num_items - 1))
read -p 'echo -n (set_color -b $budspencer_colors[2] $budspencer_colors[10])" ⌘ Goto [0"$last_item"] "(set_color -b normal $budspencer_colors[2])" "(set_color $budspencer_colors[10])' -n $input_length -l dir_num
switch $dir_num
case (seq 0 (expr $num_items - 1))
  cd $bookmarks[(expr $num_items - $dir_num)]
  return 0
end
return 1
end

#############
# => Sessions
#############
function __budspencer_delete_zombi_sessions -d 'Delete zombi sessions'
  for i in $budspencer_sessions_active_pid
    if not contains $i %fish
      set -l item (contains -i $i $budspencer_sessions_active_pid)
      set -e budspencer_sessions_active_pid[$item]
      set -e budspencer_sessions_active[$item]
  end
end
end

function __budspencer_create_new_session -d 'Create a new session'
  set -U budspencer_session_cmd_hist_$argv[1] $$cmd_hist
  set -U budspencer_session_dir_hist_$argv[1] $$dir_hist
  set -U budspencer_sessions $argv[1] $budspencer_sessions
end

function __budspencer_erase_session -d 'Erase current session'
  if [ (count $argv) -eq 1 ]
    set_color $fish_color_error[1]
    echo 'Missing argument: name of session to erase'
    return
end
if contains $argv[2] $budspencer_sessions_active
  set_color $fish_color_error[1]
  echo "Session '$argv[2]' cannot be erased because it's currently active."
  return
end
if contains $argv[2] $budspencer_sessions
  set -e budspencer_session_cmd_hist_$argv[2]
  set -e budspencer_session_dir_hist_$argv[2]
  set -e budspencer_sessions[(contains -i $argv[2] $budspencer_sessions)]
else
  set_color $fish_color_error[1]
  echo "Session '$argv[2]' not found. "(set_color normal)'Enter '(set_color $fish_color_command[1])'s '(set_color normal)'to show a list of all recorded sessions.'
end
end

function __budspencer_detach_session -d 'Detach current session'
  set cmd_hist cmd_hist_nosession
  set dir_hist dir_hist_nosession
  if [ -z $$dir_hist ] ^ /dev/null
    set $dir_hist $PWD
end
set dir_hist_val (count $$dir_hist)
set -e budspencer_sessions_active_pid[$argv] ^ /dev/null
set -e budspencer_sessions_active[$argv] ^ /dev/null
set budspencer_session_current ''
cd $$dir_hist[1][$dir_hist_val]
end

function __budspencer_attach_session -d 'Attach session'
  set argv (echo -sn $argv\n | sed 's|[^[:alnum:]]|_|g')
  if contains $argv[1] $budspencer_sessions_active
    wmctrl -a "✻ $argv[1]"
else
  wt "✻ $argv[1]"
  __budspencer_detach_session $argv[-1]
  set budspencer_sessions_active $budspencer_sessions_active $argv[1]
  set budspencer_sessions_active_pid $budspencer_sessions_active_pid %self
  set budspencer_session_current $argv[1]
  if not contains $argv[1] $budspencer_sessions
    __budspencer_create_new_session $argv[1]
  end
  set cmd_hist budspencer_session_cmd_hist_$argv[1]
  set dir_hist budspencer_session_dir_hist_$argv[1]
  if [ -z $$dir_hist ] ^ /dev/null
    set $dir_hist $PWD
  end
  set dir_hist_val (count $$dir_hist)
  cd $$dir_hist[1][$dir_hist_val] ^ /dev/null
end
end

function s -d 'Create, delete or attach session'
  __budspencer_delete_zombi_sessions
  if [ (count $argv) -eq 0 ]
    set -l active_indicator
    set -l num_items (count $budspencer_sessions)
    if [ $num_items -eq 0 ]
      set_color $fish_color_error[1]
      echo -n 'Session list is empty. '
      set_color normal
      echo -n 'Enter '
      set_color $fish_color_command[1]
      echo -n 's '
      set_color $fish_color_param[1]
      echo -n 'session-name'
      set_color normal
      echo ' to record the current session.'
      return
  end
  for i in (seq $num_items)
    if [ $budspencer_sessions[$i] = $budspencer_session_current ]
      set_color $budspencer_colors[8]
      else
        if [ (expr \( $num_items - $i \) \% 2) -eq 0 ]
          set_color normal
          else
            set_color $budspencer_colors[4]
          end
      end
      if contains $budspencer_sessions[$i] $budspencer_sessions_active
        set active_indicator '✻ '
      else
        set active_indicator '  '
      end
      echo (expr $num_items - $i)\t$active_indicator$budspencer_sessions[$i]
  end
  if [ $num_items -eq 1 ]
    set last_item ''
  else
    set last_item '-'(expr $num_items - 1)
  end
  echo -en $budspencer_cursors[3]
  set input_length (expr length (expr $num_items - 1))
  read -p 'echo -n (set_color -b $budspencer_colors[2] $budspencer_colors[8])" ✻ Attach [e|0"$last_item"] "(set_color -b normal $budspencer_colors[2])" "(set_color $budspencer_colors[8])' -n $input_length -l session_num
  switch $session_num
  case (seq 0 (expr $num_items - 1))
    set argv[1] $budspencer_sessions[(expr $num_items - $session_num)]
  case 'e'
    read -p 'echo -n (set_color -b $budspencer_colors[2] $budspencer_colors[8])" ✻ Erase [0"$last_item"] "(set_color -b normal $budspencer_colors[2])" "(set_color $budspencer_colors[8])' -n $input_length -l session_num
    __budspencer_erase_session -e $budspencer_sessions[(expr $num_items - $session_num)]
    return 0
  case '*'
    return 1
  end
end
set -l item (contains -i %self $budspencer_sessions_active_pid ^ /dev/null)
switch $argv[1]
case '-e'
  __budspencer_erase_session $argv
case '-d'
  wt 'fish'
  __budspencer_detach_session $item
case '-*'
  set_color $fish_color_error[1]
  echo "Invalid argument: $argv[1]"
case '*'
  __budspencer_attach_session $argv $item
end
end

#####################################
# => Commandline editing with $EDITOR
#####################################
function __budspencer_edit_commandline -d 'Open current commandline with your editor'
  commandline | fish_indent > $budspencer_tmpfile
  eval $EDITOR $budspencer_tmpfile
  if [ -s $budspencer_tmpfile ]
    commandline (sed 's|^\s*||' $budspencer_tmpfile)
else
  commandline ''
end
rm $budspencer_tmpfile
end

################
# => Git segment
################
function __budspencer_prompt_git_branch -d 'Return the current branch name'
  set -l branch (command git symbolic-ref HEAD ^ /dev/null | sed -e 's|^refs/heads/||')
  if not test $branch > /dev/null
    set -l position (command git describe --contains --all HEAD ^ /dev/null)
    if not test $position > /dev/null
      set -l commit (command git rev-parse HEAD ^ /dev/null | sed  's|\(^.......\).*|\1|')
      echo -n (set_color -b $budspencer_colors[11])''(set_color $budspencer_colors[1])' ➦ '$commit' '(set_color $budspencer_colors[11])
  else
    echo -n (set_color -b $budspencer_colors[9])''(set_color $budspencer_colors[1])'  '$position' '(set_color $budspencer_colors[9])
  end
else
  echo -n (set_color -b $budspencer_colors[3])''(set_color $budspencer_colors[1])'  '$branch' '(set_color $budspencer_colors[3])
end
end

######################
# => Bind-mode segment
######################
function __budspencer_prompt_bindmode -d 'Displays the current mode'
  switch $fish_bind_mode
case default
  set budspencer_current_bindmode_color $budspencer_colors[10]
  echo -en $budspencer_cursors[1]
case insert
  set budspencer_current_bindmode_color $budspencer_colors[5]
  echo -en $budspencer_cursors[2]
  if [ "$pwd_hist_lock" = true ]
    set pwd_hist_lock false
    __budspencer_create_dir_hist
  end
case visual
  set budspencer_current_bindmode_color $budspencer_colors[8]
  echo -en $budspencer_cursors[3]
end
if [ (count $budspencer_prompt_error) -eq 1 ]
  set budspencer_current_bindmode_color $budspencer_colors[7]
end
set_color -b $budspencer_current_bindmode_color $budspencer_colors[1]
echo -n " $pcount "
set_color -b $budspencer_colors[1] $budspencer_current_bindmode_color
end

####################
# => Symbols segment
####################
function __budspencer_prompt_left_symbols -d 'Display symbols'
  set -l jobs (jobs | wc -l)
  if [ -e ~/.taskrc ]
    set todo (task due.before:tomorrow ^ /dev/null | tail -1 | cut -f1 -d' ')
end
if [ -e ~/.reminders ]
  set appointments (rem -a | cut -f1 -d' ')
end
if [ (count $todo) -eq 0 ]
  set todo 0
end
if [ (count $appointments) -eq 0 ]
  set appointments 0
end
set_color -b $budspencer_colors[2]
echo -n ''

if [ $symbols_style = 'symbols' ]
  if [ $budspencer_session_current != '' ]
    set_color -o $budspencer_colors[8]
    echo -n ' ✻'
  end
  if contains $PWD $bookmarks
    set_color -o $budspencer_colors[10]
    echo -n ' ⌘'
  end
  if set -q -x VIM
    set_color -o $budspencer_colors[9]
    echo -n ' V'
  end
  if set -q -x RANGER_LEVEL
    set_color -o $budspencer_colors[9]
    echo -n ' R'
  end
  if [ $jobs -gt 0 ]
    set_color -o $budspencer_colors[11]
    echo -n ' ⚙'
  end
  if [ ! -w . ]
    set_color -o $budspencer_colors[6]
    echo -n ' '
  end
  if [ $todo -gt 0 ]
    set_color -o $budspencer_colors[4]
    echo -n ' ⚔'
  end
  if [ $appointments -gt 0 ]
    set_color -o $budspencer_colors[5]
    echo -n ' ⚑'
  end
  if [ $last_status -eq 0 ]
    set_color -o $budspencer_colors[12]
    echo -n ' ✔'
  else
    set_color -o $budspencer_colors[7]
    echo -n ' ✘'
  end
  if [ $USER = 'root' ]
    set_color -o $budspencer_colors[6]
    echo -n ' ⚡'
  end
else
  if [ $budspencer_session_current != '' ] ^ /dev/null
    set_color $budspencer_colors[8]
    echo -n ' '(expr (count $budspencer_sessions) - (contains -i $budspencer_session_current $budspencer_sessions))
  end
  if contains $PWD $bookmarks
    set_color $budspencer_colors[10]
    echo -n ' '(expr (count $bookmarks) - (contains -i $PWD $bookmarks))
  end
  if set -q -x VIM
    set_color -o $budspencer_colors[9]
    echo -n ' V'
    set_color normal
  end
  if set -q -x RANGER_LEVEL
    set_color -b $budspencer_colors[2] $budspencer_colors[9]
    echo -n ' '$RANGER_LEVEL
  end
  if [ $jobs -gt 0 ]
    set_color -b $budspencer_colors[2] $budspencer_colors[11]
    echo -n ' '$jobs
  end
  if [ ! -w . ]
    set_color -o $budspencer_colors[6]
    echo -n ' '
    set_color normal
  end
  if [ $todo -gt 0 ]
    set_color -b $budspencer_colors[2] $budspencer_colors[4]
    echo -n " $todo"
  end
  if [ $appointments -gt 0 ]
    set_color -b $budspencer_colors[2] $budspencer_colors[5]
    echo -n " $appointments"
  end
  if [ $last_status -eq 0 ]
    set_color -b $budspencer_colors[2] $budspencer_colors[12]
    echo -n ' '$last_status
  else
    set_color -b $budspencer_colors[2] $budspencer_colors[7]
    echo -n ' '$last_status
  end
  if [ $USER = 'root' ]
    set_color -o $budspencer_colors[6]
    echo -n ' ⚡'
  end
end
echo -n ' '
set_color -b normal $budspencer_colors[2]
end

###############################################################################
# => Prompt initialization
###############################################################################

# Initialize some global variables
set -g budspencer_prompt_error
set -g budspencer_current_bindmode_color
set -U budspencer_sessions_active $budspencer_sessions_active
set -U budspencer_sessions_active_pid $budspencer_sessions_active_pid
set -g budspencer_session_current ''
set -g cmd_hist_nosession
set -g cmd_hist cmd_hist_nosession
set -g CMD_DURATION 0
set -g dir_hist_nosession
set -g dir_hist dir_hist_nosession
set -g pwd_hist_lock false
set -g pcount 1
set -g prompt_hist
set -g symbols_style 'symbols'

# Load user defined key bindings
if functions --query fish_user_key_bindings
  fish_user_key_bindings
end

# Set favorite editor
if not set -q EDITOR
  set -g EDITOR vi
end

# Source config file
if [ -e $budspencer_config ]
  source $budspencer_config
end

# Don't save in command history
if not set -q budspencer_nocmdhist
  set -U budspencer_nocmdhist 'c' 'd' 'll' 'ls' 'm' 's'
end

# Set PWD segment style
if not set -q budspencer_pwdstyle
  set -U budspencer_pwdstyle short long none
end
set pwd_style $budspencer_pwdstyle[1]

# Cd to newest bookmark if this is a login shell
if not begin
  set -q -x LOGIN
  or set -q -x RANGER_LEVEL
  or set -q -x VIM
end ^ /dev/null
cd $bookmarks[1]
end
set -x LOGIN $USER

###############################################################################
# => Left prompt
###############################################################################

function fish_prompt -d 'Write out the left prompt of the budspencer theme'
  set -g last_status $status
  echo -n -s (__budspencer_prompt_bindmode) (__budspencer_prompt_git_branch) (__budspencer_prompt_left_symbols) '' ' '
end