aboutsummaryrefslogtreecommitdiff
path: root/contrib/plaintext.py
blob: 522055bf5a3d17a170aea1dc85dc386351733f3e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#!/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 .*, .* wrote:")
quote_re = re.compile(r">+")

sys.stdin.reconfigure(encoding='utf-8', errors='ignore')
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)