path: root/modules/composer-autosave/e-composer-registry.c

/*
 * e-composer-registry.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/>
 *
 */

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

#include <glib/gstdio.h>
#include <shell/e-shell.h>
#include <shell/e-shell-window.h>
#include <e-util/e-extension.h>
#include <e-util/e-alert-dialog.h>
#include <composer/e-msg-composer.h>

#include "e-autosave-utils.h"

/* Standard GObject macros */
#define E_TYPE_COMPOSER_REGISTRY \
    (e_composer_registry_get_type ())
#define E_COMPOSER_REGISTRY(obj) \
    (G_TYPE_CHECK_INSTANCE_CAST \
    ((obj), E_TYPE_COMPOSER_REGISTRY, EComposerRegistry))

typedef struct _EComposerRegistry EComposerRegistry;
typedef struct _EComposerRegistryClass EComposerRegistryClass;

struct _EComposerRegistry {
    EExtension parent;
    GQueue composers;
    gboolean orphans_restored;
};

struct _EComposerRegistryClass {
    EExtensionClass parent_class;
};

/* Forward Declarations */
GType e_composer_registry_get_type (void);
void e_composer_registry_type_register (GTypeModule *type_module);

G_DEFINE_DYNAMIC_TYPE (
    EComposerRegistry,
    e_composer_registry,
    E_TYPE_EXTENSION)

static void
composer_registry_recovered_cb (EShell *shell,
                                GAsyncResult *result,
                                EComposerRegistry *registry)
{
    EMsgComposer *composer;
    GError *error = NULL;

    composer = e_composer_load_snapshot_finish (shell, result, &error);

    if (error != NULL) {
        /* FIXME Show an alert dialog here explaining
         *       why we could not recover the message.
         *       Will need a new error XML entry. */
        g_warn_if_fail (composer == NULL);
        g_warning ("%s", error->message);
        g_error_free (error);
        goto exit;
    }

    gtk_widget_show (GTK_WIDGET (composer));

    g_object_unref (composer);

exit:
    g_object_unref (registry);
}

static gboolean
composer_registry_map_event_cb (GtkWindow *parent,
                                GdkEvent *event,
                                EComposerRegistry *registry)
{
    EExtensible *extensible;
    GList *orphans;
    gint response;
    GError *error = NULL;

    extensible = e_extension_get_extensible (E_EXTENSION (registry));

    /* Look for orphaned auto-save files. */
    orphans = e_composer_find_orphans (
        &registry->composers, &error);
    if (orphans == NULL) {
        if (error != NULL) {
            g_warning ("%s", error->message);
            g_error_free (error);
        }
        goto exit;
    }

    /* Ask if the user wants to recover the orphaned files. */
    response = e_alert_run_dialog_for_args (
        parent, "mail-composer:recover-autosave", NULL);

    /* Based on the user's reponse, recover or delete them. */
    while (orphans != NULL) {
        GFile *file = orphans->data;

        if (response == GTK_RESPONSE_YES)
            e_composer_load_snapshot (
                E_SHELL (extensible),
                file, NULL, (GAsyncReadyCallback)
                composer_registry_recovered_cb,
                g_object_ref (registry));
        else
            g_file_delete (file, NULL, NULL);

        g_object_unref (file);

        orphans = g_list_delete_link (orphans, orphans);
    }

exit:
    registry->orphans_restored = TRUE;

    return FALSE;
}

static void
composer_registry_notify_cb (EComposerRegistry *registry,
                             GObject *where_the_object_was)
{
    /* Remove the finalized composer from the registry. */
    g_queue_remove (&registry->composers, where_the_object_was);

    g_object_unref (registry);
}

static void
composer_registry_window_created_cb (EShell *shell,
                                     GtkWindow *window,
                                     EComposerRegistry *registry)
{
    /* Offer to restore any orphaned auto-save files from the
     * previous session once the first EShellWindow is mapped. */
    if (E_IS_SHELL_WINDOW (window) && !registry->orphans_restored)
        g_signal_connect (
            window, "map-event",
            G_CALLBACK (composer_registry_map_event_cb),
            registry);

    /* Track the new composer window. */
    else if (E_IS_MSG_COMPOSER (window)) {
        g_queue_push_tail (&registry->composers, window);
        g_object_weak_ref (
            G_OBJECT (window), (GWeakNotify)
            composer_registry_notify_cb,
            g_object_ref (registry));
    }
}

static void
composer_registry_finalize (GObject *object)
{
    GObjectClass *parent_class;
    EComposerRegistry *registry;

    registry = E_COMPOSER_REGISTRY (object);

    /* All composers should have been finalized by now. */
    g_warn_if_fail (g_queue_is_empty (&registry->composers));

    /* Chain up to parent's finalize() method. */
    parent_class = G_OBJECT_CLASS (e_composer_registry_parent_class);
    parent_class->finalize (object);
}

static void
composer_registry_constructed (GObject *object)
{
    EExtensible *extensible;
    GObjectClass *parent_class;

    /* Chain up to parent's constructed() method. */
    parent_class = G_OBJECT_CLASS (e_composer_registry_parent_class);
    parent_class->constructed (object);

    extensible = e_extension_get_extensible (E_EXTENSION (object));

    /* Listen for new watched windows. */
    g_signal_connect (
        extensible, "window-created",
        G_CALLBACK (composer_registry_window_created_cb),
        object);
}

static void
e_composer_registry_class_init (EComposerRegistryClass *class)
{
    GObjectClass *object_class;
    EExtensionClass *extension_class;

    object_class = G_OBJECT_CLASS (class);
    object_class->finalize = composer_registry_finalize;
    object_class->constructed = composer_registry_constructed;

    extension_class = E_EXTENSION_CLASS (class);
    extension_class->extensible_type = E_TYPE_SHELL;
}

static void
e_composer_registry_class_finalize (EComposerRegistryClass *class)
{
}

static void
e_composer_registry_init (EComposerRegistry *registry)
{
    g_queue_init (&registry->composers);
}

void
e_composer_registry_type_register (GTypeModule *type_module)
{
    /* XXX G_DEFINE_DYNAMIC_TYPE declares a static type registration
     *     function, so we have to wrap it with a public function in
     *     order to register types from a separate compilation unit. */
    e_composer_registry_register_type (type_module);
}