/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/* pine-importer.c
 * 
 * Authors: 
 *    Iain Holmes  <iain@ximian.com>
 *
 * Copyright 2001 Ximian, Inc. (www.ximian.com)
 *
 * This program is free software; you can redistribute it and/or 
 * modify it under the terms of the GNU General Public License as 
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA
 */

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

#include <stdio.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <dirent.h>

#include <glib.h>

#include <bonobo/bonobo-object.h>
#include <bonobo/bonobo-generic-factory.h>
#include <bonobo/bonobo-control.h>
#include <bonobo/bonobo-context.h>
#include <bonobo/bonobo-main.h>
#include <bonobo/bonobo-exception.h>
#include <bonobo/bonobo-moniker-util.h>
#include <bonobo-conf/bonobo-config-database.h>

#include <importer/evolution-intelligent-importer.h>
#include <importer/evolution-importer-client.h>
#include <importer/GNOME_Evolution_Importer.h>

#include <e-book.h>
#include <e-card-simple.h>

#define PINE_INTELLIGENT_IMPORTER_IID "OAFIID:GNOME_Evolution_Pine_Intelligent_Importer_Factory"
#define MBOX_IMPORTER_IID "OAFIID:GNOME_Evolution_Mail_Mbox_Importer"
#define KEY "pine-mail-imported"

/*#define SUPER_IMPORTER_DEBUG*/
#ifdef SUPER_IMPORTER_DEBUG
#define d(x) x
#else
#define d(x)
#endif

typedef struct {
	GList *dir_list;

	GNOME_Evolution_Importer importer;
	EvolutionImporterListener *listener;

	GtkWidget *mail;
	gboolean do_mail;
	GtkWidget *address;
	gboolean do_address;

	EBook *book;

	Bonobo_ConfigDatabase db;
} PineImporter;

typedef struct {
	char *parent;
	char *foldername;
	char *path;
} PineFolder;

static void import_next (PineImporter *importer);

static void
pine_store_settings (PineImporter *importer)
{
	bonobo_config_set_boolean (importer->db, "/Importer/Pine/mail", 
				   importer->do_mail, NULL);
	bonobo_config_set_boolean (importer->db, "/Importer/Pine/address", 
				   importer->do_address, NULL);
}

static void
pine_restore_settings (PineImporter *importer)
{
	importer->do_mail = bonobo_config_get_boolean_with_default (
		importer->db, "/Importer/Pine/mail", TRUE, NULL);

	importer->do_address = bonobo_config_get_boolean_with_default (
		importer->db, "/Importer/Pine/address", TRUE, NULL);

}

/* Pass in handle so we can get the next line if we need to */
static char *
get_field (char **start,
	   FILE *handle)
{
	char line[4096];
	char *end, *field;
	int length;

	/* if this was the last line, just return */
	if (*start == NULL) {
		return NULL;
	}

	/* if start is just \n then we need the next line */
	if (**start == '\n') {
		d(g_warning ("Need next line"));
		if (fgets (line, 4096, handle) == NULL) {
			g_warning ("Hmmm, no next line");
			return NULL;
		}

		if (line[0] != ' ' || line[1] != ' ' || line[2] != ' ') {
			g_warning ("Next line was not a continuation line\n"
				   "Oppps: %s", line);
			return NULL;
		}

		*start = line + 3;
	}

	if (**start == '\t') {
		/* Blank field */

		*start += 1;
		return NULL;
	}

	end = strchr (*start, '\t');
	if (end == NULL) {
		/* the line was the last comment, so just return the line */
		length = strlen (line);
		line[length - 1] = 0;

		field = g_strdup (*start);
		
		*start = NULL;
	} else {
		/* Null the end */
		*end = 0;

		field = g_strdup (*start);
		*start = end + 1;
	}

	d(g_warning ("Found %s", field));
	return field;
}

/* A very basic address spliter.
   Returns the first email address
   denoted by <address> */
static char *
parse_address (const char *address)
{
	char *addr_dup, *result, *start, *end;

	addr_dup = g_strdup (address);
	start = strchr (addr_dup, '<');
	if (start == NULL) {
		/* Whole line is an address */
		return addr_dup;
	}

	start += 1;
	end = strchr (start, '>');
	if (end == NULL) {
		result = g_strdup (start);
		g_free (addr_dup);

		return result;
	}

	*end = 0;
	result = strdup (start);
	g_free (addr_dup);

	return result;
}

static void
import_addressfile (EBook *book,
		    EBookStatus status,
		    gpointer user_data)
{
	char *addressbook;
	FILE *handle;
	char line[4096];

	PineImporter *importer = user_data;

	addressbook = gnome_util_prepend_user_home (".addressbook");
	handle = fopen (addressbook, "r");
	g_free (addressbook);

	if (handle == NULL) {
		g_warning ("Cannot open .addressbook");
		return;
	}

	while (fgets (line, 4096, handle) != NULL) {
		char *nick, *fullname, *address, *comment, *fcc, *email = NULL;
		char *start;
		gboolean distrib = FALSE;

		start = line;
		nick = get_field (&start, handle);
		d(g_print ("Nick: %s\n", nick));

		fullname = get_field (&start, handle);
		d(g_print ("Fullname: %s\n", fullname));

		address = get_field (&start, handle);
		d(g_print ("Address: %s\n", address));

		if (*address == '(') {
			char nextline[4096];
			/* Fun, it's a distribution list */
			distrib = TRUE;
			
			/* if the last char of address == ) then this is the
			   full list */
			while (fgets (nextline, 4096, handle)) {
				char *bracket;
				if (nextline[0] != ' ' ||
				    nextline[1] != ' ' ||
				    nextline[2] != ' ') {
					/* Not a continuation line */
					start = nextline;
					break;
				}

				bracket = strchr (nextline, ')');
				if (bracket != NULL &&
				    *(bracket + 1) == '\t') {

					*(bracket + 1) = 0;
					d(g_print ("More addresses: %s\n", nextline));
					start = bracket + 2;
					break;
				} else {
					d(g_print ("More addresses: %s\n", nextline));
				}
			}
		} else {
			email = parse_address (address);
		}

		fcc = get_field (&start, handle);
		d(g_print ("FCC: %s\n", fcc));

		comment = get_field (&start, handle);
		d(g_print ("Comment: %s\n", comment));

		if (distrib == FALSE) {
			/* This was not a distribution list...
			   Evolution doesn't handle that yet (070501)
			   Fixme when it does */
			ECard *card = e_card_new ("");
			ECardSimple *simple = e_card_simple_new (card);

			if (fullname != NULL)
				e_card_simple_set (simple, E_CARD_SIMPLE_FIELD_FULL_NAME,
						   fullname);
			else
				e_card_simple_set (simple, E_CARD_SIMPLE_FIELD_FILE_AS,
						   nick);

			e_card_simple_set (simple, E_CARD_SIMPLE_FIELD_EMAIL,
					   email);
			e_card_simple_set (simple, E_CARD_SIMPLE_FIELD_NOTE,
					   comment);
			e_card_simple_set (simple, E_CARD_SIMPLE_FIELD_NICKNAME,
					   nick);
			e_book_add_card (importer->book, simple->card, NULL, NULL);
		}
		
		g_free (nick);
		g_free (email);
		g_free (address);
		g_free (comment);
		g_free (fcc);
		g_free (fullname);
	}
	
	fclose (handle);
	if (importer->do_mail == FALSE) {
		if (importer->db != CORBA_OBJECT_NIL) {
			bonobo_object_release_unref (importer->db, NULL);
		}
		gtk_main_quit ();
	}
}

static void
import_addressbook (PineImporter *importer)
{
	char *path, *uri;

	importer->book = e_book_new ();
	if (importer->book == NULL) {
		g_warning ("Could not create EBook.");
		return;
	}

	path = g_concat_dir_and_file (g_get_home_dir (),
				      "evolution/local/Contacts/addressbook.db");
	uri = g_strdup_printf ("file://%s", path);
	g_free (path);

	if (!e_book_load_uri (importer->book, uri, import_addressfile, importer)) {
		g_warning ("Error calling load_uri");
	}
	g_free (uri);
}

static void
importer_cb (EvolutionImporterListener *listener,
	     EvolutionImporterResult result,
	     gboolean more_items,
	     void *data)
{
	PineImporter *importer = (PineImporter *) data;
	CORBA_Object objref;
	CORBA_Environment ev;

	if (more_items) {
		CORBA_exception_init (&ev);
		objref = bonobo_object_corba_objref (BONOBO_OBJECT (importer->listener));
		GNOME_Evolution_Importer_processItem (importer->importer,
						      objref, &ev);
		if (ev._major != CORBA_NO_EXCEPTION) {
			g_warning ("Exception: %s", CORBA_exception_id (&ev));
			CORBA_exception_free (&ev);
			return;
		}
		CORBA_exception_free (&ev);
		return;
	}

	if (importer->dir_list) {
		import_next (importer);
	} else {
		bonobo_object_release_unref (importer->importer, NULL);
		if (importer->db != CORBA_OBJECT_NIL) {
			bonobo_object_release_unref (importer->db, NULL);
		}
		gtk_main_quit ();
	}
}

static gboolean
pine_import_file (PineImporter *importer,
		  const char *path,
		  const char *folderpath)
{
	CORBA_boolean result;
	CORBA_Environment ev;
	CORBA_Object objref;

	CORBA_exception_init (&ev);
	result = GNOME_Evolution_Importer_loadFile (importer->importer, path,
						    folderpath, &ev);
	if (ev._major != CORBA_NO_EXCEPTION || result == FALSE) {
		g_warning ("Exception here: %s", CORBA_exception_id (&ev));
		CORBA_Object_release (importer->importer, &ev);
		CORBA_exception_free (&ev);
		return FALSE;
	}

	importer->listener = evolution_importer_listener_new (importer_cb,
							      importer);
	objref = bonobo_object_corba_objref (BONOBO_OBJECT (importer->listener));
	GNOME_Evolution_Importer_processItem (importer->importer, objref, &ev);
	if (ev._major != CORBA_NO_EXCEPTION) {
		g_warning ("Exception: %s", CORBA_exception_id (&ev));
		CORBA_exception_free (&ev);
	}
	CORBA_exception_free (&ev);

	return TRUE;
}

static gboolean
pine_can_import (EvolutionIntelligentImporter *ii,
		 void *closure)
{
	PineImporter *importer = closure;
	char *maildir, *addrfile;
	gboolean mail;
	gboolean md_exists, addr_exists;

	/* Already imported */

	mail = bonobo_config_get_boolean_with_default (importer->db,
                "/Importer/Pine/mail-imported", FALSE, NULL);

	if (mail)
		return FALSE;

	importer->do_mail = !mail;

	maildir = gnome_util_prepend_user_home ("mail");
	md_exists = g_file_exists (maildir);
	importer->do_mail = md_exists;
	gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (importer->mail),
				      importer->do_mail);
	
	gtk_widget_set_sensitive (importer->mail, md_exists);
	g_free (maildir);

	addrfile = gnome_util_prepend_user_home (".addressbook");
	addr_exists = g_file_exists (addrfile);
	g_free (addrfile);
	gtk_widget_set_sensitive (importer->address, addr_exists);

	return md_exists || addr_exists;
}

static void
import_next (PineImporter *importer)
{
	PineFolder *data;

	if (importer->dir_list) {
		char *folder;

		data = importer->dir_list->data;

		folder = g_concat_dir_and_file (data->parent, data->foldername);
		pine_import_file (importer, data->path, folder);
		g_free (folder);
		g_free (data->parent);
		g_free (data->path);
		g_free (data->foldername);
		g_free (data);
		importer->dir_list = importer->dir_list->next;
	}

}

/* Pine uses sent-mail and saved-mail whereas Evolution uses Sent and Drafts */
static char *
maybe_replace_name (const char *original_name)
{
	if (strcmp (original_name, "sent-mail") == 0) {
		return g_strdup ("Sent");
	} else if (strcmp (original_name, "saved-messages") == 0) {
		return g_strdup ("Drafts");
	} 

	return g_strdup (original_name);
}

static void
scan_dir (PineImporter *importer,
	  const char *dirname,
	  const char *orig_parent)
{
	DIR *maildir;
	struct stat buf;
	struct dirent *current;
	
	maildir = opendir (dirname);
	if (maildir == NULL) {
		g_warning ("Could not open %s\nopendir returned: %s",
			   dirname, g_strerror (errno));
		return;
	}
	
	current = readdir (maildir);
	while (current) {
		PineFolder *pf;
		char *fullname, *foldername;
		
		/* Ignore . and .. */
		if (current->d_name[0] == '.') {
			if (current->d_name[1] == '\0' ||
			    (current->d_name[1] == '.' && current->d_name[2] == '\0')) {
				current = readdir (maildir);
				continue;
			}
		}

		if (*orig_parent == '/') {
			foldername = maybe_replace_name (current->d_name);
		} else {
			foldername = g_strdup (current->d_name);
		}
			
		fullname = g_concat_dir_and_file (dirname, current->d_name);
		if (stat (fullname, &buf) == -1) {
			g_warning ("Could not stat %s\nstat returned: %s",
				   fullname, g_strerror (errno));
			current = readdir (maildir);
			g_free (fullname);
			continue;
		}
		
		if (S_ISREG (buf.st_mode)) {
			pf = g_new (PineFolder, 1);
			pf->path = g_strdup (fullname);
			pf->parent = g_strdup (orig_parent);
			pf->foldername = g_strdup (foldername);
			importer->dir_list = g_list_append (importer->dir_list, pf);
		} else if (S_ISDIR (buf.st_mode)) {
			char *subdir;

			pf = g_new (PineFolder, 1);
			pf->path = NULL;
			pf->parent = g_strdup (orig_parent);
			pf->foldername = g_strdup (foldername);
			importer->dir_list = g_list_append (importer->dir_list, pf);

			subdir = g_concat_dir_and_file (orig_parent, foldername);
			scan_dir (importer, fullname, subdir);
			g_free (subdir);
		}
		
		g_free (fullname);
		g_free (foldername);
		current = readdir (maildir);
	}
}

static void
pine_create_structure (EvolutionIntelligentImporter *ii,
		       void *closure)
{
	CORBA_Environment ev;
	PineImporter *importer = closure;
	char *maildir;

	bonobo_object_ref (BONOBO_OBJECT (ii));
	pine_store_settings (importer);

	if (importer->do_address == TRUE) {
		bonobo_config_set_boolean (importer->db, 
                "/Importer/Pine/address-imported", TRUE, NULL);

		import_addressbook (importer);
	}

	if (importer->do_mail == TRUE) {
		bonobo_config_set_boolean (importer->db, 
                "/Importer/Pine/mail-imported", TRUE, NULL);

		maildir = gnome_util_prepend_user_home ("mail");
		scan_dir (importer, maildir, "/");
		g_free (maildir);

		/* Import them */
		import_next (importer);
	}

	CORBA_exception_init (&ev);
	Bonobo_ConfigDatabase_sync (importer->db, &ev);
	CORBA_exception_free (&ev);

	if (importer->do_mail == FALSE && importer->do_address == FALSE) {
		/* Destroy it here if we weren't importing mail
		   otherwise the mail importer destroys itself
		   once the mail is imported */

		/* Hmmm, this needs fixed badly */
		bonobo_object_unref (BONOBO_OBJECT (ii));
	}
}

static void
pine_destroy_cb (GtkObject *object,
		 PineImporter *importer)
{
	CORBA_Environment ev;

	pine_store_settings (importer);

	CORBA_exception_init (&ev);
	Bonobo_ConfigDatabase_sync (importer->db, &ev);
	CORBA_exception_free (&ev);

	if (importer->db != CORBA_OBJECT_NIL) {
		bonobo_object_release_unref (importer->db, NULL);
	}
	importer->db = CORBA_OBJECT_NIL;

	if (importer->importer != CORBA_OBJECT_NIL) {
		bonobo_object_release_unref (importer->importer, NULL);
	}
	gtk_main_quit ();
}

/* Fun inity stuff */

/* Fun control stuff */
static void
checkbox_toggle_cb (GtkToggleButton *tb,
		    gboolean *do_item)
{
	*do_item = gtk_toggle_button_get_active (tb);
}

static BonoboControl *
create_checkboxes_control (PineImporter *importer)
{
	GtkWidget *container, *vbox;
	BonoboControl *control;

	container = gtk_frame_new (_("Import"));
	vbox = gtk_vbox_new (FALSE, 2);
	gtk_container_set_border_width (GTK_CONTAINER (container), 2);
	gtk_container_add (GTK_CONTAINER (container), vbox);

	importer->mail = gtk_check_button_new_with_label (_("Mail"));
	gtk_signal_connect (GTK_OBJECT (importer->mail), "toggled",
			    GTK_SIGNAL_FUNC (checkbox_toggle_cb),
			    &importer->do_mail);

	importer->address = gtk_check_button_new_with_label (_("Addressbook"));
	gtk_signal_connect (GTK_OBJECT (importer->address), "toggled",
			    GTK_SIGNAL_FUNC (checkbox_toggle_cb),
			    &importer->do_address);

	gtk_box_pack_start (GTK_BOX (vbox), importer->mail, FALSE, FALSE, 0);
	gtk_box_pack_start (GTK_BOX (vbox), importer->address, FALSE, FALSE, 0);

	gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (importer->mail), 
				      importer->do_mail);
	gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (importer->address),
				      importer->do_address);

	gtk_widget_show_all (container);
	control = bonobo_control_new (container);
	return control;
}

static BonoboObject *
factory_fn (BonoboGenericFactory *_factory,
	    void *closure)
{
	EvolutionIntelligentImporter *importer;
	BonoboControl *control;
	PineImporter *pine;
	CORBA_Environment ev;
	char *message = N_("Evolution has found Pine mail files.\n"
			   "Would you like to import them into Evolution?");

	pine = g_new0 (PineImporter, 1);

	CORBA_exception_init (&ev);

	pine->db = bonobo_get_object ("wombat:", "Bonobo/ConfigDatabase", &ev);

	if (BONOBO_EX (&ev) || pine->db == CORBA_OBJECT_NIL) {
		g_free (pine);
		CORBA_exception_free (&ev);
		return NULL;
 	}

	pine_restore_settings (pine);

	pine->importer = oaf_activate_from_id (MBOX_IMPORTER_IID, 0, NULL, &ev);
	if (ev._major != CORBA_NO_EXCEPTION) {
		g_warning ("Could not start MBox importer\n%s",
			   CORBA_exception_id (&ev));
		CORBA_exception_free (&ev);
		return NULL;
	}
	CORBA_exception_free (&ev);

	importer = evolution_intelligent_importer_new (pine_can_import,
						       pine_create_structure,
						       _("Pine mail"),
						       _(message), pine);
	gtk_signal_connect (GTK_OBJECT (importer), "destroy",
			    GTK_SIGNAL_FUNC (pine_destroy_cb), pine);

	control = create_checkboxes_control (pine);
	bonobo_object_add_interface (BONOBO_OBJECT (importer),
				     BONOBO_OBJECT (control));
	return BONOBO_OBJECT (importer);
}

static void
importer_init (void)
{
	BonoboGenericFactory *factory;

	factory = bonobo_generic_factory_new (PINE_INTELLIGENT_IMPORTER_IID,
					      factory_fn, NULL);
	if (factory == NULL) {
		g_warning ("Could not initialise Pine Intelligent Mail Importer.");
		exit (0);
	}
	
	bonobo_running_context_auto_exit_unref (BONOBO_OBJECT (factory));
}

int
main (int argc,
      char **argv)
{
	CORBA_ORB orb;

	bindtextdomain(PACKAGE, EVOLUTION_LOCALEDIR);
	textdomain(PACKAGE);
	
	gnome_init_with_popt_table ("Evolution-Pine-Importer",
				    VERSION, argc, argv, oaf_popt_options, 0,
				    NULL);
	orb = oaf_init (argc, argv);
	if (bonobo_init (orb, CORBA_OBJECT_NIL, CORBA_OBJECT_NIL) == FALSE) {
		g_error ("Could not initialise Bonobo");
		exit (0);
	}

	importer_init ();
	bonobo_main ();
	
	return 0;
}