## -*- mode:sh -*- #
################################################################################
## Shell -- Functions.

## Functions that can be useful outside a shell (like in Emacs or Ranger) should
## be written in stand-alone scripts instead.

## Colored man pager.
man () {
	# Justify man pager to current window size.
	local width=$(tput cols)
	[ $width -gt $MANWIDTH ] && width=$MANWIDTH

	# mb = ?
	# md = bold (titles, commands)
	# so = status bar
	# us = italic (arguments, files)
	env \
		MANWIDTH=$width \
		LESS_TERMCAP_mb=$(printf "\e[1;31m") \
		LESS_TERMCAP_md=$(printf "\e[0;36m") \
		LESS_TERMCAP_me=$(printf "\e[0m") \
		LESS_TERMCAP_se=$(printf "\e[0m") \
		LESS_TERMCAP_so=$(printf "\e[1;4;37m") \
		LESS_TERMCAP_ue=$(printf "\e[0m") \
		LESS_TERMCAP_us=$(printf "\e[0;33m") \
		man "$@"
}

## Make directories and change into the first one.
mkcd () {
	mkdir -p "$@"
	cd "$1"
}

## Quick search.
search () {
	find . -iname "*$1*"
}

## Vim-only: search the vim reference manual for a keyword.
## Usage: :h <keyword>
if command -v vim >/dev/null 2>&1; then
	:h ()
	{
		vim --cmd ":silent help $@" --cmd "only";
	}
fi

## Term properties
termcolors256 () {
	local x y o
	x=$(tput op)
	y=$(printf %76s)
	for i in {0..256}; do
		o=00$i
		echo -e "${o:${#o}-3:3} $(tput setaf $i;tput setab $i)${y// /=}$x"
	done
}

## Will not work for Bourne Shell.
termcolors () {
	local T

	# The test text. Must be 3 letters.
	T='Doh'

	echo -e "\n                 40m     41m     42m     43m\
     44m     45m     46m     47m";

	for FGs in \
	'    m' '   1m' '  30m' '1;30m' '  31m' '1;31m' '  32m' \
		'1;32m' '  33m' '1;33m' '  34m' '1;34m' '  35m' '1;35m' \
		'  36m' '1;36m' '  37m' '1;37m'; do
		FG=${FGs// /}
		echo -en " $FGs \033[$FG  $T  "
		for BG in 40m 41m 42m 43m 44m 45m 46m 47m; do
			echo -en "$EINS \033[$FG\033[$BG  $T  \033[0m";
		done
		echo;
	done
	echo
}

termprop () {
	infocmp -1 | sed -nu 's/^[ \000\t]*//;s/[ \000\t]*$//;/[^ \t\000]\{1,\}/!d;/acsc/d;s/=.*,//p'| column -c80
}

termsupport () {
	for T in `find "/usr/share/terminfo" -type f -printf '%f\n'`;do
		echo "$T `tput -T $T colors`"
	done | sort -nk2
}

if command -v highlight >/dev/null 2>&1; then
	highlight-gallery () {
		local SAMPLE DUMMY

		if [ $# -gt 0 ]; then
			cat<<EOF
Usage: $1

Highlight themes viewer for 256-colors terminals.

highlight is a tool for listing printing. This script displays a preview for
every highlight's theme.  Press ENTER to display next preview. Press anything
else then ENTER to quit.
EOF
			return
		fi

		SAMPLE="$(cat <<EOF
void function(char *pointer, int &ref) {
	int integer=3;
	char* string="Hello!";

	// Line comment
	for (i=0; i<10; i++)
	{
		printf("Sum: %d\n", i*( integer+14 ) );
	}
}
EOF
)"

		for i in $(highlight -w | grep -v "\s\|^$"); do
			echo "==> $i"
			highlight -S c -O xterm256 <(echo "$SAMPLE") -s "$i"
			read DUMMY
			if [ -n "$DUMMY" ]; then
				return
			fi
		done
	}
fi

## Keymap view in terminal
echokeymap () {
	if [ $# -ne 1 ]; then
		1="$HOME/.xkb/symbols/usim"
	fi

	awk '/━|┃|│|─/ {line=substr($0,4); print line}' "$1"
}

## Extended info
typex () {
	local xpath cmd

	if [ $# -lt 1 ]; then
		echo >&2 "Provide an existing command as parameter."
		return
	fi

	for i; do
		cmd="$(command -v "$i" 2>/dev/null)"
		if [ $? -ne 0 ]; then
			echo "$i does not exist."
		elif [ -z "$(echo "$cmd" | grep ^/)" ]; then
			type "$i"
		else
			xpath="$(type $1 | cut -d' ' -f3)"
			ls -l "$xpath"
			file "$xpath" | cut -d':' -f2 | cut -b 2-
		fi
	done
}

# Automatically change the directory in shell after closing ranger.
# Compatible with ranger >= 1.4.2.
#
# To undo the effect of this function, you can type "cd -" to return to the
# original directory.
browse () {
	local tempfile="$(mktemp)"
	ranger --choosedir="$tempfile" "${@:-$(pwd)}"
	if [ -f "$tempfile" ] && [ "$(cat -- "$tempfile")" != "$(echo -n `pwd`)" ]; then
		cd -- "$(cat "$tempfile")"
	fi
	rm -f -- "$tempfile"
}

## Webcam testing with an mplayer-like player.
webcam () {
	mpv --tv driver=v4l2:width=640:height=480:device=/dev/video0 tv://
}

xscreen () {
	## Does not work?
	cvt $1 $2
	echo  "$(cvt $1 $2 | tail -1 | cut -d' ' -f2-)"
	xrandr --newmode "$(cvt $1 $2 | tail -1 | cut -d' ' -f2-)"
}

## Xrandr
xmulti () {
	if [ $# -ne 1 ]; then
		echo "Usage: $0 VIDEOPORT"
		return
	fi

	xrandr --output "$1" --auto --right-of $(xrandr | sed -n "/current/{n;p;q}" | cut -f1 -d" ")
}

xmultioff () {
	if [ $# -ne 1 ]; then
		echo "Usage: $0 VIDEOPORT"
		return
	fi

	xrandr --output "$1" --off
}

##==============================================================================
## Pacman functions

if command -v pacman >/dev/null 2>&1; then
	## Get detailed file list for the specified packages, omitting folders.
	pql () {
		pacman -Ql "$@" | grep -v "/$"
	}

	pli () {
		expac -t '%F %T' '%-8l %n' | sort -rn | head -${1:-30}
	}

	pacfiles () {
		pacman -Qlq "$@" | grep -v '/$' | xargs du -cbh | sort -h
	}
fi

##==============================================================================
## FreeBSD functions

if [ "$(uname)" = FreeBSD ]; then
	## TODO: Useless FreeBSD function?
	# mqs () {
	#     local str="$(echo "$@" | sed -E 's/[ \t]+/\\|/g')"
	#     (cd /usr/ports && make quicksearch name="$str")
	# }

	port-list () {
		pkg_info | cut -f1 -d' '
	}

	port-size () {
		pkg_info -Qsb "$@" | sed 's/\([^:]*\):\(.*\)/\2 KiB	\1/g'
	}

	mcd () {
		local DEST=$(pkg_search -r "$@" | head -n1 | awk '{print $2}')
		cd -- "$DEST"
	}

fi

##==============================================================================
## TeXlive functions

## It is important to set umask properly, otherwise users might not have
## authorization to use the packages.

if [ -d "/usr/local/texlive" ]; then
	tli () {
		sudo sh -c 'umask 022 && tlmgr install $0 "$@"' "$@"
	}

	tlr () {
		sudo sh -c 'umask 022 && tlmgr remove $0 "$@"' "$@"
	}
fi

##==============================================================================
## Luarocks functions

if command -v luarocks >/dev/null 2>&1; then
	lri () {
		sudo sh -c 'umask 022 && luarocks install $0 $@' "$@"
	}
fi