.bash_functions

#!/usr/bin/env bash

# example output:
#  GNU/Linux Debian 11 (bullseye) {5.10.0-18-amd64}
#  GNU/Linux Raspbian 10 (buster) {5.10.103-v7+} armv7l
#  GNU/Linux Ubuntu 22.04 (jammy) {5.15.0-47-generic} amd64
#  Android {4.4.302-perf+} aarch64
get_sys_info() {
	local DIST=$(uname -o)
	if [[ "${OSTYPE}" == linux* ]]; then
		if command -v lsb_release > /dev/null; then
			DIST="${DIST} $(lsb_release -iscr | tr "\n" " " | sed -r 's/([^ ]+) ([^ ]+) ([^ ]+)/\1 \2 (\3)/')"
		fi
	fi
	echo -n ${DIST}

	local ARCH="$(uname -m)"
	if [[ "${ARCH}" == x86_64 ]]; then
		ARCH="amd64"
	fi

	local KERNEL=$(uname -r)
	echo -n " {${KERNEL}}"
	if [[ "${KERNEL}" != *"${ARCH}"* ]]; then
		echo -n " $ARCH"
	fi
}

hello_msg() {
	if [[ "${TERM}" == "linux" ]]; then
		echo
	fi
	echo -e "$(tput bold)PID:$(tput sgr0)  $(tput setaf 5)$$$(tput sgr0)"
	echo -e "$(tput bold)DATE:$(tput sgr0) $(tput setaf 6)$(date)$(tput sgr0)"
	echo -e "$(tput bold)UPT:$(tput sgr0)  $(tput setaf 9)$(uptime -p)$(tput sgr0)"
	echo -e "$(tput bold)HOME:$(tput sgr0) $(tput setaf 3)${HOME}$(tput sgr0)"
	echo -e "$(tput bold)SYS:$(tput sgr0)  $(tput setaf 4)$(get_sys_info)$(tput sgr0)"

	if [[ -f /sys/firmware/devicetree/base/model ]]; then
		# example output:
		#  Raspberry Pi 3 Model B Rev 1.2 (744M)
		#  Raspberry Pi 2 Model B Rev 1.1 (999M)
		#  Xunlong Orange Pi One (999M)
		#  OrangePi 3 LTS (2,0G)
		#  Qualcomm Technologies, Inc. SM 8350 v2.1 MTP (7.7G)
		local dt_model="$(tr -d '\0' < /sys/firmware/devicetree/base/model)"
		local dt_mem="$(free -h --si | awk '/^Mem:/{print $2}')"
		echo -e "$(tput bold)DTM:$(tput sgr0)  $(tput setaf 13)${dt_model} $(tput bold)(${dt_mem})$(tput sgr0)"
	fi

	# '--all-ip-addresses' unrecognized option in darwin/termux
	#local ips="$(hostname --all-ip-addresses)"
	if command -v ip > /dev/null; then
		local ips="$(ip -o addr show scope global | awk '{gsub(/\/.*/, " ",$4); print $4}' | tr  '\n' ' ')"
	fi

	if [[ -n "${ips}" ]]; then
		ips=$(echo -e ${ips} | sed "s/ /$(echo -e $(tput sgr0)),\n      $(echo -e $(tput setaf 1))/g") # separator color fix
		echo -e "$(tput bold)IPs:$(tput sgr0)  $(tput setaf 1)${ips}$(tput sgr0)"
	fi

	if [[ -n "${SSH_CLIENT}" ]]; then
		echo -e "$(tput bold)SSH:$(tput sgr0)  $(tput setaf 2)${SSH_CLIENT}$(tput sgr0)"
	fi
}

if [[ -n "${SHOW_HELLO_MSG}" ]]; then
	hello_msg
fi


# @depends: xttitle
if [[ "${TERM}" == "xterm" || "${TERM}" == "xterm-256color" ]] && command -v xttitle > /dev/null; then
	cd() {
		builtin cd "$@"
		xttitle "$$ [${USER}@${HOSTNAME}] ${PWD}" 2> /dev/null
	}

	xttitle "$$ [${USER}@${HOSTNAME}] ${PWD}" 2> /dev/null
fi

# fork
f() {
	setsid uxterm &> /dev/null
}

# colorful ls
# @depends: gawk
l() {
	ls -lAh --color=always --group-directories-first $* | \
		awk \
			-v black=$(tput setaf 0) \
			-v red=$(tput setaf 1) \
			-v green=$(tput setaf 2) \
			-v yellow=$(tput setaf 3) \
			-v blue=$(tput setaf 4) \
			-v magenta=$(tput setaf 5) \
			-v cyan=$(tput setaf 6) \
			-v white=$(tput setaf 7) \
			-v reset=$(tput sgr0) \
			-v user=$USER \
		'
		# for trim first space
		function ltrim(s) {
			sub(/^ /, "", s);
			return s;
		}
		# colorize owner
		function get_owner_color(owner) {
			if (owner == "root") {
				return red;
			} else if (owner == user) {
				return yellow;
			} else {
				return "";
			}
		}
		BEGIN {
			FPAT = "([[:space:]]*[^[:space:]]+)";
			OFS = "";
		}
		{
			if (NR > 1) { # skip 1st line \w total
				# permissions
				k = 0; for (i = 0; i <= 8; ++i) { # "-rw-r--r--" -> "644"
					k += ((substr($1, i+2, 1) ~ /[rwx]/) * 2 ^ (8 - i));
				}
				if (k) printf("%0o ", k); # mode
				#$1 = magenta$1reset;     # default
				# link count
				$2 = blue$2reset;
				# owner (wrap "[]")
				# user
				$3 = ltrim($3);
				$3 = " ["get_owner_color($3)$3reset;
				# group
				$4 = ltrim($4);
				$4 = " "get_owner_color($4)$4reset"]";
				# size
				$5 = green$5reset;
				# datetime (wrap "()")
				$6 = ltrim($6);
				$6 = " ("cyan$6reset; # month
				$7 = cyan$7reset;     # day
				$8 = cyan$8reset")";  # time
				print
			}
		}
		'
}
alias ll="l"

# colorful ps
# @depends: gawk
p() {
	if [[ -n "$*" ]]; then
		args=$*
	else
		args="aux"
	fi

	ps $args | \
		awk \
			-v bbold=$(tput smso) \
			-v bold=$(tput bold) \
			-v black=$(tput setaf 0) \
			-v red=$(tput setaf 1) \
			-v green=$(tput setaf 2) \
			-v yellow=$(tput setaf 3) \
			-v blue=$(tput setaf 4) \
			-v magenta=$(tput setaf 5) \
			-v cyan=$(tput setaf 6) \
			-v white=$(tput setaf 7) \
			-v reset=$(tput sgr0) \
			-v violet=$(tput setaf 13) \
			-v user=$USER \
		'
		# for trim first space
		function ltrim(s) {
			sub(/^ /, "", s);
			return s;
		}
		# colorize owner
		function get_owner_color(owner) {
			if (owner == "root") {
				return red;
			} else if (owner == user) {
				return yellow;
			} else {
				return "";
			}
		}
		BEGIN {
			FPAT = "([[:space:]]*[^[:space:]]+)";
			OFS = "";
		}
		{
			if (NR > 1) { # skip 1st line \w header
				# USER
				$1 = get_owner_color($1)$1reset;
				# PID
				$2 = magenta$2reset;
				# CPU
				$3 = green$3reset;
				# MEM
				$4 = violet$4reset;
				# VSZ
				#$5
				# RSS
				#$6
				# TTY
				#$7
				# STAT
				$8 = bold$8reset;
				# START
				$9 = blue$9reset;
				# TIME
				$10 = cyan$10reset;
				# COMMAND
				$11 = white$11reset;
				print;
			} else {
				$0 = bbold$0reset;
				print;
			}
		}
		'
}

# colorful memory usage
function pm() {
	p -eo user,pid,%cpu,size,vsz,rss,tty,stat,start,time,command --sort -rss | head -n 10
}

# colorful mount
# @depends: gawk
function _mount() {
	mount | column -t $* | \
		awk \
			-v bbold=$(tput smso) \
			-v bold=$(tput bold) \
			-v black=$(tput setaf 0) \
			-v red=$(tput setaf 1) \
			-v green=$(tput setaf 2) \
			-v yellow=$(tput setaf 3) \
			-v blue=$(tput setaf 4) \
			-v magenta=$(tput setaf 5) \
			-v cyan=$(tput setaf 6) \
			-v white=$(tput setaf 7) \
			-v reset=$(tput sgr0) \
			-v violet=$(tput setaf 13) \
			-v user=$USER \
		'
		BEGIN {
			FPAT = "([[:space:]]*[^[:space:]]+)";
			OFS = "";
		}
		{
			# DEV
			$1 = blue$1reset;
			# "on"
			#$2 = $2;
			# MOUNT POINT
			$3 = yellow$3reset;
			# "type"
			#$4 = $4;
			# FS
			$5 = white$5reset;
			# ARGS
			#$6 = white$6reset;
			print;
		}
		'
}
alias mountl="_mount"

# colorful df
# @depends: gawk
function _df() {
	if [[ -n "$*" ]]; then
		args=$*
	else
		args="-Th --total"
	fi

	df $args | \
		awk \
			-v bbold=$(tput smso) \
			-v bold=$(tput bold) \
			-v black=$(tput setaf 0) \
			-v red=$(tput setaf 1) \
			-v green=$(tput setaf 2) \
			-v yellow=$(tput setaf 3) \
			-v blue=$(tput setaf 4) \
			-v magenta=$(tput setaf 5) \
			-v cyan=$(tput setaf 6) \
			-v white=$(tput setaf 7) \
			-v reset=$(tput sgr0) \
		'
		BEGIN {
			FPAT = "([[:space:]]*[^[:space:]]+)";
			OFS = "";
		}
		{
			if (NR > 1) { # skip 1st line \w header
				# FS
				$1 = blue$1reset;
				# TYPE
				$2 = white$2reset;
				# SIZE
				$3 = magenta$3reset;
				# USED
				$4 = cyan$4reset;
				# AVAILABLE
				$5 = green$5reset;
				# USED %
				$6 = cyan$6reset;
				# MOUNT POINT
				$7 = yellow$7reset;
				print;
			} else {
				$0 = bbold$0reset;
				print;
			}
		}
		'
}
alias df="_df"

# extract any archive
extract() {
	if [[ -z $1 ]]; then
		# if no parameters given display usage
		echo "Usage: extract <filename>.<tar.bz2|tar.gz|tar.xz|lzma|bz2|rar|gz|tar|tbz2|tgz|zip|Z|7z|xz>"
	else
		if [[ -f "$1" ]]; then
			case $1 in
				*.tar.bz2) tar xvjf $1   ;;
				*.tar.gz)  tar xvzf $1   ;;
				*.tar.xz)  tar xvJf $1   ;;
				*.tar.zst) tar --use-compress-program=unzstd -xvf $1 ;;
				*.lzma)    unlzma $1     ;;
				*.bz2)     bunzip2 $1    ;;
				*.rar)     unrar x $1    ;;
				*.gz)      gunzip $1     ;;
				*.tar)     tar xvf $1    ;;
				*.tbz2)    tar xvjf $1   ;;
				*.tgz)     tar xvzf $1   ;;
				*.zip)     unzip $1      ;;
				*.Z)       uncompress $1 ;;
				*.7z)      7z x $1       ;;
				*.xz)      unxz $1       ;;
				*)         echo "[E] File extension is not supported!" ;;
			esac
		else
			echo "[E] File \"$1\" not found!"
		fi
	fi
}

# swap 2 files
swap() {
	local tmp_file="tmp_file.$$"
	mv "$1" "${tmp_file}"
	mv "$2" "$1"
	mv "${tmp_file}" "$2"
}

# backup file/dir
#   -r - delete origin path
bak() {
	if [[ "$1" == "-r" ]]; then
		local need_rm=1
		local orig_path="$2"
	else
		local need_rm=0
		local orig_path="$1"
	fi
	if [[ "${need_rm}" == 1 ]]; then
		mv -i "${orig_path}" "${orig_path%/}.bak"
	else
		cp -ri "${orig_path}" "${orig_path%/}.bak"
	fi
}

# print markdown
md() {
	${MD_PRINT_CMD:-cat} $*
}

# pretty-print
# @depends: pygments, pygments-solarized-style, perl
function pp() {
	local print_lines=0
	for arg do
		shift
		if [[ "$arg" == "-l" ]]; then
			print_lines=1
			continue
		fi
		set -- "$@" "$arg"
	done

	local cmd="pygmentize -f terminal16m -O bg=dark,style=solarizeddark -g"

	if [[ "${print_lines}" == 1 ]]; then
		${cmd} "$*" | perl -e 'for(<>){print sprintf("%3s %s", ++$i,$_);}'
	else
		${cmd} "$*"
	fi
}

# apt-history <install|upgrade|remove|rollback>
# @depends: awk
apt-history() {
	case "$1" in
		install)
			cat /var/log/dpkg.log | grep 'install '
			;;
		upgrade|remove)
			cat /var/log/dpkg.log | grep $1
			;;
		rollback)
			cat /var/log/dpkg.log | grep upgrade | \
				grep "$2" -A10000000 | \
				grep "$3" -B10000000 | \
				awk '{print $4"="$5}'
			;;
		*)
			cat /var/log/dpkg.log
			;;
	esac
}

# colorful log
# @depends: awk
tailf-monolog() {
	if [[ -z "$1" ]]; then
		echo "Please specify a monolog file for monitoring"
		return
	fi

	tail -f $1 | \
		awk \
			-v black=$(tput setaf 0) \
			-v red=$(tput setaf 1) \
			-v green=$(tput setaf 2) \
			-v yellow=$(tput setaf 3) \
			-v blue=$(tput setaf 4) \
			-v magenta=$(tput setaf 5) \
			-v cyan=$(tput setaf 6) \
			-v white=$(tput setaf 7) \
			-v reset=$(tput sgr0) \
		'
		{matched=0}
		/DEBUG/     {matched=1; print white $0 reset}
		/INFO/      {matched=1; print green $0 reset}
		/NOTICE/    {matched=1; print cyan $0 reset}
		/WARNING/   {matched=1; print yellow $0 reset}
		/ERROR/     {matched=1; print red $0 reset}
		/CRITICAL/  {matched=1; print red $0 reset}
		/ALERT/     {matched=1; print magenta $0 reset}
		/EMERGENCY/ {matched=1; print blue $0 reset}
		matched==0              {print $0}
		'
}

# get cert info
# @depends: openssl
function get-cert-names() {
	if [[ -z "${1}" ]]; then
		echo "ERROR: No domain specified.";
		return 1;
	fi;

	local domain="${1}";
	echo "Testing ${domain}...";
	echo "";

	local tmp=$(echo -e "GET / HTTP/1.0\nEOT" | openssl s_client -connect "${domain}:443" -servername "${domain}" 2>&1);

	if [[ "${tmp}" = *"-----BEGIN CERTIFICATE-----"* ]]; then
		local certText=$(echo "${tmp}" | openssl x509 -text -certopt "no_aux, no_header, no_issuer, no_pubkey, no_serial, no_sigdump, no_signame, no_validity, no_version");
		echo "Common Name:";
		echo "";
		echo "${certText}" | grep "Subject:" | sed -e "s/^.*CN=//" | sed -e "s/\/emailAddress=.*//";
		echo "";
		echo "Subject Alternative Name(s):";
		echo "";
		echo "${certText}" | grep -A 1 "Subject Alternative Name:" | sed -e "2s/DNS://g" -e "s/ //g" | tr "," "\n" | tail -n +2;
		return 0;
	else
		echo "ERROR: Certificate not found.";
		return 1;
	fi;
}

function less-tree() {
	tree -aC -I '.git|node_modules|bower_components|vendor|.idea' --dirsfirst "$@" | less -FRNX;
}

# @depends: openssl
function data-url() {
	local mimeType=$(file -b --mime-type "$1");
	if [[ $mimeType == text/* ]]; then
		mimeType="${mimeType};charset=utf-8";
	fi
	echo "data:${mimeType};base64,$(openssl base64 -in "$1" | tr -d '\n')";
}

function decode-url() {
	echo $1 | python3 -c "import sys, urllib.parse as p; print(''); print(p.unquote(sys.stdin.read()));"
}

function decode-base64 {
	#echo $1 | python3 -c "import sys, base64; print(base64.b64decode(sys.stdin.read()));";
	echo $1 | base64 -d $1
}

# @depends: iconv
function decode-imap-folder-name() {
	# for i in *; do echo -n "$i == "; decode-imap-folder-name $i; done
	echo $1 | tr '&' '+' | tr ',' '/' | iconv -f UTF-7 -t UTF-8
}

function convert-utf8-to-3byte() {
	echo -n $1 | od -An -tx1 | sed 's/ /\\x/g'
}

#function youtube-dl-sst() {
#    # depends: youtube-dl, ffmpeg
#    # args:
#    # * 1 - url
#    # * 2 - start time [hh:mm:ss]
#    # * 3 - duration (not end time!) [hh:mm:ss]
#    # * 4 - format (default: best. mb buggy, then use manual youtube-dl -F <url>)
#    local format=${4:-best}
#    local url=$(youtube-dl --get-url --format $format "$1")
#    local filename=$(youtube-dl --get-filename --format $format "$1")
#    echo $format
#    ffmpeg -ss $2 -i "$url" -t $3 -c copy "$filename" && echo "[Done]. File: $PWD/$filename"
#}

function set-tor-proxy() {
	export https_proxy="socks5://127.0.0.1:9050"
	export http_proxy="socks5://127.0.0.1:9050"
}

function unset-tor-proxy() {
	unset https_proxy
	unset http_proxy
}


##
# simple notes
##

NOTES_DIR="${NOTES_DIR:-$HOME/.notes/}"

_ngetfilepath() {
	local file_name="default.md"

	if [[ -n "$*" ]]; then
		file_name="$*"
	fi

	echo "${NOTES_DIR}${file_name}"
}

n() {
	$EDITOR $(_ngetfilepath $*)
}

# @depends: tree, awk
nls() {
	tree -CR --noreport ${NOTES_DIR} | \
		awk \
		'
		{
			if (NR > 1) { # skip 1st line \w notes dir
				if (NF == 1) {
					print $1;
				} else if (NF == 2) { # 1 lvl
					printf "⊳ %s\n", $2;
				} else if (NF == 3) { # 2 lvl
					printf "    ⊳ %s\n", $3;
				}
			}
		}
		'
}

nrm() {
	rm -i $(_ngetfilepath $*)
}

# @depends: pandoc
nprint() {
	${NOTES_PRINT_CMD:-pandoc -t plain} $(_ngetfilepath $*)
}
alias np="nprint"
alias npp="nprint"

ncat() {
	cat $(_ngetfilepath $*)
}

# custom
mount-crypt() {
	sudo cryptsetup open $1 crypt && sudo mount /dev/mapper/crypt /media/crypt
}
umount-crypt() {
	sudo umount /media/crypt && sudo cryptsetup close crypt
}