Add powerline
This commit is contained in:
Bob Mottram
parent
d55f12d79b
commit
2508d63134
|
|
@ -0,0 +1,357 @@
|
||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
# Based on https://github.com/undu/bash-powerline
|
||||||
|
|
||||||
|
__powerline() {
|
||||||
|
|
||||||
|
# User config variables,
|
||||||
|
# it's recommended to override those variables through .bashrc or similar
|
||||||
|
#
|
||||||
|
# Use powerline mode
|
||||||
|
# readonly POWERLINE_FONT=''
|
||||||
|
#
|
||||||
|
# Always show user in the prompt
|
||||||
|
# readonly SHOW_USER=''
|
||||||
|
#
|
||||||
|
# Never show a default user
|
||||||
|
# readonly DEFAULT_USER='user'
|
||||||
|
|
||||||
|
# Default background and foreground ANSI colours
|
||||||
|
readonly DEFAULT_BG=0
|
||||||
|
readonly DEFAULT_FG=7
|
||||||
|
|
||||||
|
# Max length of full path
|
||||||
|
readonly MAX_PATH_LENGTH=30
|
||||||
|
|
||||||
|
# Unicode symbols
|
||||||
|
if [ -z "${POWERLINE_FONT+x}" ]; then
|
||||||
|
readonly GIT_BRANCH_SYMBOL='⑂'
|
||||||
|
else
|
||||||
|
readonly GIT_BRANCH_SYMBOL=''
|
||||||
|
fi
|
||||||
|
readonly GIT_BRANCH_CHANGED_SYMBOL='Δ'
|
||||||
|
readonly GIT_NEED_PUSH_SYMBOL='↑'
|
||||||
|
readonly GIT_NEED_PULL_SYMBOL='↓'
|
||||||
|
|
||||||
|
# Powerline symbols
|
||||||
|
readonly BLOCK_START=''
|
||||||
|
|
||||||
|
# ANSI Colours
|
||||||
|
readonly BLACK=0
|
||||||
|
readonly RED=1
|
||||||
|
readonly GREEN=2
|
||||||
|
readonly YELLOW=3
|
||||||
|
readonly BLUE=4
|
||||||
|
readonly MAGENTA=5
|
||||||
|
readonly CYAN=6
|
||||||
|
readonly WHITE=7
|
||||||
|
|
||||||
|
readonly BLACK_BRIGHT=8
|
||||||
|
readonly RED_BRIGHT=9
|
||||||
|
readonly GREEN_BRIGHT=10
|
||||||
|
readonly YELLOW_BRIGHT=11
|
||||||
|
readonly BLUE_BRIGHT=12
|
||||||
|
readonly MAGENTA_BRIGHT=13
|
||||||
|
readonly CYAN_BRIGHT=14
|
||||||
|
readonly WHITE_BRIGHT=15
|
||||||
|
|
||||||
|
# Font effects
|
||||||
|
readonly DIM="\[$(tput dim)\]"
|
||||||
|
readonly REVERSE="\[$(tput rev)\]"
|
||||||
|
readonly RESET="\[$(tput sgr0)\]"
|
||||||
|
readonly BOLD="\[$(tput bold)\]"
|
||||||
|
|
||||||
|
# Generate terminal colour codes
|
||||||
|
# $1 is an int (a colour) and $2 must be 'fg' or 'bg'
|
||||||
|
__colour() {
|
||||||
|
case "$2" in
|
||||||
|
'fg'*)
|
||||||
|
echo "\[$(tput setaf "$1")\]"
|
||||||
|
;;
|
||||||
|
'bg'*)
|
||||||
|
echo "\[$(tput setab "$1")\]"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "\[$(tput setab "$1")\]"
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
|
# Generate a single-coloured block for the prompt
|
||||||
|
__prompt_block() {
|
||||||
|
local bg; local fg
|
||||||
|
if [ ! -z "${1+x}" ]; then
|
||||||
|
bg=$1
|
||||||
|
else
|
||||||
|
if [ ! -z "$last_bg" ]; then
|
||||||
|
bg=$last_bg
|
||||||
|
else
|
||||||
|
bg=$DEFAULT_BG
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
if [ ! -z "${2+x}" ]; then
|
||||||
|
fg=$2
|
||||||
|
else
|
||||||
|
fg=$DEFAULT_FG
|
||||||
|
fi
|
||||||
|
|
||||||
|
local block
|
||||||
|
|
||||||
|
# Need to generate a separator if the background changes
|
||||||
|
if [[ ! -z "$last_bg" && "$bg" != "$last_bg" && ! -z "${POWERLINE_FONT+x}" ]]; then
|
||||||
|
block+="$(__colour "$bg" 'bg')"
|
||||||
|
block+="$(__colour "$last_bg" 'fg')"
|
||||||
|
block+="$BLOCK_START $RESET"
|
||||||
|
block+="$(__colour "$bg" 'bg')"
|
||||||
|
block+="$(__colour "$fg" 'fg')"
|
||||||
|
else
|
||||||
|
block+="$(__colour "$bg" 'bg')"
|
||||||
|
block+="$(__colour "$fg" 'fg')"
|
||||||
|
block+=" "
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ ! -z "${3+x}" ]; then
|
||||||
|
block+="$3 $RESET"
|
||||||
|
fi
|
||||||
|
|
||||||
|
last_bg=$bg
|
||||||
|
|
||||||
|
__block_text="$block"
|
||||||
|
}
|
||||||
|
|
||||||
|
function __end_block() {
|
||||||
|
__block_text=''
|
||||||
|
if [ ! -z "$last_bg" ]; then
|
||||||
|
if [ ! -z "${POWERLINE_FONT+x}" ]; then
|
||||||
|
__block_text+="$(__colour $DEFAULT_BG 'bg')"
|
||||||
|
__block_text+="$(__colour "$last_bg" 'fg')"
|
||||||
|
__block_text+="$BLOCK_START$RESET"
|
||||||
|
__block_text+="$(__colour $DEFAULT_BG 'bg')"
|
||||||
|
__block_text+="$(__colour "$DEFAULT_FG" 'fg')"
|
||||||
|
else
|
||||||
|
__block_text+="$(__colour $DEFAULT_BG 'bg')"
|
||||||
|
__block_text+="$(__colour "$DEFAULT_FG" 'fg')"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
__block_text+=' '
|
||||||
|
}
|
||||||
|
|
||||||
|
### Prompt components
|
||||||
|
|
||||||
|
__git_block() {
|
||||||
|
if ! command -V git > /dev/null; then
|
||||||
|
# git not found
|
||||||
|
__block_text=''
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
# force git output in English to make our work easier
|
||||||
|
local git_eng="env LANG=C git"
|
||||||
|
|
||||||
|
# check if pwd is under git
|
||||||
|
if ! git rev-parse --is-inside-git-dir > /dev/null 2> /dev/null; then
|
||||||
|
# not in a git repo, bail out
|
||||||
|
__block_text=''
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
# get current branch name or short SHA1 hash for detached head
|
||||||
|
local branch; local ref_symbol
|
||||||
|
branch="$($git_eng symbolic-ref --short HEAD 2>/dev/null)"
|
||||||
|
# shellcheck disable=SC2181
|
||||||
|
if [ $? != 0 ]; then
|
||||||
|
branch="$($git_eng describe --tags --always 2>/dev/null)"
|
||||||
|
ref_symbol='➦'
|
||||||
|
else
|
||||||
|
ref_symbol=$GIT_BRANCH_SYMBOL
|
||||||
|
fi
|
||||||
|
|
||||||
|
# In pcmode (and only pcmode) the contents of
|
||||||
|
# $gitstring are subject to expansion by the shell.
|
||||||
|
# Avoid putting the raw ref name in the prompt to
|
||||||
|
# protect the user from arbitrary code execution via
|
||||||
|
# specially crafted ref names (e.g., a ref named
|
||||||
|
# '$(IFS=_;cmd=sudo_rm_-rf_/;$cmd)' would execute
|
||||||
|
# 'sudo rm -rf /' when the prompt is drawn). Instead,
|
||||||
|
# put the ref name in a new global variable (in the
|
||||||
|
# __git_ps1_* namespace to avoid colliding with the
|
||||||
|
# user's environment) and reference that variable from
|
||||||
|
# PS1.
|
||||||
|
# note that the $ is escaped -- the variable will be
|
||||||
|
# expanded later (when it's time to draw the prompt)
|
||||||
|
if shopt -q promptvars; then
|
||||||
|
export __git_ps1_block="$branch"
|
||||||
|
ref="$ref_symbol \${__git_ps1_block}"
|
||||||
|
else
|
||||||
|
ref="$ref_symbol $branch"
|
||||||
|
fi
|
||||||
|
|
||||||
|
local marks
|
||||||
|
|
||||||
|
# check if HEAD is dirty
|
||||||
|
if [ -n "$($git_eng status --porcelain 2>/dev/null)" ]; then
|
||||||
|
dirty='y'
|
||||||
|
marks+=" $GIT_BRANCH_CHANGED_SYMBOL"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# how many commits local branch is ahead/behind of remote?
|
||||||
|
local stat; local aheadN; local behindN
|
||||||
|
stat="$($git_eng status --porcelain --branch 2>/dev/null | grep '^##' | grep -o '\[.\+\]$')"
|
||||||
|
aheadN="$(echo "$stat" | grep -o 'ahead [[:digit:]]\+' | grep -o '[[:digit:]]\+')"
|
||||||
|
behindN="$(echo "$stat" | grep -o 'behind [[:digit:]]\+' | grep -o '[[:digit:]]\+')"
|
||||||
|
[ -n "$aheadN" ] && marks+=" $GIT_NEED_PUSH_SYMBOL$aheadN"
|
||||||
|
[ -n "$behindN" ] && marks+=" $GIT_NEED_PULL_SYMBOL$behindN"
|
||||||
|
|
||||||
|
local bg; local fg
|
||||||
|
fg=$BLACK
|
||||||
|
if [ -z "$dirty" ]; then
|
||||||
|
bg=$GREEN
|
||||||
|
else
|
||||||
|
bg=$YELLOW
|
||||||
|
fi
|
||||||
|
|
||||||
|
__prompt_block $bg $fg "$ref$marks"
|
||||||
|
}
|
||||||
|
|
||||||
|
__virtualenv_block() {
|
||||||
|
# Copied from Python virtualenv's activate.sh script.
|
||||||
|
# https://github.com/pypa/virtualenv/blob/a9b4e673559a5beb24bac1a8fb81446dd84ec6ed/virtualenv_embedded/activate.sh#L62
|
||||||
|
# License: MIT
|
||||||
|
if [ -n "$VIRTUAL_ENV" ]; then
|
||||||
|
local venv
|
||||||
|
# In pcmode (and only pcmode) the contents of
|
||||||
|
# $gitstring are subject to expansion by the shell.
|
||||||
|
# Avoid putting the raw ref name in the prompt to
|
||||||
|
# protect the user from arbitrary code execution via
|
||||||
|
# specially crafted ref names (e.g., a ref named
|
||||||
|
# '$(IFS=_;cmd=sudo_rm_-rf_/;$cmd)' would execute
|
||||||
|
# 'sudo rm -rf /' when the prompt is drawn). Instead,
|
||||||
|
# put the ref name in a new global variable (in the
|
||||||
|
# __git_ps1_* namespace to avoid colliding with the
|
||||||
|
# user's environment) and reference that variable from
|
||||||
|
# PS1.
|
||||||
|
# note that the $ is escaped -- the variable will be
|
||||||
|
# expanded later (when it's time to draw the prompt)
|
||||||
|
if shopt -q promptvars; then
|
||||||
|
export __venv_ps1_block
|
||||||
|
__venv_ps1_block=$(basename "$VIRTUAL_ENV")
|
||||||
|
venv="$ref_symbol \${__venv_ps1_block}"
|
||||||
|
else
|
||||||
|
venv="$(basename "$VIRTUAL_ENV")"
|
||||||
|
fi
|
||||||
|
__prompt_block $WHITE $BLACK "$venv"
|
||||||
|
else
|
||||||
|
__block_text=''
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
__pwd_block() {
|
||||||
|
# Use ~ to represent $HOME prefix
|
||||||
|
local pwd; pwd=$(pwd | sed -e "s|^$HOME|~|")
|
||||||
|
# shellcheck disable=SC1001,SC2088
|
||||||
|
if [[ ( $pwd = ~\/*\/* || $pwd = \/*\/*/* ) && ${#pwd} -gt $MAX_PATH_LENGTH ]]; then
|
||||||
|
local IFS='/'
|
||||||
|
read -ra split <<< "$pwd"
|
||||||
|
if [[ $pwd = ~* ]]; then
|
||||||
|
pwd="~/${split[1]}/.../${split[*]:(-2):1}/${split[*]:(-1)}"
|
||||||
|
else
|
||||||
|
pwd="/${split[1]}/.../${split[*]:(-2):1}/${split[*]:(-1)}"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
__prompt_block $BLACK_BRIGHT $WHITE_BRIGHT "$pwd"
|
||||||
|
}
|
||||||
|
|
||||||
|
# superuser or not, here I go!
|
||||||
|
__user_block() {
|
||||||
|
# Colours to use
|
||||||
|
local fg=$WHITE_BRIGHT
|
||||||
|
local bg=$BLUE
|
||||||
|
|
||||||
|
if [[ ! -z "$SSH_CLIENT" ]]; then
|
||||||
|
local show_host="y"
|
||||||
|
bg=$CYAN
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "$(id -u "$USER")" ]; then
|
||||||
|
bg=$RED
|
||||||
|
fi
|
||||||
|
|
||||||
|
# shellcheck disable=SC2153
|
||||||
|
if [[ ! -z "${SHOW_USER+x}" || ( ! -z "${DEFAULT_USER+x}" && "$DEFAULT_USER" != "$(whoami)" ) ]]; then
|
||||||
|
local show_user="y"
|
||||||
|
fi
|
||||||
|
|
||||||
|
local text
|
||||||
|
if [ ! -z ${show_user+x} ]; then
|
||||||
|
text+="$BOLD$(whoami)"
|
||||||
|
fi
|
||||||
|
if [ ! -z ${show_host+x} ]; then
|
||||||
|
if [ ! -z "${text+x}" ]; then
|
||||||
|
text+="@"
|
||||||
|
fi
|
||||||
|
text+="\h"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ ! -z ${text+x} ]; then
|
||||||
|
__prompt_block $bg $fg $text
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
__status_block() {
|
||||||
|
local text
|
||||||
|
if [ "$exit_code" != 0 ]; then
|
||||||
|
__prompt_block $BLACK $RED '✘'
|
||||||
|
text+=$__block_text
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ "$(id -u "$USER")" == 0 ]; then
|
||||||
|
__prompt_block $BLACK $YELLOW '⚡'
|
||||||
|
text+=$__block_text
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ "$(jobs -l | wc -l)" != 0 ]; then
|
||||||
|
__prompt_block $BLACK $CYAN '⚙'
|
||||||
|
text+=$__block_text
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ ! -z "$text" ]; then
|
||||||
|
__block_text=$text
|
||||||
|
else
|
||||||
|
__block_text=''
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Build the prompt
|
||||||
|
prompt() {
|
||||||
|
# I don't like bash; execute first to capture correct status code
|
||||||
|
local exit_code=$?
|
||||||
|
# shellcheck disable=SC2091
|
||||||
|
$(history -a ; history -n)
|
||||||
|
|
||||||
|
last_bg=''
|
||||||
|
|
||||||
|
PS1=''
|
||||||
|
|
||||||
|
__status_block
|
||||||
|
PS1+=$__block_text
|
||||||
|
|
||||||
|
__virtualenv_block
|
||||||
|
PS1+=$__block_text
|
||||||
|
|
||||||
|
__user_block
|
||||||
|
PS1+=$__block_text
|
||||||
|
|
||||||
|
__pwd_block
|
||||||
|
PS1+=$__block_text
|
||||||
|
|
||||||
|
__git_block
|
||||||
|
PS1+=$__block_text
|
||||||
|
|
||||||
|
__end_block
|
||||||
|
PS1+=$__block_text
|
||||||
|
}
|
||||||
|
|
||||||
|
PROMPT_COMMAND=prompt
|
||||||
|
}
|
||||||
|
|
||||||
|
__powerline
|
||||||
|
unset __powerline
|
||||||
|
|
@ -627,6 +627,28 @@ function setup_firewall {
|
||||||
firewall_block_bad_ip_ranges
|
firewall_block_bad_ip_ranges
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function setup_powerline {
|
||||||
|
if [ -d ~/${PROJECT_NAME}/src/${PROJECT_NAME}-powerline ]; then
|
||||||
|
cp ~/${PROJECT_NAME}/src/${PROJECT_NAME}-powerline ~/.powerline.bash
|
||||||
|
cp ~/${PROJECT_NAME}/src/${PROJECT_NAME}-powerline /etc/skel/.powerline.bash
|
||||||
|
else
|
||||||
|
if [ -d /home/${MY_USERNAME}/${PROJECT_NAME}/src/${PROJECT_NAME}-powerline ]; then
|
||||||
|
cp /home/${MY_USERNAME}/${PROJECT_NAME}/src/${PROJECT_NAME}-powerline ~/.powerline.bash
|
||||||
|
cp /home/${MY_USERNAME}/${PROJECT_NAME}/src/${PROJECT_NAME}-powerline /etc/skel/.powerline.bash
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
if ! grep -q "powerline" ~/.bashrc; then
|
||||||
|
if [ -f ~/.powerline.bash ]; then
|
||||||
|
echo 'source ~/.powerline.bash' >> ~/.bashrc
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
if ! grep -q "powerline" /etc/skel/.bashrc; then
|
||||||
|
if [ -f /etc/skel/.powerline.bash ]; then
|
||||||
|
echo 'source ~/.powerline.bash' >> /etc/skel/.bashrc
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
function setup_utils {
|
function setup_utils {
|
||||||
read_config_param "PROJECT_REPO"
|
read_config_param "PROJECT_REPO"
|
||||||
write_config_param "PROJECT_REPO" "$PROJECT_REPO"
|
write_config_param "PROJECT_REPO" "$PROJECT_REPO"
|
||||||
|
|
@ -828,6 +850,9 @@ function setup_utils {
|
||||||
|
|
||||||
function_check create_usb_canary
|
function_check create_usb_canary
|
||||||
create_usb_canary
|
create_usb_canary
|
||||||
|
|
||||||
|
function_check setup_powerline
|
||||||
|
setup_powerline
|
||||||
}
|
}
|
||||||
|
|
||||||
function setup_email {
|
function setup_email {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue