src/imap/imap-sync.c

	imap-sync.c revision e60a349c641bb2f4723e4a395a25f55531682d2b
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen/* Copyright (C) 2002-2004 Timo Sirainen */
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen#include "common.h"
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen#include "str.h"
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen#include "mail-storage.h"
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen#include "imap-util.h"
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen#include "imap-sync.h"
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen#include "commands.h"
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainenstruct cmd_sync_context {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    const char *tagline;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct imap_sync_context *sync_ctx;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen};
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainenstruct imap_sync_context {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct client *client;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct mailbox *box;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct mailbox_transaction_context *t;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct mailbox_sync_context *sync_ctx;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct mailbox_sync_rec sync_rec;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    uint32_t seq;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    unsigned int messages_count;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    int failed;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen};
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainenstruct imap_sync_context *
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainenimap_sync_init(struct client *client, struct mailbox *box,
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen           enum mailbox_sync_flags flags)
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen{
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct imap_sync_context *ctx;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    i_assert(client->mailbox == box);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    ctx = i_new(struct imap_sync_context, 1);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    ctx->client = client;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    ctx->box = box;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    ctx->sync_ctx = mailbox_sync_init(box, flags);
1b3bb8d39686ed24730cbc31cc9a33dc62c8c6c3Timo Sirainen    ctx->t = mailbox_transaction_begin(box, FALSE);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    ctx->messages_count = client->messages_count;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    return ctx;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen}
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainenint imap_sync_deinit(struct imap_sync_context *ctx)
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen{
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen    struct mailbox_status status;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    if (mailbox_sync_deinit(ctx->sync_ctx, &status) < 0 || ctx->failed) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        mailbox_transaction_rollback(ctx->t);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        i_free(ctx);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        return -1;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
b2c1349cf07410aefab0f5b17153af9e5cfcf48fTimo Sirainen    mailbox_transaction_commit(ctx->t, 0);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    t_push();
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    ctx->client->messages_count = status.messages;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    if (status.messages != ctx->messages_count) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        client_send_line(ctx->client,
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen            t_strdup_printf("* %u EXISTS", status.messages));
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    if (status.recent != ctx->client->recent_count) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                ctx->client->recent_count = status.recent;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        client_send_line(ctx->client,
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen            t_strdup_printf("* %u RECENT", status.recent));
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    /*FIXME:client_save_keywords(&client->keywords, keywords, keywords_count);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    client_send_mailbox_flags(client, mailbox, keywords, keywords_count);*/
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    t_pop();
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    i_free(ctx);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    return 0;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen}
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainenint imap_sync_more(struct imap_sync_context *ctx)
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen{
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen    struct mail *mail;
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen    enum mail_flags flags;
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen    const char *const *keywords;
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen    string_t *str;
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen    int ret = 1;
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen    t_push();
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen    str = t_str_new(256);
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    for (;;) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        if (ctx->seq == 0) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen            /* get next one */
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen            ret = mailbox_sync_next(ctx->sync_ctx,
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                        &ctx->sync_rec);
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen            if (ret <= 0) {
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                if (ret == 0) {
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                    /* all finished ok */
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                    ret = 1;
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                break;
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen            }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        switch (ctx->sync_rec.type) {
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen        case MAILBOX_SYNC_TYPE_FLAGS:
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen        case MAILBOX_SYNC_TYPE_KEYWORDS:
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen            if (ctx->seq == 0)
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                ctx->seq = ctx->sync_rec.seq1;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen            for (; ctx->seq <= ctx->sync_rec.seq2; ctx->seq++) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                mail = mailbox_fetch(ctx->t, ctx->seq,
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                             MAIL_FETCH_FLAGS);
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen                flags = mail->get_flags(mail);
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen                keywords = mail->get_keywords(mail);
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen                str_truncate(str, 0);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                str_printfa(str, "* %u FETCH (FLAGS (",
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                        ctx->seq);
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen                imap_write_flags(str, flags, keywords);
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen                str_append(str, "))");
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                ret = client_send_line(ctx->client, str_c(str));
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                if (ret <= 0) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                    t_pop();
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                    return ret;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                }
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen            }
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen            break;
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen        case MAILBOX_SYNC_TYPE_EXPUNGE:
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen            if (ctx->seq == 0) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                ctx->seq = ctx->sync_rec.seq2;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                ctx->messages_count -=
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                    ctx->sync_rec.seq2 -
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                    ctx->sync_rec.seq1 + 1;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen            }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen            for (; ctx->seq >= ctx->sync_rec.seq1; ctx->seq--) {
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen                str_truncate(str, 0);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                str_printfa(str, "* %u EXPUNGE", ctx->seq);
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                ret = client_send_line(ctx->client, str_c(str));
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                if (ret <= 0) {
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                    t_pop();
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen                    return ret;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen                }
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen            }
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen            break;
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen        }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        ctx->seq = 0;
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen    }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    t_pop();
68efcccb384f2d6871164b072457e87473502c51Timo Sirainen    return ret;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen}
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainenint imap_sync_nonselected(struct mailbox *box, enum mailbox_sync_flags flags)
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen{
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct mailbox_sync_context *ctx;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        struct mailbox_sync_rec sync_rec;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    struct mailbox_status status;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    ctx = mailbox_sync_init(box, flags);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    while (mailbox_sync_next(ctx, &sync_rec) > 0)
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        ;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    return mailbox_sync_deinit(ctx, &status);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen}
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainenstatic int cmd_sync_continue(struct client_command_context *cmd)
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen{
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    struct cmd_sync_context *ctx = cmd->context;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    if (imap_sync_more(ctx->sync_ctx) == 0)
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        return FALSE;
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    if (imap_sync_deinit(ctx->sync_ctx) < 0) {
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen        client_send_untagged_storage_error(cmd->client,
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen            mailbox_get_storage(cmd->client->mailbox));
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen    }
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    client_send_tagline(cmd, ctx->tagline);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen    return TRUE;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen}
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainenint cmd_sync(struct client_command_context *cmd, enum mailbox_sync_flags flags,
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen         const char *tagline)
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen{
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        struct cmd_sync_context *ctx;
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    if (cmd->client->mailbox == NULL) {
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen        client_send_tagline(cmd, tagline);
c4267cf4c40fb1f866b5958ff122ef836b8c5dfbTimo Sirainen        return TRUE;
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen    }
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    ctx = p_new(cmd->pool, struct cmd_sync_context, 1);
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    ctx->tagline = p_strdup(cmd->pool, tagline);
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    ctx->sync_ctx = imap_sync_init(cmd->client, cmd->client->mailbox,
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen                       flags);
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    cmd->func = cmd_sync_continue;
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    cmd->context = ctx;
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    cmd->client->command_pending = TRUE;
e60a349c641bb2f4723e4a395a25f55531682d2bTimo Sirainen    return cmd_sync_continue(cmd);
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen}