Add plaintext highlighter, escape ANSI in source

1 files changed, 21 insertions, 0 deletions

diff --git a/contrib/plaintext.py b/contrib/plaintext.py
new file mode 100755
index 0000000..3502aff
--- /dev/null
+++ b/contrib/plaintext.py
@@ -0,0 +1,21 @@
+#!/usr/bin/env python3
+from colorama import Fore, Style
+import sys
+import re
+
+# TODO: Wrap text to terminal width?
+
+ansi_escape = re.compile(r'\x1B\[[0-?]*[ -/]*[@-~]')
+# TODO: I guess this might vary from MUA to MUA. I've definitely seen localized
+# versions in the wild
+quote_prefix_re = re.compile(r"On [0-9: -]+( (AM|PM))?, .* wrote:")
+quote_re = re.compile(r">+ ")
+
+mail = sys.stdin.read().replace("\r\n", "\n")
+mail = ansi_escape.sub('', mail)
+
+for line in mail.split("\n"):
+    if quote_re.match(line) or quote_prefix_re.match(line):
+        print(f"{Style.DIM}{Fore.CYAN}{line}{Style.RESET_ALL}")
+    else:
+        print(line)