diff options
Diffstat (limited to 'mkshrc')
| -rw-r--r-- | mkshrc | 677 |
1 files changed, 677 insertions, 0 deletions
@@ -0,0 +1,677 @@ +# Copyright 2021 Mitchell Riedstra + +# Permission to use, copy, modify, and/or distribute this software for any +# purpose with or without fee is hereby granted, provided that the above +# copyright notice and this permission notice appear in all copies. +# +# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH +# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY +# AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, +# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM +# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR +# OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR +# PERFORMANCE OF THIS SOFTWARE. +setaliases() { + alias ls="ls -F"; + alias lt="ls -tF"; + alias ll="ls -lhF"; + alias llt="ls -tlhF"; + alias g="grep"; + alias m="more"; + alias j="jobs"; + alias p="pwd"; + alias c="column" + alias ct="column -t" + alias eg='grep -E' + alias egv='grep -Ev' + alias search='grep -niRIE' + alias rot13="tr 'A-Za-z' 'N-ZA-Mn-za-m'" + alias setxkbmap-list="man xkeyboard-config" + showsshhosts() { + cat ~/.ssh/config ~/.ssh/inc/* | awk '/^[hH]ost /{$1=""; print $0}' + } + # docker-getip $container_id + alias docker-getip="docker inspect -f '{{range .NetworkSettings.Networks}}{{.IPAddress}}{{end}}'" + alias xa='xargs -I{}' + alias gdb='gdb -q' + alias python='/usr/bin/env python3 -q' + alias ncmpc='ncmpc -C -M' + # Fun fact, Gnucash doesn't play nice when built against musl as a libc... + # go figure + alias gnucash_locale_fix='export LC_ALL=C; unset LANGUAGE; gnucash' + pwgenForBanks() { + set -x + pwgen -r '[-#{};[:"'"'"'^()?><.|\/,`=]' -ncsy "$@" + set +x + } + + if [ "$(uname)" = "OpenBSD" ] ; then + sockstat() { + fstat | awk ' + BEGIN { + OFS="\t"; + } + { + if ( NR == 1 ) { + print($1, $2, $3, $5, $7, "ADDR"); + } + if( $0 ~ /tcp|udp/ ) { + if(!($0 ~ /[<-]-[->]/)) { + l = $1 "\t" $2 "\t" $3 "\t" $5 "\t" $7 "\t"; + if($7 == "tcp") { + services[$3] = l $9; + } else if ($7 == "udp") { + services[$3] = l $8; + } + } + } + } + END { + for(service in services) { + print(services[service]); + } + } + ' | column -t + } + ! [ -x "$(which seq 2>/dev/null)" ] && alias seq="echo use jot(1)" + fi + + if [ "$(uname)" = "Linux" ] ; then + # Sets capslock to be another control key, only works with PS2 keyboards + alias unfuck_capslock='setkeycodes 3a 29' + + alias bc='bc -q' + fi +} + +setcoloraliases() { + if [ "$(uname)" = "Linux" ] ; then + alias ls="ls --color=auto -F"; + alias lm='ls -lh --color=force | more'; + elif [ "$(uname)" = "FreeBSD" ] ; then + alias ls="ls -FG"; + alias lm='CLICOLOR_FORCE="1" ls -l | less -r'; + fi + + if echo "color test" | grep -q --color=auto "color test" >/dev/null 2>&1 ; then + alias grep="grep --color=auto"; + alias egrep='grep -E --color=auto' + alias search='grep -E --color=auto -rnI'; + fi + + #shellcheck disable=SC2139 + alias diff="$(command -v diff colordiff 2>/dev/null | tail -n 1)"; + alias t='tree -CdL' +} + +setsudoaliases() { + # Prefer doas if available, as it's likely to be configured by me over sudo + _sudo="$(command -v doas sudo 2>/dev/null | sed 1q)" + if [ -x "$_sudo" ] ; then + # Package managers + # Isn't it amazing how many distros you end up using? + alias apt-get="$_sudo apt-get" + alias apk="$_sudo apk" + alias apt="$_sudo apt" + alias apt-file="$_sudo apt-file" + alias yum="$_sudo yum" + alias dnf="$_sudo dnf" + alias pacman="$_sudo pacman" + alias xbps-install="$_sudo xbps-install" + alias xbps-remove="$_sudo xbps-remove" + alias pkg="$_sudo pkg" + alias pkg_add="$_sudo pkg_add" + + # Linux power utilities + alias tlp='$_sudo tlp' + alias tlp-stat='$_sudo tlp-stat' + + # Filesystem utilities + alias zfs="$_sudo zfs" + alias zpool="$_sudo zpool" + alias btrfs="$_sudo btrfs" + alias mount="$_sudo mount" + alias umount="$_sudo umount" + fi + case $_sudo in + *doas*) alias sudo='doas'; ;; + esac +} + +# E.g. after `.mkshrc` is called, perhaps in `~/.mkshrc.local` +# export GOPATH=$HOME/code/gopath +# export GOROOT=$HOME/go/1.13.5 +# set_go +set_go() { + export GOPATH="${GOPATH:-$HOME/go}" + export PATH="$GOPATH/bin:$PATH" + + GOROOT="${GOROOT:-/usr/local/go}" + _gobin="${GOROOT}/bin" + if [ -e "$_gobin" ] && ! echo "$PATH" | grep -q "$_gobin" ; then + export PATH="$_gobin:$PATH" + fi +} +_set_editor() { + export EDITOR="$1" + export VISUAL="$1" + alias vi="$EDITOR" + alias vim="$EDITOR" + alias edit="$EDITOR" + alias e="$EDITOR" +} +set_editor() { + for editor in vim vi ed ; do + unalias "$editor" >/dev/null 2>&1 + e="$(command -v "$editor")" + if [ -n "$e" ] && [ -x "$e" ] ; then + _set_editor "$e" + fi + break + done +} +set_pager() { + #shellcheck disable=SC2155 + export PAGER="$(command -v cat more less 2>/dev/null | tail -n1)" +} +set_lang() { +export CHARSET="$1"; +export LANG="$1"; +} + +# Usage debugstarttls $ipaddr:$port +debugstarttls() { + openssl s_client -starttls smtp -crlf -connect "$1" "$2" +} + +randmacgen() { + if [ "$(uname)" = "FreeBSD" ] ; then + LC_COLLATE=c + fi + tr -c -d '0123456789abcdef' < /dev/urandom \ + | head -c 12 \ + | sed 's!^M$!!;s!\-!!g;s!\.!!g;s!\(..\)!\1:!g;s!:$!!' + echo "" +} +# First argument is the interface on which to scan +ipv6_local_hosts() { +if [ -n "$1" ] ; then + interface="$1" + ping6 -c 3 "ff02::2%$interface" \ + | grep 'bytes from' \ + | awk '{print $4}' \ + | sort \ + | uniq \ + | sed -e's/:$//' +else + echo "You need to specify an interface" +fi +} +set_nocaps() { + setxkbmap -layout "us,de" + setxkbmap -option ctrl:nocaps + setxkbmap -option shift:breaks_caps + setxkbmap -option numpad:microsoft + setxkbmap -option grp:alt_space_toggle +} +# Usage: dir_oct:file_oct path +setperms() { + _perm="$1"; shift + dir_perms=$(echo "$_perm" | awk -F: '{print $1}') + file_perms=$(echo "$_perm" | awk -F: '{print $2}') + find "$@" -type f -print0 | xargs -0 chmod "$file_perms" + find "$@" -type d -print0 | xargs -0 chmod "$dir_perms" +} + +set_title() { + printf $'\033]0;'"%s"$'\007' "$1" +} + +set_my_title() { + _load="Load Avg: $(_sys_load)" + _free_mem="M Free: $(_sys_memory)" + _uptime="Uptime: $(_sys_uptime)" + set_title "$(id -un)[$(hostname)] -- $_load $_free_mem $_uptime" +} + +_sys_memory() { + ! [ -e "$(command -v vmstat)" ] && return + if [ "$(uname)" = "Linux" ] ; then + vmstat | tail -n1 | awk '{print $4/1024}' | sed -e's/\..*$//g' + elif [ "$(uname)" = "FreeBSD" ] ; then + vmstat | tail -n1 | awk '{print $5/1024}' | sed -e's/\..*$//g' + elif [ "$(uname)" = "OpenBSD" ] ; then + vmstat | sed -n '$p' | awk '{print $4}' | sed 's/M$//' + fi +} + +_sys_load() { + # Works on BSD too, neat. + uptime | sed -E 's/^.*load averages?: ([0-9]+\.[0-9]+).*$/\1/g' +} + +_sys_uptime() { + # Took awhile to figure out, but this works on BSD as well + uptime | sed -E -e's/^.*up[ ]+//g' -e's/^([^,]*),.*/\1/g' -e's/^([^ ]+)[ ]*(.).*/\1\2/g' +} + + +timestamp() { + date +%m.%d.%y_%H.%M.%S +} + +checkSSHAgent() { + ssh_agent_conf="$HOME/.ssh/agent" + if [ -e "$ssh_agent_conf" ] ; then + #shellcheck disable=SC1090 + . "$ssh_agent_conf" + fi + #shellcheck disable=SC2009 + if ! ps -eo pid | grep -q "$SSH_AGENT_PID" \ + || ! [ -e "$ssh_agent_conf" ] \ + || [ -z "$SSH_AGENT_PID" ] ; \ + then + ssh-agent -s | grep -v echo > "$ssh_agent_conf" + #shellcheck disable=SC1090 + . "$ssh_agent_conf" + fi +} + +checkSSHAgentGPG() { + export SSH_AUTH_SOCK="$(gpgconf --list-dirs agent-ssh-socket)" +} + +passenv() { + _f="$(mktemp)" + pass "$@" > "$_f" + #shellcheck disable=SC1090 + . "$_f" + rm -f "$_f" +} + + +###################################################################### +# Tmux utilities +_tmux_session() { +session="" +window_name="" +working_directory="$(pwd)" +while [ $# -gt 0 ] ; do case $1 in +--) shift ; break;; +-s) session="$2" ; shift ; shift ;; +-n) window_name="$2" ; shift ; shift ;; +-w) working_directory="$2" ; shift ; shift ;; +*)echo "Invalid option $1" ; return ;; +esac ; done + +cd "$working_directory" || { echo "Cannot change to: $working_directory" ; return; } + +tmux new-session -s "$session" -n "$window_name" -c "$working_directory" \;\ + "$@" +} + + +_tmux_dev() { +session="" +working_directory="$(pwd)" +window_name="editor" +while [ $# -gt 0 ] ; do case $1 in +--) shift ; break;; +-s) session="$2" ; shift ; shift ;; +-n) window_name="$2" ; shift ; shift ;; +-w) working_directory="$2" ; shift ; shift ;; +*)echo "Invalid option $1" ; return ;; +esac ; done + +cd "$working_directory" || { echo "Cannot change to: $working_directory" ; return; } + +_tmux_session -s "$session" -w "$working_directory" -n "$window_name" \ + -- \ + send-keys -t 0 "$EDITOR" \; send-keys -t 0 Enter \;\ + new-window -n "shell" \;\ + new-window -n "git" \;\ + send-keys -t 0 "git status" \; send-keys -t 0 Enter \;\ + "$@" + +} + +_tmux_servers_split_commands() { +n=0 +server_cmds="" +while [ $# -gt 0 ] ; do + if [ $n -eq 0 ] ; then + server_cmds='send-keys "ssh \"'"$1"'\"" \; send-keys Enter \; ' + else + server_cmds="${server_cmds} split-window \; "'send-keys "ssh \"'"$1"'\"" \; send-keys Enter \; ' + fi + n="$(echo "$n+1" | bc)" + shift +done +echo "$server_cmds" +} + +_tmux_servers() { +session="" +working_directory="$HOME" +servers="" +layout="even-vertical" +while [ $# -gt 0 ] ; do case $1 in +--) shift ; break;; +-s) session="$2" ; shift ; shift ;; +-w) working_directory="$2" ; shift ; shift ;; +-servers) servers="$2"; shift ; shift ;; +-l) layout="$2"; shift ; shift ;; +*)echo "Invalid option $1" ; return ;; +esac ; done +cd "$working_directory" || { echo "Cannot change to: $working_directory" ; return; } + +layout="select-layout $layout ';' set-window-option synchronize-panes on ';'" + +eval _tmux_session -s "\$session" -w "\$working_directory" -n "main" \ + -- \ + $(_tmux_servers_split_commands $servers) \ + "$layout" \ + "\$@" \ + +} + +gitdate() { + date="" + while [ $# -gt 0 ] ; do case $1 in + -d) date="$2" ; shift ; shift ;; + *) printf "Bad option: $1... usage:\n$0 -d \"\$date_string\"\n" ; return ;; + esac ; done + + date="$(date --date "$date")" + + export GIT_AUTHOR_DATE="$date" + export GIT_COMMITTER_DATE="$date" + echo export GIT_AUTHOR_DATE="$date" + echo export GIT_COMMITTER_DATE="$date" +} + +gitdateUnset() { + unset GIT_AUTHOR_DATE + unset GIT_COMMITTER_DATE +} + +gpgTmpDir() { + set -x + if echo "$GNUPGHOME" | grep -q "/dev/shm/"; then + rm -rvf "$GNUPGHOME" + fi + + #shellcheck disable=SC2155 + export GNUPGHOME="$(mktemp -d /dev/shm/GPG.XXXXXX)" + set +x +} + +setcolorterminal() { + case $TERM in + xterm-kitty) + # Prevent xterm-kitty from being turned into xterm-256color + return; + ;; + xterm*) + export TERM=xterm-256color + # printf $'\033]0;'"$(hostname)"$'\007'; + ;; + screen*) + export TERM=screen-256color + # printf $'\033k'"$(whoami)"$'\033\\'; + ;; + esac +} + +setlscolors() { + if [ "$(uname)" = "Linux" ] ; then + export LS_COLORS='di=36;40:ln=1;31;40:so=32;40:pi=33;40:ex=1;32;40:bd=34;46:cd=34;43:su=0;41:sg=0;46:tw=0;42:ow=0;43:' + elif [ "$(uname)" = "FreeBSD" ] ; then + export LSCOLORS='gxBxcxdxCxegedabagacad' + fi +} + +colormanpages() { + export LESS_TERMCAP_mb="[01;33m" + export LESS_TERMCAP_md="[01;33m" + export LESS_TERMCAP_me="[0m" + export LESS_TERMCAP_se="[0m" + export LESS_TERMCAP_so="[01;44;33m" + export LESS_TERMCAP_ue="[0m" + export LESS_TERMCAP_us="[01;36m" + + ## Gentoo is nuts + export GROFF_NO_SGR=1 +} + +pkg_switch_branch() { + _f="/etc/pkg/FreeBSD.conf" + cp $_f ${_f}.orig + sed -e's/quarterly/latest/' < ${_f}.orig > ${_f} +} +# System setup `chroot`, useful mainly on Linux +syschroot() { + CHDIR=$1 + PROG=$2 + CUR_PS1="$PS1" + PS1="(CHROOT) # " + mount -t proc none "$CHDIR/proc" + mount -o bind /dev "$CHDIR/dev" + mount -o bind /dev/pts "$CHDIR/dev/pts" + mount -o bind /sys "$CHDIR/sys" + mount -o bind /run "$CHDIR/run" + if [ "$PROG" = "/bin/bash" ] ; then + PS1="(CHROOT) [ \u@\h \w ] # " + fi + chroot "$CHDIR" "$PROG" + umount "$CHDIR/dev/pts" "$CHDIR/dev" "$CHDIR/sys" "$CHDIR/proc" \ + "$CHDIR/run" + PS1="$CUR_PS1" +} +pull() { + _f="" + _u="" + while [ $# -gt 0 ] ; do case $1 in + -f) _f="$2"; shift; shift ;; + -u) _u="$2"; shift; shift ;; + *) echo "pull usage: -f <output> -u <url>"; return ;; + esac ; done + echo "file: $_f" + echo "url: $_u" + if [ -z "$_f" ] || [ -z "$_u" ] ; then + echo "pull usage: -f <output> -u <url>"; return + fi + + if [ -x "$(command -v wget)" ] ; then + wget "$_u" -O "$_f" && return + fi + + if [ -x "$(command -v curl)" ] ; then + curl "$_u" -o "$_f" && return + fi + + if [ -x "$(command -v ftp)" ] && [ "$(uname)" = "OpenBSD" ] ; then + ftp "$_u" -o "$_f" && return + fi + + if [ -x "$(command -v fetch)" ] && [ "$(uname)" = "FreeBSD" ] ; then + fetch "$_u" -o "$_f" && return + fi + + + echo "Unable to fetch $_u" >&2 +} +pulldotfile() { + file="$1" + if [ -z "$file" ] ; then + echo "pulldotfile usage: <dotfilename>" + echo "Will pull from ${UPDATE_URL}/<dotfilename> and save to:" + #shellcheck disable=SC2088 + echo "~/.<dotfilename>" + return + fi + destfile="$(basename "${file}")" + pull -u "${UPDATE_URL}${file}" -f "$HOME/.${destfile}" +} +_make_ssh_dir_if_not_exists() { + auth_keys="$HOME/.ssh/authorized_keys" + if ! [ -d "$HOME/.ssh" ] ; then mkdir "$HOME/.ssh" ; fi + chmod 700 "$HOME/.ssh" + chmod 600 "$auth_keys" +} +_pullkeys() { + url="$1"; shift + _timestamp="$(timestamp)" + _make_ssh_dir_if_not_exists + auth_keys="$HOME/.ssh/authorized_keys" + mv "$auth_keys" "${auth_keys}.${_timestamp}" || \ + echo "Authorized Keys do not currently exist" + pull -u "$url" -f "$auth_keys" || \ + mv "${auth_keys}.${_timestamp}" "$auth_keys" + ssh-keygen -lf "$auth_keys" || \ + mv "${auth_keys}.${_timestamp}" "$auth_keys" + + chmod 600 "$auth_keys" +} +pullkeys() { + file="keys/$1"; shift; + url="${UPDATE_URL}${file}" + _pullkeys "$url" +} +pullkeys_github() { + _username="$1"; shift; + url="https://github.com/${_username}.keys" + _pullkeys "$url" +} +pulltermcolors() { + file="$HOME/bin/terminal-colors" + if ! [ -d "$HOME/bin" ] ; then + mkdir "$HOME/bin" + fi + pull -u "${UPDATE_URL}/util/terminal-colors" -f "$file" + chmod +x "$file" +} +updatetmuxconf() { + pulldotfile "tmux/tmux.conf" + pulldotfile "tmux/tmux_helper.sh" + if ! [ -e "$HOME/.tmux.conf.local" ] && [ "$(id -u)" -eq 0 ] ; then + # If I ever change the color from green this will have to be updated I guess + grep 'green' "$HOME/.tmux.conf" | sed -e's/green/red/g' > "$HOME/.tmux.conf.local" + elif ! [ -e "$HOME/.tmux.conf.local" ] ; then + touch "$HOME/.tmux.conf.local" + fi +} +updategitconf() { + file="gitconfig/$1" + pull -u "${UPDATE_URL}${file}" -f "$HOME/.gitconfig" +} +updateshell() { + pulldotfile "kshrc" + #shellcheck disable=SC1090 + . "$HOME/.kshrc" +} +updatevimrc() { + OLDWD="$(pwd)" + #shellcheck disable=SC2164 + cd "$HOME" + git clone --recurse-submodules "$VIM_GIT_URL" .vim + exit="$?" + cd "$HOME/.vim" || { echo "Cannot cd to .vim dir" ; return; } + if [ "$exit" -gt 0 ] ; then + git pull origin master + git submodule update + fi + ./link-vimrc.sh + #shellcheck disable=SC2164 + cd "$OLDWD" +} +updatevimrc_basic() { + pull -u "$VIM_BASIC_URL" -f ~/.vimrc +} + + +SHELL_CONFIG=".kshrc" +COLOR=0 +case "$TERM" in + xterm*) COLOR=1;; + *256color) COLOR=1;; +esac + +if ! echo "$PATH" | grep -q "$HOME/bin" ; then + export PATH="$HOME/bin:$PATH" +fi + +UPDATE_URL="https://git.riedstra.dev/mitch/dotfiles/plain/" +VIM_GIT_URL="https://git.riedstra.dev/mitch/vim-cfg" +VIM_BASIC_URL="https://git.riedstra.dev/mitch/dotfiles/plain/vimrc" + +PRELOAD="$HOME/${SHELL_CONFIG}.pre" +LOCAL="$HOME/${SHELL_CONFIG}.local" + +# Don't put any variables below this, as they can't be overridden before the +# functions if you do that + +# Load pre config +[ -r "$PRELOAD" ] && . "$PRELOAD" + +set_pager + +# note that the use of set_editor and _set_editor on oksh may +# change the input mode. ( set -o emacs ) after to go back +set_editor # Also, opinionated, hope you like vim/vi +# set_go + +export PASSWORD_STORE_ENABLE_EXTENSIONS=true + +setaliases +setsudoaliases + +# checkSSHAgent + + +# Support various different shells +if [ -n "$KSH_VERSION" ] ; then + export HISTFILE="$HOME/.ksh_history" + export HISTSIZE=10000 + + set -o emacs + bind ^L=clear-screen + #shellcheck disable=SC2155 + + # if you want PWD + # export PS1='\u@\h \$PWD \$ ' + export PS1='\u@\h\$ ' + + if [ $(id -u) -eq 0 ] ; then + export PS1='\h\$ ' + fi + +elif [ "$SHELL" = "/bin/sh" ] && [ "$(uname)" = "FreeBSD" ] ; then + set -o emacs + #shellcheck disable=SC2155 + export PS1="$(whoami)"'@\h:\w \$ ' +elif [ -n "$BASH" ] ; then + HISTFILE="$HOME/.bash_history" + HISTSIZE=10000 + HISTFILESIZE=10000 + shopt -s histappend + export PROMPT_COMMAND="history -a; history -c; history -r" + + # if you want PWD + # export PS1='\u@\h \$PWD \$ ' + export PS1='\u@\h\$ ' + + if [ $(id -u) -eq 0 ] ; then + export PS1='\h\$ ' + fi +fi + +if [ $COLOR -eq 1 ] ; then + colormanpages + setcolorterminal + setcoloraliases + setlscolors +fi + +# Include local configuration options +[ -r "$LOCAL" ] && . "$LOCAL" |