summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorKali Kaneko <kali@leap.se>2014-01-13 13:20:00 -0400
committerKali Kaneko <kali@leap.se>2014-01-13 13:20:00 -0400
commit51eaab77deedf0c923fe40cf3d346fa879bf2ae3 (patch)
tree6650583e59baf09ceca794ecc6ba080d6c7aaa62 /src
parent2d77a64c7cf3e70070e25af4ef75af7de7a3a5b1 (diff)
Add check for uniqueness when adding mails.
Check by mbox + content-hash
Diffstat (limited to 'src')
-rw-r--r--src/leap/mail/imap/fields.py4
-rw-r--r--src/leap/mail/imap/mailbox.py6
-rw-r--r--src/leap/mail/imap/messages.py50
3 files changed, 57 insertions, 3 deletions
diff --git a/src/leap/mail/imap/fields.py b/src/leap/mail/imap/fields.py
index 2545adf..70af61f 100644
--- a/src/leap/mail/imap/fields.py
+++ b/src/leap/mail/imap/fields.py
@@ -99,6 +99,7 @@ class WithMsgFields(object):
TYPE_MBOX_SEEN_IDX = 'by-type-and-mbox-and-seen'
TYPE_MBOX_RECT_IDX = 'by-type-and-mbox-and-recent'
TYPE_MBOX_DEL_IDX = 'by-type-and-mbox-and-deleted'
+ TYPE_MBOX_C_HASH_IDX = 'by-type-and-mbox-and-contenthash'
TYPE_C_HASH_IDX = 'by-type-and-contenthash'
TYPE_C_HASH_PART_IDX = 'by-type-and-contenthash-and-partnumber'
TYPE_P_HASH_IDX = 'by-type-and-payloadhash'
@@ -121,6 +122,9 @@ class WithMsgFields(object):
# mailboxes
TYPE_SUBS_IDX: [KTYPE, 'bool(subscribed)'],
+ # fdocs uniqueness
+ TYPE_MBOX_C_HASH_IDX: [KTYPE, MBOX_VAL, CHASH_VAL],
+
# content, headers doc
TYPE_C_HASH_IDX: [KTYPE, CHASH_VAL],
diff --git a/src/leap/mail/imap/mailbox.py b/src/leap/mail/imap/mailbox.py
index 7c01490..c9e8684 100644
--- a/src/leap/mail/imap/mailbox.py
+++ b/src/leap/mail/imap/mailbox.py
@@ -125,7 +125,7 @@ class SoledadMailbox(WithMsgFields, MBoxParser):
def addListener(self, listener):
"""
- Adds a listener to the listeners queue.
+ Add a listener to the listeners queue.
The server adds itself as a listener when there is a SELECT,
so it can send EXIST commands.
@@ -137,7 +137,7 @@ class SoledadMailbox(WithMsgFields, MBoxParser):
def removeListener(self, listener):
"""
- Removes a listener from the listeners queue.
+ Remove a listener from the listeners queue.
:param listener: listener to remove
:type listener: an object that implements IMailboxListener
@@ -146,7 +146,7 @@ class SoledadMailbox(WithMsgFields, MBoxParser):
def _get_mbox(self):
"""
- Returns mailbox document.
+ Return mailbox document.
:return: A SoledadDocument containing this mailbox, or None if
the query failed.
diff --git a/src/leap/mail/imap/messages.py b/src/leap/mail/imap/messages.py
index 37e4311..a3fcd87 100644
--- a/src/leap/mail/imap/messages.py
+++ b/src/leap/mail/imap/messages.py
@@ -1064,10 +1064,27 @@ class MessageCollection(WithMsgFields, IndexedDB, MailParser, MBoxParser):
hd[self.DATE_KEY] = date
return hd
+ def _fdoc_already_exists(self, chash):
+ """
+ Check whether we can find a flags doc for this mailbox with the
+ given content-hash. It enforces that we can only have the same maessage
+ listed once for a a given mailbox.
+
+ :param chash: the content-hash to check about.
+ :type chash: basestring
+ :return: False, if it does not exist, or UID.
+ """
+ exist = self._get_fdoc_from_chash(chash)
+ if exist:
+ return exist.content.get(fields.UID_KEY, "unknown-uid")
+ else:
+ return False
+
@deferred
def add_msg(self, raw, subject=None, flags=None, date=None, uid=1):
"""
Creates a new message document.
+ Here lives the magic of the leap mail. Well, in soledad, really.
:param raw: the raw message
:type raw: str
@@ -1097,6 +1114,14 @@ class MessageCollection(WithMsgFields, IndexedDB, MailParser, MBoxParser):
# parse
msg, chash, size, multi = self._do_parse(raw)
+ # check for uniqueness.
+ if self._fdoc_already_exists(chash):
+ logger.warning("We already have that message in this mailbox.")
+ # note that this operation will leave holes in the UID sequence,
+ # but we're gonna change that all the same for a local-only table.
+ # so not touch it by the moment.
+ return False
+
fd = self._populate_flags(flags, uid, chash, size, multi)
hd = self._populate_headr(msg, chash, subject, date)
@@ -1156,6 +1181,31 @@ class MessageCollection(WithMsgFields, IndexedDB, MailParser, MBoxParser):
# getters
+ def _get_fdoc_from_chash(self, chash):
+ """
+ Return a flags document for this mailbox with a given chash.
+
+ :return: A SoledadDocument containing the Flags Document, or None if
+ the query failed.
+ :rtype: SoledadDocument or None.
+ """
+ try:
+ query = self._soledad.get_from_index(
+ fields.TYPE_MBOX_C_HASH_IDX,
+ fields.TYPE_FLAGS_VAL, self.mbox, chash)
+ if query:
+ if len(query) > 1:
+ logger.warning(
+ "More than one fdoc found for this chash, "
+ "we got a duplicate!!")
+ # XXX we could take action, like trigger a background
+ # process to kill dupes.
+ return query.pop()
+ else:
+ return None
+ except Exception as exc:
+ logger.exception("Unhandled error %r" % exc)
+
def get_msg_by_uid(self, uid):
"""
Retrieves a LeapMessage by UID.