evolution/plugins/backup-restore/backup.c

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

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/wait.h>

#include <glib/gi18n.h>
#include <gtk/gtk.h>
#include <libgnome/gnome-util.h>

#define EVOLUTION "evolution-" BASE_VERSION
#define EVOLUTION_DIR "~/.evolution/"
#define EVOLUTION_DIR_BACKUP "~/.evolution-old/"
#define GCONF_DUMP_FILE "backup-restore-gconf.xml"
#define GCONF_DUMP_PATH EVOLUTION_DIR GCONF_DUMP_FILE
#define GCONF_DIR "/apps/evolution"
#define ARCHIVE_NAME "evolution-backup.tar.gz"

static gboolean backup_op = FALSE;
static char *bk_file;
static gboolean restore_op = FALSE;
static char *res_file;
static gboolean check_op = FALSE;
static char *chk_file;
static gboolean restart_arg = FALSE;
static gboolean gui_arg = FALSE;
static int result=0;
static GtkWidget *progress_dialog;
static GtkWidget *pbar;
static char *txt = NULL;
gboolean complete = FALSE;

#define d(x) x

/* #define s(x) system (x) */
#define s(x) G_STMT_START { g_message (x); system (x); } G_STMT_END

#define CANCEL(x) if (x) return;

static void
backup (const char *filename) 
{
	char *command;

	CANCEL (complete);
	txt = _("Shutting down Evolution");
	/* FIXME Will the versioned setting always work? */
	s (EVOLUTION " --force-shutdown");

	CANCEL (complete);
	txt = _("Backing Evolution accounts and settings");
	s ("gconftool-2 --dump " GCONF_DIR " > " GCONF_DUMP_PATH);


	CANCEL (complete);
	txt = _("Backing Evolution data (Mails, Contacts, Calendar, Tasks, Memos)");

	/* FIXME stay on this file system ,other options?" */
	/* FIXME compression type?" */
	/* FIXME date/time stamp?" */
	/* FIXME archive location?" */
	command = g_strdup_printf ("cd ~ && tar zpcf %s .evolution .camel_certs", filename);
	s (command);
	g_free (command);

	txt = _("Backup complete");

	if (restart_arg) {

		CANCEL (complete);
		txt = _("Restarting Evolution");
		complete=TRUE;
		
		s (EVOLUTION);
	}

}

static void
restore (const char *filename) 
{
	char *command;
	
	/* FIXME Will the versioned setting always work? */
	CANCEL (complete);
	txt = _("Shutting down Evolution");
	s (EVOLUTION " --force-shutdown");

	CANCEL (complete);
	txt = _("Backup current Evolution data");
	s ("mv " EVOLUTION_DIR " " EVOLUTION_DIR_BACKUP);
	s ("mv ~/.camel_certs ~/.camel_certs_old");

	CANCEL (complete);
	txt = _("Extracting files from the archive");
	command = g_strdup_printf ("cd ~ && tar zxf %s", filename);
	s (command);
	g_free (command);

	CANCEL (complete);
	txt = _("Loading Evolution settings");
	s ("gconftool-2 --load " GCONF_DUMP_PATH);

	CANCEL (complete);
	txt = _("Removing temporary backup files");
	s ("rm -rf " GCONF_DUMP_PATH);
	s ("rm -rf " EVOLUTION_DIR_BACKUP);
	s ("rm -rf ~/.camel_certs_old");
	
	if (restart_arg) {
		CANCEL (complete);
		txt = _("Restarting Evolution");
		complete=TRUE;
		s (EVOLUTION);
	}

}

static void
check (const char *filename) 
{
	char *command;

	command = g_strdup_printf ("tar ztf %s | grep -e \"^\\.evolution/$\"", filename);
	result = system (command);
	g_free (command);
	
	g_message ("First result %d", result);
	if (result)
		exit (result);

	command = g_strdup_printf ("tar ztf %s | grep -e \"^\\.evolution/%s$\"", filename, GCONF_DUMP_FILE);
	result = system (command);
	g_free (command);

	g_message ("Second result %d", result);

}

static gboolean 
pbar_update()
{
	if (!complete) {
		 gtk_progress_bar_pulse ((GtkProgressBar *)pbar);
		 gtk_progress_bar_set_text ((GtkProgressBar *)pbar, txt);
		return TRUE;
	}

	gtk_main_quit ();
	return FALSE;
}

static gpointer 
thread_start (gpointer data)
{
	if (backup_op)
		backup (bk_file);
	else if (restore_op)
		restore (res_file);
	else if (check_op)
		check (chk_file);

	complete = TRUE;
	
	return GINT_TO_POINTER(result);
}

static gboolean 
idle_cb(gpointer data)
{
	GThread *t;

	if (gui_arg) {
		/* Show progress dialog */
		 gtk_progress_bar_pulse ((GtkProgressBar *)pbar);
		g_timeout_add (50, pbar_update, NULL);
	}

	t = g_thread_create (thread_start, NULL, FALSE, NULL);

	return FALSE;
}

static void
dlg_response (GtkWidget *dlg, gint response, gpointer data)
{
	/* We will cancel only backup/restore operations and not the check operation */
	complete = TRUE;
	
	/* If the response is not of delete_event then destroy the event */
	if (response != GTK_RESPONSE_NONE)
		gtk_widget_destroy (dlg);
	
	/* We will kill just the tar operation. Rest of the them will be just a second of microseconds.*/
	s ("pkill tar");

	gtk_main_quit ();
}
int
main (int argc, char **argv)
{
	GValue popt_context_value = { 0, };
	GnomeProgram *program;
	poptContext popt_context;
	const char **args;
	char *file = NULL, *oper = NULL;

	struct poptOption options[] = {
		{ "backup", '\0', POPT_ARG_NONE, &backup_op, 0, 
		  N_("Backup Evolution directory"), NULL },
		{ "restore", '\0', POPT_ARG_NONE, &restore_op, 0, 
		  N_("Restore Evolution directory"), NULL },
		{ "check", '\0', POPT_ARG_NONE, &check_op, 0, 
		  N_("Check Evolution archive"), NULL },
		{ "restart", '\0', POPT_ARG_NONE, &restart_arg, 0, 
		  N_("Restart Evolution"), NULL },
		{ "gui", '\0', POPT_ARG_NONE, &gui_arg, 0, 
		  N_("With GUI"), NULL },
		{ NULL, '\0', 0, NULL, 0, NULL, NULL }
	};

	gtk_init (&argc, &argv);
	bindtextdomain (GETTEXT_PACKAGE, EVOLUTION_LOCALEDIR);
	bind_textdomain_codeset (GETTEXT_PACKAGE, "UTF-8");
	textdomain (GETTEXT_PACKAGE);

	program = gnome_program_init (PACKAGE, VERSION, LIBGNOME_MODULE, argc, argv, 
				      GNOME_PROGRAM_STANDARD_PROPERTIES,
				      GNOME_PARAM_POPT_TABLE, options,
				      NULL);

	g_value_init (&popt_context_value, G_TYPE_POINTER);
	g_object_get_property (G_OBJECT (program), GNOME_PARAM_POPT_CONTEXT, &popt_context_value);
	popt_context = g_value_get_pointer (&popt_context_value);
	args = poptGetArgs (popt_context);


	if (args != NULL) {
		const char **p;
		
		for (p = args; *p != NULL; p++) {
			if (backup_op) {
				oper = _("Backing up to %s");
				d(g_message ("Backing up to %s", (char *) *p));
				bk_file = g_strdup ((char *) *p);
				file = bk_file;
			} else if (restore_op) {
				oper = _("Restoring from %s");
				d(g_message ("Restoring from %s", (char *) *p));
				res_file = g_strdup ((char *) *p);
				file = res_file;
			} else if (check_op) {
				d(g_message ("Checking %s", (char *) *p));
				chk_file = g_strdup ((char *) *p);
			}
		}
	}

	if (gui_arg) {
		GtkWidget *vbox, *hbox, *label;
		char *str=NULL;

		/* Backup / Restore only can have GUI. We should restrict the rest */
	 	progress_dialog = gtk_dialog_new_with_buttons (backup_op ? _("Evolution Backup"): _("Evolution Restore"),
        	                                          NULL,
                	                                  GTK_DIALOG_MODAL,
                        	                          GTK_STOCK_CANCEL,
                                	                  GTK_RESPONSE_REJECT,
                                        	          NULL);
		if (oper && file) 
			str = g_strdup_printf(oper, file);
	
		vbox = gtk_vbox_new (FALSE, 6);
		if (str) {
			hbox = gtk_hbox_new (FALSE, 12);
			label = gtk_label_new (str);
			g_free (str);
			gtk_box_pack_start ((GtkBox *)hbox, label, FALSE, FALSE, 6);
			gtk_box_pack_start ((GtkBox *)vbox, hbox, FALSE, FALSE, 6);
		}
		hbox = gtk_hbox_new (FALSE, 12);
		pbar = gtk_progress_bar_new ();
		gtk_box_pack_start ((GtkBox *)hbox, pbar, TRUE, TRUE, 6);



		gtk_box_pack_start ((GtkBox *)vbox, hbox, FALSE, FALSE, 0);
		
		gtk_container_add (GTK_CONTAINER (GTK_DIALOG(progress_dialog)->vbox), vbox);
		gtk_window_set_default_size ((GtkWindow *) progress_dialog,450, 120);
		g_signal_connect (progress_dialog, "response", G_CALLBACK(dlg_response), NULL);
		gtk_widget_show_all (progress_dialog);
	} else if (check_op) {
		 /* For sanity we don't need gui */
		 check (chk_file);
		 exit (result);
	}

	g_idle_add (idle_cb, NULL);
	gtk_main ();

	g_value_unset (&popt_context_value);
	
	return result;
}