Cleanup for stow ---STOW

[dotfiles/.git] / .local / lib / python3.9 / site-packages / pywal / sequences.py

diff --git a/.local/lib/python3.9/site-packages/pywal/sequences.py b/.local/lib/python3.9/site-packages/pywal/sequences.py
new file mode 100644 (file)
index 0000000..b42c3b7
--- /dev/null
+++ b/.local/lib/python3.9/site-packages/pywal/sequences.py
@@ -0,0 +1,88 @@
+"""
+Send sequences to all open terminals.
+"""
+import glob
+import logging
+import os
+
+from .settings import CACHE_DIR, OS
+from . import util
+
+
+def set_special(index, color, iterm_name="h", alpha=100):
+    """Convert a hex color to a special sequence."""
+    if OS == "Darwin":
+        return "\033]P%s%s\033\\" % (iterm_name, color.strip("#"))
+
+    if index in [11, 708] and alpha != "100":
+        return "\033]%s;[%s]%s\033\\" % (index, alpha, color)
+
+    return "\033]%s;%s\033\\" % (index, color)
+
+
+def set_color(index, color):
+    """Convert a hex color to a text color sequence."""
+    if OS == "Darwin" and index < 20:
+        return "\033]P%1x%s\033\\" % (index, color.strip("#"))
+
+    return "\033]4;%s;%s\033\\" % (index, color)
+
+
+def set_iterm_tab_color(color):
+    """Set iTerm2 tab/window color"""
+    return ("\033]6;1;bg;red;brightness;%s\a"
+            "\033]6;1;bg;green;brightness;%s\a"
+            "\033]6;1;bg;blue;brightness;%s\a") % (*util.hex_to_rgb(color),)
+
+
+def create_sequences(colors, vte_fix=False):
+    """Create the escape sequences."""
+    alpha = colors["alpha"]
+
+    # Colors 0-15.
+    sequences = [set_color(index, colors["colors"]["color%s" % index])
+                 for index in range(16)]
+
+    # Special colors.
+    # Source: https://goo.gl/KcoQgP
+    # 10 = foreground, 11 = background, 12 = cursor foregound
+    # 13 = mouse foreground, 708 = background border color.
+    sequences.extend([
+        set_special(10, colors["special"]["foreground"], "g"),
+        set_special(11, colors["special"]["background"], "h", alpha),
+        set_special(12, colors["special"]["cursor"], "l"),
+        set_special(13, colors["special"]["foreground"], "l"),
+        set_special(17, colors["special"]["foreground"], "l"),
+        set_special(19, colors["special"]["background"], "l"),
+        set_color(232, colors["special"]["background"]),
+        set_color(256, colors["special"]["foreground"])
+    ])
+
+    if not vte_fix:
+        sequences.extend(
+            set_special(708, colors["special"]["background"], "l", alpha)
+        )
+
+    if OS == "Darwin":
+        sequences += set_iterm_tab_color(colors["special"]["background"])
+
+    return "".join(sequences)
+
+
+def send(colors, cache_dir=CACHE_DIR, to_send=True, vte_fix=False):
+    """Send colors to all open terminals."""
+    if OS == "Darwin":
+        tty_pattern = "/dev/ttys00[0-9]*"
+
+    else:
+        tty_pattern = "/dev/pts/[0-9]*"
+
+    sequences = create_sequences(colors, vte_fix)
+
+    # Writing to "/dev/pts/[0-9] lets you send data to open terminals.
+    if to_send:
+        for term in glob.glob(tty_pattern):
+            util.save_file(sequences, term)
+
+    util.save_file(sequences, os.path.join(cache_dir, "sequences"))
+    logging.info("Set terminal colors.")