Add check to determine display server without flags

1 files changed, 13 insertions, 0 deletions

diff --git a/lwc b/lwc
index ae472c8..29daf45 100755
--- a/lwc
+++ b/lwc
@@ -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 ~/.librewolf/profiles.ini ]; then
   NATIVE_PROFILES=`grep 'Path=' ~/.librewolf/profiles.ini | cut -d'.' -f2`