path: root/plugins/mail-remote/evolution-mail-store.c

/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */
/* 
 * Copyright (C) 2005  Novell, Inc.
 *
 * Authors: Michael Zucchi <notzed@novell.com>
 *
 * 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., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

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

#include <errno.h>
#include <string.h>
#include <bonobo/bonobo-shlib-factory.h>
#include <bonobo/bonobo-control.h>
#include <bonobo/bonobo-i18n.h>
#include <bonobo/bonobo-exception.h>
#include <bonobo/bonobo-arg.h>
#include <bonobo/bonobo-main.h>

#include "evolution-mail-session.h"
#include "evolution-mail-store.h"
#include "evolution-mail-folder.h"

#include "e-corba-utils.h"

#include <camel/camel-store.h>
#include <camel/camel-session.h>
#include <camel/camel-stream-mem.h>
#include <camel/camel-mime-message.h>
#include <camel/camel-folder.h>

#include <libedataserver/e-account.h>

#include "mail/mail-component.h"
#include "mail/mail-send-recv.h"

#define PARENT_TYPE bonobo_object_get_type ()

static BonoboObjectClass *parent_class = NULL;

#define _PRIVATE(o) (g_type_instance_get_private ((GTypeInstance *)o, evolution_mail_store_get_type()))

struct _EvolutionMailStorePrivate {
    CamelStore *store;

    GHashTable *folders;
    /* sorted array of folders by full_name */
    GPtrArray *folders_array;

    guint32 folder_opened;
    guint32 folder_created;
    guint32 folder_deleted;
    guint32 folder_renamed;
    guint32 folder_subscribed;
    guint32 folder_unsubscribed;

    EDList listeners;
};

/* GObject methods */

static void
impl_dispose (GObject *object)
{
    EvolutionMailStore *ems = (EvolutionMailStore *)object;
    struct _EvolutionMailStorePrivate *p = _PRIVATE(object);

    /* FIXME: unref store
       unhook events */

    if (ems->account) {
        g_object_unref(ems->account);
        ems->account = NULL;
    }

    e_mail_listener_free(&p->listeners);

    (* G_OBJECT_CLASS (parent_class)->dispose) (object);
}

static void
impl_finalize (GObject *object)
{
    struct _EvolutionMailStorePrivate *p = _PRIVATE(object);

    g_warning("EvolutionMailStore is finalised!\n");

    if (p->folders) {
        /* FIXME: bonobo unref? */
        g_hash_table_foreach(p->folders, (GHFunc)g_object_unref, NULL);
        g_hash_table_destroy(p->folders);
        g_ptr_array_free(p->folders_array, TRUE);
    }

    (* G_OBJECT_CLASS (parent_class)->finalize) (object);
}

/* Evolution.Mail.Store */

static CORBA_boolean
impl_getProperties(PortableServer_Servant _servant,
           const Evolution_Mail_PropertyNames* names,
           Evolution_Mail_Properties **propsp,
           CORBA_Environment * ev)
{
    EvolutionMailStore *ems = (EvolutionMailStore *)bonobo_object_from_servant(_servant);
    int i;
    Evolution_Mail_Properties *props;
    /*struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);*/
    CORBA_boolean ok = CORBA_TRUE;

    *propsp = props = Evolution_Mail_Properties__alloc();
    props->_length = names->_length;
    props->_maximum = props->_length;
    props->_buffer = Evolution_Mail_Properties_allocbuf(props->_maximum);
    CORBA_sequence_set_release(props, CORBA_TRUE);

    for (i=0;i<names->_length;i++) {
        const CORBA_char *name = names->_buffer[i];
        Evolution_Mail_Property *prop = &props->_buffer[i];

        d(printf("getting property '%s'\n", name));

        if (!strcmp(name, "name")) {
            e_mail_property_set_string(prop, name, evolution_mail_store_get_name(ems));
        } else if (!strcmp(name, "uid")) {
            e_mail_property_set_string(prop, name, evolution_mail_store_get_uid(ems));
        } else {
            e_mail_property_set_null(prop, name);
            ok = CORBA_FALSE;
        }
    }

    return ok;
}

static void
ems_add_folders(EvolutionMailStore *ems, CamelFolderInfo *fi, GPtrArray *added)
{
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);

    while (fi) {
        if (g_hash_table_lookup(p->folders, fi->full_name) == NULL) {
            EvolutionMailFolder *emf = evolution_mail_folder_new(ems, fi->name, fi->full_name);

            g_hash_table_insert(p->folders, emf->full_name, emf);
            g_ptr_array_add(p->folders_array, emf);
            if (added) {
                g_object_ref(emf);
                g_ptr_array_add(added, emf);
            }
        }

        if (fi->child)
            ems_add_folders(ems, fi->child, added);

        fi = fi->next;
    }
}

static void
ems_remove_folders(EvolutionMailStore *ems, CamelFolderInfo *fi, GPtrArray *removed)
{
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);
    EvolutionMailFolder *emf;

    while (fi) {
        emf = g_hash_table_lookup(p->folders, fi->full_name);
        if (emf) {
            g_hash_table_remove(p->folders, fi->full_name);
            g_ptr_array_remove(p->folders_array, emf);
            if (removed)
                g_ptr_array_add(removed, emf);
            else
                g_object_unref(emf);
        } else {
            g_warning("Folder removed I didn't know existed '%s'\n", fi->full_name);
        }

        if (fi->child)
            ems_remove_folders(ems, fi->child, removed);

        fi = fi->next;
    }
}

static int
ems_sort_folders_cmp(const void *ap, const void *bp)
{
    const EvolutionMailFolder *a = ((const EvolutionMailFolder **)ap)[0];
    const EvolutionMailFolder *b = ((const EvolutionMailFolder **)bp)[0];

    return strcmp(a->full_name, b->full_name);
}

static void
ems_sort_folders(struct _EvolutionMailStorePrivate *p)
{
    qsort(p->folders_array->pdata, p->folders_array->len, sizeof(p->folders_array->pdata[0]), ems_sort_folders_cmp);
}

static void
ems_set_changes(Evolution_Mail_StoreChange *change, Evolution_Mail_ChangeType how, GPtrArray *changed)
{
    int i;

    change->type = how;
    change->folders._maximum = changed->len;
    change->folders._length = changed->len;
    change->folders._buffer = Evolution_Mail_FolderInfos_allocbuf(change->folders._maximum);
    CORBA_sequence_set_release(&change->folders, TRUE);

    for (i=0;i<changed->len;i++)
        e_mail_folderinfo_set_folder(&change->folders._buffer[i], changed->pdata[i]);
}

static Evolution_Mail_StoreChanges *
ems_create_changes(EvolutionMailStore *ems, Evolution_Mail_ChangeType how, GPtrArray *changed)
{
    Evolution_Mail_StoreChanges *changes;

    /* NB: we only ever create 1 changetype at the moment */

    changes = Evolution_Mail_StoreChanges__alloc();
    changes->_maximum = 1;
    changes->_length = 1;
    changes->_buffer = Evolution_Mail_StoreChanges_allocbuf(1);
    CORBA_sequence_set_release(changes, TRUE);

    ems_set_changes(&changes->_buffer[0], how, changed);

    return changes;
}

static void
ems_folder_opened(CamelObject *o, void *d, void *data)
{
    EvolutionMailStore *ems = data;
    CamelFolder *folder = d;

    ems = ems;
    folder = folder;
    /* noop */
}

static void
ems_folder_subscribed(CamelObject *o, void *d, void *data)
{
    EvolutionMailStore *ems = data;
    CamelFolderInfo *fi = d;
    GPtrArray *added;
    int i;

    added = g_ptr_array_new();
    ems_add_folders(ems, fi, added);

    if (added) {
        if (added->len) {
            Evolution_Mail_StoreChanges *changes = ems_create_changes(ems, Evolution_Mail_ADDED, added);

            evolution_mail_store_changed(ems, changes);
            CORBA_free(changes);

            for (i=0;i<added->len;i++)
                g_object_unref(added->pdata[i]);
        }
        g_ptr_array_free(added, TRUE);
    }
}

static void
ems_folder_unsubscribed(CamelObject *o, void *d, void *data)
{
    EvolutionMailStore *ems = data;
    CamelFolderInfo *fi = d;
    GPtrArray *removed = NULL;
    int i;

    removed = g_ptr_array_new();
    ems_remove_folders(ems, fi, removed);

    if (removed) {
        if (removed->len) {
            Evolution_Mail_StoreChanges *changes = ems_create_changes(ems, Evolution_Mail_REMOVED, removed);

            evolution_mail_store_changed(ems, changes);
            CORBA_free(changes);

            for (i=0;i<removed->len;i++)
                g_object_unref(removed->pdata[i]);
        }
        g_ptr_array_free(removed, TRUE);
    }
}

static void
ems_folder_created(CamelObject *o, void *d, void *data)
{
    CamelStore *store = (CamelStore *)o;

    if (!camel_store_supports_subscriptions(store))
        ems_folder_subscribed(o, d, data);
}

static void
ems_folder_deleted(CamelObject *o, void *d, void *data)
{
    CamelStore *store = (CamelStore *)o;

    if (!camel_store_supports_subscriptions(store))
        ems_folder_unsubscribed(o, d, data);
}

static void
get_folders(CamelFolderInfo *fi, GPtrArray *folders)
{
    while (fi) {
        g_ptr_array_add(folders, fi);

        if (fi->child)
            get_folders(fi->child, folders);

        fi = fi->next;
    }
}

static int
folder_cmp(const void *ap, const void *bp)
{
    const CamelFolderInfo *a = ((CamelFolderInfo **)ap)[0];
    const CamelFolderInfo *b = ((CamelFolderInfo **)bp)[0];

    return strcmp(a->full_name, b->full_name);
}

static void
ems_folder_renamed(CamelObject *o, void *d, void *data)
{
    EvolutionMailStore *ems = data;
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);
    CamelRenameInfo *reninfo = d;
    int i, oldlen, newlen;
    GPtrArray *renamed = g_ptr_array_new(), *folders = g_ptr_array_new();
    CamelFolderInfo *top;
    GString *name = g_string_new("");

    /* flatten/sort folders to make sure they're in the right order */
    get_folders(reninfo->new, folders);
    qsort(folders->pdata, folders->len, sizeof(folders->pdata[0]), folder_cmp);
    top = folders->pdata[0];

    oldlen = strlen(reninfo->old_base);
    newlen = strlen(top->full_name);

    for (i=0;i<folders->len;i++) {
        CamelFolderInfo *fi = folders->pdata[i];
        EvolutionMailFolder *emf;

        if (strlen(fi->full_name) >= newlen) {
            g_string_printf(name, "%s%s", reninfo->old_base, fi->full_name + newlen);
            if ((emf = g_hash_table_lookup(p->folders, name->str))) {
                /* FIXME: locking / or api to rename */
                g_hash_table_remove(p->folders, emf->full_name);
                g_free(emf->full_name);
                g_free(emf->name);
                emf->full_name = g_strdup(fi->full_name);
                emf->name = g_strdup(fi->name);
                g_hash_table_insert(p->folders, emf->full_name, emf);

                g_object_ref(emf);
                g_ptr_array_add(renamed, emf);
            }
        }
    }

    g_string_free(name, TRUE);
    g_ptr_array_free(folders, TRUE);

    if (renamed) {
        if (renamed->len) {
            Evolution_Mail_StoreChanges *changes = ems_create_changes(ems, Evolution_Mail_CHANGED, renamed);

            evolution_mail_store_changed(ems, changes);
            CORBA_free(changes);

            for (i=0;i<renamed->len;i++)
                g_object_unref(renamed->pdata[i]);
        }
        g_ptr_array_free(renamed, TRUE);
    }
}

static Evolution_Mail_FolderInfos *
impl_getFolders(PortableServer_Servant _servant,
        const CORBA_char * pattern,
        const Evolution_Mail_FolderListener listener,
        CORBA_Environment * ev)
{
    EvolutionMailStore *ems = (EvolutionMailStore *)bonobo_object_from_servant(_servant);
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);
    CamelFolderInfo *fi;
    CamelException ex = { 0 };
    Evolution_Mail_FolderInfos *folders = NULL;
    int i;
    CamelStore *store;

    store = evolution_mail_store_get_store(ems, ev);
    if (store == NULL) {
        return CORBA_OBJECT_NIL;
    }

    if (p->folders == NULL) {
        fi = camel_store_get_folder_info(store, "", CAMEL_STORE_FOLDER_INFO_RECURSIVE|CAMEL_STORE_FOLDER_INFO_FAST, &ex);

        if (fi) {
            p->folders = g_hash_table_new(g_str_hash, g_str_equal);
            p->folders_array = g_ptr_array_new();
            ems_add_folders(ems, fi, NULL);
            camel_store_free_folder_info(store, fi);
            ems_sort_folders(p);
        } else {
            e_mail_exception_xfer_camel(ev, &ex);
            camel_object_unref(store);
            return CORBA_OBJECT_NIL;
        }
    }

    folders = Evolution_Mail_FolderInfos__alloc();
    folders->_length = p->folders_array->len;
    folders->_maximum = folders->_length;
    folders->_buffer = Evolution_Mail_FolderInfos_allocbuf(folders->_maximum);
    CORBA_sequence_set_release(folders, CORBA_TRUE);

    for (i=0;i<p->folders_array->len;i++) {
        EvolutionMailFolder *emf = p->folders_array->pdata[i];

        evolution_mail_folder_addlistener(emf, listener);
        e_mail_folderinfo_set_folder(&folders->_buffer[i], emf);
    }

    camel_object_unref(store);

    return folders;
}

static void
impl_sendMessage(PortableServer_Servant _servant,
         const Evolution_Mail_MessageStream message,
         CORBA_Environment * ev)
{
    EvolutionMailStore *ems = (EvolutionMailStore *)bonobo_object_from_servant(_servant);
    CamelException ex = { 0 };
    CamelMimeMessage *msg;
    CamelInternetAddress *from;
    CamelMessageInfo *info;
    CORBA_Environment wev = { 0 };

    if (ems->account == NULL
        || ems->account->transport == NULL
        || ems->account->transport->url == NULL) {
        e_mail_exception_set(ev, Evolution_Mail_NOT_SUPPORTED, _("Account cannot send e-mail"));
        goto done;
    }

    msg = e_messagestream_to_message(message, ev);
    if (msg == NULL)
        goto done;

    from = camel_internet_address_new();
    camel_internet_address_add(from, ems->account->id->name, ems->account->id->address);
    camel_mime_message_set_from(msg, from);
    camel_object_unref(from);

    camel_medium_set_header((CamelMedium *)msg, "X-Evolution-Account", ems->account->uid);

    if (msg->date == 0)
        camel_mime_message_set_date(msg, CAMEL_MESSAGE_DATE_CURRENT, 0);

    info = camel_message_info_new(NULL);
    camel_message_info_set_flags(info, CAMEL_MESSAGE_SEEN, ~0);

    camel_folder_append_message(mail_component_get_folder(NULL, MAIL_COMPONENT_FOLDER_OUTBOX), msg, info, NULL, &ex);
    camel_message_info_free(info);

    if (camel_exception_is_set(&ex)) {
        e_mail_exception_xfer_camel(ev, &ex);
    } else {
        mail_send();
    }

    camel_object_unref(msg);
done:
    Evolution_Mail_MessageStream_dispose(message, &wev);
    if (wev._major != CORBA_NO_EXCEPTION)
        CORBA_exception_free(&wev);
}

/* Initialization */

static void
evolution_mail_store_class_init (EvolutionMailStoreClass *klass)
{
    POA_Evolution_Mail_Store__epv *epv = &klass->epv;
    GObjectClass *object_class = G_OBJECT_CLASS (klass);

    parent_class = g_type_class_peek_parent (klass);

    epv->getProperties = impl_getProperties;
    epv->getFolders = impl_getFolders;
    epv->sendMessage = impl_sendMessage;

    object_class->dispose = impl_dispose;
    object_class->finalize = impl_finalize;

    g_type_class_add_private(klass, sizeof(struct _EvolutionMailStorePrivate));
}

static void
evolution_mail_store_init(EvolutionMailStore *ems, EvolutionMailStoreClass *klass)
{
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);

    bonobo_object_set_immortal((BonoboObject *)ems, TRUE);
    e_dlist_init(&p->listeners);
}

BONOBO_TYPE_FUNC_FULL (EvolutionMailStore, Evolution_Mail_Store, PARENT_TYPE, evolution_mail_store)

EvolutionMailStore *
evolution_mail_store_new(struct _EvolutionMailSession *s, struct _EAccount *ea)
{
    EvolutionMailStore *ems;
    struct _EvolutionMailStorePrivate *p;
    static PortableServer_POA poa = NULL;

    d(printf("EvolutionMailStore.new(\"%s\")\n", ea?ea->name:"local"));

    if (poa == NULL)
        poa = bonobo_poa_get_threaded (ORBIT_THREAD_HINT_PER_REQUEST, NULL);

    ems = g_object_new (EVOLUTION_MAIL_TYPE_STORE, "poa", poa, NULL);
    p = _PRIVATE(ems);

    if (ea) {
        ems->account = ea;
        g_object_ref(ea);
    }

    ems->session = s;

    return ems;
}

const char *evolution_mail_store_get_name(EvolutionMailStore *ems)
{
    if (ems->account)
        return ems->account->name;
    else
        return ("On This Computer");
}

const char *evolution_mail_store_get_uid(EvolutionMailStore *ems)
{
    if (ems->account)
        return ems->account->uid;
    else
        return "local@local";
}

CamelStore *evolution_mail_store_get_store(EvolutionMailStore *ems, CORBA_Environment *ev)
{
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);

    if (p->store == NULL) {
        if (ems->account == NULL) {
            p->store = mail_component_peek_local_store(NULL);
            camel_object_ref(p->store);
        } else {
            const char *uri;
            CamelException ex = { 0 };

            uri = e_account_get_string(ems->account, E_ACCOUNT_SOURCE_URL);
            if (uri && *uri) {
                p->store = camel_session_get_store(ems->session->session, uri, &ex);
                if (camel_exception_is_set(&ex)) {
                    e_mail_exception_xfer_camel(ev, &ex);
                    return NULL;
                }
            } else {
                e_mail_exception_set(ev, Evolution_Mail_NOT_SUPPORTED, _("No store available"));
                return NULL;
            }
        }

        p->folder_opened = camel_object_hook_event(p->store, "folder_opened", ems_folder_opened, ems);
        p->folder_created = camel_object_hook_event(p->store, "folder_created", ems_folder_created, ems);
        p->folder_deleted = camel_object_hook_event(p->store, "folder_deleted", ems_folder_deleted, ems);
        p->folder_renamed = camel_object_hook_event(p->store, "folder_renamed", ems_folder_renamed, ems);
        p->folder_subscribed = camel_object_hook_event(p->store, "folder_subscribed", ems_folder_subscribed, ems);
        p->folder_unsubscribed = camel_object_hook_event(p->store, "folder_unsubscribed", ems_folder_unsubscribed, ems);
    }

    camel_object_ref(p->store);
    return p->store;
}

int evolution_mail_store_close_store(EvolutionMailStore *ems)
{
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);

    /* FIXME: locking */
    if (p->store) {
        if (!e_dlist_empty(&p->listeners))
            return -1;

        camel_object_remove_event(p->store, p->folder_opened);
        camel_object_remove_event(p->store, p->folder_created);
        camel_object_remove_event(p->store, p->folder_deleted);
        camel_object_remove_event(p->store, p->folder_renamed);
        camel_object_remove_event(p->store, p->folder_subscribed);
        camel_object_remove_event(p->store, p->folder_unsubscribed);
        camel_object_unref(p->store);
        p->store = NULL;
    }

    /* FIXME: need to close of sub-folders too? */

    return 0;
}

void
evolution_mail_store_addlistener(EvolutionMailStore *ems, Evolution_Mail_StoreListener listener)
{
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);

    /* FIXME: locking */
    e_mail_listener_add(&p->listeners, listener);
}

void
evolution_mail_store_changed(EvolutionMailStore *ems, Evolution_Mail_StoreChanges *changes)
{
    struct _EvolutionMailStorePrivate *p = _PRIVATE(ems);

    if (!e_mail_listener_emit(&p->listeners, (EMailListenerChanged)Evolution_Mail_StoreListener_changed,
                  bonobo_object_corba_objref((BonoboObject *)ems), changes)) {
        evolution_mail_store_close_store(ems);
        w(printf("No more listeners for store, could dispose store object now?\n"));
    }
}