.bashrc

#!/bin/bash

# If not running interactively, don't do anything
[ -z "$PS1" ] && return

# Set up colors
if [ -z "$SSH_CLIENT" ] && [ -z "$SSH_TTY" ] && [ -z "$VIM_TERMINAL" ]; then
	# shellcheck disable=SC1090
	. ~/.kkrc/base16-colors.dark.sh 2>/dev/null
fi

# If there was a .bashrc we moved away at install, run that first
# (so we can override)
# shellcheck disable=SC1090
if [ -e ~/.bashrc.orig ]; then . ~/.bashrc.orig; fi

# Ignore both duplicated and whitespace
# (This is inserted separately to the front of the the inject file)
HISTCONTROL=ignoreboth

# OS X specifics, before
if [[ "$OSTYPE" == "darwin"* ]]; then
	# Force load bash-completion on OS X
	# We need to do this BEFORE the generic part, or else `compopt` is unknown
	# shellcheck disable=SC1091
	. "/opt/homebrew/etc/profile.d/bash_completion.sh" 2>/dev/null
	# If we have `brew install coreutils`, then use the linux-compatible `ls`
	# NOTE: Have this before defining `lll`, so that uses gls too
	[ -f /opt/homebrew/bin/gls ] && alias ls="gls --color"
fi

# START of part to be injected

# I need VI keys
set -o vi

# Disable Debian/Ubuntu's annoying "command not found" script
unset -f command_not_found_handle

# Disable history
unset HISTFILE

# Clear out other people's stupid history from my session
#history -c

# Make history show date and time
HISTTIMEFORMAT="%F %T "

# NOTE: We're not exporting the prompt: this is important especially in tmux
# inject.  This is because in a subshell, other important things (most
# notably HISTFILE, aliases, etc) are NOT exported anyway or might be
# overwritten, and the plain prompt is a reminder that the environment is NOT
# fully set up.
#
# Colored prompt. Displays user@host, current dir, and job count. Same as KKRC's zsh prompt with RPROMPT turned off. Root detection only on setup.
# NOTE: If you want a different colored hostname somewhere, copy this line to .bashrc.local, and change the "32m" part to, say "34m".
PS1='\[\033[00;'$([[ "$UID" -eq 0 ]]&&echo -n 31||echo -n 34)'m\]\u\[\033[00m\]@\[\033[00;32m\]\h \[\033[00;33m\]\w \[\033[00;36m\][\j]\[\033[00m\]\$ '
# Or, if ANSI is problematic:
#PS1="\u@\h \w [\j]\$ "

# Set GNU screen title to hostname
# Shellcheck mis-detects this, so:
# shellcheck disable=SC1003
[ -z "$STY" ] && printf '\ek%s\e\\' "${HOSTNAME:${#HOSTNAME}<11?0:-10}"

# "bell" before prompt. Separated from PS1 so it's easier to turn off when needed,
# and at least clears any erroneous local PROMPT_COMMAND.
export PROMPT_COMMAND="printf '\a'"

# Set up some necessary environment variables
export EDITOR=vim
export PAGER=less
export LC_ALL="en_US.UTF-8" # No, don't try export LC_ALL="C", will mess up your UTF-8
export LANG="en_US.UTF-8"
# Not supported widely enough yet :(
#export LESS="--mouse"
export LESSHISTFILE=/dev/null

# Colors! :)
export CLICOLOR=1
# BSD colors
export LSCOLORS=ExFxCxDxBxegedabagacad
# Linux colors -- set always because of zsh's list-colors
export LS_COLORS="di=1;34:ln=1;35:so=1;32:pi=1;33:ex=1;31:bd=34;46:cd=34;43:su=30;41:sg=30;46:tw=30;42:ow=30;43"

# Most useful quoting style, because you can copy/paste it
export QUOTING_STYLE=shell-escape

# Debian/Ubuntu should just restart things during upgrade.
# It's pointless and dangerous to ask the user (because the update gets stuck).
export NEEDRESTART_MODE=a

# FOR INJECT: Lighter vim
# NOTE: this prevented the use of kkrc on the server side, since if there is a
# VIMINIT env var, then ~/.vimrc doesn't get read. But I leave it here because
# it might be useful if the server has a wildly configured vim.
#export VIMINIT=":set nobackup noswapfile encoding=utf8 viminfo="
# No swapfile, no viminfo and a few other things
# Fortunately packadd does not give an error on pre-9.1 versions (before editorconfig was added to vim by default)
alias vim='vim -n -i NONE "+set nobackup noswapfile encoding=utf8 mouse=a" +"packadd editorconfig"'

# Use bash-completion (but show no errors if not found).
# shellcheck disable=SC1091
. /usr/share/bash-completion/bash_completion 2>/dev/null

# Set up some handy aliases
#alias l="ls -lrt" # Shows only non-hidden files.
alias l="ls -lrtA" # Always shows hidden files too.
alias la="ls -lrtA -I*" # For Linux. Hidden files ONLY.
#alias la="ls -lrtd .*" # For stupider systems (OS X, ash, etc), works only in current dir
# A version of ls that is still quick to type, BUT uses "less" automatically BUT exits it immediately if text is not long enough.
# NOTE: use this in the "function ..." form, because if there is an ll alias, that causes an error (even if we unalias in the previos lik
unalias ll 2>/dev/null # Many systems has an ll alias, THIS IS NOT TEMPORARY
# No, we WANT to use ls, so:
# shellcheck disable=2012
function ll() { ls -lrtA --color "$@" | less -FXRn +G ; }
#alias lr="ls -AR1 -I .git|awk '/:$/{gsub(/[^\/]+\//,\"--\",\$0);printf(\"%d files\n%s \t\",p-2,\$0);p=0}{p++}END{print p \" files\"}'|less -FXn" # Cut -FX in ash
#alias bell="printf '\a'" # either echo -ne '\007' or printf '\a'" or tput bel
alias h="history"
# PostgreSQL with readline
function psql { INPUTRC=/dev/fd/9 command psql 9<<<'set editing-mode vi' "$@" ; }
# PostgreSQL as above, but as postgres user
function ppsql { sudo -u postgres -- bash -c "$(declare -f psql); psql \"\$@\"" bash "$@" ; }
# MySQL with readline
alias mysql="INPUTRC=/dev/fd/9 mysql 9<<<'set editing-mode vi'"
# This is getting even uglier, but must have on remote machines
alias tig='TIGRC_USER=/dev/fd/9 tig 9<<<"set main-options = --all${IFS}set main-view = line-number:no,interval=5 id:yes date:relative author:abbreviated commit-title:yes,graph,refs,overflow=no"'
alias ts="tig status"
alias gl="git log --graph --pretty=format:'%Cred%h%Creset -%C(auto)%d%Creset %s %Cgreen(%cr) %C(cyan)<%an>%Creset' --abbrev-commit --date=relative --all --date-order"
alias gs="git status -sb"
alias gf="git fetch --all -v"
alias gp="git pull --ff-only -v"
alias gclean="git reset --hard && git clean -f -d -x"

# Recursive git
function G { find . -name .git -type d | while read -r a; do a="${a%.git}"; tput smso; echo -e "\n$a"; tput rmso; if [ "$#" -lt 1 ]; then command git -C "$a" status -sb; else command git -C "$a" "$@"; fi; done ; }
# Recursive git status. This is to quickly find uncommitted changes, not a detailed view
function GS { find . -name .git -type d | while read -r a; do a="${a%.git}"; tput smso; echo -e "$a"; tput rmso; if ! command git -C "$a" diff-index --quiet --ignore-submodules HEAD --; then if [ "$#" -lt 3 ]; then command git -C "$a" status -sb; else command git -C "$a" "$@"; fi; fi ; done ; }

# Show .gitignore-d files, all of them
alias gii="git ls-files --exclude-standard --ignored --others"
# Show .gitignore-d files except vendor and node_modules, because that's TMI
alias gi="gii | egrep -v '^vendor/|^node_modules/'"
# Quick grep, with case and ignore case.
# NOTE: GR_EXCLUDE is an array!
export GR_EXCLUDE=(-I --exclude-dir=.git --exclude=*.min.* --exclude=*.sql --exclude=*.log --exclude=tags --exclude-dir=cache --exclude-dir=vendor --exclude-dir=node_modules --exclude-dir=storage)
function gr { command grep --color=auto -r "${GR_EXCLUDE[@]}" "$@" . 2>/dev/null ; }
function ggr { command grep --color=force -r "${GR_EXCLUDE[@]}" "$@" . 2>/dev/null | less -FSXnr ; }
# Better git grep
function gg { git grep -I "$@" -- :^vendor/ :^public/vendor/ :^node_modules/ :^*.sql :^*.min.* ; }

# GNU screen with SSH_AUTH_SOCK filename transfer, to be used with `CTRL+A` `:paste s`, tmux-bound to <c-q><c-r>a
# You can append a PID if there are multiple screen-s.
unalias s 2>/dev/null # XXX Tempotarily, while I have running sessions having the alias
# We WANT this to expand when defined, so:
# shellcheck disable=SC2139
function s() { screen ${1:+-S} $1 -X register s " export SSH_AUTH_SOCK=\"$SSH_AUTH_SOCK\"" ; screen -xR "$1" ; }

# Watch out for using git as a different user than the repository. Avoid mandatory reconfiguration of git with user/email for hotfixes.
function git { if [[ -O "$(command git rev-parse --show-toplevel 2>/dev/null)/.git" || " config grep log blame diff show status init clone " == *" $1 "* ]]; then command git -c user.email="$USER@$HOSTNAME" -c user.name="$USER" "$@"; else echo "Please use the unix user that owns .git"; return 1; fi }
# Anyone else here remember when `mount` and `df` were 2-3 actual disks...?
M() { mount "$@" | grep '^\/dev\/' ; }
D() { df -h "$@" | grep -v 'snap\|^tmpfs\|^udev\|^none\|^overlay\|^shm' ; }
# Free memory
F() { free -h ; }
# Process list overview (for Linux)
alias P="ps axfwwo pid,user,start,rss,stat,cmd | less -SXRn"
# lxd/lxc list conatiners
alias lxl='lxc list -c ns46tSbNm,image.release'
# lxd/lxc show port forwards XXX not pretty if no forwards
function lxp { for ip in $(lxc network forward list lxdbr0 -f csv | awk -F, -e '{ print $1 }'); do lxc network forward show lxdbr0 "$ip"; done ; }
# With MAC addresses:
#alias lxl='lxc list -c ns46tSbNm,image.release,volatile.lxdbr0.hwaddr:lxdbr0,volatile.eth0.hwaddr:eth0'
# If can't use docker as current user, try sudo
# We define a shell command, but we don't mind that sudo will not use it, so:
# shellcheck disable=SC2033,SC2032
function docker { if [[ -r /var/run/docker.sock ]] ; then command docker "$@" ; else sudo docker "$@" ; fi ; }
# shellcheck disable=SC2033,SC2032
function docker-compose { if [[ -r /var/run/docker.sock ]] ; then command docker-compose "$@" ; else sudo docker-compose "$@" ; fi ; }
# Docker containers overview
alias C="docker ps -as"
# Docker-compose
alias dc="docker compose"
_completion_loader docker-compose 2>/dev/null # This is some new dynamic style? Show no error if not found.
complete -F _docker_compose dc
alias dc1="dc up -d; dcs"
alias dc0="dc down"
alias dcs="dc ps -a"
alias dci="dc images; echo; docker images"
# List docker and docker-compose images
alias I="docker images"
# Docker Swarm
alias S="docker service ls"
# Kubernetes overview. Using an alias instead of a function because often kubectl is an alias itself... (minikube, etc)
alias K="kubectl get all --output=wide"
# Kubernetes overview, including system
alias KA="kubectl get all --output=wide --all-namespaces"
# Kubernetes nodes (not included in "all")
alias KN="kubectl get nodes --output=wide"
# Firefox private window
alias ff="firefox --new-instance --private-window &"
# LVM
alias lv="pvs; echo; vgs; echo; lvs -o lv_name,pool_lv,size,data_percent,metadata_percent,origin"
# lsblk
alias lb="lsblk -Mf"
# Replicate zsh's "vared" command (with autocompletion)
function vared { read -r -e -p "$1=" -i "${!1}" "$1" ; }
complete -v vared

# Only if not on busybox
[ -L "$(type -p grep)" ] || alias grep="grep --color"

# Shell options.
# histverify (NOT USED NOW): Poor man's history expansion (which bash doesn't do on TAB)
# globstar: To have "**" as in zsh
# cdspell, dirspell: Spell checking on tab expansion
# checkwinsize: Set LINES and COLUMNS
# cmdhist: Save multi-line commands as one command
# extglob (NOT USED to avoid scripting mistakes): better/more parameter substitution pattern matching options, for example *([0-9])
shopt -s globstar cdspell dirspell checkwinsize cmdhist

# Configure readline
bind 'TAB:menu-complete'
bind '"\e[Z": menu-complete-backward'
bind 'set menu-complete-display-prefix on'
bind 'set show-all-if-ambiguous on'
bind 'set show-all-if-unmodified on'
bind 'set completion-ignore-case on'
bind 'set match-hidden-files off'
bind 'set colored-stats on'
bind 'set colored-completion-prefix on'
bind 'set visible-stats on'
bind 'set completion-prefix-display-length 1'
bind 'set skip-completed-text on'
bind 'set history-preserve-point on'
bind 'set mark-symlinked-directories on'
# History expansion on space
bind 'space:magic-space'

# Better history stepping, both in insert and command mode
# HACK: Assigning functions to non-existent keys, so a few lines down we can do two things on one keypress. This is all about the jump to end-of-line
bind -m vi '"\200":previous-history'
bind -m vi '"\201":next-history'
bind -m vi '"\202":end-of-line'
bind -m vi 'k:"\200\202"'
bind -m vi 'j:"\201\202"'
# NOTE: use `tput rmkx` or `reset` when the terminal is sending \eOA instead of \e[A (stuck in "app mode")
bind -m vi '"\e[A":"\200\202"'
bind -m vi '"\e[B":"\201\202"'
bind -m vi-insert '"\e[A":history-search-backward'
bind -m vi-insert '"\e[B":history-search-forward'
# And I still want CTRL-L in insert mode
bind -m vi-insert "C-l:clear-screen"

# Restore default completion for cd, since bash-completion doesn't handle wildcards
compopt -o bashdefault cd

# START of better systemd.
# Makes it almost usable, works around brain-dead-ness, even add some comfort:
# - Unlike simple aliases, we still bash-complete.
# - We use $SC to store (automatically) the unit we're working on, so no need for typing all the time, or history athletics. One rarely works on multiple services at once.
# - Functions start, reload and restart show the status of how it went, and also tails the log/journal afterwards. Exit tail with CTRL-c.

# Generic helper functions to set up bash-autocomplete for our aliases
_scx() { COMP_WORDS=("systemctl" "$1" "${COMP_WORDS[@]:1}") ; ((COMP_CWORD++)) ; _systemctl ; }
_jcx() { COMP_WORDS=("journalctl" "$1" "${COMP_WORDS[@]:1}") ; ((COMP_CWORD++)) ; _journalctl ; }

# "SystemCtl", generic alias, because who has time to type THAT much?
sc() { SC="${*: -1}" ; systemctl "$@" ; }
complete -F _systemctl sc
# "User-level SystemCtl" alias, becuase, there's NOT even a short-option or something...
usc() { SC="${*: -1}" ; systemctl --user "$@" ; }
_usc() { _scx --user ; }
complete -F _usc usc
# "SystemCtl List" services, because it should be simple
scl() { systemctl list-units --type service --all ; }
# Systemctl Daemon-Reload", because systemd can't do it on its own... Do user or system depending on UID
sdr() { if [ $UID -eq 0 ]; then systemctl daemon-reload; else systemctl --user daemon-reload; fi ; }

# JournalCtl for Unit
jc() { SC="${1:-${SC}}" ; journalctl -xu "$SC" ; }
_jc() { _jcx "--unit" ; }
complete -F _jc jc
# ...with "tail -f"
jcf() { SC="${1:-${SC}}" ; journalctl -xefu "$SC" ; }
complete -F _jc jcf
# User-level JournalCtrl for Unit
ujc() { SC="${1:-${SC}}" ; journalctl -x --user-unit "$SC" ; }
_ujc() { _jcx "--user-unit" ; }
complete -F _ujc ujc

# Reusable command to show status afterwards, and tail the log during reload. Exit with CTRL-c
# - We use daemon-reload automatically, because very often the thing you modified for a restart is the unit file.
# - We use reset-failed, because for some incomprehensible reason, the restart-count limit imposed by StartLimitInterval[Sec] also affects manual restarts.
stail() { SC="${2:-${SC}}" ; sdr ; systemctl reset-failed "$SC" ; journalctl -n 0 -xfu "$SC" & systemctl "$1" "$SC" ; scs ; fg %journalctl ; }
# Param #1: desired function name
# Param #2: command to call
# Param #3: command action to call
# Param #4+: any further options to pass to action
_mksc() {
	eval "$1() { SC=\"\${1:-\${SC}}\" ; $2 $3 ${*:4} \"\$SC\" ; }"
	eval "_$1() { _scx \"$3\" ; }"
	eval "complete -F _$1 $1"
}
# Definitions:
_mksc scs systemctl status -l --no-pager # status, but don't trim lines
_mksc sc0 systemctl stop
_mksc sc1 stail start
_mksc scr stail reload-or-restart
_mksc scR stail restart
# Override sc0 to add an additional `scs` to show status afterwards
sc0() { SC="${1:-${SC}}" ; systemctl stop "$SC" ; scs ; }

# Now fix bash competion for our systemd aliases.
# Even without bash-completion, most linux package managers put these there from the systemd packages - take advantage.
# Display no error if not found, since it's not fully standard.
# shellcheck disable=SC1091
. /usr/share/bash-completion/completions/systemctl 2>/dev/null
# shellcheck disable=SC1091
. /usr/share/bash-completion/completions/journalctl 2>/dev/null
# END of better systemd.

# END of part to be injected

# Commands which are not required in remote inject
alias tmux="tmux -2"
# When I don't want to pollute my known_hosts file (temporary VMs, etc)
alias sssh="ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no"
# SSH reverse-compatible with RSA (routers with dropbear, etc)
alias ssh-with-rsa-enabled="ssh -o PubkeyAcceptedAlgorithms=+ssh-rsa -o HostkeyAlgorithms=+ssh-rsa"

# Locally we don't need these (but leave them in the inject part)
unalias tig
#unset VIMINIT

# Linux hardware keyboard setup options.
# Reset to US keyboard
alias kbd-reset="setxkbmap -layout us"
# A) For PC-style hardware (Order: Win, Alt, Space)
alias kbd-pc="kbd-reset;xmodmap -e 'keysym Super_L = Mode_switch' ~/.Xmodmap"
# B) For MAC-style hardware (Order: Alt, Win, Space)
alias kbd-mac="kbd-reset;xmodmap -e 'keysym Alt_L = Mode_switch' -e 'keysym Super_L = Alt_L' ~/.Xmodmap"

# Quickly create/list/delete VMs on DigitalOcean.
# NOTE: You can set "export DIGITALOCEAN_ACCESS_TOKEN=..." in ~/.bashrc.local , or use `doctl auth` to log in
do-mk() { doctl compute droplet create "${1:-tmp1}" --region ams3 --ssh-keys "$(doctl compute ssh-key list --format=ID --no-header | paste -sd "," -)" --image "${2:-ubuntu-22-04-x64}" --size "${3:-s-2vcpu-2gb}" --wait -v ; }
do-ssh() { ( HN="${1:-tmp1}"; shift; ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no "root@$(doctl compute droplet list --format=PublicIPv4 --no-header "${HN}")" "$@" ) ; }
do-ls() { doctl compute droplet list --format Name,ID,PublicIPv4,Memory,VCPUs,Disk,Region,Status; doctl account get --format Email,DropletLimit,Status ; }
do-rm() { doctl compute droplet delete "$(doctl compute droplet list --format=ID --no-header "${1:-tmp1}")" ; }

# hl - highlight command
# shellcheck disable=SC1090
source ~/.kkrc/hl
export -f hl

# Automatically set TMUX window title on SSH
ssh() {
	# Only if running under TMUX
	if [ -n "$TMUX" ]; then
		# Store current window name
		#local SAVED=$(tmux display-message -p '#W')
		local ARGS=("$@")
		local NAME="ssh"
		local I
		# Try to find the server name
		for I in "${ARGS[@]}" ; do
			[[ $I == "--" ]] && break
			NAME="$I"
		done
		# Set window name
		tmux rename-window "${NAME}" >/dev/null 2>&1
	fi
	# Do it
	command ssh "$@"
	# To Restore window name automatically:
	#tmux rename-window "$SAVED" >/dev/null 2>/dev/null
	# To switch back to auto-renaming after disconnection:
	#tmux set-window-option automatic-rename "on" >/dev/null 2>/dev/null
	# Restore from alternate mode (if set),
	# and move cursor to the last line (so if ssh lost connection in the middle of a full-screen app like VI, then don't leave the cursor in the middle of some content).
	printf '\e[?47l\e[99B'
}

# Strictly NOT in inject, just LOCAL: open files from vim :term back in VIM.
if [ -n "$VIM_TERMINAL" ]; then
	unalias vim
	function vim() {
		printf '\e]51;["drop", "%s"]\g' "$(realpath "$1")"
	}
	export -f vim
fi

# Display screens if any
screen -ls | grep -v "Socket"

# Turn CAPS LOCK into CTRL on Linux.
#setxkbmap -option ctrl:nocaps
# Or. on Ubuntu, put this into /etc/default/keyboard: XKBOPTIONS="ctrl:nocaps"

# Local commands
# shellcheck disable=SC1090
if [ -e ~/.bashrc.local ]; then . ~/.bashrc.local; fi