1 files changed, 452 insertions, 28 deletions

diff --git a/mail/e-mail-shell-module.c b/mail/e-mail-shell-module.c
index cd74a76cd9..7ee718a9de 100644
--- a/mail/e-mail-shell-module.c
+++ b/mail/e-mail-shell-module.c
@@ -20,23 +20,365 @@
  */
 
 #include <glib/gi18n.h>
+#include <camel/camel-session.h>
+#include <camel/camel-url.h>
 
+#include "e-util/e-import.h"
+#include "e-util/e-util.h"
 #include "shell/e-shell.h"
-#include "shell/e-shell-module.h"
 #include "shell/e-shell-window.h"
 
 #include "e-mail-shell-view.h"
 #include "e-mail-shell-module.h"
 #include "e-mail-shell-module-migrate.h"
 
+#include "em-config.h"
+#include "em-event.h"
+#include "em-folder-tree-model.h"
+#include "em-format-hook.h"
+#include "em-format-html-display.h"
+#include "em-junk-hook.h"
+#include "mail-config.h"
+#include "mail-folder-cache.h"
+#include "mail-mt.h"
+#include "mail-session.h"
+#include "importers/mail-importer.h"
+
 #define MODULE_NAME		"mail"
 #define MODULE_ALIASES		""
 #define MODULE_SCHEMES		"mailto:email"
 #define MODULE_SORT_ORDER	200
 
+typedef struct _StoreInfo StoreInfo;
+
+/* XXX Temporary */
+CamelStore *vfolder_store;
+
+struct _StoreInfo {
+	CamelStore *store;
+	gint ref_count;
+	gchar *name;
+
+	/* Keep a reference to these so they remain around for the session. */
+	CamelFolder *vtrash;
+	CamelFolder *vjunk;
+
+	/* Initialization callback. */
+	void (*done) (CamelStore *store,
+	              CamelFolderInfo *info,
+	              gpointer user_data);
+	gpointer done_user_data;
+
+	guint removed : 1;
+};
+
 /* Module Entry Point */
 void e_shell_module_init (GTypeModule *type_module);
 
+/* The array elements correspond to EMailFolderType. */
+static struct {
+	gchar *name;
+	gchar *uri;
+	CamelFolder *folder;
+} default_local_folders[] = {
+	{ N_("Inbox") },
+	{ N_("Drafts") },
+	{ N_("Outbox") },
+	{ N_("Sent") },
+	{ N_("Templates") },
+	{ "Inbox" }  /* "always local" inbox */
+};
+
+/* XXX So many things need the shell module that it's
+ *     just easier to make it globally available. */ 
+EShellModule *mail_shell_module = NULL;
+
+static GHashTable *store_hash;
+static MailAsyncEvent *async_event;
+static EMFolderTreeModel *folder_tree_model;
+static CamelStore *local_store;
+
+G_LOCK_DEFINE_STATIC (local_store);
+
+static StoreInfo *
+store_info_new (CamelStore *store,
+                const gchar *name)
+{
+	CamelService *service;
+	StoreInfo *si;
+
+	g_return_val_if_fail (CAMEL_IS_STORE (store), NULL);
+
+	service = CAMEL_SERVICE (store);
+
+	si = g_slice_new0 (StoreInfo);
+	si->ref_count = 1;
+
+	if (name == NULL)
+		si->name = camel_service_get_name (service, TRUE);
+	else
+		si->name = g_strdup (name);
+
+	si->store = store;
+	camel_object_ref (store);
+
+	/* If these are vfolders then they need to be opened now,
+	 * otherwise they won't keep track of all folders. */
+	if (store->flags & CAMEL_STORE_VTRASH)
+		si->vtrash = camel_store_get_trash (store, NULL);
+	if (store->flags & CAMEL_STORE_VJUNK)
+		si->vjunk = camel_store_get_junk (store, NULL);
+
+	return si;
+}
+
+static StoreInfo *
+store_info_ref (StoreInfo *si)
+{
+	g_return_val_if_fail (si != NULL, si);
+	g_return_val_if_fail (si->ref_count > 0, si);
+
+	g_atomic_int_add (&si->ref_count, 1);
+
+	return si;
+}
+
+static void
+store_info_unref (StoreInfo *si)
+{
+	g_return_if_fail (si != NULL);
+	g_return_if_fail (si->ref_count > 0);
+
+	if (g_atomic_int_exchange_and_add (&si->ref_count, -1) > 1)
+		return;
+
+	if (si->vtrash != NULL)
+		camel_object_unref (si->vtrash);
+	if (si->vjunk != NULL)
+		camel_object_unref (si->vjunk);
+	camel_object_unref (si->store);
+	g_free (si->name);
+
+	g_slice_free (StoreInfo, si);
+}
+
+static void
+store_hash_free (StoreInfo *si)
+{
+	si->removed = 1;
+	store_info_unref (si);
+}
+
+static gboolean
+mail_shell_module_add_store_done (CamelStore *store,
+                                  CamelFolderInfo *info,
+                                  gpointer user_data)
+{
+	StoreInfo *si = user_data;
+
+	if (si->done != NULL)
+		si->done (store, info, si);
+
+	if (!si->removed) {
+		/* Let the counters know about the already-opened
+		 * junk and trash folders. */
+		if (si->vtrash != NULL)
+			mail_note_folder (si->vtrash);
+		if (si->vjunk != NULL)
+			mail_note_folder (si->vjunk);
+	}
+
+	store_info_unref (si);
+
+	return TRUE;
+}
+
+static void
+mail_shell_module_add_store (EShellModule *shell_module,
+                             CamelStore *store,
+                             const gchar *name,
+                             void (*done) (CamelStore *store,
+                                           CamelFolderInfo *info,
+                                           gpointer user_data))
+{
+	StoreInfo *si;
+
+	si = store_info_new (store, name);
+	si->done = done;
+	g_hash_table_insert (store_hash, store, si);
+
+	em_folder_tree_model_add_store (folder_tree_model, store, si->name);
+
+	mail_note_store (
+		shell_module, store, NULL,
+		mail_shell_module_add_store_done, store_info_ref (si));
+}
+
+static void
+mail_shell_module_add_local_store_done (CamelStore *store,
+                                        CamelFolderInfo *info,
+                                        gpointer unused)
+{
+	gint ii;
+
+	for (ii = 0; ii < G_N_ELEMENTS (default_local_folders); ii++) {
+		if (default_local_folders[ii].folder != NULL)
+			mail_note_folder (default_local_folders[ii].folder);
+	}
+}
+
+static void
+mail_shell_module_add_local_store (EShellModule *shell_module,
+                                   CamelStore *local_store,
+                                   const gchar *name)
+{
+	mail_shell_module_add_store (
+		shell_module, local_store, name,
+		mail_shell_module_add_local_store_done);
+}
+
+static void
+mail_shell_module_init_hooks (void)
+{
+	e_plugin_hook_register_type (em_config_hook_get_type ());
+	e_plugin_hook_register_type (em_event_hook_get_type ());
+	e_plugin_hook_register_type (em_format_hook_get_type ());
+	e_plugin_hook_register_type (em_junk_hook_get_type ());
+
+	em_format_hook_register_type (em_format_get_type ());
+	em_format_hook_register_type (em_format_html_get_type ());
+	em_format_hook_register_type (em_format_html_display_get_type ());
+
+	em_junk_hook_register_type (emj_get_type ());
+}
+
+static void
+mail_shell_module_init_importers (void)
+{
+	EImportClass *import_class;
+	EImportImporter *importer;
+
+	import_class = g_type_class_ref (e_import_get_type ());
+
+	importer = mbox_importer_peek ();
+	e_import_class_add_importer (import_class, importer, NULL, NULL);
+
+	importer = elm_importer_peek ();
+	e_import_class_add_importer (import_class, importer, NULL, NULL);
+
+	importer = pine_importer_peek ();
+	e_import_class_add_importer (import_class, importer, NULL, NULL);
+}
+
+static void
+mail_shell_module_init_local_store (EShellModule *shell_module)
+{
+	CamelException ex;
+	CamelService *service;
+	CamelURL *url;
+	const gchar *data_dir;
+	gchar *temp;
+	gint ii;
+
+	camel_exception_init (&ex);
+
+	url = camel_url_new ("mbox:", NULL);
+	data_dir = e_shell_module_get_data_dir (shell_module);
+	temp = g_build_filename (data_dir, "local", NULL);
+	camel_url_set_path (url, temp);
+	g_free (temp);
+
+	temp = camel_url_to_string (url, 0);
+	service = camel_session_get_service (
+		session, temp, CAMEL_PROVIDER_STORE, &ex);
+	g_free (temp);
+
+	if (service == NULL)
+		goto fail;
+
+	for (ii = 0; ii < G_N_ELEMENTS (default_local_folders); ii++) {
+		/* FIXME Should this URI be account relative? */
+		camel_url_set_fragment (url, default_local_folders[ii].name);
+		default_local_folders[ii].uri = camel_url_to_string (url, 0);
+		default_local_folders[ii].folder = camel_store_get_folder (
+			CAMEL_STORE (service), default_local_folders[ii].name,
+			CAMEL_STORE_FOLDER_CREATE, &ex);
+		camel_exception_clear (&ex);
+	}
+
+	camel_url_free (url);
+
+	camel_object_ref (service);
+	g_object_ref (shell_module);
+
+	mail_async_event_emit (
+		async_event, MAIL_ASYNC_GUI,
+		(MailAsyncFunc) mail_shell_module_add_local_store,
+		shell_module, service, _("On This Computer"));
+
+	local_store = CAMEL_STORE (service);
+
+	return;
+
+fail:
+	g_warning ("Could not initialize local store/folder: %s", ex.desc);
+
+	camel_exception_clear (&ex);
+	camel_url_free (url);
+}
+
+static void
+mail_shell_module_load_accounts (EShellModule *shell_module)
+{
+	EAccountList *account_list;
+	EIterator *iter;
+
+	account_list = mail_config_get_accounts ();
+
+	for (iter = e_list_get_iterator ((EList *) account_list);
+		e_iterator_is_valid (iter); e_iterator_next (iter)) {
+
+		EAccountService *service;
+		EAccount *account;
+		const gchar *name;
+		const gchar *url;
+
+		account = (EAccount *) e_iterator_get (iter);
+		service = account->source;
+		name = account->name;
+		url = service->url;
+
+		if (!account->enabled)
+			continue;
+
+		if (url == NULL || *url == '\0')
+			continue;
+
+		/* HACK: mbox URL's are handled by the local store setup
+		 *       above.  Any that come through as account sources
+		 *       are really movemail sources! */
+		if (g_str_has_prefix (url, "mbox:"))
+			continue;
+
+		e_mail_shell_module_load_store_by_uri (
+			shell_module, url, name);
+	}
+
+	g_object_unref (iter);
+}
+
+static void
+mail_shell_module_new_mail_cb (EShellWindow *shell_window)
+{
+	GtkAction *action;
+
+	action = e_shell_window_get_shell_view_action (
+		shell_window, MODULE_NAME);
+	g_object_set (action, "icon-name", "mail-unread", NULL);
+
+	g_print ("Shell Event: new-mail\n");
+}
+
 static void
 action_mail_folder_new_cb (GtkAction *action,
                            EShellWindow *shell_window)
@@ -83,8 +425,10 @@ static void
 mail_module_window_created (EShellModule *shell_module,
                             EShellWindow *shell_window)
 {
+	EShell *shell;
 	const gchar *module_name;
 
+	shell = e_shell_module_get_shell (shell_module);
 	module_name = G_TYPE_MODULE (shell_module)->name;
 
 	e_shell_window_register_new_item_actions (
@@ -94,6 +438,10 @@ mail_module_window_created (EShellModule *shell_module,
 	e_shell_window_register_new_source_actions (
 		shell_window, module_name,
 		source_entries, G_N_ELEMENTS (source_entries));
+
+	g_signal_connect_swapped (
+		shell, "event::new-mail",
+		G_CALLBACK (mail_shell_module_new_mail_cb), shell_window);
 }
 
 static EShellModuleInfo module_info = {
@@ -121,6 +469,24 @@ e_shell_module_init (GTypeModule *type_module)
 		shell_module, &module_info,
 		e_mail_shell_view_get_type (type_module));
 
+	/* This also initializes Camel, so it needs to happen early. */
+	mail_session_init (shell_module);
+
+	mail_shell_module_init_hooks ();
+	mail_shell_module_init_importers ();
+
+	/* XXX This never gets unreffed. */
+	mail_shell_module = g_object_ref (shell_module);
+
+	store_hash = g_hash_table_new_full (
+		g_direct_hash, g_direct_equal,
+		(GDestroyNotify) NULL,
+		(GDestroyNotify) store_hash_free);
+
+	async_event = mail_async_event_new ();
+
+	folder_tree_model = em_folder_tree_model_new (shell_module);
+
 	g_signal_connect_swapped (
 		shell, "handle-uri",
 		G_CALLBACK (mail_module_handle_uri), shell_module);
@@ -128,48 +494,106 @@ e_shell_module_init (GTypeModule *type_module)
 	g_signal_connect_swapped (
 		shell, "window-created",
 		G_CALLBACK (mail_module_window_created), shell_module);
+
+	mail_config_init ();
+	mail_msg_init ();
+
+	mail_shell_module_init_local_store (shell_module);
+	mail_shell_module_load_accounts (shell_module);
 }
 
-/**
- * e_mail_shell_module_load_store_by_uri:
- * @uri: URI of the #CamelStore
- * @name: name of the #CamelStore (for display purposes)
- *
- * Returns the newly added #CamelStore, or %NULL if a store could not
- * be loaded for @uri.
- *
- * Returns: the newly added #CamelStore, or %NULL
- **/
-CamelStore *
-e_mail_shell_module_load_store_by_uri (const gchar *uri,
-                                       const gchar *name)
+/******************************** Public API *********************************/
+
+CamelFolder *
+e_mail_shell_module_get_folder (EShellModule *shell_module,
+                                EMailFolderType folder_type)
 {
-	return NULL; /* FIXME */
+	g_return_val_if_fail (E_IS_SHELL_MODULE (shell_module), NULL);
+
+	return default_local_folders[folder_type].folder;
 }
 
-EAccountList *
-mail_config_get_accounts (void)
+const gchar *
+e_mail_shell_module_get_folder_uri (EShellModule *shell_module,
+                                    EMailFolderType folder_type)
 {
-	/* XXX Temporary placeholder. */
-	return NULL;
+	g_return_val_if_fail (E_IS_SHELL_MODULE (shell_module), NULL);
+
+	return default_local_folders[folder_type].uri;
+}
+
+EMFolderTreeModel *
+e_mail_shell_module_get_folder_tree_model (EShellModule *shell_module)
+{
+	/* Require a shell module in case we need it in the future. */
+	g_return_val_if_fail (E_IS_SHELL_MODULE (shell_module), NULL);
+
+	return folder_tree_model;
 }
 
 void
-mail_config_save_accounts (void)
+e_mail_shell_module_add_store (EShellModule *shell_module,
+                               CamelStore *store,
+                               const gchar *name)
 {
-	/* XXX Temporary placeholder. */
+	g_return_if_fail (E_IS_SHELL_MODULE (shell_module));
+	g_return_if_fail (CAMEL_IS_STORE (store));
+	g_return_if_fail (name != NULL);
+
+	mail_shell_module_add_store (shell_module, store, name, NULL);
 }
 
-ESignatureList *
-mail_config_get_signatures (void)
+CamelStore *
+e_mail_shell_module_get_local_store (EShellModule *shell_module)
 {
-	/* XXX Temporary placeholder. */
-	return NULL;
+	g_return_val_if_fail (E_IS_SHELL_MODULE (shell_module), NULL);
+	g_return_val_if_fail (local_store != NULL, NULL);
+
+	return local_store;
 }
 
-gchar *
-em_uri_from_camel (const gchar *curi)
+CamelStore *
+e_mail_shell_module_load_store_by_uri (EShellModule *shell_module,
+                                       const gchar *uri,
+                                       const gchar *name)
 {
-	/* XXX Temporary placeholder. */
+	CamelStore *store;
+	CamelProvider *provider;
+	CamelException ex;
+
+	g_return_val_if_fail (E_IS_SHELL_MODULE (shell_module), NULL);
+	g_return_val_if_fail (uri != NULL, NULL);
+	g_return_val_if_fail (name != NULL, NULL);
+
+	camel_exception_init (&ex);
+
+	/* Load the service, but don't connect.  Check its provider,
+	 * and if this belongs in the shell's folder list, add it. */
+
+	provider = camel_provider_get (uri, &ex);
+	if (provider == NULL)
+		goto fail;
+
+	if (!(provider->flags & CAMEL_PROVIDER_IS_STORAGE))
+		return NULL;
+
+	store = (CamelStore *) camel_session_get_service (
+		session, uri, CAMEL_PROVIDER_STORE, &ex);
+	if (store == NULL)
+		goto fail;
+
+	e_mail_shell_module_add_store (shell_module, store, name);
+
+	camel_object_unref (store);
+
+	return store;
+
+fail:
+	/* FIXME: Show an error dialog. */
+	g_warning (
+		"Couldn't get service: %s: %s", uri,
+		camel_exception_get_description (&ex));
+	camel_exception_clear (&ex);
+
 	return NULL;
 }