mirror of
https://github.com/itme-brain/nixos.git
synced 2026-03-24 08:39:42 -04:00
60 lines
1.6 KiB
Nix
60 lines
1.6 KiB
Nix
''
|
|
eval "$(direnv hook bash)"
|
|
|
|
is_ssh_session() {
|
|
if [ -n "$SSH_CLIENT" ] || [ -n "$SSH_TTY" ]; then
|
|
return 0
|
|
else
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
function set_ps1_prompt() {
|
|
local git_branch=""
|
|
local flake_icon=""
|
|
local python_icon=""
|
|
local cur_dir=""
|
|
|
|
if git rev-parse --is-inside-work-tree >/dev/null 2>&1; then
|
|
git_branch="$(git symbolic-ref --short HEAD 2>/dev/null)"
|
|
|
|
if [ $? -ne 0 ]; then
|
|
git_branch="$(git rev-parse --short HEAD 2>/dev/null)"
|
|
fi
|
|
|
|
git_branch="\[\033[01;31m\]$git_branch :\[\033[00m\]"
|
|
|
|
if [ -f "$(git rev-parse --show-toplevel)/flake.nix" ]; then
|
|
# If it exists, set the flake icon and color it blue
|
|
flake_icon="\[\033[01;34m\] \[\033[00m\]"
|
|
fi
|
|
|
|
git_root="$(basename "$(git rev-parse --show-toplevel)")"
|
|
|
|
cur_dir=$(realpath --relative-to=$(git rev-parse --show-toplevel) .)
|
|
if [ "$cur_dir" == "." ]; then
|
|
cur_dir="\[\033[01;34m\] $git_root\[\033[00m\]"
|
|
else
|
|
cur_dir="\[\033[01;34m\] $git_root/$cur_dir\[\033[00m\]"
|
|
fi
|
|
else
|
|
cur_dir="\[\033[01;34m\]\w\[\033[00m\]"
|
|
fi
|
|
|
|
if [[ -n "$VIRTUAL_ENV" ]]; then
|
|
python_icon="\[\033[01;33m\] \[\033[00m\]"
|
|
fi
|
|
|
|
if [ -n "''${IN_NIX_SHELL:+x}" ]; then
|
|
PS1="$cur_dir\n$flake_icon$python_icon\[\033[01;32m\]nixShell>$git_branch\[\033[00m\]"
|
|
else
|
|
if ! is_ssh_session; then
|
|
PS1="\n$cur_dir\n$flake_icon$python_icon\[\033[01;32m\]>$git_branch\[\033[00m\]"
|
|
else
|
|
PS1="\n\[\033[01;34m\]\w\[\033[00m\]\n\[\033[01;32m\]\u@\h:\[\033[00m\] "
|
|
fi
|
|
fi
|
|
unset flake_icon
|
|
}
|
|
PROMPT_COMMAND="set_ps1_prompt; $PROMPT_COMMAND"
|
|
''
|