Bugfix for prezto: In prezto the theme gets called by function (via zstyle). In that case we hardcode the path to the theme as there seems no way to find the location so that we can source our other functions.

1 files changed, 14 insertions, 2 deletions

diff --git a/powerlevel9k.zsh-theme b/powerlevel9k.zsh-theme
index 32a03b94..b35dafda 100755
--- a/powerlevel9k.zsh-theme
+++ b/powerlevel9k.zsh-theme
@@ -17,11 +17,23 @@
 #zstyle ':vcs_info:*+*:*' debug true
 #set -o xtrace
 
+# Check if the theme was called as a function.
+if [[ $(whence -w prompt_powerlevel9k_setup) =~ "function" ]]; then
+  # Script is a function! We assume this to happen only in
+  # prezto, as they use the zstyle-builtin to set the theme.
+  0="${ZDOTDIR:-$HOME}/.zprezto/modules/prompt/functions/prompt_powerlevel9k_setup"
+fi
+
 # Check if filename is a symlink.
-if [[ -L "$0" ]]; then
+if [[ -L $0 ]]; then
+  # Script is a symlink
   filename="$(realpath -P $0 2>/dev/null || readlink -f $0 2>/dev/null)"
-else
+elif [[ -f $0 ]]; then
+  # Script is a file
   filename="$0"
+else
+  print -P "%F{red}Script location could not be found!%f"
+  exit 1
 fi
 script_location="$(dirname $filename)"