[+] Add padding and color options to print_ascii

2023-03-08 02:00:19 -05:00
parent 15a2ba419a
commit ec4bd85477
1 changed files with 8 additions and 1 deletions
@@ -167,12 +167,19 @@ async def shell(reader: TelnetReaderUnicode, writer: TelnetWriterUnicode):
        return height, width

    # Print ascii art
-    def print_ascii(asc: str, x: int, y: int):
+    def print_ascii(asc: str | AsciiArt, x: int, y: int, color: RGB | None = None, pad: int = 0):
+        if isinstance(asc, AsciiArt):
+            asc = asc.art
        asc = asc.strip('\n')
        # Write ascii line by line
        for i, line in enumerate(asc.splitlines()):
            writer.write(f'\x1b[{y + i};{x}H')
+            writer.write(' ' * pad)
+            if color is not None:
+                writer.write(color.to_ansi_rgb())
            writer.write(line)
+            writer.write(' ' * pad)
+            writer.write('\x1b[0m')

    # Clear the screen
    def clear():