Implemented correct number of unread messages for multi-mailbox accounts
[modest] / src / dbus_api / modest-dbus-callbacks.c
index 1811ab7..90bc890 100644 (file)
@@ -67,6 +67,8 @@
 
 #include <modest-text-utils.h>
 
+#define DISABLE_GET_UNREAD_MSGS_FOR_MULTI_MAILBOX 1
+
 typedef struct 
 {
        gchar *to;
@@ -1529,8 +1531,17 @@ on_idle_show_memory_low (gpointer user_data)
 static gboolean
 on_idle_present_modal (gpointer user_data)
 {
+       GtkWindow *current, *transient;
        gdk_threads_enter ();
-       gtk_window_present (user_data);
+       current = (GtkWindow *) user_data;
+       while (GTK_IS_DIALOG (current)) {
+               transient = gtk_window_get_transient_for (GTK_WINDOW (current));
+               if (transient == NULL)
+                       break;
+               else
+                       current = transient;
+       }
+       gtk_window_present (current);
        gdk_threads_leave ();
 
        return FALSE;
@@ -1630,6 +1641,23 @@ modest_dbus_req_handler(const gchar * interface, const gchar * method,
        DBUS_TYPE_INT64_AS_STRING /* timestamp */ \
        DBUS_STRUCT_END_CHAR_AS_STRING
 
+#define ACCOUNT_HIT_DBUS_TYPE \
+       DBUS_STRUCT_BEGIN_CHAR_AS_STRING \
+       DBUS_TYPE_INT64_AS_STRING /* timestamp */ \
+       DBUS_TYPE_STRING_AS_STRING /* subject */ \
+       DBUS_STRUCT_END_CHAR_AS_STRING
+
+
+#define ACCOUNTS_HIT_DBUS_TYPE \
+       DBUS_STRUCT_BEGIN_CHAR_AS_STRING \
+       DBUS_TYPE_STRING_AS_STRING \
+       DBUS_TYPE_STRING_AS_STRING \
+       DBUS_TYPE_STRING_AS_STRING \
+       DBUS_TYPE_INT64_AS_STRING \
+       DBUS_TYPE_ARRAY_AS_STRING \
+       ACCOUNT_HIT_DBUS_TYPE \
+       DBUS_STRUCT_END_CHAR_AS_STRING
+
 static DBusMessage *
 search_result_to_message (DBusMessage *reply,
                           GList       *hits)
@@ -1862,6 +1890,456 @@ on_dbus_method_search (DBusConnection *con, DBusMessage *message)
        modest_search_all_accounts (search, search_all_cb, helper);
 }
 
+static gint
+headers_cmp (TnyHeader *a, TnyHeader *b)
+{
+       time_t date_a, date_b;
+       date_a = tny_header_get_date_received (a);
+       date_b = tny_header_get_date_received (b);
+
+       return date_a - date_b;
+}
+
+/**
+  * The helper structure for handling GetUnreadMessages DBus request
+  */
+typedef struct {
+       TnyList *accounts_list;
+       TnyList *inboxes_list;
+       gint unread_msgs_count;
+       DBusConnection *con;
+       DBusMessage *message;
+       GList *account_hits_list;
+       ModestMailOperation *mail_op;
+       guint folder_requests_total; /**< Total get-folder requests number for multi-mailbox accounts */
+       guint folder_requests_done; /**< Done get-folder requests number for multi-mailbox accounts */
+} GetUnreadMessagesHelper;
+
+typedef struct {
+       gchar *account_id;
+       gchar *account_name;
+       gchar *store_protocol;
+       gchar *mailbox_id;
+       gint unread_count;
+       GList *header_list;
+} AccountHits;
+
+static void return_results (GetUnreadMessagesHelper *helper)
+{
+       DBusMessage *reply;
+
+       reply = dbus_message_new_method_return (helper->message);
+       if (reply) {
+               dbus_uint32_t serial = 0;
+               GList *node;
+               DBusMessageIter iter;
+               DBusMessageIter array_iter;
+
+               dbus_message_iter_init_append (reply, &iter);
+               dbus_message_iter_open_container (&iter,
+                                                 DBUS_TYPE_ARRAY,
+                                                 ACCOUNTS_HIT_DBUS_TYPE,
+                                                 &array_iter);
+               for (node = helper->account_hits_list; node != NULL; node = g_list_next (node)) {
+                       AccountHits *ah = (AccountHits *) node->data;
+                       const char *account_id;
+                       const char *account_name;
+                       const char *store_protocol;
+                       gint64 unread_count;
+                       DBusMessageIter ah_struct_iter;
+                       DBusMessageIter sh_array_iter;
+                       GList *result_node;
+
+                       dbus_message_iter_open_container (&array_iter,
+                                                         DBUS_TYPE_STRUCT,
+                                                         NULL,
+                                                         &ah_struct_iter);
+                       account_id = ah->account_id;
+                       account_name = ah->account_name;
+                       store_protocol = ah->store_protocol;
+                       unread_count = ah->unread_count;
+                       dbus_message_iter_append_basic (&ah_struct_iter,
+                                                       DBUS_TYPE_STRING,
+                                                       &account_id);
+                       dbus_message_iter_append_basic (&ah_struct_iter,
+                                                       DBUS_TYPE_STRING,
+                                                       &account_name);
+                       dbus_message_iter_append_basic (&ah_struct_iter,
+                                                       DBUS_TYPE_STRING,
+                                                       &store_protocol);
+                       dbus_message_iter_append_basic (&ah_struct_iter,
+                                                       DBUS_TYPE_INT64,
+                                                       &unread_count);
+
+                       dbus_message_iter_open_container (&ah_struct_iter,
+                                                         DBUS_TYPE_ARRAY,
+                                                         ACCOUNT_HIT_DBUS_TYPE,
+                                                         &sh_array_iter);
+                       for (result_node = ah->header_list; result_node != NULL; result_node = g_list_next (result_node)) {
+                               TnyHeader *header = (TnyHeader *) result_node->data;
+                               DBusMessageIter sh_struct_iter;
+                               gint64 ts = tny_header_get_date_received (header);
+                               gchar *subject = tny_header_dup_subject (header);
+
+                               dbus_message_iter_open_container (&sh_array_iter,
+                                                                 DBUS_TYPE_STRUCT,
+                                                                 NULL,
+                                                                 &sh_struct_iter);
+                               dbus_message_iter_append_basic (&sh_struct_iter,
+                                                               DBUS_TYPE_INT64,
+                                                               &ts);
+                               dbus_message_iter_append_basic (&sh_struct_iter,
+                                                               DBUS_TYPE_STRING,
+                                                               (const gchar **) &subject); 
+
+                               dbus_message_iter_close_container (&sh_array_iter,
+                                                          &sh_struct_iter); 
+                               g_object_unref (header);
+                       }
+                       dbus_message_iter_close_container (&ah_struct_iter,
+                                                          &sh_array_iter); 
+
+                       dbus_message_iter_close_container (&array_iter,
+                                                          &ah_struct_iter); 
+                       g_free (ah->account_id);
+                       g_free (ah->account_name);
+                       g_free (ah->store_protocol);
+                       g_list_free (ah->header_list);
+               }
+
+               dbus_message_iter_close_container (&iter,
+                                                  &array_iter); 
+               dbus_connection_send (helper->con, reply, &serial);
+               dbus_connection_flush (helper->con);
+               dbus_message_unref (reply);
+
+       }
+       g_list_free (helper->account_hits_list);
+       dbus_message_unref (helper->message);
+       g_object_unref (helper->accounts_list);
+       modest_mail_operation_queue_remove (modest_runtime_get_mail_operation_queue (),
+                                           helper->mail_op);
+       g_object_unref (helper->mail_op);
+       g_slice_free (GetUnreadMessagesHelper, helper);
+}
+
+static void get_unread_messages_get_account (GetUnreadMessagesHelper *helper);
+static void get_unread_messages_get_headers (GetUnreadMessagesHelper *helper);
+
+
+static void get_unread_messages_get_headers_cb (TnyFolder *self,
+                                               gboolean cancelled, 
+                                               TnyList *headers,
+                                               GError *err, 
+                                               GetUnreadMessagesHelper *helper)
+{
+       TnyIterator *acc_iterator;
+       TnyAccount *account;
+       TnyIterator *headers_iterator;
+       GList *result_list = NULL;
+       gint members_count = 0;
+       AccountHits *account_hits;
+       const gchar *folder_id;
+       const gchar *bar;
+       ModestProtocolType store_protocol_type;
+       ModestProtocol *store_protocol;
+       gint unread_count;
+       ModestProtocolRegistry *registry;
+
+       acc_iterator = tny_list_create_iterator (helper->accounts_list);
+       account = TNY_ACCOUNT (tny_iterator_get_current (acc_iterator));
+
+       headers_iterator = tny_list_create_iterator (headers);
+       unread_count = 0;
+       while (!tny_iterator_is_done (headers_iterator)) {
+               TnyHeader *header;
+               TnyHeaderFlags flags;
+
+               header = TNY_HEADER (tny_iterator_get_current (headers_iterator));
+               flags = tny_header_get_flags (header);
+               if (!(flags & TNY_HEADER_FLAG_SEEN)) {
+                       unread_count++;
+                       result_list = g_list_insert_sorted (result_list, g_object_ref (header), (GCompareFunc) headers_cmp);
+                       if (members_count == helper->unread_msgs_count) {
+                               g_object_unref (result_list->data);
+                               result_list = g_list_delete_link (result_list, result_list);
+                       } else {
+                               members_count++;
+                       }
+               }
+
+               g_object_unref (header);
+               tny_iterator_next (headers_iterator);
+       }
+       g_object_unref (headers_iterator);
+
+       registry = modest_runtime_get_protocol_registry ();
+       store_protocol_type = modest_tny_account_get_protocol_type (account);
+
+       /* Get the number of unread messages for plug-in based accounts */
+       if (modest_protocol_registry_protocol_type_is_provider (registry, store_protocol_type)) {
+               unread_count = tny_folder_get_unread_count (self);
+       }
+
+       account_hits = g_slice_new (AccountHits);
+       account_hits->account_id = g_strdup (modest_tny_account_get_parent_modest_account_name_for_server_account (account));
+       account_hits->account_name = g_strdup (tny_account_get_name (account));
+       store_protocol = modest_protocol_registry_get_protocol_by_type (registry, store_protocol_type);
+       account_hits->store_protocol = g_strdup (modest_protocol_get_name (store_protocol));
+       account_hits->header_list = result_list;
+       account_hits->mailbox_id = NULL;
+       account_hits->unread_count = unread_count;
+
+       folder_id = tny_folder_get_id (self);
+       bar = g_strstr_len (folder_id, -1, "/");
+       if (bar) {
+               gchar *prefix;
+               prefix = g_strndup (folder_id, bar - folder_id);
+               if (g_strstr_len (prefix, -1, "@")) {
+                       account_hits->mailbox_id = g_strdup (prefix);
+               }
+               g_free (prefix);
+       }
+
+       helper->account_hits_list = g_list_prepend (helper->account_hits_list, account_hits);
+
+       get_unread_messages_get_headers (helper);
+
+}
+
+static void
+get_unread_messages_get_headers (GetUnreadMessagesHelper *helper)
+{
+       TnyIterator *iterator;
+
+       iterator = tny_list_create_iterator (helper->inboxes_list);
+       if (tny_iterator_is_done (iterator)) {
+               TnyIterator *accounts_iter;
+               TnyAccount *account;
+               g_object_unref (helper->inboxes_list);
+               helper->inboxes_list = NULL;
+
+               accounts_iter = tny_list_create_iterator (helper->accounts_list);
+               account = TNY_ACCOUNT (tny_iterator_get_current (accounts_iter));
+               g_object_unref (accounts_iter);
+
+               tny_list_remove (helper->accounts_list, (GObject *) account);
+               g_object_unref (account);
+               get_unread_messages_get_account (helper);
+       } else {
+               TnyFolder *folder;
+
+               folder = TNY_FOLDER (tny_iterator_get_current (iterator));
+               if (folder) {
+                       TnyList *headers_list;
+
+                       headers_list = TNY_LIST (tny_simple_list_new ());
+                       tny_folder_get_headers_async (folder, headers_list, FALSE, (TnyGetHeadersCallback) get_unread_messages_get_headers_cb, NULL, helper);
+                       g_object_unref (headers_list);
+                       tny_list_remove (helper->inboxes_list, G_OBJECT (folder));
+                       g_object_unref (folder);
+               }
+       }
+       g_object_unref (iterator);
+}
+
+static void get_account_folders_cb (TnyFolderStore *self, gboolean cancelled, TnyList *list, GError *err, gpointer user_data)
+{
+       GetUnreadMessagesHelper *helper = (GetUnreadMessagesHelper *) user_data;
+       TnyIterator *iterator;
+
+       iterator = tny_list_create_iterator (list);
+       while (!tny_iterator_is_done (iterator)) {
+               TnyFolder *folder;
+
+               folder = TNY_FOLDER (tny_iterator_get_current (iterator));
+               if (tny_folder_get_folder_type (folder) == TNY_FOLDER_TYPE_INBOX) {
+                       tny_list_prepend (helper->inboxes_list, G_OBJECT (folder));
+                       g_object_unref (folder);
+                       break;
+               }
+               g_object_unref (folder);
+               tny_iterator_next (iterator);
+       }
+       g_object_unref (iterator);
+       
+       get_unread_messages_get_headers (helper);
+}
+
+static void get_multi_mailbox_account_folders_cb (TnyFolderStore *self, gboolean cancelled, TnyList *list, GError *err, gpointer user_data)
+{
+       GetUnreadMessagesHelper *helper = (GetUnreadMessagesHelper *) user_data;
+       TnyIterator *iterator;
+
+       /* another request has been finished */
+       helper->folder_requests_done += 1;
+
+       /* analize the folders we got */
+       iterator = tny_list_create_iterator (list);
+       while (!tny_iterator_is_done (iterator)) {
+               TnyFolder *folder;
+
+               folder = TNY_FOLDER (tny_iterator_get_current (iterator));
+               if (tny_folder_get_folder_type (folder) == TNY_FOLDER_TYPE_INBOX) {
+                       tny_list_prepend (helper->inboxes_list, G_OBJECT (folder));
+               }
+
+               /* Try to check if we have inbox sub-folders in this folder */
+               if (TNY_IS_FOLDER_STORE (folder)) {
+                       TnyFolderStore *folder_store;
+
+                       folder_store = TNY_FOLDER_STORE (folder);
+                       if (folder_store) {
+                               TnyList *folders_list;
+
+                               helper->folder_requests_total += 1;
+                               folders_list = tny_simple_list_new ();
+                               tny_folder_store_get_folders_async (folder_store, folders_list,
+                                       NULL, FALSE, get_multi_mailbox_account_folders_cb, NULL, helper);
+                               g_object_unref (folders_list);
+                       }
+               }
+               g_object_unref (folder);
+               tny_iterator_next (iterator);
+       }
+       g_object_unref (iterator);
+
+       /* Check if we have handled all the inbox folders for multi-mailbox accounts */
+       if (helper->folder_requests_done == helper->folder_requests_total) {
+               TnyIterator *inboxes_it;
+               TnyIterator *accounts_it;
+               TnyAccount *account;
+               guint unread_messages;
+               AccountHits* account_hits;
+               ModestProtocol *store_protocol;
+               ModestProtocolType store_protocol_type;
+
+               /* Store the number of unread messages for the handled account */
+               unread_messages = 0;
+               accounts_it = tny_list_create_iterator (helper->accounts_list);
+               account = TNY_ACCOUNT (tny_iterator_get_current (accounts_it));
+               g_object_unref (accounts_it);
+               inboxes_it = tny_list_create_iterator (helper->inboxes_list);
+               while (!tny_iterator_is_done (inboxes_it)) {
+                       TnyFolder *folder;
+
+                       folder = TNY_FOLDER (tny_iterator_get_current (inboxes_it));
+                       if (folder) {
+                               unread_messages += tny_folder_get_unread_count (folder);
+                       }
+                       tny_iterator_next (inboxes_it);
+               }
+               g_object_unref (inboxes_it);
+               g_object_unref (helper->inboxes_list);
+               helper->inboxes_list = NULL;
+               account_hits = g_slice_new (AccountHits);
+               account_hits->account_id = g_strdup (
+                       modest_tny_account_get_parent_modest_account_name_for_server_account (account));
+               account_hits->account_name = g_strdup (tny_account_get_name (account));
+               store_protocol_type = modest_tny_account_get_protocol_type (account);
+               store_protocol = modest_protocol_registry_get_protocol_by_type (
+                       modest_runtime_get_protocol_registry (), store_protocol_type);
+               account_hits->store_protocol = g_strdup (modest_protocol_get_name (store_protocol));
+               account_hits->mailbox_id = NULL;
+               account_hits->header_list = NULL;
+               account_hits->unread_count = unread_messages;
+               helper->account_hits_list = g_list_prepend (helper->account_hits_list, account_hits);
+
+               /* remove the handled account from the accounts_list */
+               tny_list_remove (helper->accounts_list, G_OBJECT (account));
+               g_object_unref (account);
+
+               get_unread_messages_get_account (helper);
+       }
+}
+
+static void
+get_unread_messages_get_account (GetUnreadMessagesHelper *helper)
+{
+       TnyIterator *iterator;
+
+       /* Search through all accounts */
+       iterator = tny_list_create_iterator (helper->accounts_list);
+       if (tny_iterator_is_done (iterator)) {
+               /* all results, then finish */
+               return_results (helper);
+       } else {
+               TnyAccount *account;
+               TnyList *folders_list;
+               ModestProtocolType protocol_type;
+
+               account = TNY_ACCOUNT (tny_iterator_get_current (iterator));
+               protocol_type = modest_tny_account_get_protocol_type (account);
+
+               folders_list = tny_simple_list_new ();
+               helper->inboxes_list =  tny_simple_list_new ();
+               if (MODEST_PROTOCOL_REGISTRY_TYPE_INVALID != protocol_type &&
+                       modest_protocol_registry_protocol_type_has_tag (modest_runtime_get_protocol_registry (),
+                               protocol_type, MODEST_PROTOCOL_REGISTRY_MULTI_MAILBOX_PROVIDER_PROTOCOLS)) {
+                       /* For multi-mailbox protocols we only can get the number of unread messages,
+                          we will not even try to get their email headers */
+                       helper->folder_requests_done = 0;
+                       helper->folder_requests_total = 1;
+                       tny_folder_store_get_folders_async (TNY_FOLDER_STORE (account),
+                               folders_list, NULL, FALSE, get_multi_mailbox_account_folders_cb, NULL, helper);
+               }
+               else {
+                       /* For non-multi-mailbox protocols we will get their email headers */
+                       tny_folder_store_get_folders_async (TNY_FOLDER_STORE (account),
+                               folders_list, NULL, FALSE, get_account_folders_cb, NULL, helper);
+               }
+               g_object_unref (folders_list);
+               g_object_unref (account);
+
+       }
+       g_object_unref (iterator);
+}
+
+static gboolean
+on_idle_get_unread_messages (GetUnreadMessagesHelper *helper)
+{
+       ModestTnyAccountStore *astore;
+       astore = modest_runtime_get_account_store ();
+
+       tny_account_store_get_accounts (TNY_ACCOUNT_STORE (astore),
+                                       helper->accounts_list,
+                                       TNY_ACCOUNT_STORE_STORE_ACCOUNTS);
+
+       get_unread_messages_get_account (helper);
+
+       return FALSE;
+}
+
+static void
+on_dbus_method_get_unread_messages (DBusConnection *con, DBusMessage *message)
+{
+       dbus_bool_t  res;
+       dbus_int64_t unread_msgs_count_v;
+       DBusError error;
+       GetUnreadMessagesHelper *helper;
+
+       dbus_error_init (&error);
+
+       res = dbus_message_get_args (message,
+                                    &error,
+                                    DBUS_TYPE_INT32, &unread_msgs_count_v,
+                                    DBUS_TYPE_INVALID);
+
+       helper = g_slice_new (GetUnreadMessagesHelper);
+       helper->unread_msgs_count = unread_msgs_count_v;
+       dbus_message_ref (message);
+       helper->message = message;
+       helper->con = con;
+       helper->account_hits_list = NULL;
+       helper->inboxes_list = NULL;
+       helper->accounts_list = TNY_LIST (tny_simple_list_new ());
+       helper->mail_op = modest_mail_operation_new (NULL);
+       modest_mail_operation_queue_add (modest_runtime_get_mail_operation_queue (),
+                                        helper->mail_op);
+
+       g_idle_add ((GSourceFunc) on_idle_get_unread_messages, helper);
+}
+
 
 /* A complex D-Bus type (like a struct),
  * used to return various information about a folder.
@@ -2213,6 +2691,22 @@ modest_dbus_req_filter (DBusConnection *con,
                }
                                
        } else if (dbus_message_is_method_call (message,
+                                        MODEST_DBUS_IFACE,
+                                        MODEST_DBUS_METHOD_GET_UNREAD_MESSAGES)) {
+               
+       /* don't try to get unread messages when there not enough mem */
+               if (modest_platform_check_memory_low (NULL, TRUE)) {
+                       g_warning ("%s: not enough memory for searching",
+                                  __FUNCTION__);
+                       reply_empty_results (con, message);
+                       handled = TRUE;
+
+               } else {
+                       on_dbus_method_get_unread_messages (con, message);
+                       handled = TRUE;
+               }
+                               
+       } else if (dbus_message_is_method_call (message,
                                                MODEST_DBUS_IFACE,
                                                MODEST_DBUS_METHOD_GET_FOLDERS)) {
                on_dbus_method_get_folders (con, message);