shrc.sh

#!/bin/bash
# shellcheck disable=SC2155

# Colourful manpages
export LESS_TERMCAP_mb=$'\E[01;31m'
export LESS_TERMCAP_md=$'\E[01;31m'
export LESS_TERMCAP_me=$'\E[0m'
export LESS_TERMCAP_se=$'\E[0m'
export LESS_TERMCAP_so=$'\E[01;44;33m'
export LESS_TERMCAP_ue=$'\E[0m'
export LESS_TERMCAP_us=$'\E[01;32m'

# Set to avoid `env` output from changing console colour
export LESS_TERMEND=$'\E[0m'

# Print field by number
field() {
  ruby -ane "puts \$F[$1]"
}

# Setup PATH

# Remove from anywhere in PATH
remove_from_path() {
  [[ -d "$1" ]] || return
  PATHSUB=":${PATH}:"
  PATHSUB=${PATHSUB//:$1:/:}
  PATHSUB=${PATHSUB#:}
  PATHSUB=${PATHSUB%:}
  export PATH="${PATHSUB}"
}

# Add to the start of PATH if it exists
add_to_path_start() {
  [[ -d "$1" ]] || return
  remove_from_path "$1"
  export PATH="$1:${PATH}"
}

# Add to the end of PATH if it exists
add_to_path_end() {
  [[ -d "$1" ]] || return
  remove_from_path "$1"
  export PATH="${PATH}:$1"
}

# Add to PATH even if it doesn't exist
force_add_to_path_start() {
  remove_from_path "$1"
  export PATH="$1:${PATH}"
}

quiet_which() {
  command -v "$1" >/dev/null
}

if [[ -n "${MACOS}" ]]
then
  add_to_path_start "/opt/homebrew/bin"
  add_to_path_start "/opt/workbrew/bin"
elif [[ -n "${LINUX}" ]]
then
  add_to_path_start "/home/linuxbrew/.linuxbrew/bin"
fi

add_to_path_start "/usr/local/bin"
add_to_path_end "${HOME}/.dotfiles/bin"

# Setup Go development
export GOPATH="${HOME}/.gopath"
add_to_path_end "${GOPATH}/bin"

# Aliases
alias mkdir="mkdir -vp"
alias df="df -H"
alias rm="rm -iv"
alias mv="mv -iv"
alias cp="cp -irv"
alias du="du -sh"
alias make="nice make"
alias less="less --ignore-case --raw-control-chars"
alias rsync="rsync --partial --progress --human-readable --compress"
alias rg="rg --colors 'match:style:nobold' --colors 'path:style:nobold'"
alias be="bundle exec"
alias sha256="shasum -a 256"
alias perlsed="perl -p -e"

# Command-specific stuff
if quiet_which brew
then
  eval "$(brew shellenv)"

  export HOMEBREW_DEVELOPER=1
  export HOMEBREW_BOOTSNAP=1
  export HOMEBREW_BUNDLE_INSTALL_CLEANUP=1
  export HOMEBREW_BUNDLE_DUMP_DESCRIBE=1
  export HOMEBREW_NO_ENV_HINTS=1
  export HOMEBREW_CLEANUP_PERIODIC_FULL_DAYS=1
  export HOMEBREW_CLEANUP_MAX_AGE_DAYS=30
  export HOMEBREW_ENFORCE_SBOM=1
  export HOMEBREW_NO_VERIFY_ATTESTATIONS=1

  add_to_path_end "${HOMEBREW_PREFIX}/Library/Homebrew/shims/gems"

  # Specifically want this to expand when defined, not when run.
  # shellcheck disable=SC2139
  alias homebrew="${HOMEBREW_PREFIX}/bin/brew"
  alias portableruby="${HOMEBREW_PREFIX}/Library/Homebrew/vendor/portable-ruby/current/bin/ruby"
  alias portablebundle="${HOMEBREW_PREFIX}/Library/Homebrew/vendor/portable-ruby/current/bin/bundle"

  alias workbrew='/opt/workbrew/bin/brew'
  alias workbrewdo='sudo --set-home --preserve-env --user=workbrew --'
  alias workbrewpermissions='sudo chown -R workbrew /opt/homebrew; sudo chmod -R g+w /opt/homebrew; sudo chmod -R g-w /opt/homebrew/var'
  alias youtube-dl='yt-dlp'

  alias hbc='cd $HOMEBREW_REPOSITORY/Library/Taps/homebrew/homebrew-core'
fi

if quiet_which delta
then
  export GIT_PAGER='delta'
else
  # shellcheck disable=SC2016
  export GIT_PAGER='less -+$LESS -RX'
fi

if quiet_which eza
then
  alias ls="eza --classify --group --git"
elif [[ -n "${MACOS}" ]]
then
  alias ls="ls -F"
else
  alias ls="ls -F --color=auto"
fi

if quiet_which bat
then
  export BAT_THEME="ansi"
  alias cat="bat"
  export HOMEBREW_BAT=1
fi

if quiet_which prettyping
then
  alias ping="prettyping --nolegend"
fi

if quiet_which dust
then
  alias du="dust"
fi

if quiet_which duf
then
  alias df="duf"
fi

if quiet_which htop
then
  alias top="sudo htop"
fi

# Configure environment
export CLICOLOR=1

# OS-specific configuration
if [[ -n "${MACOS}" ]]
then
  export GREP_OPTIONS="--color=auto"
  export VAGRANT_DEFAULT_PROVIDER="vmware_fusion"

  add_to_path_end "${HOMEBREW_PREFIX}/opt/git/share/git-core/contrib/diff-highlight"
  add_to_path_end "/Applications/Visual Studio Code.app/Contents/Resources/app/bin"

  alias fork="/Applications/Fork.app/Contents/Resources/fork_cli"

  alias locate="mdfind -name"
  alias finder-hide="setfile -a V"

  # Load GITHUB_TOKEN from macOS keychain
  export GITHUB_TOKEN=$(
    printf "protocol=https\\nhost=github.com\\n" |
      git credential fill |
      perl -lne '/password=(gho_.+)/ && print "$1"'
  )
  export HOMEBREW_GITHUB_API_TOKEN="${GITHUB_TOKEN}"
  export JEKYLL_GITHUB_TOKEN="${GITHUB_TOKEN}"

  # output what's listening on the supplied port
  on-port() {
    sudo lsof -nP -i4TCP:"$1"
  }

  # make no-argument find Just Work.
  find() {
    local arg
    local path_arg
    local dot_arg

    for arg
    do
      [[ ${arg} =~ "^-" ]] && break
      path_arg="${arg}"
    done

    [[ -z "${path_arg}" ]] && dot_arg="."

    command find "${dot_arg}" "$@"
  }

  # Only run this if it's not already running
  pgrep -fq touchid-enable-pam-sudo || touchid-enable-pam-sudo --quiet
elif [[ -n "${LINUX}" ]]
then
  quiet_which keychain && eval "$(keychain -q --eval --agents ssh id_rsa)"

  # Run dircolors if it exists
  quiet_which dircolors && eval "$(dircolors -b)"

  add_to_path_end "/data/github/shell/bin"
  add_to_path_start "/workspaces/github/bin"

  alias su="/bin/su -"
  alias open="xdg-open"
elif [[ -n "${WINDOWS}" ]]
then
  open() {
    # shellcheck disable=SC2145
    cmd /C"$@"
  }
fi
# Run rbenv/nodenv if they exist
if quiet_which rbenv
then
  shims="$(rbenv root)/shims"
  if ! [[ -d "${shims}" ]]
  then
    rbenv rehash
  fi
  add_to_path_start "${shims}"
fi

if quiet_which nodenv
then
  shims="$(nodenv root)/shims"
  if ! [[ -d "${shims}" ]]
  then
    nodenv rehash
  fi
  add_to_path_start "${shims}"
fi

# Set up editor
if quiet_which code
then
  export EDITOR="code"
  export GIT_EDITOR="${EDITOR} -w"
  export SVN_EDITOR="${GIT_EDITOR}"

  code() {
    local arg

    # mkdir/touch any files that don't exist because the `open` hack doesn't work for them.
    for arg
    do
      [[ -e "${arg}" ]] && break

      command mkdir -p "$(dirname "${arg}")"
      touch "${arg}"
    done

    open -b "com.microsoft.VSCode" "$@"
  }

  # Edit Rails credentials in VSCode
  rails-credentials-edit-production() {
    EDITOR="code -w" bundle exec rails credentials:edit --environment production
  }
  rails-credentials-edit-development() {
    EDITOR="code -w" bundle exec rails credentials:edit --environment development
  }
else
  export EDITOR="vim"
fi

# Save directory changes
cd() {
  builtin cd "$@" || return
  [[ -n "${TERMINALAPP}" ]] && command -v set_terminal_app_pwd >/dev/null &&
    set_terminal_app_pwd
  pwd >"${HOME}/.lastpwd"
  ls
}

# Use ruby-prof to generate a call stack
ruby-call-stack() {
  ruby-prof --printer=call_stack --file=call_stack.html -- "$@"
}

# Pretty-print JSON files
json() {
  [[ -n "$1" ]] || return
  cat "$1" | jq .
}

# Pretty-print Homebrew install receipts
receipt() {
  [[ -n "$1" ]] || return
  json "${HOMEBREW_PREFIX}/opt/$1/INSTALL_RECEIPT.json"
}

# Move files to the Trash folder
trash() {
  mv "$@" "${HOME}/.Trash/"
}

# GitHub API shortcut
github-api-curl() {
  curl -H "Authorization: token ${GITHUB_TOKEN}" "https://api.github.com/$1" | jq .
}

# Spit out Okta keychain password
okta-keychain() {
  security find-generic-password -l device_trust '-w'
}

# Clear entire screen buffer
clearer() {
  tput reset
}

# Sit/stand desk
desk_stand() {
  curl -X POST http://10.0.0.45/button/desk_preset_1/press
}
desk_sit() {
  curl -X POST http://10.0.0.45/button/desk_preset_2/press
}