From bd2783012e2630b7c7ab2fe20060957a7a9a7858 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Amadeusz=20=C5=BBo=C5=82nowski?= <aidecoe@aidecoe.name>
Date: Sun, 26 Nov 2017 15:10:12 +0000
Subject: [PATCH] Add filter tagging mail sent directly to me

Add filter tagging mail sent directly to any of addresses defined in
Notmuch config file: `primary_email` or `other_email`.

Default tag is `to-me` and can be customized with `me_tag` option.
---
 afew/filters/MeFilter.py    | 46 +++++++++++++++++++++++++++++++++++++
 afew/tests/test_settings.py |  1 +
 setup.py                    |  1 +
 3 files changed, 48 insertions(+)
 create mode 100644 afew/filters/MeFilter.py

diff --git a/afew/filters/MeFilter.py b/afew/filters/MeFilter.py
new file mode 100644
index 0000000..3c8623f
--- /dev/null
+++ b/afew/filters/MeFilter.py
@@ -0,0 +1,46 @@
+# coding=utf-8
+from __future__ import print_function, absolute_import, unicode_literals
+
+#
+# Copyright (c) Amadeusz Zolnowski <aidecoe@aidecoe.name>
+#
+# Permission to use, copy, modify, and distribute this software for any
+# purpose with or without fee is hereby granted, provided that the above
+# copyright notice and this permission notice appear in all copies.
+#
+# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+#
+
+import re
+
+from ..utils import filter_compat
+from .BaseFilter import Filter
+from ..NotmuchSettings import notmuch_settings
+
+
+class MeFilter(Filter):
+    message = 'Tagging all mails sent directly to myself'
+    _bare_email_re = re.compile(r"[^<]*<(?P<email>[^@<>]+@[^@<>]+)>")
+
+    def __init__(self, database, me_tag='to-me'):
+        super(MeFilter, self).__init__(database)
+
+        my_addresses = set()
+        my_addresses.add(notmuch_settings.get('user', 'primary_email'))
+        if notmuch_settings.has_option('user', 'other_email'):
+            other_emails = notmuch_settings.get('user', 'other_email').split(';')
+            my_addresses.update(filter_compat(None, other_emails))
+
+        self.query = ' OR '.join('to:"%s"' % address
+                                 for address in my_addresses)
+
+        self.me_tag = me_tag
+
+    def handle_message(self, message):
+        self.add_tags(message, self.me_tag)
diff --git a/afew/tests/test_settings.py b/afew/tests/test_settings.py
index f098927..4261c09 100644
--- a/afew/tests/test_settings.py
+++ b/afew/tests/test_settings.py
@@ -43,6 +43,7 @@ def test_all_builtin_FilterRegistrys_exist(self):
                            'SpamFilter',
                            'Filter',
                            'KillThreadsFilter',
+                           'MeFilter',
                            'SentMailsFilter',
                            'HeaderMatchingFilter',
                            'ListMailsFilter']),
diff --git a/setup.py b/setup.py
index 83ed7d8..b1a868c 100755
--- a/setup.py
+++ b/setup.py
@@ -46,6 +46,7 @@ def get_requires():
             'InboxFilter = afew.filters.InboxFilter:InboxFilter',
             'KillThreadsFilter = afew.filters.KillThreadsFilter:KillThreadsFilter',
             'ListMailsFilter = afew.filters.ListMailsFilter:ListMailsFilter',
+            'MeFilter = afew.filters.MeFilter:MeFilter',
             'SentMailsFilter = afew.filters.SentMailsFilter:SentMailsFilter',
             'SpamFilter = afew.filters.SpamFilter:SpamFilter',
         ],