be more tolerant towards misconfigured locales; users with non-utf-8 locales will still have broken shells but at least p10k will work

3 files changed, 38 insertions, 17 deletions

diff --git a/internal/configure.zsh b/internal/configure.zsh
index 1c69fa85..5b054291 100644
--- a/internal/configure.zsh
+++ b/internal/configure.zsh
@@ -11,36 +11,48 @@ typeset -gr __p9k_root_dir_u=${${${(q-)__p9k_root_dir}/#(#b)$HOME(|\/*)/'~'$matc
 
 function _p9k_can_configure() {
   emulate -L zsh
-  setopt err_return extended_glob no_prompt_{bang,subst} prompt_{cr,percent,sp}
+  setopt extended_glob no_prompt_{bang,subst} prompt_{cr,percent,sp}
   [[ $1 == '-q' ]] && local -i q=1 || local -i q=0
   function $0_error() {
     (( q )) || print -rP "%1F[ERROR]%f %Bp10k configure%b: $1" >&2
-    return 1
   }
   {
-    [[ -t 0 && -t 1 ]]                                || $0_error "no TTY"
-    [[ -o multibyte ]]                                || $0_error "multibyte option is not set"
-    [[ "${#$(print '\u276F' 2>/dev/null)}" == 1 ]]    || $0_error "shell doesn't support unicode"
-    [[ -e $__p9k_zd ]]                                || $0_error "$__p9k_zd_u does not exist"
-    [[ -d $__p9k_zd ]]                                || $0_error "$__p9k_zd_u is not a directory"
-    [[ -w $__p9k_zd ]]                                || $0_error "$__p9k_zd_u is not writable"
-    [[ ! -d $__p9k_cfg_path ]]                        || $0_error "$__p9k_cfg_path_u is a directory"
-    [[ ! -d $__p9k_zshrc ]]                           || $0_error "$__p9k_zshrc_u is a directory"
+    [[ -t 0 && -t 1 ]]         || { $0_error "no TTY";                           return 1 }
+    [[ -o multibyte ]]         || { $0_error "multibyte option is not set";      return 1 }
+    [[ -e $__p9k_zd ]]         || { $0_error "$__p9k_zd_u does not exist";       return 1 }
+    [[ -d $__p9k_zd ]]         || { $0_error "$__p9k_zd_u is not a directory";   return 1 }
+    [[ -w $__p9k_zd ]]         || { $0_error "$__p9k_zd_u is not writable";      return 1 }
+    [[ ! -d $__p9k_cfg_path ]] || { $0_error "$__p9k_cfg_path_u is a directory"; return 1 }
+    [[ ! -d $__p9k_zshrc ]]    || { $0_error "$__p9k_zshrc_u is a directory";    return 1 }
 
-    [[ ! -e $__p9k_cfg_path || -f $__p9k_cfg_path || -h $__p9k_cfg_path ]] ||
+    [[ ! -e $__p9k_cfg_path || -f $__p9k_cfg_path || -h $__p9k_cfg_path ]] || {
       $0_error "$__p9k_cfg_path_u is a special file"
-    [[ -r $__p9k_root_dir/config/p10k-lean.zsh ]]                          ||
+      return 1
+    }
+    [[ -r $__p9k_root_dir/config/p10k-lean.zsh ]]                          || {
       $0_error "cannot read $__p9k_root_dir_u/config/p10k-lean.zsh"
-    [[ -r $__p9k_root_dir/config/p10k-classic.zsh ]]                       ||
+      return 1
+    }
+    [[ -r $__p9k_root_dir/config/p10k-classic.zsh ]]                       || {
       $0_error "cannot read $__p9k_root_dir_u/config/p10k-classic.zsh"
-    [[ ! -e $__p9k_zshrc || -f $__p9k_zshrc || -h $__p9k_zshrc ]]          ||
+      return 1
+    }
+    [[ ! -e $__p9k_zshrc || -f $__p9k_zshrc || -h $__p9k_zshrc ]]          || {
       $0_error "$__p9k_zshrc_u a special file"
-    [[ ! -e $__p9k_zshrc || -r $__p9k_zshrc ]]                             ||
+      return 1
+    }
+    [[ ! -e $__p9k_zshrc || -r $__p9k_zshrc ]]                             || {
       $0_error "$__p9k_zshrc_u is not readable"
-    [[ ! -e $__p9k_zshrc || -w $__p9k_zshrc ]]                             ||
+      return 1
+    }
+    [[ ! -e $__p9k_zshrc || -w $__p9k_zshrc ]]                             || {
       $0_error "$__p9k_zshrc_u is not writable"
-    (( LINES >= __p9k_wizard_lines && COLUMNS >= __p9k_wizard_columns ))   ||
+      return 1
+    }
+    (( LINES >= __p9k_wizard_lines && COLUMNS >= __p9k_wizard_columns ))   || {
       $0_error "terminal size too small; must be at least $__p9k_wizard_columns x $__p9k_wizard_lines"
+      return 1
+    }
   } always {
     unfunction $0_error
   }
diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index 4196e18a..2a751e65 100755
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -147,6 +147,7 @@ function _p9k_declare() {
       (( set )) && typeset -g _$2=${(P)2} || typeset -g _$2=$3
       ;;
     -e)
+      local LC_ALL=C.UTF-8
       if (( set )); then
         local v=${(P)2}
         typeset -g _$2=${(g::)v}
@@ -329,6 +330,7 @@ _p9k_get_icon() {
       if [[ $_p9k_ret == $'\1'* ]]; then
         _p9k_ret=${_p9k_ret[2,-1]}
       else
+        local LC_ALL=C.UTF-8
         _p9k_ret=${(g::)_p9k_ret}
         [[ $_p9k_ret != $'\b'? ]] || _p9k_ret="%{$_p9k_ret%}"  # penance for past sins
       fi
@@ -549,6 +551,7 @@ _p9k_left_prompt_segment() {
 
     p+='${${_p9k_e:#00}:+${${_p9k_t[$_p9k_n]/'$ss'/$_p9k_ss}/'$s'/$_p9k_s}'
 
+    local LC_ALL=C.UTF-8
     _p9k_param $1 ICON_BEFORE_CONTENT ''
     if [[ $_p9k_ret != false ]]; then
       _p9k_param $1 PREFIX ''
@@ -749,6 +752,7 @@ _p9k_right_prompt_segment() {
 
     p+='${${_p9k_e:#00}:+${_p9k_t[$_p9k_n]/'$w'/$_p9k_w}'
 
+    local LC_ALL=C.UTF-8
     _p9k_param $1 ICON_BEFORE_CONTENT ''
     if [[ $_p9k_ret != true ]]; then
       _p9k_param $1 PREFIX ''
@@ -1087,6 +1091,7 @@ prompt_context() {
       if [[ -z $content ]]; then
         local var=POWERLEVEL9K_CONTEXT_${state}_TEMPLATE
         if (( $+parameters[$var] )); then
+          local LC_ALL=C.UTF-8
           content=${(P)var}
           content=${(g::)content}
         else
@@ -3508,6 +3513,7 @@ _p9k_init_params() {
   if [[ $parameters[POWERLEVEL9K_BATTERY_STAGES] == scalar ]]; then
     _p9k_declare -e POWERLEVEL9K_BATTERY_STAGES
   else
+    local LC_ALL=C.UTF-8
     _p9k_declare -a POWERLEVEL9K_BATTERY_STAGES --
     _POWERLEVEL9K_BATTERY_STAGES=("${(@g::)_POWERLEVEL9K_BATTERY_STAGES}")
   fi
@@ -3765,6 +3771,7 @@ _p9k_build_gap_post() {
 }
 
 _p9k_init_lines() {
+  local LC_ALL=C.UTF-8
   local -a left_segments=($_POWERLEVEL9K_LEFT_PROMPT_ELEMENTS)
   local -a right_segments=($_POWERLEVEL9K_RIGHT_PROMPT_ELEMENTS)
 
@@ -4167,6 +4174,7 @@ _p9k_init() {
 
   if _p9k_segment_in_use dir; then
     if (( $+_POWERLEVEL9K_DIR_CLASSES )); then
+      local LC_ALL=C.UTF-8
       local -i i=3
       for ((; i <= $#_POWERLEVEL9K_DIR_CLASSES; i+=3)); do
         _POWERLEVEL9K_DIR_CLASSES[i]=${(g::)_POWERLEVEL9K_DIR_CLASSES[i]}
diff --git a/internal/wizard.zsh b/internal/wizard.zsh
index ae8322a4..c59e0edf 100755
--- a/internal/wizard.zsh
+++ b/internal/wizard.zsh
@@ -5,6 +5,7 @@ setopt noaliases
 
 () {
 setopt extended_glob no_prompt_{bang,subst} prompt_{cr,percent,sp}
+local LC_ALL=C.UTF-8
 
 typeset -g __p9k_root_dir
 typeset -gi force=0