#!/bin/sh

[ "$1" = "privilege" ] && export cmd_prefix='sudo -A ' && export dmenu_args='-p Privilege -sb red' && shift 1;
[ "$1" = "terminal" ] && export cmd_prefix='urxvt -hold -e ' && export dmenu_args='-p Terminal -sb gray' && shift 1;

# https://tools.suckless.org/dmenu/scripts/

cachedir=${XDG_CACHE_HOME:-"$HOME/.cache"}
if [ -d "$cachedir" ]; then
	cache=$cachedir/dmenu_run
	historyfile=$XDG_DATA_HOME/dmenu_history
else
	cache=$HOME/.dmenu_cache
	historyfile=$HOME/.dmenu_history
fi

if [ "$1" = "update" ] || [ ! -f "$cache" ]; then
  IFS=:
  stest -dqr -n "$cache" $PATH && stest -flx $PATH | sort -u > "$cache";
  unset IFS;
  exit;
fi

awk -v histfile=$historyfile '
	BEGIN {
		while( (getline < histfile) > 0 ) {
			sub("^[0-9]+\t","")
			print
			x[$0]=1
		}
	} !x[$0]++ ' "$cache" \
	| dmenu -b ${dmenu_args:--p Launch} "$@" \
	| awk -v histfile=$historyfile '
		BEGIN {
			FS=OFS="\t"
			while ( (getline < histfile) > 0 ) {
				count=$1
				sub("^[0-9]+\t","")
				fname=$0
				history[fname]=count
			}
			close(histfile)
		}

		{
			history[$0]++
			print
		}

		END {
			if(!NR) exit
			for (f in history)
				print history[f],f | "sort -t '\t' -k1rn >" histfile
		}
	' \
	| while read -r cmd; do ${SHELL:-"/bin/sh"} -c "$cmd_prefix $cmd" & done