#!/bin/sh

TOFI_CONF="$XDG_CONFIG_HOME/tofi/themes/dmenu"
if echo "$@" | grep -qE -- '-l [0-9]+'; then
  TOFI_CONF="$XDG_CONFIG_HOME/tofi/themes/dmenu_vertical"
  ARGS="$(echo "$@" | grep -qE -- '-l [0-9]+' | sed 's/-l [0-9]+//' )"
else
  ARGS="$@"
fi

if [ -n "$WAYLAND_DISPLAY" ]; then
  # Wayland
  tofi -c "$TOFI_CONF" -d $ARGS 2>/dev/null
  unset ARGS
elif [ -n "$DISPLAY" ]; then
  # X11
  dmenu $@
fi
unset TOFI_CONF