path: root/addressbook/backend/pas/pas-book-factory.c

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/*
 *
 * Author:
 *   Nat Friedman (nat@ximian.com)
 *
 * Copyright 2000, Ximian, Inc.
 */

#include <config.h>
#include <string.h>

#include "addressbook.h"
#include "pas-book-factory.h"
#include "pas-marshal.h"
#include <bonobo-activation/bonobo-activation.h>
#include <bonobo/bonobo-main.h>

#define DEFAULT_PAS_BOOK_FACTORY_OAF_ID "OAFIID:GNOME_Evolution_Wombat_ServerFactory"

static BonoboObjectClass          *pas_book_factory_parent_class;

typedef struct {
    char                              *uri;
    GNOME_Evolution_Addressbook_BookListener             listener;
} PASBookFactoryQueuedRequest;

struct _PASBookFactoryPrivate {
    gint        idle_id;
    GHashTable *backends;
    GHashTable *active_server_map;
    GList      *queued_requests;

    /* OAFIID of the factory */
    char *iid;

    /* Whether the factory has been registered with OAF yet */
    guint       registered : 1;
};

/* Signal IDs */
enum {
    LAST_BOOK_GONE,
    LAST_SIGNAL
};

static guint factory_signals[LAST_SIGNAL];

static char *
pas_book_factory_canonicalize_uri (const char *uri)
{
    /* FIXME: What do I do here? */

    return g_strdup (uri);
}

static char *
pas_book_factory_extract_proto_from_uri (const char *uri)
{
    char *proto;
    char *p;

    p = strchr (uri, ':');

    if (p == NULL)
        return NULL;

    proto = g_malloc0 (p - uri + 1);

    strncpy (proto, uri, p - uri);

    return proto;
}

/**
 * pas_book_factory_register_backend:
 * @factory:
 * @proto:
 * @backend:
 */
void
pas_book_factory_register_backend (PASBookFactory      *factory,
                   const char          *proto,
                   PASBackendFactoryFn  backend)
{
    g_return_if_fail (factory != NULL);
    g_return_if_fail (PAS_IS_BOOK_FACTORY (factory));
    g_return_if_fail (proto != NULL);
    g_return_if_fail (backend != NULL);

    if (g_hash_table_lookup (factory->priv->backends, proto) != NULL) {
        g_warning ("pas_book_factory_register_backend: "
               "Proto \"%s\" already registered!\n", proto);
    }

    g_hash_table_insert (factory->priv->backends,
                 g_strdup (proto), backend);
}

/**
 * pas_book_factory_get_n_backends:
 * @factory: An addressbook factory.
 * 
 * Queries the number of running addressbook backends in an addressbook factory.
 * 
 * Return value: Number of running backends.
 **/
int
pas_book_factory_get_n_backends (PASBookFactory *factory)
{
    g_return_val_if_fail (factory != NULL, -1);
    g_return_val_if_fail (PAS_IS_BOOK_FACTORY (factory), -1);

    return g_hash_table_size (factory->priv->active_server_map);
}

static void
dump_active_server_map_entry (gpointer key, gpointer value, gpointer data)
{
    char *uri;
    PASBackend *backend;

    uri = key;
    backend = PAS_BACKEND (value);

    g_message ("  %s: %p", uri, backend);
}

void
pas_book_factory_dump_active_backends (PASBookFactory *factory)
{
    g_message ("Active PAS backends");

    g_hash_table_foreach (factory->priv->active_server_map,
                  dump_active_server_map_entry,
                  NULL);

}

/* Callback used when a backend loses its last connected client */
static void
backend_last_client_gone_cb (PASBackend *backend, gpointer data)
{
    PASBookFactory *factory;
    const char *uri;

    factory = PAS_BOOK_FACTORY (data);

    /* Remove the backend from the active server map */

    uri = pas_backend_get_uri (backend);
    if (uri) {
        gpointer orig_key;
        gboolean result;
        char *orig_uri;

        result = g_hash_table_lookup_extended (factory->priv->active_server_map, uri,
                               &orig_key, NULL);
        g_assert (result != FALSE);

        orig_uri = orig_key;

        g_hash_table_remove (factory->priv->active_server_map, orig_uri);
        g_free (orig_uri);

        g_object_unref (backend);

        /* Notify upstream if there are no more backends */
    }

    if (g_hash_table_size (factory->priv->active_server_map) == 0)
        g_signal_emit (G_OBJECT (factory), factory_signals[LAST_BOOK_GONE], 0);
}

static PASBackendFactoryFn
pas_book_factory_lookup_backend_factory (PASBookFactory *factory,
                     const char     *uri)
{
    PASBackendFactoryFn  backend_fn;
    char                *proto;
    char                *canonical_uri;

    g_assert (factory != NULL);
    g_assert (PAS_IS_BOOK_FACTORY (factory));
    g_assert (uri != NULL);

    canonical_uri = pas_book_factory_canonicalize_uri (uri);
    if (canonical_uri == NULL)
        return NULL;

    proto = pas_book_factory_extract_proto_from_uri (canonical_uri);
    if (proto == NULL) {
        g_free (canonical_uri);
        return NULL;
    }

    backend_fn = g_hash_table_lookup (factory->priv->backends, proto);

    g_free (proto); 
    g_free (canonical_uri);

    return backend_fn;
}

static PASBackend *
pas_book_factory_launch_backend (PASBookFactory              *factory,
                 GNOME_Evolution_Addressbook_BookListener       listener,
                 const char                  *uri)
{
    PASBackendFactoryFn  backend_factory;
    PASBackend          *backend;

    backend_factory = pas_book_factory_lookup_backend_factory (
        factory, uri);

    if (!backend_factory) {
        CORBA_Environment ev;

        CORBA_exception_init (&ev);
        GNOME_Evolution_Addressbook_BookListener_notifyBookOpened (
            listener,
            GNOME_Evolution_Addressbook_BookListener_ProtocolNotSupported,
            CORBA_OBJECT_NIL,
            &ev);

        if (ev._major != CORBA_NO_EXCEPTION)
            g_message ("pas_book_factory_launch_backend(): could not notify "
                   "the listener");

        CORBA_exception_free (&ev);
        return NULL;
    }

    backend = (* backend_factory) ();
    if (!backend) {
        CORBA_Environment ev;

        CORBA_exception_init (&ev);
        GNOME_Evolution_Addressbook_BookListener_notifyBookOpened (
            listener,
            GNOME_Evolution_Addressbook_BookListener_OtherError,
            CORBA_OBJECT_NIL,
            &ev);

        if (ev._major != CORBA_NO_EXCEPTION)
            g_message ("pas_book_factory_launch_backend(): could not notify "
                   "the listener");

        CORBA_exception_free (&ev);
        return NULL;
    }

    g_hash_table_insert (factory->priv->active_server_map,
                 g_strdup (uri),
                 backend);

    g_signal_connect (backend, "last_client_gone",
              G_CALLBACK (backend_last_client_gone_cb),
              factory);

    return backend;
}

static void
pas_book_factory_process_request (PASBookFactory              *factory,
                  PASBookFactoryQueuedRequest *request)
{
    PASBackend *backend;
    char *uri;
    GNOME_Evolution_Addressbook_BookListener listener;
    CORBA_Environment ev;

    uri = request->uri;
    listener = request->listener;
    g_free (request);

    /* Look up the backend and create one if needed */

    backend = g_hash_table_lookup (factory->priv->active_server_map, uri);

    if (!backend) {
        GNOME_Evolution_Addressbook_BookListener_CallStatus status;

        backend = pas_book_factory_launch_backend (factory, listener, uri);
        if (!backend)
            goto out;

        status = pas_backend_load_uri (backend, uri);
        if (status != GNOME_Evolution_Addressbook_BookListener_Success) {
            /* tell the listener that we failed to open the book */
            CORBA_exception_init (&ev);

            GNOME_Evolution_Addressbook_BookListener_notifyBookOpened (
                       listener, status,
                       CORBA_OBJECT_NIL,
                       &ev);

            if (ev._major != CORBA_NO_EXCEPTION) {
                g_warning ("pas_book_respond_open: Exception "
                       "responding to BookListener!\n");
            }

            CORBA_exception_free (&ev);
            
            backend_last_client_gone_cb (backend, factory);

            goto out;
        }

        pas_backend_add_client (backend, listener);

        goto out;
    }

    pas_backend_add_client (backend, listener);

 out:
    g_free (uri);

    CORBA_exception_init (&ev);
    CORBA_Object_release (listener, &ev);

    if (ev._major != CORBA_NO_EXCEPTION)
        g_message ("pas_book_factory_process_request(): could not release the listener");

    CORBA_exception_free (&ev);
}

static gboolean
pas_book_factory_process_queue (PASBookFactory *factory)
{
    /* Process pending Book-creation requests. */
    if (factory->priv->queued_requests != NULL) {
        PASBookFactoryQueuedRequest  *request;
        GList *l;

        l = factory->priv->queued_requests;
        request = l->data;

        pas_book_factory_process_request (factory, request);

        factory->priv->queued_requests = g_list_remove_link (
            factory->priv->queued_requests, l);
        g_list_free_1 (l);
    }

    if (factory->priv->queued_requests == NULL) {

        factory->priv->idle_id = 0;
        return FALSE;
    }

    return TRUE;
}

static void
pas_book_factory_queue_request (PASBookFactory               *factory,
                const char                   *uri,
                const GNOME_Evolution_Addressbook_BookListener  listener)
{
    PASBookFactoryQueuedRequest *request;
    GNOME_Evolution_Addressbook_BookListener       listener_copy;
    CORBA_Environment            ev;

    CORBA_exception_init (&ev);

    listener_copy = CORBA_Object_duplicate (listener, &ev);

    if (ev._major != CORBA_NO_EXCEPTION) {
        g_warning ("PASBookFactory: Could not duplicate BookListener!\n");
        CORBA_exception_free (&ev);
        return;
    }

    CORBA_exception_free (&ev);

    request           = g_new0 (PASBookFactoryQueuedRequest, 1);
    request->listener = listener_copy;
    request->uri      = g_strdup (uri);

    factory->priv->queued_requests =
        g_list_prepend (factory->priv->queued_requests, request);

    if (! factory->priv->idle_id) {
        factory->priv->idle_id =
            g_idle_add ((GSourceFunc) pas_book_factory_process_queue, factory);
    }
}

static void
impl_GNOME_Evolution_Addressbook_BookFactory_openBook (PortableServer_Servant        servant,
                      const CORBA_char             *uri,
                      const GNOME_Evolution_Addressbook_BookListener  listener,
                      CORBA_Environment            *ev)
{
    PASBookFactory      *factory = PAS_BOOK_FACTORY (bonobo_object (servant));
    PASBackendFactoryFn  backend_factory;

    backend_factory = pas_book_factory_lookup_backend_factory (factory, uri);

    if (backend_factory == NULL) {
        GNOME_Evolution_Addressbook_BookListener_notifyBookOpened (
            listener,
            GNOME_Evolution_Addressbook_BookListener_ProtocolNotSupported,
            CORBA_OBJECT_NIL,
            ev);

        return;
    }

    pas_book_factory_queue_request (factory, uri, listener);
}

static void
pas_book_factory_construct (PASBookFactory *factory)
{
    /* nothing to do here.. */
}

/**
 * pas_book_factory_new:
 */
PASBookFactory *
pas_book_factory_new (void)
{
    PASBookFactory *factory;

    factory = g_object_new (PAS_TYPE_BOOK_FACTORY, NULL);
    
    pas_book_factory_construct (factory);

    return factory;
}

/**
 * pas_book_factory_activate:
 */
gboolean
pas_book_factory_activate (PASBookFactory *factory, const char *iid)
{
    PASBookFactoryPrivate *priv;
    CORBA_Object obj;
    Bonobo_RegistrationResult result;
    char *tmp_iid;

    g_return_val_if_fail (factory != NULL, FALSE);
    g_return_val_if_fail (PAS_IS_BOOK_FACTORY (factory), FALSE);

    priv = factory->priv;

    g_return_val_if_fail (!priv->registered, FALSE);

    /* if iid is NULL, use the default factory OAFIID */
    if (iid)
        tmp_iid = g_strdup (iid);
    else
        tmp_iid = g_strdup (DEFAULT_PAS_BOOK_FACTORY_OAF_ID);

    result = bonobo_activation_active_server_register (tmp_iid, bonobo_object_corba_objref (BONOBO_OBJECT (factory)));

    switch (result) {
    case Bonobo_ACTIVATION_REG_SUCCESS:
        priv->registered = TRUE;
        priv->iid = tmp_iid;
        return TRUE;
    case Bonobo_ACTIVATION_REG_NOT_LISTED:
        g_message ("Error registering the PAS factory: not listed");
        break;
    case Bonobo_ACTIVATION_REG_ALREADY_ACTIVE:
        g_message ("Error registering the PAS factory: already active");
        break;
    case Bonobo_ACTIVATION_REG_ERROR:
    default:
        g_message ("Error registering the PAS factory: generic error");
        break;
    }

    g_free (tmp_iid);
    return FALSE;
}

static void
pas_book_factory_init (PASBookFactory *factory)
{
    factory->priv = g_new0 (PASBookFactoryPrivate, 1);

    factory->priv->active_server_map = g_hash_table_new (g_str_hash, g_str_equal);
    factory->priv->backends          = g_hash_table_new (g_str_hash, g_str_equal);
    factory->priv->queued_requests   = NULL;
    factory->priv->registered        = FALSE;
}

static void
free_active_server_map_entry (gpointer key, gpointer value, gpointer data)
{
    char *uri;
    PASBackend *backend;

    uri = key;
    g_free (uri);

    backend = PAS_BACKEND (value);
    g_object_unref (backend);
}

static void
remove_backends_entry (gpointer key, gpointer value, gpointer data)
{
    char *uri;

    uri = key;
    g_free (uri);
}

static void
pas_book_factory_dispose (GObject *object)
{
    PASBookFactory *factory = PAS_BOOK_FACTORY (object);

    if (factory->priv) {
        PASBookFactoryPrivate *priv = factory->priv;
        GList          *l;

        for (l = priv->queued_requests; l != NULL; l = l->next) {
            PASBookFactoryQueuedRequest *request = l->data;
            CORBA_Environment ev;

            g_free (request->uri);

            CORBA_exception_init (&ev);
            CORBA_Object_release (request->listener, &ev);
            CORBA_exception_free (&ev);

            g_free (request);
        }
        g_list_free (priv->queued_requests);
        priv->queued_requests = NULL;

        g_hash_table_foreach (priv->active_server_map,
                      free_active_server_map_entry,
                      NULL);
        g_hash_table_destroy (priv->active_server_map);
        priv->active_server_map = NULL;

        g_hash_table_foreach (priv->backends,
                      remove_backends_entry,
                      NULL);
        g_hash_table_destroy (priv->backends);
        priv->backends = NULL;

        if (priv->registered) {
            bonobo_activation_active_server_unregister (priv->iid,
                                bonobo_object_corba_objref (BONOBO_OBJECT (factory)));
            priv->registered = FALSE;
        }

        g_free (priv->iid);
    
        g_free (priv);
        factory->priv = NULL;
    }

    if (G_OBJECT_CLASS (pas_book_factory_parent_class)->dispose)
        G_OBJECT_CLASS (pas_book_factory_parent_class)->dispose (object);
}

static void
pas_book_factory_class_init (PASBookFactoryClass *klass)
{
    GObjectClass *object_class = G_OBJECT_CLASS (klass);
    POA_GNOME_Evolution_Addressbook_BookFactory__epv *epv;

    pas_book_factory_parent_class = g_type_class_peek_parent (klass);

    object_class->dispose = pas_book_factory_dispose;

    factory_signals[LAST_BOOK_GONE] =
        g_signal_new ("last_book_gone",
                  G_OBJECT_CLASS_TYPE (object_class),
                  G_SIGNAL_RUN_FIRST,
                  G_STRUCT_OFFSET (PASBookFactoryClass, last_book_gone),
                  NULL, NULL,
                  pas_marshal_NONE__NONE,
                  G_TYPE_NONE, 0);


    epv = &klass->epv;

    epv->openBook = impl_GNOME_Evolution_Addressbook_BookFactory_openBook;
}

BONOBO_TYPE_FUNC_FULL (
               PASBookFactory,
               GNOME_Evolution_Addressbook_BookFactory,
               BONOBO_TYPE_OBJECT,
               pas_book_factory);