path: root/modules/backup-restore/evolution-backup-restore.c

/*
 * evolution-backup-restore.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/>
 *
 */

#include <config.h>
#include <unistd.h>
#include <sys/types.h>
#ifdef HAVE_SYS_WAIT_H
#  include <sys/wait.h>
#endif
#include <stdlib.h>

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

#include <libebackend/libebackend.h>

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

#include <shell/e-shell-utils.h>
#include <shell/e-shell-window.h>

#include <mail/e-mail-config-assistant.h>

#include "e-mail-config-restore-page.h"
#include "e-mail-config-restore-ready-page.h"

#ifdef G_OS_WIN32
#ifdef localtime_r
#undef localtime_r
#endif
/* The localtime() in Microsoft's C library *is* thread-safe */
#define localtime_r(timep, result) \
    (localtime (timep) ? memcpy ( \
    (result), localtime (timep), sizeof (*(result))) : 0)
#endif

typedef EExtension EvolutionBackupRestoreAssistant;
typedef EExtensionClass EvolutionBackupRestoreAssistantClass;

typedef EExtension EvolutionBackupRestoreMenuItems;
typedef EExtensionClass EvolutionBackupRestoreMenuItemsClass;

/* Module Entry Points */
void e_module_load (GTypeModule *type_module);
void e_module_unload (GTypeModule *type_module);

/* Forward Declarations */
GType evolution_backup_restore_assistant_get_type (void);
GType evolution_backup_restore_menu_items_get_type (void);

static const gchar *ui =
"<ui>"
"  <menubar name='main-menu'>"
"    <menu action='file-menu'>"
"      <placeholder name='file-actions'>"
"        <menuitem action='settings-backup'/>"
"        <menuitem action='settings-restore'/>"
"      </placeholder>"
"    </menu>"
"  </menubar>"
"</ui>";

G_DEFINE_DYNAMIC_TYPE (
    EvolutionBackupRestoreAssistant,
    evolution_backup_restore_assistant,
    E_TYPE_EXTENSION)

G_DEFINE_DYNAMIC_TYPE (
    EvolutionBackupRestoreMenuItems,
    evolution_backup_restore_menu_items,
    E_TYPE_EXTENSION)

enum {
    BR_OK    = 1 << 0,
    BR_START = 1 << 1
};

static void
backup (const gchar *filename,
        gboolean restart)
{
    if (restart)
        execl (
            EVOLUTION_TOOLSDIR "/evolution-backup",
            "evolution-backup",
            "--gui",
            "--backup",
            "--restart",
            filename,
            NULL);
    else
        execl (
            EVOLUTION_TOOLSDIR "/evolution-backup",
            "evolution-backup",
            "--gui",
            "--backup",
            filename,
            NULL);
}

static void
restore (const gchar *filename,
         gboolean restart)
{
    if (restart)
        execl (
            EVOLUTION_TOOLSDIR "/evolution-backup",
            "evolution-backup",
            "--gui",
            "--restore",
            "--restart",
            filename,
            NULL);
    else
        execl (
            EVOLUTION_TOOLSDIR "/evolution-backup",
            "evolution-backup",
            "--gui",
            "--restore",
            filename,
            NULL);
}

static guint32
dialog_prompt_user (GtkWindow *parent,
                    const gchar *string,
                    const gchar *tag,
                    ...)
{
    GtkWidget *dialog;
    GtkWidget *check = NULL;
    GtkWidget *container;
    va_list ap;
    gint button;
    guint32 mask = 0;
    EAlert *alert = NULL;

    va_start (ap, tag);
    alert = e_alert_new_valist (tag, ap);
    va_end (ap);

    dialog = e_alert_dialog_new (parent, alert);
    g_object_unref (alert);

    container = e_alert_dialog_get_content_area (E_ALERT_DIALOG (dialog));

    check = gtk_check_button_new_with_mnemonic (string);
    gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (check), TRUE);
    gtk_box_pack_start (GTK_BOX (container), check, FALSE, FALSE, 0);
    gtk_widget_show (check);

    button = gtk_dialog_run (GTK_DIALOG (dialog));

    if (button == GTK_RESPONSE_YES)
        mask |= BR_OK;
    if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (check)))
        mask |= BR_START;

    gtk_widget_destroy (dialog);

    return mask;
}

static void
set_local_only (GtkFileChooser *file_chooser)
{
    /* XXX Has to be a local file, since the backup utility
     *     takes a filename argument, not a URI. */
    gtk_file_chooser_set_local_only (file_chooser, TRUE);
}

static gchar *
suggest_file_name (void)
{
    time_t t;
    struct tm tm;

    t = time (NULL);
    localtime_r (&t, &tm);

    return g_strdup_printf (
        "evolution-backup-%04d%02d%02d.tar.gz",
        tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday);
}

static void
action_settings_backup_cb (GtkAction *action,
                           EShellWindow *shell_window)
{
    GFile *file;
    GFile *parent;
    GFileInfo *file_info;
    const gchar *attribute;
    GError *error = NULL;
    gchar *suggest;

    suggest = suggest_file_name ();

    file = e_shell_run_save_dialog (
        e_shell_window_get_shell (shell_window),
        _("Select name of the Evolution backup file"),
        suggest, "*.tar.gz", (GtkCallback)
        set_local_only, NULL);

    g_free (suggest);

    if (file == NULL)
        return;

    /* Make sure the parent directory can be written to. */

    parent = g_file_get_parent (file);
    attribute = G_FILE_ATTRIBUTE_ACCESS_CAN_WRITE;

    /* XXX The query operation blocks the main loop but we
     *     know it's a local file, so let it slide for now. */
    file_info = g_file_query_info (
        parent, attribute, G_FILE_QUERY_INFO_NONE, NULL, &error);

    g_object_unref (parent);

    if (error != NULL) {
        g_warning ("%s", error->message);
        g_error_free (error);
        return;
    }

    if (g_file_info_get_attribute_boolean (file_info, attribute)) {
        guint32 mask;
        gchar *path;

        mask = dialog_prompt_user (
            GTK_WINDOW (shell_window),
            _("_Restart Evolution after backup"),
            "org.gnome.backup-restore:backup-confirm", NULL);
        if (mask & BR_OK) {
            path = g_file_get_path (file);
            backup (path, (mask & BR_START) ? TRUE: FALSE);
            g_free (path);
        }
    } else {
        e_alert_run_dialog_for_args (
            GTK_WINDOW (shell_window),
            "org.gnome.backup-restore:insufficient-permissions",
            NULL);
    }

    g_object_unref (file_info);
    g_object_unref (file);
}

static void
action_settings_restore_cb (GtkAction *action,
                            EShellWindow *shell_window)
{
    GFile *file;
    gchar *path;

    file = e_shell_run_open_dialog (
        e_shell_window_get_shell (shell_window),
        _("Select name of the Evolution backup file to restore"),
        (GtkCallback) set_local_only, NULL);

    if (file == NULL)
        return;

    path = g_file_get_path (file);

    if (evolution_backup_restore_validate_backup_file (path)) {
        guint32 mask;

        mask = dialog_prompt_user (
            GTK_WINDOW (shell_window),
            _("_Restart Evolution after restore"),
            "org.gnome.backup-restore:restore-confirm", NULL);
        if (mask & BR_OK)
            restore (path, mask & BR_START);
    } else {
        e_alert_run_dialog_for_args (
            GTK_WINDOW (shell_window),
            "org.gnome.backup-restore:invalid-backup", NULL);
    }

    g_object_unref (file);
    g_free (path);
}

static GtkActionEntry entries[] = {

    { "settings-backup",
      NULL,
      N_("_Back up Evolution Data..."),
      NULL,
      N_("Back up Evolution data and settings to an archive file"),
      G_CALLBACK (action_settings_backup_cb) },

    { "settings-restore",
      NULL,
      N_("R_estore Evolution Data..."),
      NULL,
      N_("Restore Evolution data and settings from an archive file"),
      G_CALLBACK (action_settings_restore_cb) }
};

static gboolean
evolution_backup_restore_filename_to_visible (GBinding *binding,
                                              const GValue *source_value,
                                              GValue *target_value,
                                              gpointer unused)
{
    const gchar *filename;
    gboolean visible;

    filename = g_value_get_string (source_value);
    visible = (filename != NULL && *filename != '\0');
    g_value_set_boolean (target_value, visible);

    return TRUE;
}

static void
evolution_backup_restore_prepare_cb (GtkAssistant *assistant,
                                     GtkWidget *page,
                                     EMailConfigRestorePage *restore_page)
{
    const gchar *filename;

    /* If we've landed on the EMailConfigRestoreReadyPage, that
     * means the user has chosen a valid backup file to restore
     * so start the "evolution-backup" tool immediately. */

    filename = e_mail_config_restore_page_get_filename (restore_page);

    if (E_IS_MAIL_CONFIG_RESTORE_READY_PAGE (page))
        restore (filename, TRUE);
}

static void
evolution_backup_restore_assistant_constructed (GObject *object)
{
    EExtension *extension;
    EExtensible *extensible;
    EMailConfigAssistant *assistant;
    const gchar *type_name;

    extension = E_EXTENSION (object);
    extensible = e_extension_get_extensible (extension);

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

    assistant = E_MAIL_CONFIG_ASSISTANT (extensible);

    /* XXX We only want to add the EMailConfigRestorePage to an
     *     EStartupAssistant instance, not a normal EMailConfigAssistant.
     *     But EStartupAssistant is defined in the "startup-wizard" module
     *     and we can't access its GType without knowing its type name, so
     *     just hard-code the type name. */
    type_name = G_OBJECT_TYPE_NAME (assistant);
    if (g_strcmp0 (type_name, "EStartupAssistant") == 0) {
        EMailConfigPage *restore_page;
        EMailConfigPage *ready_page;

        restore_page = e_mail_config_restore_page_new ();
        e_mail_config_assistant_add_page (assistant, restore_page);

        ready_page = e_mail_config_restore_ready_page_new ();
        e_mail_config_assistant_add_page (assistant, ready_page);

        g_object_bind_property_full (
            restore_page, "filename",
            ready_page, "visible",
            G_BINDING_SYNC_CREATE,
            evolution_backup_restore_filename_to_visible,
            NULL,
            NULL, (GDestroyNotify) NULL);

        g_signal_connect (
            assistant, "prepare",
            G_CALLBACK (evolution_backup_restore_prepare_cb),
            restore_page);
    }
}

static void
evolution_backup_restore_assistant_class_init (EExtensionClass *class)
{
    GObjectClass *object_class;

    object_class = G_OBJECT_CLASS (class);
    object_class->constructed = evolution_backup_restore_assistant_constructed;

    class->extensible_type = E_TYPE_MAIL_CONFIG_ASSISTANT;
}

static void
evolution_backup_restore_assistant_class_finalize (EExtensionClass *class)
{
}

static void
evolution_backup_restore_assistant_init (EExtension *extension)
{
}

static void
evolution_backup_restore_menu_items_constructed (GObject *object)
{
    EExtension *extension;
    EExtensible *extensible;
    EShellWindow *shell_window;
    GtkActionGroup *action_group;
    GtkUIManager *ui_manager;
    GError *error = NULL;

    extension = E_EXTENSION (object);
    extensible = e_extension_get_extensible (extension);

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

    shell_window = E_SHELL_WINDOW (extensible);
    action_group = e_shell_window_get_action_group (shell_window, "shell");

    /* Add actions to the "shell" action group. */
    gtk_action_group_add_actions (
        action_group, entries,
        G_N_ELEMENTS (entries), shell_window);

    /* Because we are loading from a hard-coded string, there is
     * no chance of I/O errors.  Failure here implies a malformed
     * UI definition.  Full stop. */
    ui_manager = e_shell_window_get_ui_manager (shell_window);
    gtk_ui_manager_add_ui_from_string (ui_manager, ui, -1, &error);
    if (error != NULL)
        g_error ("%s", error->message);
}

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

    object_class = G_OBJECT_CLASS (class);
    object_class->constructed = evolution_backup_restore_menu_items_constructed;

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

static void
evolution_backup_restore_menu_items_class_finalize (EExtensionClass *class)
{
}

static void
evolution_backup_restore_menu_items_init (EExtension *extension)
{
}

G_MODULE_EXPORT void
e_module_load (GTypeModule *type_module)
{
    evolution_backup_restore_assistant_register_type (type_module);
    evolution_backup_restore_menu_items_register_type (type_module);

    e_mail_config_restore_page_type_register (type_module);
    e_mail_config_restore_ready_page_type_register (type_module);
}

G_MODULE_EXPORT void
e_module_unload (GTypeModule *type_module)
{
}