#!/bin/zsh
  
  # WARP DIRECTORY
  # ==============
  # Jump to custom directories in terminal
  # because `cd` takes too long...
  #
  # @github.com/mfaerevaag/wd
  
  # version

  readonly WD_VERSION=0.4.2

  
  # colors
  readonly WD_BLUE="\033[96m"
  readonly WD_GREEN="\033[92m"
  readonly WD_YELLOW="\033[93m"
  readonly WD_RED="\033[91m"
  readonly WD_NOC="\033[m"
  
  ## functions
  
  # helpers
  wd_yesorno()
  {
      # variables
      local question="${1}"
      local prompt="${question} "
      local yes_RETVAL="0"
      local no_RETVAL="3"
      local RETVAL=""
      local answer=""
  
      # read-eval loop
      while true ; do
          printf $prompt
          read -r answer
  
          case ${answer:=${default}} in
              Y|y|YES|yes|Yes )
                  RETVAL=${yes_RETVAL} && \
                      break
                  ;;
              N|n|NO|no|No )
                  RETVAL=${no_RETVAL} && \
                      break
                  ;;
              * )
                  echo "Please provide a valid answer (y or n)"
                  ;;
          esac
      done
  
      return ${RETVAL}
  }
  
  wd_print_msg()
  {
      if [[ -z $wd_quiet_mode ]]
      then
          local color=$1
          local msg=$2
  
          if [[ $color == "" || $msg == "" ]]
          then
              print " ${WD_RED}*${WD_NOC} Could not print message. Sorry!"
          else
              print " ${color}*${WD_NOC} ${msg}"
          fi
      fi
  }
  
  wd_print_usage()
  {
      cat <<- EOF
  Usage: wd [command] <point>
  
  Commands:
  	add <point>	Adds the current working directory to your warp points
  	add! <point>	Overwrites existing warp point
  	rm <point>	Removes the given warp point
  	show		Print warp points to current directory
  	show <point>	Print path to given warp point

  	list	        Print all stored warp points
  ls  <point>     Show files from given warp point
  path <point>    Show the path to given warp point

  	clean!		Remove points warping to nonexistent directories
  
  	-v | --version	Print version
  	-d | --debug	Exit after execution with exit codes (for testing)
  	-c | --config	Specify config file (default ~/.warprc)
  	-q | --quiet	Suppress all output
  
  	help		Show this extremely helpful text
  EOF
  }
  
  wd_exit_fail()
  {
      local msg=$1

      wd_print_msg $WD_RED $msg

      WD_EXIT_CODE=1
  }
  
  wd_exit_warn()
  {
      local msg=$1
  
      wd_print_msg $WD_YELLOW $msg
      WD_EXIT_CODE=1
  }

  wd_getdir()
  {
      local name_arg=$1
  
      point=$(wd_show $name_arg)
      dir=${point:28+$#name_arg+7}
  
      if [[ -z $name_arg ]]; then
          wd_exit_fail "You must enter a warp point"
          break
      elif [[ -z $dir ]]; then
          wd_exit_fail "Unknown warp point '${name_arg}'"
          break
      fi
  }

  # core
  
  wd_warp()
  {
      local point=$1
  
      if [[ $point =~ "^\.+$" ]]
      then
          if [ $#1 < 2 ]
          then
              wd_exit_warn "Warping to current directory?"
          else
              (( n = $#1 - 1 ))
              cd -$n > /dev/null
          fi
      elif [[ ${points[$point]} != "" ]]
      then

          cd ${points[$point]/#\~/$HOME}

      else
          wd_exit_fail "Unknown warp point '${point}'"
      fi
  }
  
  wd_add()
  {
      local force=$1
      local point=$2
  
      if [[ $point =~ "^[\.]+$" ]]
      then
          wd_exit_fail "Warp point cannot be just dots"
      elif [[ $point =~ "[[:space:]]+" ]]
      then
          wd_exit_fail "Warp point should not contain whitespace"
      elif [[ $point == *:* ]]
      then
          wd_exit_fail "Warp point cannot contain colons"
      elif [[ $point == "" ]]
      then
          wd_exit_fail "Warp point cannot be empty"
      elif [[ ${points[$2]} == "" ]] || $force
      then
          wd_remove $point > /dev/null

          printf "%q:%s
  " "${point}" "${PWD/#$HOME/~}" >> $WD_CONFIG

  
          wd_print_msg $WD_GREEN "Warp point added"
  
          # override exit code in case wd_remove did not remove any points
          # TODO: we should handle this kind of logic better
          WD_EXIT_CODE=0
      else
          wd_exit_warn "Warp point '${point}' already exists. Use 'add!' to overwrite."
      fi
  }
  
  wd_remove()
  {
      local point=$1
  
      if [[ ${points[$point]} != "" ]]
      then
          local config_tmp=$WD_CONFIG.tmp
          if sed -n "/^${point}:.*$/!p" $WD_CONFIG > $config_tmp && mv $config_tmp $WD_CONFIG
          then
              wd_print_msg $WD_GREEN "Warp point removed"
          else
              wd_exit_fail "Something bad happened! Sorry."
          fi
      else
          wd_exit_fail "Warp point was not found"
      fi
  }
  
  wd_list_all()
  {
      wd_print_msg $WD_BLUE "All warp points:"

      entries=$(sed "s:${HOME}:~:g" $WD_CONFIG)
  
      max_warp_point_length=0
      while IFS= read -r line
      do
          arr=(${(s,:,)line})
          key=${arr[1]}
  
          length=${#key}
          if [[ length -gt max_warp_point_length ]]
          then
              max_warp_point_length=$length
          fi
      done <<< $entries

      while IFS= read -r line
      do
          if [[ $line != "" ]]
          then
              arr=(${(s,:,)line})
              key=${arr[1]}
              val=${arr[2]}
  
              if [[ -z $wd_quiet_mode ]]
              then

                  printf "%${max_warp_point_length}s  ->  %s
  " $key $val

              fi
          fi

      done <<< $entries

  }

  wd_ls()
  {
      wd_getdir $1

      ls ${dir/#\~/$HOME}

  }
  
  wd_path()
  {
      wd_getdir $1
      echo $(echo $dir | sed "s:${HOME}:~:g")
  }

  wd_show()
  {
      local name_arg=$1
      # if there's an argument we look up the value
      if [[ ! -z $name_arg ]]
      then
          if [[ -z $points[$name_arg] ]]
          then
              wd_print_msg $WD_BLUE "No warp point named $name_arg"
          else
              wd_print_msg $WD_GREEN "Warp point: ${WD_GREEN}$name_arg${WD_NOC} -> $points[$name_arg]"
          fi
      else
          # hax to create a local empty array
          local wd_matches
          wd_matches=()
          # do a reverse lookup to check whether PWD is in $points

          PWD="${PWD/$HOME/~}"

          if [[ ${points[(r)$PWD]} == $PWD ]]
          then
              for name in ${(k)points}
              do
                  if [[ $points[$name] == $PWD ]]
                  then
                      wd_matches[$(($#wd_matches+1))]=$name
                  fi
              done
  
              wd_print_msg $WD_BLUE "$#wd_matches warp point(s) to current directory: ${WD_GREEN}$wd_matches${WD_NOC}"
          else
              wd_print_msg $WD_YELLOW "No warp point to $(echo $PWD | sed "s:$HOME:~:")"
          fi
      fi
  }
  
  wd_clean() {
      local force=$1
      local count=0
      local wd_tmp=""
  
      while read line
      do
          if [[ $line != "" ]]
          then
              arr=(${(s,:,)line})
              key=${arr[1]}
              val=${arr[2]}
  
              if [ -d "$val" ]
              then
                  wd_tmp=$wd_tmp"
  "`echo $line`
              else
                  wd_print_msg $WD_YELLOW "Nonexistent directory: ${key} -> ${val}"
                  count=$((count+1))
              fi
          fi
      done < $WD_CONFIG
  
      if [[ $count -eq 0 ]]
      then
          wd_print_msg $WD_BLUE "No warp points to clean, carry on!"
      else
          if $force || wd_yesorno "Removing ${count} warp points. Continue? (Y/n)"
          then
              echo $wd_tmp >! $WD_CONFIG
              wd_print_msg $WD_GREEN "Cleanup complete. ${count} warp point(s) removed"
          else
              wd_print_msg $WD_BLUE "Cleanup aborted"
          fi
      fi
  }
  
  local WD_CONFIG=$HOME/.warprc
  local WD_QUIET=0
  local WD_EXIT_CODE=0
  local WD_DEBUG=0
  
  # Parse 'meta' options first to avoid the need to have them before
  # other commands. The `-D` flag consumes recognized options so that
  # the actual command parsing won't be affected.
  
  zparseopts -D -E \
      c:=wd_alt_config -config:=wd_alt_config \
      q=wd_quiet_mode -quiet=wd_quiet_mode \
      v=wd_print_version -version=wd_print_version \
      d=wd_debug_mode -debug=wd_debug_mode
  
  if [[ ! -z $wd_print_version ]]
  then
      echo "wd version $WD_VERSION"
  fi
  
  if [[ ! -z $wd_alt_config ]]
  then
      WD_CONFIG=$wd_alt_config[2]
  fi
  
  # check if config file exists
  if [ ! -e $WD_CONFIG ]
  then
      # if not, create config file
      touch $WD_CONFIG
  fi
  
  # load warp points
  typeset -A points
  while read -r line
  do
      arr=(${(s,:,)line})
      key=${arr[1]}
      val=${arr[2]}
  
      points[$key]=$val
  done < $WD_CONFIG
  
  # get opts

  args=$(getopt -o a:r:c:lhs -l add:,rm:,clean\!,list,ls:,path:,help,show -- $*)

  
  # check if no arguments were given, and that version is not set
  if [[ ($? -ne 0 || $#* -eq 0) && -z $wd_print_version ]]
  then
      wd_print_usage
  
      # check if config file is writeable
  elif [ ! -w $WD_CONFIG ]
  then
      # do nothing
      # can't run `exit`, as this would exit the executing shell
      wd_exit_fail "\'$WD_CONFIG\' is not writeable."
  
  else
  
      # parse rest of options
      for o
      do
          case "$o"
              in
              -a|--add|add)
                  wd_add false $2
                  break
                  ;;
              -a!|--add!|add!)
                  wd_add true $2
                  break
                  ;;
              -r|--remove|rm)
                  wd_remove $2
                  break
                  ;;

              -l|list)

                  wd_list_all
                  break
                  ;;

              -ls|ls)
                  wd_ls $2
                  break
                  ;;
              -p|--path|path)
                  wd_path $2
                  break
                  ;;

              -h|--help|help)
                  wd_print_usage
                  break
                  ;;
              -s|--show|show)
                  wd_show $2
                  break
                  ;;
              -c|--clean|clean)
                  wd_clean false
                  break
                  ;;
              -c!|--clean!|clean!)
                  wd_clean true
                  break
                  ;;
              *)
                  wd_warp $o
                  break
                  ;;
              --)
                  break
                  ;;
          esac
      done
  fi
  
  ## garbage collection
  # if not, next time warp will pick up variables from this run
  # remember, there's no sub shell
  
  unset wd_warp
  unset wd_add
  unset wd_remove
  unset wd_show
  unset wd_list_all
  unset wd_print_msg
  unset wd_yesorno
  unset wd_print_usage
  unset wd_alt_config
  unset wd_quiet_mode
  unset wd_print_version
  
  unset args
  unset points
  unset val &> /dev/null # fixes issue #1
  
  if [[ ! -z $wd_debug_mode ]]
  then
      exit $WD_EXIT_CODE
  else
      unset wd_debug_mode
  fi