diff options
Diffstat (limited to 'src/plugins/notify/notify-storage.c')
| -rw-r--r-- | src/plugins/notify/notify-storage.c | 269 |
1 files changed, 269 insertions, 0 deletions
diff --git a/src/plugins/notify/notify-storage.c b/src/plugins/notify/notify-storage.c new file mode 100644 index 0000000..b7a6a3a --- /dev/null +++ b/src/plugins/notify/notify-storage.c @@ -0,0 +1,269 @@ +/* Copyright (c) 2013-2018 Dovecot authors, see the included COPYING file */ + +#include "lib.h" +#include "array.h" +#include "mail-storage-private.h" +#include "mailbox-list-private.h" +#include "notify-plugin-private.h" + +#define NOTIFY_CONTEXT(obj) \ + MODULE_CONTEXT_REQUIRE(obj, notify_storage_module) +#define NOTIFY_MAIL_CONTEXT(obj) \ + MODULE_CONTEXT_REQUIRE(obj, notify_mail_module) + +static MODULE_CONTEXT_DEFINE_INIT(notify_storage_module, + &mail_storage_module_register); +static MODULE_CONTEXT_DEFINE_INIT(notify_mail_module, + &mail_module_register); + +static void +notify_mail_expunge(struct mail *_mail) +{ + struct mail_private *mail = (struct mail_private *)_mail; + union mail_module_context *lmail = NOTIFY_MAIL_CONTEXT(mail); + + notify_contexts_mail_expunge(_mail); + lmail->super.expunge(_mail); +} + +static void +notify_mail_update_flags(struct mail *_mail, enum modify_type modify_type, + enum mail_flags flags) +{ + struct mail_private *mail = (struct mail_private *)_mail; + union mail_module_context *lmail = NOTIFY_MAIL_CONTEXT(mail); + enum mail_flags old_flags, new_flags; + + old_flags = mail_get_flags(_mail); + lmail->super.update_flags(_mail, modify_type, flags); + new_flags = mail_get_flags(_mail); + + if ((old_flags ^ new_flags) == 0) + return; + + notify_contexts_mail_update_flags(_mail, old_flags); +} + +static void +notify_mail_update_keywords(struct mail *_mail, enum modify_type modify_type, + struct mail_keywords *keywords) +{ + struct mail_private *mail = (struct mail_private *)_mail; + union mail_module_context *lmail = NOTIFY_MAIL_CONTEXT(mail); + const char *const *old_keywords, *const *new_keywords; + unsigned int i; + + old_keywords = mail_get_keywords(_mail); + lmail->super.update_keywords(_mail, modify_type, keywords); + new_keywords = mail_get_keywords(_mail); + + for (i = 0; old_keywords[i] != NULL && new_keywords[i] != NULL; i++) { + if (strcmp(old_keywords[i], new_keywords[i]) != 0) + break; + } + + if (old_keywords[i] == NULL && new_keywords[i] == NULL) + return; + + notify_contexts_mail_update_keywords(_mail, old_keywords); +} + +static void notify_mail_allocated(struct mail *_mail) +{ + struct mail_private *mail = (struct mail_private *)_mail; + struct mail_vfuncs *v = mail->vlast; + union mail_module_context *lmail; + + if ((_mail->transaction->flags & MAILBOX_TRANSACTION_FLAG_NO_NOTIFY) != 0) + return; + + lmail = p_new(mail->pool, union mail_module_context, 1); + lmail->super = *v; + mail->vlast = &lmail->super; + + v->expunge = notify_mail_expunge; + v->update_flags = notify_mail_update_flags; + v->update_keywords = notify_mail_update_keywords; + MODULE_CONTEXT_SET_SELF(mail, notify_mail_module, lmail); +} + +static int +notify_copy(struct mail_save_context *ctx, struct mail *mail) +{ + union mailbox_module_context *lbox = + NOTIFY_CONTEXT(ctx->transaction->box); + int ret; + + if ((ret = lbox->super.copy(ctx, mail)) < 0) + return -1; + + if ((ctx->transaction->flags & MAILBOX_TRANSACTION_FLAG_NO_NOTIFY) != 0) { + /* no notifications */ + } else if (ctx->saving) { + /* we came from mailbox_save_using_mail() */ + notify_contexts_mail_save(ctx->dest_mail); + } else { + notify_contexts_mail_copy(mail, ctx->dest_mail); + } + return ret; +} + +static int +notify_save_finish(struct mail_save_context *ctx) +{ + union mailbox_module_context *lbox = + NOTIFY_CONTEXT(ctx->transaction->box); + struct mail *dest_mail = ctx->copying_via_save ? NULL : ctx->dest_mail; + + if (lbox->super.save_finish(ctx) < 0) + return -1; + if (dest_mail != NULL && + (ctx->transaction->flags & MAILBOX_TRANSACTION_FLAG_NO_NOTIFY) == 0) + notify_contexts_mail_save(dest_mail); + return 0; +} + +static struct mailbox_transaction_context * +notify_transaction_begin(struct mailbox *box, + enum mailbox_transaction_flags flags, + const char *reason) +{ + union mailbox_module_context *lbox = NOTIFY_CONTEXT(box); + struct mailbox_transaction_context *t; + + t = lbox->super.transaction_begin(box, flags, reason); + + if ((t->flags & MAILBOX_TRANSACTION_FLAG_NO_NOTIFY) == 0) + notify_contexts_mail_transaction_begin(t); + return t; +} + +static int +notify_transaction_commit(struct mailbox_transaction_context *t, + struct mail_transaction_commit_changes *changes_r) +{ + union mailbox_module_context *lbox = NOTIFY_CONTEXT(t->box); + bool no_notify = (t->flags & MAILBOX_TRANSACTION_FLAG_NO_NOTIFY) != 0; + + if ((lbox->super.transaction_commit(t, changes_r)) < 0) { + if (!no_notify) + notify_contexts_mail_transaction_rollback(t); + return -1; + } + + /* FIXME: note that t is already freed at this stage. it's not actually + being dereferenced anymore though. still, a bit unsafe.. */ + if (!no_notify) + notify_contexts_mail_transaction_commit(t, changes_r); + return 0; +} + +static void +notify_transaction_rollback(struct mailbox_transaction_context *t) +{ + union mailbox_module_context *lbox = NOTIFY_CONTEXT(t->box); + + if ((t->flags & MAILBOX_TRANSACTION_FLAG_NO_NOTIFY) == 0) + notify_contexts_mail_transaction_rollback(t); + lbox->super.transaction_rollback(t); +} + +static int +notify_mailbox_create(struct mailbox *box, const struct mailbox_update *update, + bool directory) +{ + union mailbox_module_context *lbox = NOTIFY_CONTEXT(box); + + if (lbox->super.create_box(box, update, directory) < 0) + return -1; + + notify_contexts_mailbox_create(box); + return 0; +} + +static int +notify_mailbox_update(struct mailbox *box, const struct mailbox_update *update) +{ + union mailbox_module_context *lbox = NOTIFY_CONTEXT(box); + + if (lbox->super.update_box(box, update) < 0) + return -1; + + notify_contexts_mailbox_update(box); + return 0; +} + +static int +notify_mailbox_delete(struct mailbox *box) +{ + union mailbox_module_context *lbox = NOTIFY_CONTEXT(box); + + notify_contexts_mailbox_delete_begin(box); + if (lbox->super.delete_box(box) < 0) { + notify_contexts_mailbox_delete_rollback(); + return -1; + } + notify_contexts_mailbox_delete_commit(box); + return 0; +} + +static int +notify_mailbox_rename(struct mailbox *src, struct mailbox *dest) +{ + union mailbox_module_context *lbox = NOTIFY_CONTEXT(src); + + if (lbox->super.rename_box(src, dest) < 0) + return -1; + + notify_contexts_mailbox_rename(src, dest); + return 0; +} + +static int +notify_mailbox_set_subscribed(struct mailbox *box, bool set) +{ + union mailbox_module_context *lbox = NOTIFY_CONTEXT(box); + + if (lbox->super.set_subscribed(box, set) < 0) + return -1; + + notify_contexts_mailbox_set_subscribed(box, set); + return 0; +} + +static void notify_mailbox_allocated(struct mailbox *box) +{ + struct mailbox_vfuncs *v = box->vlast; + union mailbox_module_context *lbox; + + lbox = p_new(box->pool, union mailbox_module_context, 1); + lbox->super = *v; + box->vlast = &lbox->super; + + v->copy = notify_copy; + v->save_finish = notify_save_finish; + v->transaction_begin = notify_transaction_begin; + v->transaction_commit = notify_transaction_commit; + v->transaction_rollback = notify_transaction_rollback; + v->create_box = notify_mailbox_create; + v->update_box = notify_mailbox_update; + v->delete_box = notify_mailbox_delete; + v->rename_box = notify_mailbox_rename; + v->set_subscribed = notify_mailbox_set_subscribed; + MODULE_CONTEXT_SET_SELF(box, notify_storage_module, lbox); +} + +static struct mail_storage_hooks notify_mail_storage_hooks = { + .mailbox_allocated = notify_mailbox_allocated, + .mail_allocated = notify_mail_allocated +}; + +void notify_plugin_init_storage(struct module *module) +{ + mail_storage_hooks_add(module, ¬ify_mail_storage_hooks); +} + +void notify_plugin_deinit_storage(void) +{ + mail_storage_hooks_remove(¬ify_mail_storage_hooks); +} |