Add a generic --session-wrapper for non-X11 sessions.

README.md

@@ -12,6 +12,8 @@ Options:
     -v, --version       print version information
     -c, --cmd COMMAND   command to run
     -s, --sessions DIRS colon-separated list of Wayland session paths
+        --session-wrapper 'CMD [ARGS]...'
+                        wrapper command to initialize the non-X11 session
     -x, --xsessions DIRS
                         colon-separated list of X11 session paths
         --xsession-wrapper 'CMD [ARGS]...'

contrib/man/tuigreet-1.scd

@@ -24,6 +24,10 @@ tuigreet - A graphical console greeter for greetd
 	Location of desktop-files to be used as Wayland session definitions. By
 	default, Wayland sessions are fetched from */usr/share/wayland-sessions*.
 
+*--session-wrapper 'CMD [ARGS]...'*
+	Specify a wrapper command to execute instead of the session for non-X11
+	sessions. This command will receive the session command as its arguments.
+
 *-x, --xsessions DIR1[:DIR2]...*
 	Location of desktop-files to be used as X11 session definitions. By
 	default, X11 sessions are fetched from */usr/share/xsessions*.

src/greeter.rs

@@ -85,6 +85,7 @@ pub struct Greeter {
   pub session_path: Option<PathBuf>,
   pub session_paths: Vec<(PathBuf, SessionType)>,
   pub sessions: Menu<Session>,
+  pub session_wrapper: Option<String>,
   pub xsession_wrapper: Option<String>,
 
   pub username: String,
@@ -296,6 +297,7 @@ impl Greeter {
     opts.optflag("v", "version", "print version information");
     opts.optopt("c", "cmd", "command to run", "COMMAND");
     opts.optopt("s", "sessions", "colon-separated list of Wayland session paths", "DIRS");
+    opts.optopt("", "session-wrapper", "wrapper command to initialize the non-X11 session", "'CMD [ARGS]...'");
     opts.optopt("x", "xsessions", "colon-separated list of X11 session paths", "DIRS");
     opts.optopt("", "xsession-wrapper", xsession_wrapper_desc.as_str(), "'CMD [ARGS]...'");
     opts.optflag("", "no-xsession-wrapper", "do not wrap commands for X11 sessions");
@@ -415,6 +417,10 @@ impl Greeter {
       self.session_paths.extend(env::split_paths(&dirs).map(|dir| (dir, SessionType::X11)));
     }
 
+    if self.option("session-wrapper").is_some() {
+      self.session_wrapper = self.option("session-wrapper");
+    }
+
     if !self.config().opt_present("no-xsession-wrapper") {
       self.xsession_wrapper = self.option("xsession-wrapper").or_else(|| Some(DEFAULT_XSESSION_WRAPPER.to_string()));
     }

src/ipc.rs

@@ -135,6 +135,14 @@ impl Ipc {
               if let Some(ref wrap) = greeter.xsession_wrapper {
                 command = format!("{} {}", wrap, command);
               }
+            } else {
+              if let Some(ref wrap) = greeter.session_wrapper {
+                command = format!("{} {}", wrap, command);
+              }
+            }
+          } else {
+            if let Some(ref wrap) = greeter.session_wrapper {
+              command = format!("{} {}", wrap, command);
             }
           }