path: root/mail/importers/evolution-mbox-importer.c

/*
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) version 3.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with the program; if not, see <http://www.gnu.org/licenses/>
 *
 *
 * Authors:
 *      Iain Holmes <iain@ximian.com>
 *      Michael Zucchi <notzed@ximian.com>
 *
 * Copyright (C) 1999-2008 Novell, Inc. (www.novell.com)
 *
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>

#include <stdio.h>
#include <ctype.h>
#include <string.h>

#include <gtk/gtk.h>
#include <glib/gi18n.h>
#include <glib/gstdio.h>

#include <camel/camel-exception.h>

#include "shell/e-shell.h"
#include "shell/e-shell-window.h"
#include "shell/e-shell-view.h"
#include "shell/e-shell-sidebar.h"

#include "mail/e-mail-local.h"
#include "mail/e-mail-store.h"
#include "mail/em-folder-selection-button.h"
#include "mail/em-folder-tree-model.h"
#include "mail/em-folder-tree.h"
#include "mail/mail-mt.h"

#include "mail-importer.h"

#include "e-util/e-import.h"

typedef struct {
    EImport *import;
    EImportTarget *target;

    GMutex *status_lock;
    gchar *status_what;
    gint status_pc;
    gint status_timeout_id;
    CamelOperation *cancel; /* cancel/status port */

    gchar *uri;
} MboxImporter;

static void
folder_selected(EMFolderSelectionButton *button, EImportTargetURI *target)
{
    g_free(target->uri_dest);
    target->uri_dest = g_strdup(em_folder_selection_button_get_selection(button));
}

static GtkWidget *
mbox_getwidget(EImport *ei, EImportTarget *target, EImportImporter *im)
{
    GtkWindow *window;
    GtkWidget *hbox, *w;
    gchar *select_uri = NULL;

    /* preselect the folder selected in a mail view */
    window = e_shell_get_active_window (e_shell_get_default ());
    if (E_IS_SHELL_WINDOW (window)) {
        EShellWindow *shell_window;
        const gchar *view;

        shell_window = E_SHELL_WINDOW (window);
        view = e_shell_window_get_active_view (shell_window);

        if (view && g_str_equal (view, "mail")) {
            EShellView *shell_view = e_shell_window_get_shell_view (shell_window, view);

            if (shell_view) {
                EMFolderTree *folder_tree = NULL;
                EShellSidebar *shell_sidebar = e_shell_view_get_shell_sidebar (shell_view);

                g_object_get (shell_sidebar, "folder-tree", &folder_tree, NULL);

                if (folder_tree)
                    select_uri = em_folder_tree_get_selected_uri (folder_tree);
            }
        }
    }

    if (!select_uri)
        select_uri = g_strdup (e_mail_local_get_folder_uri (E_MAIL_FOLDER_INBOX));

    hbox = gtk_hbox_new(FALSE, 0);

    w = gtk_label_new(_("Destination folder:"));
    gtk_box_pack_start((GtkBox *)hbox, w, FALSE, TRUE, 6);

    w = em_folder_selection_button_new(
        _("Select folder"), _("Select folder to import into"));
    em_folder_selection_button_set_selection ((EMFolderSelectionButton *)w, select_uri);
    folder_selected (EM_FOLDER_SELECTION_BUTTON (w), (EImportTargetURI *)target);
    g_signal_connect (w, "selected", G_CALLBACK(folder_selected), target);
    gtk_box_pack_start((GtkBox *)hbox, w, FALSE, TRUE, 6);

    w = gtk_vbox_new(FALSE, 0);
    gtk_box_pack_start((GtkBox *)w, hbox, FALSE, FALSE, 0);
    gtk_widget_show_all(w);

    g_free (select_uri);

    return w;
}

static gboolean
mbox_supported(EImport *ei, EImportTarget *target, EImportImporter *im)
{
    gchar signature[6];
    gboolean ret = FALSE;
    gint fd, n;
    EImportTargetURI *s;
    gchar *filename;

    if (target->type != E_IMPORT_TARGET_URI)
        return FALSE;

    s = (EImportTargetURI *)target;
    if (s->uri_src == NULL)
        return TRUE;

    if (strncmp(s->uri_src, "file:///", strlen("file:///")) != 0)
        return FALSE;

    filename = g_filename_from_uri(s->uri_src, NULL, NULL);
    fd = g_open(filename, O_RDONLY, 0);
    g_free(filename);
    if (fd != -1) {
        n = read(fd, signature, 5);
        ret = n == 5 && memcmp(signature, "From ", 5) == 0;
        close(fd);
    }

    return ret;
}

static void
mbox_status(CamelOperation *op, const gchar *what, gint pc, gpointer data)
{
    MboxImporter *importer = data;

    if (pc == CAMEL_OPERATION_START)
        pc = 0;
    else if (pc == CAMEL_OPERATION_END)
        pc = 100;

    g_mutex_lock(importer->status_lock);
    g_free(importer->status_what);
    importer->status_what = g_strdup(what);
    importer->status_pc = pc;
    g_mutex_unlock(importer->status_lock);
}

static gboolean
mbox_status_timeout(gpointer data)
{
    MboxImporter *importer = data;
    gint pc;
    gchar *what;

    if (importer->status_what) {
        g_mutex_lock(importer->status_lock);
        what = importer->status_what;
        importer->status_what = NULL;
        pc = importer->status_pc;
        g_mutex_unlock(importer->status_lock);

        e_import_status(importer->import, (EImportTarget *)importer->target, what, pc);
    }

    return TRUE;
}

static void
mbox_import_done(gpointer data, CamelException *ex)
{
    MboxImporter *importer = data;

    g_source_remove(importer->status_timeout_id);
    g_free(importer->status_what);
    g_mutex_free(importer->status_lock);
    camel_operation_unref(importer->cancel);

    e_import_complete(importer->import, importer->target);
    g_free(importer);
}

static void
mbox_import(EImport *ei, EImportTarget *target, EImportImporter *im)
{
    MboxImporter *importer;
    gchar *filename;

    /* TODO: do we validate target? */

    importer = g_malloc0(sizeof(*importer));
    g_datalist_set_data(&target->data, "mbox-data", importer);
    importer->import = ei;
    importer->target = target;
    importer->status_lock = g_mutex_new();
    importer->status_timeout_id = g_timeout_add(100, mbox_status_timeout, importer);
    importer->cancel = camel_operation_new(mbox_status, importer);

    filename = g_filename_from_uri(((EImportTargetURI *)target)->uri_src, NULL, NULL);
    mail_importer_import_mbox (
        filename, ((EImportTargetURI *)target)->uri_dest,
        importer->cancel, mbox_import_done, importer);
    g_free(filename);
}

static void
mbox_cancel(EImport *ei, EImportTarget *target, EImportImporter *im)
{
    MboxImporter *importer = g_datalist_get_data(&target->data, "mbox-data");

    if (importer)
        camel_operation_cancel(importer->cancel);
}

static EImportImporter mbox_importer = {
    E_IMPORT_TARGET_URI,
    0,
    mbox_supported,
    mbox_getwidget,
    mbox_import,
    mbox_cancel,
};

EImportImporter *
mbox_importer_peek(void)
{
    mbox_importer.name = _("Berkeley Mailbox (mbox)");
    mbox_importer.description = _("Importer Berkeley Mailbox format folders");

    return &mbox_importer;
}