#!/bin/bash

# A prompt that will push the current directory up a line when it gets too 
# long, giving more space for commands.  Also, the current directory gets 
# truncated when it goes over a certain length

function fixlong {
  # Calculate the width of the prompt
  local newPWD="$PWD"
  local let pwdsize=$(echo -n ${newPWD} | wc -c | tr -d " ")
  if [ "$COLUMNS" ]; then
    # try to keep the environment clean
    let Xpwdmax=${COLUMNS}-5
    local pwdmax=$Xpwdmax
    unset Xpwdmax
  else
    local pwdmax=75
  fi
  if [ "$pwdsize" -gt "$pwdmax" ]; then
    let cut=${pwdsize}-${pwdmax}
    local newPWD="...$(echo -n $PWD | sed -e "s/\(^.\{$cut\}\)\(.*\)/\2/")"
    unset cut
  fi

  if [ -f ~/.iwantcolorprompt ]; then

    #
    # foreground colors:
    #
    #local BLACK='\[\033[30m\]'
    local RED='\[\033[31m\]'
    #local GREEN='\[\033[32m\]'
    local YELLOW='\[\033[33m\]'
    local BLUE='\[\033[34m\]'
    #local MAGENTA='\[\033[35m\]'
    local CYAN='\[\033[36m\]'
    local WHITE='\[\033[37m\]'

    #
    # background colors:
    #
    #local BGBLACK='\[\033[40m\]'
    #local BGRED='\[\033[41m\]'
    #local BGGREEN='\[\033[42m\]'
    #local BGYELLOW='\[\033[43m\]'
    #local BGBLUE='\[\033[44m\]'
    #local BGMAGENTA='\[\033[45m\]'
    #local BGCYAN='\[\033[46m\]'
    #local BGWHITE='\[\033[47m\]'

    #
    # attributes:
    #
    local BRIGHT='\[\033[01m\]'
    local NORMAL='\[\033[00m\]'
    #local REVERSE='\[\033[07m\]'
    #local BLINK='\[\033[05m\]'      # doesn't work in xterms
    #local UNDERSCORE='\[\033[04m\]' # only works in xterms

    # zzzzzz='\[\033[00m\]'  # last env var

    # Go into/out of line drawing mode
    local LINE='\[\016\]'
    local UNLINE='\[\017\]'

  fi

  # Add the current user@host:pwd to the xterm titlebar
  case $TERM in
    xterm*)
      local TITLEBAR='\[\033]0;(\u) \w\007\]'
      ;;
    *)
      local TITLEBAR=''
      ;;
  esac

  if [ "$EUID" = "0" ]; then
    local mycolor=$RED
  else
    local mycolor=$CYAN
  fi

  if [ "$COLUMNS" ]; then
    let Xpwdflip=$COLUMNS/2
    local pwdflip=$Xpwdflip
    unset Xpwdflip
  else
    local pwdflip=40
  fi
  if [ "$pwdsize" -gt "$pwdflip" ]; then
    PS1="$TITLEBAR\
$BRIGHT$BLUE[$YELLOW$newPWD$BLUE]$NORMAL\n\
$BRIGHT$BLUE[$mycolor\u$BLUE]$mycolor\\$""$NORMAL "
  else
    PS1="$TITLEBAR\
$BRIGHT$BLUE[$mycolor\u$BLUE]\
[$YELLOW\w$BLUE]$mycolor\\$""$NORMAL "
  fi
}

PROMPT_COMMAND=fixlong