bangs

#! /bin/sh
# Todo: add option to pass stuff via args

if [ "$1" = "--help" ] || [ "$1" = "-h" ] || [ "$1" = "help" ]; then
  printf "$ bangs
Portable custom duckduckgo bangs (tpb, lg, etc.)
Open in \$BROWSER ($BROWSER).
Bangs not in the list will be referred to duckduckgo.
Should be run via keyboard shortcut.

Commands:
  list: list all available bangs
  BANG QUERY: open the given query for the given bang

Examples:
  $ bangs
  $ bangs list
  $ bangs yt 'rofi showcase'
"; exit
fi

command -v rofi >/dev/null || { echo "rofi is not installed"; exit 1; }
command -v $BROWSER >/dev/null || { echo "$BROWSER is not installed"; exit 1; }
command -v node >/dev/null || { echo "node is not installed"; exit 1; }

DEFAULT_BANG="-" # ddg

bangList() {
  cat $(dirname $0)/banglist
}

if [ "$1" = "list" ]; then
  bangList
  exit 0
fi

prompt() {
  rofi -dmenu -l 0 -theme-str 'window {width: 300px;}' -p \!
}

urlEncode() {
  node -e "console.log( encodeURIComponent( process.argv[1] ) )" "$1"
}

if [ -z "$2" ]; then
  res="$(prompt)"
  [ -z "$res" ] && exit 1
  bang="${res%% *}"
  query="${res#* }"
  [ "$query" = "$bang" ] && bang=$DEFAULT_BANG
else
  bang=$1
  query="$2"
fi

case "$bang" in
  y)
    ~/scripts/yt $query
    exit;;
  yts)
    ~/scripts/yts $query
    exit;;
  # yt)
    #   if [ -f "$HOME/scripts/personal/youtube.html" ]; then
    #     {
    #       sleep 1 && $BROWSER $HOME/scripts/personal/youtube.html >/dev/null 2>&1
    #     } &
    #   fi;;
esac

query=`urlEncode "$query"`

if bangList | grep -m 1 "^$bang " >/dev/null; then
  bangList | grep -m 1 "^$bang " | cut -d\  -f2- | sed "s|\%s|$query|" | xargs -r -0 $BROWSER >/dev/null 2>&1 &
else
  if [ "$bang" = "$DEFAULT_BANG" ] && [ "$DEFAULT_BANG" = "-" ]; then
    $BROWSER "https://duckduckgo.com/?q=$query" >/dev/null 2>&1 &
  else
    $BROWSER "https://duckduckgo.com/?q=!$bang%20$query" >/dev/null 2>&1 &
  fi
fi