src/doveadm/doveadm-mail-altmove.c

	doveadm-mail-altmove.c revision ab406f5c4fd6373ead6c095dc1c33525b789da19
/* Copyright (c) 2010-2012 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "array.h"
#include "mail-index.h"
#include "mail-storage.h"
#include "mail-namespace.h"
#include "doveadm-mailbox-list-iter.h"
#include "doveadm-mail-iter.h"
#include "doveadm-mail.h"

struct altmove_cmd_context {
    struct doveadm_mail_cmd_context ctx;
    bool reverse;
};

static int
cmd_altmove_box(struct doveadm_mail_cmd_context *ctx,
        const struct mailbox_info *info,
        struct mail_search_args *search_args, bool reverse)
{
    struct doveadm_mail_iter *iter;
    struct mailbox_transaction_context *trans;
    struct mail *mail;
    enum modify_type modify_type =
        !reverse ? MODIFY_ADD : MODIFY_REMOVE;

    if (doveadm_mail_iter_init(ctx, info, search_args, 0, NULL,
                   &trans, &iter) < 0)
        return -1;

    while (doveadm_mail_iter_next(iter, &mail)) {
        if (doveadm_debug) {
            i_debug("altmove: box=%s uid=%u",
                info->name, mail->uid);
        }
        mail_update_flags(mail, modify_type,
            (enum mail_flags)MAIL_INDEX_MAIL_FLAG_BACKEND);
    }
    return doveadm_mail_iter_deinit_sync(&iter);
}

static int
ns_purge(struct doveadm_mail_cmd_context *ctx, struct mail_namespace *ns)
{
    if (mail_storage_purge(ns->storage) < 0) {
        i_error("Purging namespace '%s' failed: %s", ns->prefix,
            mail_storage_get_last_error(ns->storage, NULL));
        doveadm_mail_failed_storage(ctx, ns->storage);
        return -1;
    }
    return 0;
}

static int
cmd_altmove_run(struct doveadm_mail_cmd_context *_ctx, struct mail_user *user)
{
    struct altmove_cmd_context *ctx = (struct altmove_cmd_context *)_ctx;
    const enum mailbox_list_iter_flags iter_flags =
        MAILBOX_LIST_ITER_RAW_LIST |
        MAILBOX_LIST_ITER_NO_AUTO_BOXES |
        MAILBOX_LIST_ITER_RETURN_NO_FLAGS;
    struct doveadm_mailbox_list_iter *iter;
    const struct mailbox_info *info;
    struct mail_namespace *ns, *prev_ns = NULL;
    ARRAY_DEFINE(purged_storages, struct mail_storage *);
    struct mail_storage *const *storages;
    unsigned int i, count;
    int ret = 0;

    t_array_init(&purged_storages, 8);
    iter = doveadm_mailbox_list_iter_init(_ctx, user, _ctx->search_args,
                          iter_flags);
    while ((info = doveadm_mailbox_list_iter_next(iter)) != NULL) T_BEGIN {
        if (info->ns != prev_ns) {
            if (prev_ns != NULL) {
                if (ns_purge(_ctx, prev_ns) < 0)
                    ret = -1;
                array_append(&purged_storages,
                         &prev_ns->storage, 1);
            }
            prev_ns = info->ns;
        }
        if (cmd_altmove_box(_ctx, info, _ctx->search_args, ctx->reverse) < 0)
            ret = -1;
    } T_END;
    if (doveadm_mailbox_list_iter_deinit(&iter) < 0)
        ret = -1;

    if (prev_ns != NULL) {
        if (ns_purge(_ctx, prev_ns) < 0)
            ret = -1;
        array_append(&purged_storages, &prev_ns->storage, 1);
    }

    /* make sure all private storages have been purged */
    storages = array_get(&purged_storages, &count);
    for (ns = user->namespaces; ns != NULL; ns = ns->next) {
        if (ns->type != NAMESPACE_PRIVATE)
            continue;

        for (i = 0; i < count; i++) {
            if (ns->storage == storages[i])
                break;
        }
        if (i == count) {
            if (ns_purge(_ctx, ns) < 0)
                ret = -1;
            array_append(&purged_storages, &ns->storage, 1);
            storages = array_get(&purged_storages, &count);
        }
    }
    return ret;
}

static void cmd_altmove_init(struct doveadm_mail_cmd_context *ctx,
                 const char *const args[])
{
    if (args[0] == NULL)
        doveadm_mail_help_name("altmove");
    ctx->search_args = doveadm_mail_build_search_args(args);
}

static bool
cmd_mailbox_altmove_parse_arg(struct doveadm_mail_cmd_context *_ctx, int c)
{
    struct altmove_cmd_context *ctx = (struct altmove_cmd_context *)_ctx;

    switch (c) {
    case 'r':
        ctx->reverse = TRUE;
        break;
    default:
        return FALSE;
    }
    return TRUE;
}

static struct doveadm_mail_cmd_context *cmd_altmove_alloc(void)
{
    struct altmove_cmd_context *ctx;

    ctx = doveadm_mail_cmd_alloc(struct altmove_cmd_context);
    ctx->ctx.getopt_args = "r";
    ctx->ctx.v.parse_arg = cmd_mailbox_altmove_parse_arg;
    ctx->ctx.v.init = cmd_altmove_init;
    ctx->ctx.v.run = cmd_altmove_run;
    return &ctx->ctx;
}

struct doveadm_mail_cmd cmd_altmove = {
    cmd_altmove_alloc, "altmove", "[-r] <search query>"
};