path: root/mail/mail-offline-handler.c

/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */
/* mail-offline-handler.c
 *
 * Copyright (C) 2001  Ximian, Inc.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of version 2 of the GNU General Public
 * License as published by the Free Software Foundation.
 *
 * 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
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public
 * License along with this program; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 *
 * Authors:
 *   Ettore Perazzoli <ettore@ximian.com>
 *   Dan Winship <danw@ximian.com>
 */

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

#include "mail-offline-handler.h"
#include "mail.h"
#include "mail-ops.h"

#include <gtk/gtkmain.h>

#include <gal/util/e-util.h>

#define PARENT_TYPE bonobo_object_get_type ()
static BonoboObjectClass *parent_class = NULL;

struct _MailOfflineHandlerPrivate {
    GHashTable *sync_table;
};

static gboolean
service_is_relevant (CamelService *service, gboolean going_offline)
{
    if (!(service->provider->flags & CAMEL_PROVIDER_IS_REMOTE) ||
        (service->provider->flags & CAMEL_PROVIDER_IS_EXTERNAL))
        return FALSE;

    if (CAMEL_IS_DISCO_STORE (service) &&
        camel_disco_store_status (CAMEL_DISCO_STORE (service)) == CAMEL_DISCO_STORE_OFFLINE)
            return !going_offline;

    return service->status != CAMEL_SERVICE_DISCONNECTED;
}

static void
add_connection (gpointer key, gpointer data, gpointer user_data)
{
    CamelService *service = key;
    GNOME_Evolution_ConnectionList *list = user_data;

    if (!service_is_relevant (service, TRUE))
        return;

    list->_buffer[list->_length].hostName = CORBA_string_dup (service->url->host);
    list->_buffer[list->_length].type     = CORBA_string_dup (service->provider->name);
    list->_length++;
}

static GNOME_Evolution_ConnectionList *
create_connection_list (void)
{
    GNOME_Evolution_ConnectionList *list;

    list = GNOME_Evolution_ConnectionList__alloc ();
    list->_length = 0;
    list->_maximum = mail_storages_count ();
    list->_buffer = CORBA_sequence_GNOME_Evolution_Connection_allocbuf (list->_maximum);

    mail_storages_foreach (add_connection, list);

    return list;
}

/* GNOME::Evolution::Offline methods.  */

static CORBA_boolean
impl__get_isOffline (PortableServer_Servant servant,
             CORBA_Environment *ev)
{
    return !camel_session_is_online (session);
}

static void
impl_prepareForOffline (PortableServer_Servant servant,
            GNOME_Evolution_ConnectionList **active_connection_list,
            CORBA_Environment *ev)
{
    MailOfflineHandler *offline_handler;
    MailOfflineHandlerPrivate *priv;

    offline_handler = MAIL_OFFLINE_HANDLER (bonobo_object_from_servant (servant));
    priv = offline_handler->priv;

    *active_connection_list = create_connection_list ();
}

/* keep track of each sync in progress */
struct _sync_info {
    char *uri;      /* uri of folder being synced */
    CamelOperation *cancel; /* progress report/cancellation object */
    GNOME_Evolution_SyncFolderProgressListener listener;
    int pc;         /* percent complete (0-100) */
    int lastpc;     /* last percent reported, so we dont overreport */
    int id;         /* timeout id */
    GHashTable *table;      /* the hashtable that we're registered in */
};

static void
impl_cancelSyncFolder (PortableServer_Servant servant,
               const GNOME_Evolution_Folder *folder,
               CORBA_Environment *ev)
{
    MailOfflineHandler *offline_handler;
    MailOfflineHandlerPrivate *priv;
    struct _sync_info *info;

    offline_handler = MAIL_OFFLINE_HANDLER (bonobo_object_from_servant (servant));
    priv = offline_handler->priv;

    info = g_hash_table_lookup(priv->sync_table, folder->physicalUri);
    if (info)
        camel_operation_cancel(info->cancel);
    else
        g_warning("Shell tried to cancel sync of '%s': no such folder", folder->physicalUri);
}

static int sync_timeout(struct _sync_info *info)
{
    CORBA_Environment ev;

    if (info->pc != info->lastpc) {
        CORBA_exception_init(&ev);
        GNOME_Evolution_SyncFolderProgressListener_updateProgress(info->listener, info->pc/100.0, &ev);
        if (ev._major != CORBA_NO_EXCEPTION)
            g_warning("Error updating offline progress");
        CORBA_exception_free(&ev);
        info->lastpc = info->pc;
    }

    return TRUE;
}

static void sync_status(CamelOperation *op, const char *what, int pc, void *data)
{
    struct _sync_info *info = data;

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

    info->pc = pc;
}

static void
sync_done(const char *uri, void *crap)
{
    CORBA_Environment ev;
    struct _sync_info *info = crap;

    g_source_remove(info->id);

    CORBA_exception_init(&ev);
    GNOME_Evolution_SyncFolderProgressListener_reportSuccess(info->listener, &ev);
    if (ev._major != CORBA_NO_EXCEPTION)
        g_warning("Error sending offline completion: hang likely");
    CORBA_Object_release(info->listener, &ev);
    CORBA_exception_free(&ev);

    g_hash_table_remove (info->table, info->uri);
    g_free(info->uri);
    camel_operation_unref(info->cancel);
    g_free(info);
}

static void
impl_syncFolder (PortableServer_Servant servant,
         const GNOME_Evolution_Folder *folder,
         const GNOME_Evolution_SyncFolderProgressListener progress_listener,
         CORBA_Environment *ev)
{
    MailOfflineHandler *offline_handler;
    MailOfflineHandlerPrivate *priv;
    struct _sync_info *info;

    offline_handler = MAIL_OFFLINE_HANDLER(bonobo_object_from_servant (servant));
    priv = offline_handler->priv;

    info = g_malloc(sizeof(*info));
    info->listener = CORBA_Object_duplicate(progress_listener, ev);
    info->pc = 0;
    info->uri = g_strdup(folder->physicalUri);
    info->id = g_timeout_add(500, (GSourceFunc)sync_timeout, info);
    info->cancel = camel_operation_new(sync_status, info);
    info->table = priv->sync_table;

    g_hash_table_insert(priv->sync_table, info->uri, info);

    mail_prep_offline(info->uri, info->cancel, sync_done, info);
}

static void
went_offline (CamelStore *store, void *data)
{
    CORBA_Environment ev;
    GNOME_Evolution_ConnectionList *connection_list;
    GNOME_Evolution_OfflineProgressListener listener = data;

    connection_list = create_connection_list ();

    CORBA_exception_init (&ev);
    GNOME_Evolution_OfflineProgressListener_updateProgress(listener, connection_list, &ev);
    if (ev._major != CORBA_NO_EXCEPTION)
        g_warning ("Error updating offline progress");
    CORBA_Object_release(listener, &ev);

    CORBA_exception_free (&ev);
    CORBA_free (connection_list);
}

static void
storage_go_offline (gpointer key, gpointer value, gpointer data)
{
    CamelStore *store = key;
    GNOME_Evolution_OfflineProgressListener listener = data;
    CORBA_Environment ev;

    CORBA_exception_init(&ev);
    if (service_is_relevant (CAMEL_SERVICE (store), TRUE)) {
        mail_store_set_offline (store, TRUE, went_offline, CORBA_Object_duplicate(listener, &ev));
    }
    CORBA_exception_free(&ev);
}

static void
impl_goOffline (PortableServer_Servant servant,
        const GNOME_Evolution_OfflineProgressListener progress_listener,
        CORBA_Environment *ev)
{
    MailOfflineHandler *offline_handler;

    offline_handler = MAIL_OFFLINE_HANDLER (bonobo_object_from_servant (servant));

    /* This will disable further auto-mail-check action. */
    camel_session_set_online (session, FALSE);

    /* FIXME: If send/receive active, wait for it to finish */

    mail_storages_foreach (storage_go_offline, progress_listener);
}

static void
storage_go_online (gpointer key, gpointer value, gpointer data)
{
    CamelStore *store = key;

    if (service_is_relevant (CAMEL_SERVICE (store), FALSE)) {
        mail_store_set_offline (store, FALSE, NULL, NULL);
        mail_note_store (store, NULL, CORBA_OBJECT_NIL,
                 NULL, NULL);
    }
}

static void
impl_goOnline (PortableServer_Servant servant,
           CORBA_Environment *ev)
{
    MailOfflineHandler *offline_handler;
    MailOfflineHandlerPrivate *priv;

    offline_handler = MAIL_OFFLINE_HANDLER (bonobo_object_from_servant (servant));
    priv = offline_handler->priv;

    /* Enable auto-mail-checking */
    camel_session_set_online (session, TRUE);

    mail_storages_foreach (storage_go_online, NULL);
}

/* GObject methods.  */

static void
impl_finalise (GObject *object)
{
    MailOfflineHandler *offline_handler;
    MailOfflineHandlerPrivate *priv;

    offline_handler = MAIL_OFFLINE_HANDLER (object);
    priv = offline_handler->priv;
    g_hash_table_destroy(priv->sync_table);
    g_free (priv);

    if (G_OBJECT_CLASS (parent_class)->finalize != NULL)
        (* G_OBJECT_CLASS (parent_class)->finalize) (object);
}

/* GTK+ type initialization.  */

static void
mail_offline_handler_class_init (MailOfflineHandlerClass *klass)
{
    GObjectClass *object_class;
    POA_GNOME_Evolution_Offline__epv *epv;

    object_class = G_OBJECT_CLASS (klass);
    object_class->finalize = impl_finalise;

    epv = & klass->epv;
    epv->_get_isOffline    = impl__get_isOffline;
    epv->prepareForOffline = impl_prepareForOffline;
    epv->syncFolder        = impl_syncFolder;
    epv->cancelSyncFolder  = impl_cancelSyncFolder;
    epv->goOffline         = impl_goOffline;
    epv->goOnline          = impl_goOnline;

    parent_class = g_type_class_ref(PARENT_TYPE);
}

static void
mail_offline_handler_init (MailOfflineHandler *offline_handler)
{
    MailOfflineHandlerPrivate *priv;

    priv = g_new (MailOfflineHandlerPrivate, 1);
    priv->sync_table = g_hash_table_new(g_str_hash, g_str_equal);

    offline_handler->priv = priv;
}

MailOfflineHandler *
mail_offline_handler_new (void)
{
    MailOfflineHandler *new;

    new = g_object_new(mail_offline_handler_get_type (), NULL);

    return new;
}

BONOBO_TYPE_FUNC_FULL (MailOfflineHandler, GNOME_Evolution_Offline, PARENT_TYPE, mail_offline_handler);