more cats!

home/conf/xonsh/kittycatsay.py

@@ -0,0 +1,95 @@
+import os
+import sys
+
+DEFAULT_CAT = r"""  ╱|、     
+ (˚ˎ 。7 {{HBAR}}{{HBAR}}
+  |、˜〵   
+  じしˍ,)ノ"""
+
+BOXES = {
+    "rounded": {
+        "hbar": "─",
+        "vbar": "│",
+        "corner_tl": "╭",
+        "corner_tr": "╮",
+        "corner_bl": "╰",
+        "corner_br": "╯",
+        "joiner": "┤",
+    }
+    
+}
+
+def main():
+    parser = False
+    args = False
+    isatty = sys.stdin.isatty()
+    message = ""
+    if not isatty:
+        message = sys.stdin.read().strip()
+    if len(sys.argv) > 1:
+        import argparse
+        parser = argparse.ArgumentParser(description="Display a message in a speech bubble next to a kittycat.")
+        parser.add_argument("--box", type=str, choices=BOXES.keys(), default="rounded", help="The style of the speech bubble.")
+        parser.add_argument("-W", type=int, default=0, help="The maximum width of the message before wrapping. Defaults to terminal width minus kittycat width.")
+        parser.add_argument("-f", "--file", type=str, default="./kittycats/default.cat", help="The path to the kittycat file to use.")
+        if isatty:
+            parser.add_argument("message", type=str, help="The message to display.")
+            args = parser.parse_args()
+            message = args.message
+        else:
+            args = parser.parse_args()
+    if len(message) == 0:
+        message = [
+            "Meow.",
+            "We still don't support multi-line input.",
+            "Try a pipe why don't you?",
+            "Wi wi wi wi.",
+            "Weh weh weh.",
+        ][int((int.from_bytes(open("/dev/urandom", "rb").read(1), "big") / 256) * 5)]
+    kittycat = DEFAULT_CAT
+    box_style = "rounded"
+    wrapping = 0
+    if args:
+        if os.path.isfile(args.file):
+            kittycat = open(args.file).read()
+        box_style = args.box
+        wrapping = args.W
+    outcat = get_kittycat(kittycat, message, BOXES[box_style], wrap=wrapping)
+    print(outcat)
+
+def get_kittycat(cat: str, msg: str, box: dict[str, str], wrap = 0):
+    for kw in box.keys():
+        cat = cat.replace(r"{{" + kw.upper() + r"}}", box[kw])
+    cat_lines = cat.splitlines()
+    cat_width = max(len(line) for line in cat_lines)
+    max_msg_height = len(cat_lines) - 2
+    max_msg_width = os.get_terminal_size()[0] - cat_width - 2
+    if not wrap < max_msg_width:
+        raise ValueError("Wrap width exceeds maximum message width")
+    wrap = wrap if wrap else max_msg_width
+    msg = msg.replace("\n", " ")
+    msg_lines = []
+    msg_words = msg.split(" ")
+    while msg_words:
+        line = ""
+        while msg_words and len(line) + len(msg_words[0]) + 1 <= wrap:
+            line += (msg_words.pop(0) + " ")
+        msg_lines.append(line.rstrip())
+    if len(msg_lines) > max_msg_height:
+        raise ValueError(f"Message exceeds maximum height of {max_msg_height} given {wrap} character wrap")
+    longest_msg_line = max(len(line) for line in msg_lines)
+    msg_lines = [line.ljust(longest_msg_line) for line in msg_lines]
+    msg_lines = [box["joiner"] + " " + msg_lines.pop(0) + " " + box["vbar"]] + [box["vbar"] + " " + line + " " + box["vbar"] for line in msg_lines]
+    top_bar =  box["corner_tl"] + box["hbar"] * (longest_msg_line + 2) + box["corner_tr"]
+    bottom_bar = box["corner_bl"] + box["hbar"] * (longest_msg_line + 2) + box["corner_br"]
+    msg_lines = [top_bar, *msg_lines, bottom_bar]
+    outcat_lines = []
+    for i, line in enumerate(cat_lines):
+        box_line = ""
+        if i < len(msg_lines):
+            box_line = msg_lines[i]
+        outcat_lines.append(line + box_line)
+    return "\n".join(outcat_lines)
+
+if __name__ == "__main__":
+    main()