🐛 256/truecolor detection fix

haliphax · Mar 27, 2024 · 6d3f2cc · 6d3f2cc
1 parent deb082b
commit 6d3f2cc
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/xthulu/ssh/console/__init__.py b/xthulu/ssh/console/__init__.py
@@ -30,8 +30,18 @@ def __init__(
         **kwargs,
     ):
         self.encoding = encoding
+        color_term = _environ.get("COLORTERM") if _environ else None
+        term_type = (_environ.get("TERM") if _environ else None) or ""
+        color_system = (
+            "truecolor"
+            if color_term == "truecolor" or term_type.find("truecolor") >= 0
+            else "256"
+            if term_type.find("256") >= 0
+            else "standard"
+        )
         super().__init__(
             **kwargs,
+            color_system=color_system,
             file=FileWrapper(ssh_writer, encoding),
             force_interactive=True,
             force_terminal=True,