Dot_Files/.profile

#!/bin/bash

### Opts ###
ulimit -n 10000 # Allows for more open files

# shellcheck source=./.zstyle
source ~/.zstyle

. "$HOME/.cargo/env"

### Pathing ###

export PATH="$HOME/.bin:$PATH"
export PATH="$PATH:/opt/homebrew/Cellar/llvm/13.0.0_2/bin"

# Python with pyenv
export PYENV_ROOT="${HOME}/.pyenv"
export PATH="${PATH}:${PYENV_ROOT}/bin"
eval "$(pyenv init -)"

# Mac pathing beneath
if [[ "$OSTYPE" = "darwin"* ]]; then
	export PATH="/opt/homebrew/opt/coreutils/libexec/gnubin:$PATH"
	export PATH="/opt/homebrew/opt/gnu-sed/libexec/gnubin:$PATH"
	export PATH="/Applications/Firefox Developer Edition.app/Contents/MacOS:$PATH"
	export PATH="$HOME/.rustup/toolchains/stable-aarch64-apple-darwin/bin:$PATH"
	export MANPATH="/opt/homebrew/opt/coreutils/libexec/gnuman:$MANPATH"

	complete -o nospace -C /opt/homebrew/bin/terraform terraform

	# LLDB
	export PATH="/opt/homebrew/opt/llvm/bin:$PATH"
fi

### OPTS ###
setopt extendedglob
setopt nullglob

### Environment Variables ###
export LANG=en_US.UTF-8
if [[ -f "/usr/libexec/java_home" ]]; then
	export JAVA_HOME="$(/usr/libexec/java_home)"
fi
export AWS_CLI_AUTO_PROMPT=on
export ZSH_HIGHLIGHT_MAXLENGTH=10000
export ZSH_AUTOSUGGEST_BUFFER_MAX_SIZE=2
export LANG=en_US.UTF-8
export NOTES_DIR="$HOME/.notes"
export DOCKER_BUILDKIT=1

# Fzf Variables
export FZF_DEFAULT_COMMAND='fd --hidden --exclude .git --color=always'
export FZF_DEFAULT_OPTS="--height=80% --layout=reverse --info=inline --border --margin=1 --ansi"
export FZF_CTRL_T_COMMAND="${FZF_DEFAULT_COMMAND}"

# Gitlab
export GITLAB_HOST="https://gitlab.orion-technologies.io"

if command -v nvim >/dev/null 2>&1; then
	export EDITOR=nvim
	export VISUAL=nvim
	export MANPAGER="nvim +Man!"
elif command -v vim >/dev/null 2>&1; then
	export EDITOR=vim
	export VISUAL=vim
else
	export EDITOR=vi
	export VISUAL=less
fi

# Nvim Vars
export NVIM_ENVS_DIR="${HOME}/.nvim-environments"

### Terminal Specific Configuration ###
# If using the kitty terminal we want to set our TERM var to be xterm as kitty will send
# kitty-xterm which causes a fucking headache and a half due to ncurses not containing
# that by default
if [[ "$TERM" = *"xterm-kitty"* ]]; then
	alias ssh="TERM=xterm-256color ssh"
	alias icat="kitty +kitten icat"
	__kitty_complete
fi

if [[ "${TERM}" = "wezterm" ]]; then
	alias ssh="TERM=xterm-256color ssh"
fi

### Zsh Vi-Mode Configuration ###

# Makes the cursor change based on the mode
export VI_MODE_SET_CURSOR=true
# Set the editor to neovim for visual mode
export ZVM_VI_EDITOR="$EDITOR"
# This ensures that fzf completion works
zvm_after_init_commands+=('[[  -r ~/.fzf.zsh ]] && source ~/.fzf.zsh')

### Aliases ###

## Exa ls Replacement Configuration ##
if command -v exa >/dev/null 2>&1; then
	alias ls="exa --icons --group --group-directories-first --octal-permissions --classify"
	alias l="ls -alh"
	alias ll="l"
	alias tree="ls --tree"
fi

## Git Aliases ##
if command -v git >/dev/null 2>&1; then
	alias gc="git commit"
	alias gcm="git commit -m"
	alias ga="git add"
	alias gp="git push"
	alias gb="git branch"
	alias gco="git checkout"
	alias gpl="git pull"
	alias gs="git status"
	alias gst="git stash"
	alias gstc="git stash clear"
	alias gsw="git switch"
	alias gd="git diff"
	alias gr="git remote"
	alias gl="git log"
	alias gw="git worktree"
	alias gwa="git worktree add"
	alias gwr="git worktree remove"
	alias git-remote="git config --get remote.origin.url"
	alias gbc="git config remote.origin.fetch '+refs/heads/*:refs/remotes/origin/*'"
fi

gbf() {
	local remote
	local default_branch
	remote="${1:=origin}"
	default_branch="$(git remote show "$remote" | awk '/HEAD branch/ {print $NF}')"
	git fetch "$remote" "$default_branch":"$default_branch"
}
## Mac Specific Aliases ##
if [[ "$OSTYPE" = "darwin"* ]]; then
	alias c="pbcopy"
	alias p="pbpaste"
	alias grep="ggrep"
	alias find="gfind"
	alias sed="gsed"
fi

## Neovim Aliases ##
if command -v nvim >/dev/null 2>&1; then
	alias n="nvim"
fi

## Bat (Alternative to Cat) Aliases ##
if command -v bat >/dev/null 2>&1; then
	alias cat="bat"
fi

## Misc Aliases ##
alias e="$EDITOR"
alias ghidra="ghidraRun"
alias v="vim"
alias :q="exit"
alias Get-Public-IPV4="dig @resolver4.opendns.com myip.opendns.com +short -4"
alias Get-Public-IPV6="dig @resolver1.ipv6-sandbox.opendns.com AAAA myip.opendns.com +short -6"
alias :profile="$EDITOR $HOME/.profile"
alias cv="command -v"

### Completions ###

_import_custom_bash_completions() {
	local custom_comps=(
		"az"
	)
	for file in "${custom_comps[@]}"; do
		source "${HOMEBREW_PREFIX}/etc/bash_completion.d/${file}"
	done
}

# Homebrew completions
if type brew &>/dev/null; then
	# Zsh
	FPATH="$(brew --prefix)/share/zsh/site-functions:${FPATH}"
	_import_custom_bash_completions

	autoload -Uz compinit
	compinit
fi

# Completions from .zsh directory
if [[ -d "$HOME/.zsh/completions" ]]; then
	FPATH="$HOME/.zsh/completions:${FPATH}"

	autoload -Uz compinit
	compinit
fi

### Functions ###
del() {
	local BASE_TRASH_DIR
	BASE_TRASH_DIR="$HOME/.Trash/"

	local trash_del_dir
	trash_del_dir="$BASE_TRASH_DIR/del-$(date +%Y-%m-%dT%H:%M:%S)"

	mkdir -p "$trash_del_dir"

	for i in "${@}"; do
		mv "$i" "$trash_del_dir"
	done
}

DAP-Python-Gen() {
	local module
	module="${1}"
	name="${2:=Local Module}"

	mkdir -p .vscode
	cat <<__EOF__ >.vscode/launch.json
{
	"configurations": [
		{
			"name": "Launch ${name}",
			"type": "python",
			"request": "launch",
			"module": "${module}"
		}
	]
}
__EOF__
}

TCPDump-Capture() {
	tcpdump -qns 0 -X -r "${1}"
}

Known-Hosts() {
	local prev_ifs
	prev_ifs="$IFS"
	while IFS=$'\n' read -r line; do
		echo "$line" | cut -d " " -f1
	done <~/.ssh/known_hosts
	IFS="$prev_ifs"
}

Work-Hours() {
	local WORK_HOURS_PATH
	WORK_HOURS_PATH=~/Vircio/Work-Hours
	mkdir -p "$WORK_HOURS_PATH"
	"$EDITOR" "$WORK_HOURS_PATH/$(date +%Y-%m-%d)"
}

date_iso_8601() {
	date +%Y-%m-%dT%H:%M:%S
}

echo_rgb() {
	# Echo a colored string to the terminal based on rgb values
	#
	# NOTE: This function will only truly work with terminals that support TRUE COLOR, see:
	#   https://gist.github.com/XVilka/8346728
	#
	# Positional Arguments:
	#
	#   message <type: string> <position: 1> <required: true>
	#	 - The message to be printed to stdout
	#   red <type: int> <position: 2> <required: true>
	#	 - The red value from 0 to 255
	#   green <type: int> <position: 3> <required: true>
	#	 - The green value from 0 to 255
	#   blue <type: int> <position: 4> <required: true>
	#	 - The blue value from 0 to 255
	#   bg_red <type: int> <position: 5> <required: false>
	#	 - The background red value from 0 to 255
	#   bg_green <type: int> <position: 6> <required: false>
	#	 - The background green value from 0 to 255
	#   bg_blue <type: int> <position: 7> <required: false>
	#	 - The background blue value from 0 to 255
	#
	# Usage:
	#   echo_rgb "Yep" 10 80 30
	#   echo_rgb "DESTROY MY EYES" 255 0 255  0 255 0
	#
	# POSIX Compliant:
	#   N/A
	#

	local red
	local green
	local blue
	local input

	local bg_red
	local bg_green
	local bg_blue

	input="${1}"
	red="${2}"
	green="${3}"
	blue="${4}"
	bg_red="${5}"
	bg_green="${6}"
	bg_blue="${7}"

	for num in "${@:2}"; do
		[[ ! "$num" =~ [0-9] ]] &&
			printf "%s\n" "Given RGB value was not a number, received $num" 1>&2 &&
			return 1
		[[ "$num" -gt 255 ]] &&
			printf "%s\n" "Given RGB value must be less than 255, received $num" 1>&2 &&
			return 1
		[[ "$num" -lt 0 ]] &&
			printf "%s\n" "Given RGB value must be more than 0, received $num" 1>&2 &&
			return 1
	done

	if [ -t 1 ]; then
		if [ "${5}" != "" ]; then
			[[ -z "${6}" ]] && echo "A value must be passed for bg_green" && return 1
			[[ -z "${7}" ]] && echo "A value must be passed for bg_blue" && return 1
			printf "\033[38;2;%s;%s;%s;48;2;%s;%s;%sm%s\033[m\n" \
				"$red" "$green" "$blue" "$bg_red" "$bg_green" "$bg_blue" "$input"
		else
			printf "\033[0;38;2;%s;%s;%sm%s\033[m\n" "$red" "$green" "$blue" "$input"
		fi
	else
		printf "%s\n" "$input"
	fi
}

chr() {
	python3 -c "print(chr(int('${1}')))"
}

ord() {
	python3 -c "print(ord('${1}'))"
}

tmp() {
	(
		export MY_SHLVL=tmp:$MY_SHLVL
		export od=$PWD
		tmp=$(mktemp -d)
		export tmp
		trap 'rm -rf ${tmp}' 0
		if ! cd "$tmp"; then
			printf "%s \"%s\"\n" "Unable to cd into" "$tmp"
			exit 1
		fi
		if [ "$1" = "" ]; then
			"$SHELL" -l
		else
			[ "$1" = "-l" ] && {
				shift
				set "${@}" ";" "$SHELL -l"
			}
			eval "${@}"
		fi
	)
}

cdl() {
	if [[ -f "${1}" ]]; then
		cd "$(dirname "$(readlink -f "${1}")")"
	else
		echo_rgb "No such file: ${1}" 255 0 0 30 0 0
	fi
}

nvim-env() {
	# If using ZSH uncomment the line below, requires emulate
	emulate -L ksh
	local nvim_env
	local nvim_env_path
	local nvim_env_full_path

	nvim_env="${1:-primary}"
	nvim_env_path="${NVIM_ENVS_DIR}"
	nvim_env_full_path="$nvim_env_path/$nvim_env"

	declare -A nvim_syml_paths=(["$HOME/.config/nvim"]="$nvim_env_full_path/config" ["$HOME/.local/share/nvim"]="$nvim_env_full_path/share")
	mkdir -p "${nvim_env_full_path}/share"

	local red='\033[0;31m'
	local cyan='\033[0;36m'
	local green='\033[0;32m'
	local light_green='\033[1;32m'
	local reset='\033[0m'

	# Check that the given environment exists
	if [[ ! -d "$nvim_env_full_path" ]]; then
		printf "${red}Error:\n\tUnable to locate environment, ${reset}\"${cyan}%s\"\n\t\"%s\"\n${reset}" "$nvim_env" "$nvim_env_full_path" >&2
		return 1
	fi

	local symlink_src
	local symlink_dst
	for syml_key in "${!nvim_syml_paths[@]}"; do
		symlink_src="${nvim_syml_paths[$syml_key]}"
		symlink_dst="$syml_key"

		# Check that the configuration path exist from our env
		if [[ ! -d "$symlink_src" ]]; then
			printf "${red}ERROR:\n\tUnable to find configuration:${reset} \"${cyan}%s${reset}\"${red} in environment ${reset}\"${cyan}%s${reset}\"${red}\n\tConsider creating it with ${reset} \"${cyan}mkdir -p %s${reset}\"\n" "$symlink_src" "$nvim_env" "$symlink_src" >&2
			return 1
		fi

		# If a directory (not a symlink) exists in where we want to write a symlink refuse to overwrite
		if [[ -r "$symlink_dst" ]] && [[ ! -L "$symlink_dst" ]]; then
			printf "${red}ERROR:\n\tNeovim configuration to be replaced is not a symlink, please remove the files before proceeding, issue file:${reset}\n\t\"${cyan}%s${reset}\"\n" "$symlink_dst" >&2
			return 1
		fi

		# If we find an existing symlink, try to delete it
		if [[ -L "$symlink_dst" ]]; then
			local linked_sym
			linked_sym="$(readlink "$symlink_dst")"
			printf "${light_green}Attemping to remove symlink:${reset}\n\t${cyan}%s${reset} -> ${cyan}%s${reset}\n" "$linked_sym" "$symlink_dst"
			if ! rm "$symlink_dst"; then
				printf "${red}ERROR:\n\tFailed to remove symlink:${reset}\n\t${cyan}%s${reset} -> ${cyan}%s${reset}\n" "$linked_sym" "$symlink_dst" >&2
				return 1
			else
				printf "${light_green}Successfully removed symlink:${reset}\n\t${cyan}%s${reset} -> ${cyan}%s${reset}\n" "$linked_sym" "$symlink_dst"
			fi
		fi

		# Actually write the env now we're past our guards
		printf "${light_green}Linking:${reset}\n\t${cyan}%s${reset} -> ${cyan}%s${reset}\n" "$symlink_src" "$symlink_dst"
		if ! ln -s "$symlink_src" "$symlink_dst"; then
			printf "${red}ERROR:\n\tFailed to link:${reset}\n\t${cyan}%s${reset} -> ${cyan}%s${reset}\n" "$symlink_src" "$symlink_dst" >&2
			return 1
		else
			printf "${light_green}Successfully linked:${reset}\n\t${cyan}%s${reset} -> ${cyan}%s${reset}\n" "$symlink_src" "$symlink_dst"
		fi
		printf "${green}%.s─${reset}" $(seq 1 $(tput cols))
		printf "\n"
	done
}

### Used to open firefox from terminal with search ###
browser_search() {
	local browser
	local search_term
	browser="firefox"
	search_term="${*}"
	eval "$browser -search \"$search_term\""
}

### Search Aliases ###
if command -v browser_search >/dev/null 2>&1; then
	alias s="browser_search"
	if ! which search >/dev/null 2>&1; then
		alias search="browser_search"
	fi
fi

watcher() {
	local command
	local sleep_time
	local exit_code

	command="${1}"
	sleep_time="${2:-5}"
	while :; do
		clear
		printf "%s  |  %s  |  %s\n" "Command: \"$command\"" "Every $sleep_time seconds" "$(date)"
		printf "%*s\n" "${COLUMNS:-$(tput cols)}" "" | tr " " -
		eval "$command"
		exit_code="${?}"
		if ((exit_code != 0)); then
			break
		fi
		sleep "$sleep_time"
	done

	return "$((exit_code))"
}

update_cargo_crates() {
	cargo install "$(cargo install --list | grep -E '^[a-z0-9_-]+ v[0-9.]+:$' | cut -f1 -d" ")"
}

### FZF FUNCTIONS ###
# shellcheck disable=1036,1072,1056,1009,1073
wgrep() {
	local selection
	read -r selection < =(rg --color=always --line-number --no-heading --smart-case "${*:-}" | fzf --ansi \
		--color "hl:-1:underline,hl+:-1:underline:reverse" \
		--delimiter : \
		--preview 'bat --color=always {1} --highlight-line {2}' \
		--preview-window 'up,60%,border-bottom,+{2}+3/3,~3')

	# This crazy shit handles egde cases of files having colons (:) in them
	selection=$(rev <<< $selection | cut -d ':' -f 3- | rev)
	if [[ -n "${selection}" ]]; then
		[[ -r "${selection}" ]] && "${EDITOR}" "${selection}" || echo_rgb "Cannot open ${selection}" 255 0 0 80 0 0
	fi

}

pods() {
	FZF_DEFAULT_COMMAND="kubectl get pods --all-namespaces" \
		fzf --info=inline --layout=reverse --header-lines=1 \
		--prompt "$(kubectl config current-context | sed 's/-context$//')> " \
		--header $'╱ Enter (kubectl exec) ╱ CTRL-O (open log in editor) ╱ CTRL-R (reload) ╱\n\n' \
		--bind 'ctrl-/:change-preview-window(80%,border-bottom|hidden|)' \
		--bind 'enter:execute:kubectl exec -it --namespace {1} {2} -- bash > /dev/tty' \
		--bind 'ctrl-o:execute:${EDITOR:-vim} <(kubectl logs --all-containers --namespace {1} {2}) > /dev/tty' \
		--bind 'ctrl-r:reload:$FZF_DEFAULT_COMMAND' \
		--preview-window up:follow \
		--preview 'kubectl logs --follow --all-containers --tail=10000 --namespace {1} {2}' "$@"
}

killer() {
	(
		date
		ps -ef
	) |
		fzf --bind='ctrl-r:reload(date; ps -ef)' \
			--header=$'Press CTRL-R to reload\n\n' --header-lines=2 \
			--preview='echo {}' --preview-window=down,3,wrap \
			--layout=reverse --height=80% | awk '{print $2}' | xargs kill -"${1:-9}"
}
. "$HOME/.cargo/env"