src/imap/cmd-copy.c

	cmd-copy.c revision 0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834
/* Copyright (C) 2002 Timo Sirainen */

#include "common.h"
#include "commands.h"
#include "imap-search.h"

static int fetch_and_copy(struct mailbox_transaction_context *t,
              struct mailbox *srcbox,
              struct mail_search_arg *search_args)
{
    struct mail_search_context *search_ctx;
        struct mailbox_transaction_context *src_trans;
    struct mail *mail;
    int ret;

    src_trans = mailbox_transaction_begin(srcbox, FALSE);
    search_ctx = mailbox_search_init(src_trans, NULL, search_args, NULL,
                     MAIL_FETCH_STREAM_HEADER |
                     MAIL_FETCH_STREAM_BODY, NULL);
    if (search_ctx == NULL) {
        mailbox_transaction_rollback(src_trans);
        return -1;
    }

    ret = 1;
    while ((mail = mailbox_search_next(search_ctx)) != NULL) {
        if (mail->expunged) {
            ret = 0;
            break;
        }
        if (mailbox_copy(t, mail) < 0) {
            ret = -1;
            break;
        }
    }

    if (mailbox_search_deinit(search_ctx) < 0)
        ret = -1;

    if (mailbox_transaction_commit(src_trans) < 0)
        ret = -1;

    return ret;
}

int cmd_copy(struct client *client)
{
    struct mail_storage *storage;
    struct mailbox *destbox;
    struct mailbox_transaction_context *t;
        struct mail_search_arg *search_arg;
    const char *messageset, *mailbox;
    int ret;

    /* <message set> <mailbox> */
    if (!client_read_string_args(client, 2, &messageset, &mailbox))
        return FALSE;

    if (!client_verify_open_mailbox(client))
        return TRUE;

    if (!client_verify_mailbox_name(client, mailbox, TRUE, FALSE))
        return TRUE;

    /* open the destination mailbox */
    if (!client_verify_mailbox_name(client, mailbox, TRUE, FALSE))
        return TRUE;

    search_arg = imap_search_get_arg(client, messageset, client->cmd_uid);
    if (search_arg == NULL)
        return TRUE;

    storage = client_find_storage(client, mailbox);
    if (storage == NULL)
        return TRUE;

    destbox = mailbox_open(storage, mailbox,
                   mailbox_open_flags | MAILBOX_OPEN_FAST);
    if (destbox == NULL) {
        client_send_storage_error(client, storage);
        return TRUE;
    }

    t = mailbox_transaction_begin(destbox, FALSE);
    ret = fetch_and_copy(t, client->mailbox, search_arg);

    if (ret <= 0)
        mailbox_transaction_rollback(t);
    else {
        if (mailbox_transaction_commit(t) < 0)
            ret = -1;
    }

    if (ret < 0)
        client_send_storage_error(client, storage);
    else if (ret == 0) {
        /* some messages were expunged, sync them */
        client_sync_full(client);
        client_send_tagline(client,
            "NO Some of the requested messages no longer exist.");
    } else {
        if (destbox == client->mailbox)
            client_sync_full(client);
        else
            client_sync_full_fast(client);
        client_send_tagline(client, "OK Copy completed.");
    }

    mailbox_close(destbox);
    return TRUE;
}