From b9a618a514069a4076bf0fa8489213033d888e98 Mon Sep 17 00:00:00 2001
From: Vladimir Hodakov <vladimir@hodakov.me>
Date: Thu, 25 Jul 2024 14:29:40 +0500
Subject: [PATCH] Add Distrobox indication support (#15)

Distrobox is a modern containerized tool to run development workflows in
the isolated spaces.

Reviewed-on: https://code.pztrn.name/misc/zsh-config/pulls/15
Co-authored-by: Vladimir Hodakov <vladimir@hodakov.me>
Co-committed-by: Vladimir Hodakov <vladimir@hodakov.me>
---
 zsh/prompts/fat0troll.zsh | 5 ++++-
 zsh/prompts/pztrn.zsh     | 7 ++++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/zsh/prompts/fat0troll.zsh b/zsh/prompts/fat0troll.zsh
index 5266b06..5eb07de 100644
--- a/zsh/prompts/fat0troll.zsh
+++ b/zsh/prompts/fat0troll.zsh
@@ -23,10 +23,13 @@ prompt_fat0troll_setup () {
     vars['brackets_end']="%{$fg[yellow]%}]%{$reset_color%}"
 
     # Session-dependend colorizing.
-    # Local will be black, remote - yellow.
+    # Local will be black, remote and containers - yellow.
     if [[ -n $SSH_TTY ]]; then
         vars['console']="$vars['brackets_start']SSH$vars['brackets_end']"
         vars['default_color']="%{$fg[cyan]%}"
+    elif test -e /run/.containerenv -o -e /.dockerenv; then
+        vars['console']="$vars['brackets_start']📦$vars['brackets_end']"
+        vars['default_color']="%{$fg[cyan]%}"
     else
         vars['console']=""
         vars['default_color']="%{$fg[green]%}"
diff --git a/zsh/prompts/pztrn.zsh b/zsh/prompts/pztrn.zsh
index 2866945..2b5a0cb 100644
--- a/zsh/prompts/pztrn.zsh
+++ b/zsh/prompts/pztrn.zsh
@@ -21,12 +21,17 @@ prompt_pztrn_setup () {
     precmd_functions=(${precmd_functions[@]} "prompt_pztrn_gitinfo" "prompt_pztrn_createprompt")
 
     # Session-dependend colorizing.
-    # Local will be black, remote - yellow.
+    # Local will be black, remote and containers - yellow.
     if [[ -n $SSH_TTY ]]; then
         vars['brackets_start']="%{$fg[yellow]%}[%{$reset_color%}"
         vars['brackets_end']="%{$fg[yellow]%}]%{$reset_color%}"
         vars['console']="$vars['brackets_start']SSH$vars['brackets_end']"
         vars['default_color']="%{$fg[cyan]%}"
+    elif test -e /run/.containerenv -o -e /.dockerenv; then
+        vars['brackets_start']="%{$fg[yellow]%}[%{$reset_color%}"
+        vars['brackets_end']="%{$fg[yellow]%}]%{$reset_color%}"
+        vars['console']="$vars['brackets_start']📦$vars['brackets_end']"
+        vars['default_color']="%{$fg[cyan]%}"
     else
         vars['brackets_start']="%{$reset_color%}%{$fg[gray]%}[%{$reset_color%}"
         vars['brackets_end']="%{$reset_color%}%{$fg[gray]%}]%{$reset_color%}"