From 56ea7f0e430bc44f8b652ef88bedb186b6b3e0df Mon Sep 17 00:00:00 2001
From: Drew DeVault <sir@cmpwn.com>
Date: Sun, 31 Mar 2019 12:57:44 -0400
Subject: Add plaintext highlighter, escape ANSI in source

---
 contrib/plaintext.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100755 contrib/plaintext.py

(limited to 'contrib/plaintext.py')

diff --git a/contrib/plaintext.py b/contrib/plaintext.py
new file mode 100755
index 00000000..3502aff9
--- /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)
-- 
cgit