Add check to determine display server without flags

1 files changed, 13 insertions, 0 deletions

diff --git a/ffc b/ffc
index fd004be..117523c 100755
--- a/ffc
+++ b/ffc
@@ -21,6 +21,19 @@ while getopts "hwxd:u:" o; do case "${o}" in
 esac done
 #}}}
 
+#{{{ Check display server if -w or -x not provided
+if [ -z "$DMENU" ]; then
+  if [ -n "$WAYLAND_DISPLAY" ]; then
+    DMENU="bemenu"
+  elif [ -n "$DISPLAY" ]; then
+    DMENU="dmene"
+  else
+    print "Can't tell if Wayland or X; what gives?\n"
+    exit 1
+  fi
+fi
+#}}}
+
 #{{{ Native Profiles
 if [ -f ~/.mozilla/firefox/profiles.ini ]; then
   NATIVE_PROFILES=`grep 'Path=' ~/.mozilla/firefox/profiles.ini | cut -d'.' -f2`