dripfetch 0.1.0__py3-none-any.whl

dripfetch/box/types/clock.py

@@ -0,0 +1,487 @@
+import time
+
+from ..base import BaseBox
+
+DIGITS = {
+    "single": {
+        "medium": {
+            "0": ["┌────┐", "│    │", "│    │", "│    │", "└────┘"],
+            "1": ["  ┌┐  ", "  ││  ", "  ││  ", "  ││  ", "  └┘  "],
+            "2": ["┌────┐", "     │", "┌────┘", "│     ", "└────┘"],
+            "3": ["┌────┐", "     │", " ────┤", "     │", "└────┘"],
+            "4": ["┌    ┐", "│    │", "└────┤", "     │", "     ┘"],
+            "5": ["┌────┐", "│     ", "└────┐", "     │", "└────┘"],
+            "6": ["┌────┐", "│     ", "├────┐", "│    │", "└────┘"],
+            "7": ["┌────┐", "     │", "     │", "     │", "     ┘"],
+            "8": ["┌────┐", "│    │", "├────┤", "│    │", "└────┘"],
+            "9": ["┌────┐", "│    │", "└────┤", "     │", "└────┘"],
+            ":": ["  ┌┐  ", "  └┘  ", "      ", "  ┌┐  ", "  └┘  "],
+            " ": ["      ", "      ", "      ", "      ", "      "],
+            "M": ["┌┐   ┌┐", "│└┐ ┌┘│", "│ └┬┘ │", "│     │", "└     ┘"],
+            "A": ["┌────┐", "│    │", "├────┤", "│    │", "└    ┘"],
+            "P": ["┌────┐", "│    │", "├────┘", "│     ", "└     "],
+        },
+        "big": {
+            "0": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ └──────┘ │",
+                "└──────────┘",
+            ],
+            "1": [
+                "    ┌──┐    ",
+                "    │  │    ",
+                "    │  │    ",
+                "    │  │    ",
+                "    │  │    ",
+                "    │  │    ",
+                "    │  │    ",
+                "    │  │    ",
+                "    │  │    ",
+                "    └──┘    ",
+            ],
+            "2": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "└─┘      │ │",
+                "         │ │",
+                "┌────────┘ │",
+                "│ ┌────────┘",
+                "│ │         ",
+                "│ │      ┌─┐",
+                "│ └──────┘ │",
+                "└──────────┘",
+            ],
+            "3": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "└─┘      │ │",
+                "         │ │",
+                "  ┌──────┘ │",
+                "  └──────┐ │",
+                "         │ │",
+                "┌─┐      │ │",
+                "│ └──────┘ │",
+                "└──────────┘",
+            ],
+            "4": [
+                "┌──┐    ┌──┐",
+                "│ ┌┘    └┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ └──────┘ │",
+                "└────────┐ │",
+                "         │ │",
+                "         │ │",
+                "       ┌─┘ │",
+                "       └───┘",
+            ],
+            "5": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "│ │      └─┘",
+                "│ │         ",
+                "│ └────────┐",
+                "└────────┐ │",
+                "         │ │",
+                "┌─┐      │ │",
+                "│ └──────┘ │",
+                "└──────────┘",
+            ],
+            "6": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "│ │      └─┘",
+                "│ │         ",
+                "│ └────────┐",
+                "│ ┌──────┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ └──────┘ │",
+                "└──────────┘",
+            ],
+            "7": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "└─┘      │ │",
+                "         │ │",
+                "         │ │",
+                "         │ │",
+                "         │ │",
+                "         │ │",
+                "        ┌┘ │",
+                "        └──┘",
+            ],
+            "8": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ └──────┘ │",
+                "│ ┌──────┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ └──────┘ │",
+                "└──────────┘",
+            ],
+            "9": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ └──────┘ │",
+                "└────────┐ │",
+                "         │ │",
+                "┌─┐      │ │",
+                "│ └──────┘ │",
+                "└──────────┘",
+            ],
+            ":": [
+                "            ",
+                "   ┌────┐   ",
+                "   │ ┌┐ │   ",
+                "   │ └┘ │   ",
+                "   └────┘   ",
+                "   ┌────┐   ",
+                "   │ ┌┐ │   ",
+                "   │ └┘ │   ",
+                "   └────┘   ",
+                "            ",
+            ],
+            " ": ["  ", "  ", "  ", "  ", "  ", "  ", "  ", "  ", "  ", "  "],
+            "M": [
+                "┌────┐ ┌────┐",
+                "│ ┌┐ │ │ ┌┐ │",
+                "│ ││ │ │ ││ │",
+                "│ ││ └─┘ ││ │",
+                "│ │└┐   ┌┘│ │",
+                "│ │ └───┘ │ │",
+                "│ │       │ │",
+                "│ │       │ │",
+                "│ │       │ │",
+                "└─┘       └─┘",
+            ],
+            "A": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ └──────┘ │",
+                "│ ┌──────┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "└─┘      └─┘",
+            ],
+            "P": [
+                "┌──────────┐",
+                "│ ┌──────┐ │",
+                "│ │      │ │",
+                "│ │      │ │",
+                "│ └──────┘ │",
+                "│ ┌────────┘",
+                "│ │         ",
+                "│ │         ",
+                "│ │         ",
+                "└─┘         ",
+            ],
+        },
+    },
+    "double": {
+        "medium": {
+            "0": ["╔════╗", "║    ║", "║    ║", "║    ║", "╚════╝"],
+            "1": ["  ╔╗  ", "  ║║  ", "  ║║  ", "  ║║  ", "  ╚╝  "],
+            "2": ["╔════╗", "     ║", "╔════╝", "║     ", "╚════╝"],
+            "3": ["╔════╗", "     ║", " ════╣", "     ║", "╚════╝"],
+            "4": ["╔    ╗", "║    ║", "╚════╣", "     ║", "     ╝"],
+            "5": ["╔════╗", "║     ", "╚════╗", "     ║", "╚════╝"],
+            "6": ["╔════╗", "║     ", "╠════╗", "║    ║", "╚════╝"],
+            "7": ["╔════╗", "     ║", "     ║", "     ║", "     ╝"],
+            "8": ["╔════╗", "║    ║", "╠════╣", "║    ║", "╚════╝"],
+            "9": ["╔════╗", "║    ║", "╚════╣", "     ║", "╚════╝"],
+            ":": ["  ╔╗  ", "  ╚╝  ", "      ", "  ╔╗  ", "  ╚╝  "],
+            " ": ["      ", "      ", "      ", "      ", "      "],
+            "M": ["╔╗   ╔╗", "║╚╗ ╔╝║", "║ ╚╦╝ ║", "║     ║", "╚     ╝"],
+            "A": ["╔════╗", "║    ║", "╠════╣", "║    ║", "╚    ╝"],
+            "P": ["╔════╗", "║    ║", "╠════╝", "║     ", "╚     "],
+        },
+        "big": {
+            "0": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ╚══════╝ ║",
+                "╚══════════╝",
+            ],
+            "1": [
+                "    ╔══╗    ",
+                "    ║  ║    ",
+                "    ║  ║    ",
+                "    ║  ║    ",
+                "    ║  ║    ",
+                "    ║  ║    ",
+                "    ║  ║    ",
+                "    ║  ║    ",
+                "    ║  ║    ",
+                "    ╚══╝    ",
+            ],
+            "2": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "╚═╝      ║ ║",
+                "         ║ ║",
+                "╔════════╝ ║",
+                "║ ╔════════╝",
+                "║ ║         ",
+                "║ ║      ╔═╗",
+                "║ ╚══════╝ ║",
+                "╚══════════╝",
+            ],
+            "3": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "╚═╝      ║ ║",
+                "         ║ ║",
+                "  ╔══════╝ ║",
+                "  ╚══════╗ ║",
+                "         ║ ║",
+                "╔═╗      ║ ║",
+                "║ ╚══════╝ ║",
+                "╚══════════╝",
+            ],
+            "4": [
+                "╔══╗    ╔══╗",
+                "║ ╔╝    ╚╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ╚══════╝ ║",
+                "╚════════╗ ║",
+                "         ║ ║",
+                "         ║ ║",
+                "       ╔═╝ ║",
+                "       ╚═══╝",
+            ],
+            "5": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "║ ║      ╚═╝",
+                "║ ║         ",
+                "║ ╚════════╗",
+                "╚════════╗ ║",
+                "         ║ ║",
+                "╔═╗      ║ ║",
+                "║ ╚══════╝ ║",
+                "╚══════════╝",
+            ],
+            "6": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "║ ║      ╚═╝",
+                "║ ║         ",
+                "║ ╚════════╗",
+                "║ ╔══════╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ╚══════╝ ║",
+                "╚══════════╝",
+            ],
+            "7": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "╚═╝      ║ ║",
+                "         ║ ║",
+                "         ║ ║",
+                "         ║ ║",
+                "         ║ ║",
+                "         ║ ║",
+                "        ╔╝ ║",
+                "        ╚══╝",
+            ],
+            "8": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ╚══════╝ ║",
+                "║ ╔══════╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ╚══════╝ ║",
+                "╚══════════╝",
+            ],
+            "9": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ╚══════╝ ║",
+                "╚════════╗ ║",
+                "         ║ ║",
+                "╔═╗      ║ ║",
+                "║ ╚══════╝ ║",
+                "╚══════════╝",
+            ],
+            ":": [
+                "            ",
+                "   ╔════╗   ",
+                "   ║ ╔╗ ║   ",
+                "   ║ ╚╝ ║   ",
+                "   ╚════╝   ",
+                "   ╔════╗   ",
+                "   ║ ╔╗ ║   ",
+                "   ║ ╚╝ ║   ",
+                "   ╚════╝   ",
+                "            ",
+            ],
+            " ": ["  ", "  ", "  ", "  ", "  ", "  ", "  ", "  ", "  ", "  "],
+            "M": [
+                "╔════╗ ╔════╗",
+                "║ ╔╗ ║ ║ ╔╗ ║",
+                "║ ║║ ║ ║ ║║ ║",
+                "║ ║║ ╚═╝ ║║ ║",
+                "║ ║╚╗   ╔╝║ ║",
+                "║ ║ ╚═══╝ ║ ║",
+                "║ ║       ║ ║",
+                "║ ║       ║ ║",
+                "║ ║       ║ ║",
+                "╚═╝       ╚═╝",
+            ],
+            "A": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ╚══════╝ ║",
+                "║ ╔══════╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "╚═╝      ╚─╝",
+            ],
+            "P": [
+                "╔══════════╗",
+                "║ ╔══════╗ ║",
+                "║ ║      ║ ║",
+                "║ ║      ║ ║",
+                "║ ╚══════╝ ║",
+                "║ ╔════════╝",
+                "║ ║         ",
+                "║ ║         ",
+                "║ ║         ",
+                "╚═╝         ",
+            ],
+        },
+    },
+}
+
+
+class ClockBox(BaseBox):
+    def __init__(self, stdscr, config, boxes_config, colors, renderer):
+        super().__init__(stdscr, config, boxes_config, colors, renderer)
+        self.style = config.get("clock_style", "single")
+        self.size = config.get("clock_size", "medium")
+        self.clock_24h = config.get("clock_24h", True)
+        self.show_seconds = config.get("show_seconds", True)
+        self.show_am_pm = config.get("show_am_pm", True)
+        self.blink_colon = config.get("blink_colon", False)
+        self.show_date = config.get("show_date", False)
+
+    @staticmethod
+    def _blank(pattern):
+        return [" " * len(line) for line in pattern]
+
+    def _time_text(self, now):
+        hour = "%H" if self.clock_24h else "%I"
+        seconds = ":%S" if self.show_seconds else ""
+
+        text = time.strftime(f"{hour}:%M{seconds}", now)
+
+        if not self.clock_24h and self.show_am_pm:
+            text = f"{text[:-2]} {text[-2:]}"
+
+        return text
+
+    def _patterns(self, text, now):
+        digits = DIGITS[self.style][self.size]
+        show_colon = not self.blink_colon or now.tm_sec % 2 == 0
+
+        return [
+            self._blank(digits[":"])
+            if char == ":" and not show_colon
+            else digits.get(char, self._blank(digits["0"]))
+            for char in text
+        ]
+
+    def _geometry(self, now):
+        text = self._time_text(now)
+        patterns = self._patterns(text, now)
+        spacing = 2 if self.size == "big" else 1
+        width = sum(len(pattern[0]) for pattern in patterns)
+        width += (len(patterns) - 1) * spacing
+        height = len(patterns[0])
+        date = time.strftime("%A, %b %d %Y", now) if self.show_date else ""
+
+        return text, patterns, spacing, width, height, date
+
+    def dimensions(self):
+        _, _, _, width, height, date = self._geometry(time.localtime())
+        return width, height + bool(date) * 2
+
+    def draw_content(self, x, y, width, height):
+        now = time.localtime()
+
+        text, patterns, spacing, clock_width, clock_height, date = self._geometry(now)
+
+        content_height = clock_height + bool(date) * 2
+        x_start = x + max(0, (width - clock_width) // 2)
+        y_start = y + max(0, (height - content_height) // 2)
+
+        for char_index, char in enumerate(text):
+            pattern = patterns[char_index]
+
+            for row, line in enumerate(pattern):
+                offset = sum(
+                    len(patterns[index][0]) + spacing for index in range(char_index)
+                )
+
+                if char == ":":
+                    self.renderer.draw(
+                        x_start + offset,
+                        y_start + row,
+                        line,
+                        self.colors.accent,
+                    )
+                else:
+                    self.renderer.draw(
+                        x_start + offset,
+                        y_start + row,
+                        line,
+                        self.colors.text,
+                    )
+
+        if date:
+            date_x = x + max(0, (width - len(date)) // 2)
+
+            self.renderer.draw(
+                date_x,
+                y_start + clock_height + 1,
+                date[:width],
+                self.colors.text,
+            )
+
+    def update(self):
+        pass
+
+    def close(self):
+        pass

dripfetch/box/types/logo.py

@@ -0,0 +1,62 @@
+import re
+from importlib.resources import files
+
+from ...terminal import parse_color
+from ..base import BaseBox
+
+
+class LogoBox(BaseBox):
+    def __init__(self, stdscr, config, boxes, colors, renderer):
+        super().__init__(stdscr, config, boxes, colors, renderer)
+        self.logo = config.get("logo", "")
+        self.logo_colors = [parse_color(color) for color in config.get("colors", [])]
+        self.lines = self._load_logo()
+        self.parsed_lines = [self._parse_line(line) for line in self.lines]
+
+    def _load_logo(self):
+        path = files("dripfetch").joinpath("assets", "logos", f"{self.logo}.txt")
+        print(f"LOGO: {self.logo}")
+        print(f"PATH: {path}")
+        print(f"EXISTS: {path.is_file()}")
+        if not path.is_file():
+            return []
+        return path.read_text(encoding="utf-8").splitlines()
+    def _parse_line(self, line):
+        parts = re.split(r"(\$\d+)", line)
+        color = None
+        result = []
+
+        for part in parts:
+            if not part:
+                continue
+
+            if part.startswith("$"):
+                index = int(part[1:]) - 1
+                color = self.logo_colors[index] if 0 <= index < len(self.logo_colors) else None
+                continue
+
+            result.append((part, color))
+
+        return result
+
+    def draw_content(self, x, y, width, height):
+        for row, segments in enumerate(self.parsed_lines):
+            if row >= height:
+                break
+
+            offset = 0
+
+            for text, color in segments:
+                if offset >= width:
+                    break
+
+                text = text[:width - offset]
+                self.renderer.draw(x + offset, y + row, text, color)
+                offset += len(text)
+
+    def dimensions(self):
+        width = max(
+            (sum(len(text) for text, _ in line) for line in self.parsed_lines),
+            default=0,
+        )
+        return width, len(self.parsed_lines)