From 057d2253d042c193e4d01651a04f24359d1889a0 Mon Sep 17 00:00:00 2001
From: Vishwanath Martur <64204611+vishwamartur@users.noreply.github.com>
Date: Sat, 21 Dec 2024 11:42:49 +0530
Subject: [PATCH] Add support for Zellij

Related to #11

Update the environment check in `wut/wut.py` to support Zellij.

* Modify the check to include `os.environ.get("ZELLIJ")`.
* Update the error message to include Zellij, indicating that `wut` must be run inside a tmux, screen, or Zellij session.
---
 wut/wut.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/wut/wut.py b/wut/wut.py
index 3f537e9..1bb3d47 100644
--- a/wut/wut.py
+++ b/wut/wut.py
@@ -41,9 +41,9 @@ def main():
 
     with console.status("[bold green]Trying my best..."):
         # Ensure environment is set up correctly
-        if not os.environ.get("TMUX") and not os.environ.get("STY"):
+        if not os.environ.get("TMUX") and not os.environ.get("STY") and not os.environ.get("ZELLIJ"):
             console.print(
-                "[bold red]wut must be run inside a tmux or screen session.[/bold red]"
+                "[bold red]wut must be run inside a tmux, screen, or Zellij session.[/bold red]"
             )
             return
         if (