From 289de6576ef89fd0fc14caff27d37dd9dcbbe756 Mon Sep 17 00:00:00 2001
From: Jaidyn Levesque <10477760+JadedCtrl@users.noreply.github.com>
Date: Tue, 24 Jan 2023 19:26:13 -0600
Subject: [PATCH] Add addr/name-only variants of 'To' and 'From'

---
 Maildir.py | 90 +++++++++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 75 insertions(+), 15 deletions(-)

diff --git a/Maildir.py b/Maildir.py
index 1946645..80d681c 100755
--- a/Maildir.py
+++ b/Maildir.py
@@ -16,6 +16,7 @@
 #
 
 import os
+import re
 from sys import argv
 import urllib
 import email.parser
@@ -30,20 +31,41 @@ class Maildir(GObject.GObject,
               Caja.InfoProvider):
 
 
+    # Pass our columns to Caja
     def get_columns(self):
         return (
-            Caja.Column(
-                name="Maildir::from_column",
-                attribute="from",
-                label="From",
-                description=""
-            ),
             Caja.Column(
                 name="Maildir::subject_column",
                 attribute="subject",
                 label="Subject",
                 description=""
             ),
+            Caja.Column(
+                name="Maildir::sent_column",
+                attribute="sent",
+                label="Sent",
+                description=""
+            ),
+
+            Caja.Column(
+                name="Maildir::from_column",
+                attribute="from",
+                label="From",
+                description=""
+            ),
+            Caja.Column(
+                name="Maildir::from_addr_column",
+                attribute="from_addr",
+                label="From (Addr)",
+                description=""
+            ),
+            Caja.Column(
+                name="Maildir::from_name_column",
+                attribute="from_name",
+                label="From (Name)",
+                description=""
+            ),
+
             Caja.Column(
                 name="Maildir::to_column",
                 attribute="to",
@@ -51,14 +73,21 @@ class Maildir(GObject.GObject,
                 description=""
             ),
             Caja.Column(
-                name="Maildir::sent_column",
-                attribute="sent",
-                label="Sent",
+                name="Maildir::to_addr_column",
+                attribute="to_addr",
+                label="To (Addr)",
+                description=""
+            ),
+            Caja.Column(
+                name="Maildir::to_name_column",
+                attribute="to_name",
+                label="To (Name)",
                 description=""
             )
         )
 
 
+    # Implants e-mail-related attributes into a file's columns
     def update_file_info(self, file):
         if not file.is_mime_type("message/rfc822"):
             return
@@ -66,10 +95,41 @@ class Maildir(GObject.GObject,
         filename = file.get_uri()[7:]
         message = email.message_from_file(open(filename))
 
-        file.add_string_attribute('to', message.get("To"))
-        file.add_string_attribute('from', message.get("From"))
-        file.add_string_attribute('subject', message.get("Subject"))
+        sender = message.get("From")
+        to = message.get("To")
+        subject = message.get("Subject")
 
-        date = parser.parse(message.get("Date"))
-        if date:
-            file.add_string_attribute('sent', date.strftime("%Y-%m-%d %H:%M:%S %z"))
+        if to:
+            file.add_string_attribute('to', to)
+            file.add_string_attribute('to_addr', self.from_header_addr(to))
+            file.add_string_attribute('to_name', self.from_header_name(to))
+
+        if sender:
+            file.add_string_attribute('from', sender)
+            file.add_string_attribute('from_addr', self.from_header_addr(sender))
+            file.add_string_attribute('from_name', self.from_header_name(sender))
+
+        if subject:
+            file.add_string_attribute('subject', message.get("Subject"))
+
+        try:
+            date = parser.parse(message.get("Date"))
+            if date:
+                file.add_string_attribute('sent', date.strftime("%Y-%m-%d %H:%M:%S %z"))
+        except:
+            print("The date couldn't be parsed — that's alright, we didn't need it anyway.")
+
+
+    # Helper function, for parsing e-mail addresses of to/from headers
+    def from_header_addr(self, str):
+        if re.search(r'<', str):
+            return re.sub('[<|>]', '',
+                          re.search(r'<.*@.*>', str).group(0))
+        return str
+
+
+    # Helper function, for parsing names of to/from headers
+    def from_header_name(self, str):
+        if re.search(r'<', str):
+            return re.sub('<.*', '', str)
+        return ''