From 5e82c5a57e84fde658c082b9702692b1f97937e7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cenk=20G=C3=BCndo=C4=9Fan?= <cnkgndgn@gmail.com>
Date: Fri, 25 Mar 2016 22:36:34 +0100
Subject: [PATCH] [prompt] allow optional parameter for short_pwd

---
 modules/prompt/functions/short_pwd | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/modules/prompt/functions/short_pwd b/modules/prompt/functions/short_pwd
index dc93885..37950cc 100644
--- a/modules/prompt/functions/short_pwd
+++ b/modules/prompt/functions/short_pwd
@@ -1,8 +1,10 @@
 # shortens the pwd for use in prompt
 
-local current_dir="${PWD/#${HOME}/~}"
+local current_dir="${1:-${PWD}}"
 local return_dir='~'
 
+current_dir="${current_dir/#${HOME}/~}"
+
 # if we aren't in ~
 if [[ ${current_dir} != '~' ]]; then
   return_dir="${${${${(@j:/:M)${(@s:/:)current_dir}##.#?}:h}%/}//\%/%%}/${${current_dir:t}//\%/%%}"