trapd00r: look for 256-color support, not $DISPLAY

Checking if the terminal supports 256 colors is better suited for our purpose. Checking if `$DISPLAY` is set doesn't tell us if our colors will be displayed correctly.

trapd00r: look for 256-color support, not $DISPLAY
4774bc62 · Marc Cornellà · 1d26e2ab · 4774bc62
Commit 4774bc62 authored Aug 20, 2018 by Marc Cornellà
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

trapd00r.zsh-theme themes/trapd00r.zsh-theme +6 -3

No files found.
--- a/themes/trapd00r.zsh-theme
+++ b/themes/trapd00r.zsh-theme
@@ -38,6 +38,9 @@ local c13=$(printf "\e[38;5;196m\e[1m")


 zsh_path() {
+  local colors
+  colors=$(echoti colors)
+
  local -A yellow
  yellow=(
    1  '%F{228}'   2  '%F{222}'   3  '%F{192}'   4  '%F{186}'
@@ -51,7 +54,7 @@ zsh_path() {
  for c (${(s::)PWD}); do
    if [[ $c = "/" ]]; then
      if [[ $i -eq 1 ]]; then
-        if [[ -n "$DISPLAY" ]]; then
+        if [[ $colors -ge 256 ]]; then
          print -Pn '%F{065}%B /%b%f'
        else
          print -Pn '\e[31;1m /%f'
@@ -60,14 +63,14 @@ zsh_path() {
        continue
      fi

-      if [[ -n "$DISPLAY" ]]; then
+      if [[ $colors -ge 256 ]]; then
        print -Pn "${yellow[$i]:-%f} » %f"
      else
        print -Pn "%F{yellow} > %f"
      fi
      (( i += 6 ))
    else
-      if [[ -n "$DISPLAY" ]]; then
+      if [[ $colors -ge 256 ]]; then
        print -Pn "%F{065}$c%f"
      else
        print -Pn "%F{blue}$c%f"