* fixed message count in folder view
[modest] / src / modest-tny-folder-tree-view.c
1 /* modest-tny-folder-tree-view.c */
2
3 /* insert (c)/licensing information) */
4 #include <glib/gi18n.h>
5 #include <string.h>
6
7 #include <tny-account-tree-model.h>
8 #include <tny-account-store-iface.h>
9 #include <tny-account-iface.h>
10 #include <tny-msg-folder-iface.h>
11 #include <tny-summary-window-iface.h>
12
13 #include "modest-tny-folder-tree-view.h"
14
15 #include <modest-icon-names.h>
16 #include "modest-icon-factory.h"
17
18
19 /* 'private'/'protected' functions */
20 static void modest_tny_folder_tree_view_class_init  (ModestTnyFolderTreeViewClass *klass);
21 static void modest_tny_folder_tree_view_init        (ModestTnyFolderTreeView *obj);
22 static void modest_tny_folder_tree_view_finalize    (GObject *obj);
23
24 //static void modest_tny_folder_tree_view_iface_init   (gpointer iface, gpointer data);
25 static void modest_tny_folder_tree_view_set_account_store (TnySummaryWindowIface *self,
26                                                            TnyAccountStoreIface *account_store);
27 static gboolean update_model (ModestTnyFolderTreeView *self,TnyAccountStoreIface *iface);
28 static gboolean update_model_empty (ModestTnyFolderTreeView *self);
29
30 static void selection_changed (GtkTreeSelection *sel, gpointer data);
31
32 enum {
33         FOLDER_SELECTED_SIGNAL,
34         LAST_SIGNAL
35 };
36
37 typedef struct _ModestTnyFolderTreeViewPrivate ModestTnyFolderTreeViewPrivate;
38 struct _ModestTnyFolderTreeViewPrivate {
39         TnyAccountStoreIface *tny_account_store;
40         TnyMsgFolderIface *cur_folder;
41         gboolean view_is_empty;
42 };
43 #define MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(o)      (G_TYPE_INSTANCE_GET_PRIVATE((o), \
44                                                          MODEST_TYPE_TNY_FOLDER_TREE_VIEW, \
45                                                          ModestTnyFolderTreeViewPrivate))
46 /* globals */
47 static GObjectClass *parent_class = NULL;
48
49 static guint signals[LAST_SIGNAL] = {0}; 
50
51 GType
52 modest_tny_folder_tree_view_get_type (void)
53 {
54         static GType my_type = 0;
55         if (!my_type) {
56                 static const GTypeInfo my_info = {
57                         sizeof(ModestTnyFolderTreeViewClass),
58                         NULL,           /* base init */
59                         NULL,           /* base finalize */
60                         (GClassInitFunc) modest_tny_folder_tree_view_class_init,
61                         NULL,           /* class finalize */
62                         NULL,           /* class data */
63                         sizeof(ModestTnyFolderTreeView),
64                         1,              /* n_preallocs */
65                         (GInstanceInitFunc) modest_tny_folder_tree_view_init,
66                 };
67                                 
68                 my_type = g_type_register_static (GTK_TYPE_TREE_VIEW,
69                                                   "ModestTnyFolderTreeView",
70                                                   &my_info, 0);         
71         }
72         return my_type;
73 }
74
75 static void
76 modest_tny_folder_tree_view_class_init (ModestTnyFolderTreeViewClass *klass)
77 {
78         GObjectClass *gobject_class;
79         gobject_class = (GObjectClass*) klass;
80
81         parent_class            = g_type_class_peek_parent (klass);
82         gobject_class->finalize = modest_tny_folder_tree_view_finalize;
83         
84         klass->update_model = modest_tny_folder_tree_view_update_model;
85
86         g_type_class_add_private (gobject_class,
87                                   sizeof(ModestTnyFolderTreeViewPrivate));
88         
89         signals[FOLDER_SELECTED_SIGNAL] = 
90                 g_signal_new ("folder_selected",
91                               G_TYPE_FROM_CLASS (gobject_class),
92                               G_SIGNAL_RUN_FIRST,
93                               G_STRUCT_OFFSET (ModestTnyFolderTreeViewClass,folder_selected),
94                               NULL, NULL,
95                               g_cclosure_marshal_VOID__POINTER,
96                               G_TYPE_NONE, 1, G_TYPE_POINTER); 
97 }
98
99
100
101 static void
102 text_cell_data  (GtkTreeViewColumn *column,  GtkCellRenderer *renderer,
103                    GtkTreeModel *tree_model,  GtkTreeIter *iter,  gpointer data)
104 {
105         GObject *rendobj;
106         gchar *fname;
107         gint unread;
108         TnyMsgFolderType type;
109         
110         gtk_tree_model_get (tree_model, iter,
111                             TNY_ACCOUNT_TREE_MODEL_NAME_COLUMN, &fname,
112                             TNY_ACCOUNT_TREE_MODEL_TYPE_COLUMN, &type,
113                             TNY_ACCOUNT_TREE_MODEL_UNREAD_COLUMN, &unread, -1);
114         rendobj = G_OBJECT(renderer);
115
116         if (unread > 0) {
117                 gchar *folder_title = g_strdup_printf ("%s (%d)", fname, unread);
118                 g_object_set (rendobj,"text", folder_title,  "weight", 800, NULL);
119                 g_free (folder_title);
120         } else 
121                 g_object_set (rendobj,"text", fname, "weight", 400, NULL);
122                 
123         g_free (fname);
124 }
125
126 /* FIXME: move these to TnyMail */
127 enum {
128
129         TNY_MSG_FOLDER_TYPE_NOTES = TNY_MSG_FOLDER_TYPE_SENT + 1, /* urgh */
130         TNY_MSG_FOLDER_TYPE_DRAFTS,
131         TNY_MSG_FOLDER_TYPE_CONTACTS,
132         TNY_MSG_FOLDER_TYPE_CALENDAR
133 };
134         
135 static TnyMsgFolderType
136 guess_folder_type (const gchar* name)
137 {
138         TnyMsgFolderType type;
139         gchar *folder;
140
141         g_return_val_if_fail (name, TNY_MSG_FOLDER_TYPE_NORMAL);
142         
143         type = TNY_MSG_FOLDER_TYPE_NORMAL;
144         folder = g_utf8_strdown (name, strlen(name));
145
146         if (strcmp (folder, "inbox") == 0 || strcmp (folder, _("inbox")) == 0)
147             type = TNY_MSG_FOLDER_TYPE_INBOX;
148         else if (strcmp (folder, "outbox") == 0 || strcmp (folder, _("outbox")) == 0)
149                 type = TNY_MSG_FOLDER_TYPE_OUTBOX;
150         else if (g_str_has_prefix(folder, "junk") || g_str_has_prefix(folder, _("junk")))
151                 type = TNY_MSG_FOLDER_TYPE_JUNK;
152         else if (g_str_has_prefix(folder, "trash") || g_str_has_prefix(folder, _("trash")))
153                 type = TNY_MSG_FOLDER_TYPE_JUNK;
154         else if (g_str_has_prefix(folder, "sent") || g_str_has_prefix(folder, _("sent")))
155                 type = TNY_MSG_FOLDER_TYPE_SENT;
156
157         /* these are not *really* TNY_ types */
158         else if (g_str_has_prefix(folder, "draft") || g_str_has_prefix(folder, _("draft")))
159                 type = TNY_MSG_FOLDER_TYPE_DRAFTS;
160         else if (g_str_has_prefix(folder, "notes") || g_str_has_prefix(folder, _("notes")))
161                 type = TNY_MSG_FOLDER_TYPE_NOTES;
162         else if (g_str_has_prefix(folder, "contacts") || g_str_has_prefix(folder, _("contacts")))
163                 type = TNY_MSG_FOLDER_TYPE_CONTACTS;
164         else if (g_str_has_prefix(folder, "calendar") || g_str_has_prefix(folder, _("calendar")))
165                 type = TNY_MSG_FOLDER_TYPE_CALENDAR;
166         
167         g_free (folder);
168         return type;
169 }
170
171
172 static void
173 icon_cell_data  (GtkTreeViewColumn *column,  GtkCellRenderer *renderer,
174                  GtkTreeModel *tree_model,  GtkTreeIter *iter, gpointer data)
175 {
176         GObject *rendobj;
177         GdkPixbuf *pixbuf;
178         TnyMsgFolderType type;
179         gchar *fname = NULL;
180         gint unread;
181         
182         rendobj = G_OBJECT(renderer);
183         gtk_tree_model_get (tree_model, iter,
184                             TNY_ACCOUNT_TREE_MODEL_TYPE_COLUMN, &type,
185                             TNY_ACCOUNT_TREE_MODEL_NAME_COLUMN, &fname,
186                             TNY_ACCOUNT_TREE_MODEL_UNREAD_COLUMN, &unread, -1);
187         rendobj = G_OBJECT(renderer);
188         
189         if (type == TNY_MSG_FOLDER_TYPE_NORMAL)
190                 type = guess_folder_type (fname);
191         
192         if (fname);
193                 g_free (fname);
194
195         switch (type) {
196         case TNY_MSG_FOLDER_TYPE_INBOX:
197                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_INBOX);
198                 break;
199         case TNY_MSG_FOLDER_TYPE_OUTBOX:
200                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_OUTBOX);
201                 break;
202         case TNY_MSG_FOLDER_TYPE_JUNK:
203                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_JUNK);
204                 break;
205         case TNY_MSG_FOLDER_TYPE_SENT:
206                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_SENT);
207                 break;
208         case TNY_MSG_FOLDER_TYPE_DRAFTS:
209                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_DRAFTS);
210                 break;
211         case TNY_MSG_FOLDER_TYPE_NOTES:
212                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_NOTES);
213                 break;
214         case TNY_MSG_FOLDER_TYPE_CALENDAR:
215                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_CALENDAR);
216                 break;
217         case TNY_MSG_FOLDER_TYPE_CONTACTS:
218                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_CONTACTS);
219                 break;
220                 
221         case TNY_MSG_FOLDER_TYPE_NORMAL:
222         default:
223                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_NORMAL);
224                 break;
225         }
226
227         g_object_set (rendobj,
228                       "pixbuf-expander-open",
229                       modest_icon_factory_get_icon (MODEST_FOLDER_ICON_OPEN),
230                       "pixbuf-expander-closed",
231                       modest_icon_factory_get_icon (MODEST_FOLDER_ICON_CLOSED),
232                       "pixbuf", pixbuf,
233                       NULL);
234 }
235
236 static void
237 modest_tny_folder_tree_view_init (ModestTnyFolderTreeView *obj)
238 {
239         ModestTnyFolderTreeViewPrivate *priv;
240         GtkTreeViewColumn *column;
241         GtkCellRenderer *renderer;
242         GtkTreeSelection *sel;
243         
244         priv =  MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(obj);
245         
246         priv->view_is_empty     = TRUE;
247         priv->tny_account_store = NULL;
248         priv->cur_folder = NULL;
249
250         column = gtk_tree_view_column_new ();
251         gtk_tree_view_column_set_title (column,
252                                         _("All Mail Folders"));
253         
254         gtk_tree_view_append_column (GTK_TREE_VIEW(obj),
255                                      column);
256         
257         renderer = gtk_cell_renderer_pixbuf_new();
258         gtk_tree_view_column_pack_start (column, renderer, FALSE);
259         gtk_tree_view_column_set_cell_data_func(column, renderer,
260                                                 icon_cell_data, NULL, NULL);
261         
262         renderer = gtk_cell_renderer_text_new();
263         gtk_tree_view_column_pack_start (column, renderer, FALSE);
264         gtk_tree_view_column_set_cell_data_func(column, renderer,
265                                                 text_cell_data, NULL, NULL);
266         
267         sel = gtk_tree_view_get_selection (GTK_TREE_VIEW(obj));
268         gtk_tree_selection_set_mode (sel, GTK_SELECTION_SINGLE);
269
270         gtk_tree_view_column_set_spacing (column, 2);
271         gtk_tree_view_column_set_resizable (column, TRUE);
272         gtk_tree_view_column_set_fixed_width (column, TRUE);            
273         gtk_tree_view_set_headers_visible   (GTK_TREE_VIEW(obj), TRUE);
274         gtk_tree_view_set_headers_clickable (GTK_TREE_VIEW(obj), FALSE);
275
276 }
277
278
279 static void
280 modest_tny_folder_tree_view_finalize (GObject *obj)
281 {
282         ModestTnyFolderTreeViewPrivate *priv;
283
284         g_return_if_fail (obj);
285         
286         priv =  MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(obj);
287         if (priv->tny_account_store) {
288                 g_object_unref (G_OBJECT(priv->tny_account_store));
289                 priv->tny_account_store = NULL;
290         }
291
292         (*parent_class->finalize)(obj);
293 }
294
295
296 static void
297 modest_tny_folder_tree_view_set_account_store (TnySummaryWindowIface *self,
298                                                TnyAccountStoreIface *account_store)
299 {
300         ModestTnyFolderTreeViewPrivate *priv;
301
302         g_return_if_fail (self);
303         g_return_if_fail (account_store);
304         
305         priv = MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(self);
306         if (priv->tny_account_store) {
307                 g_object_unref (priv->tny_account_store);
308                 priv->tny_account_store = NULL;
309         }
310
311         g_object_ref (G_OBJECT(priv->tny_account_store = account_store));
312 }
313
314
315
316 GtkWidget*
317 modest_tny_folder_tree_view_new (TnyAccountStoreIface *iface)
318 {
319         GObject *self;
320         ModestTnyFolderTreeViewPrivate *priv;
321         GtkTreeSelection *sel;
322
323         self = G_OBJECT(g_object_new(MODEST_TYPE_TNY_FOLDER_TREE_VIEW, NULL));
324         priv = MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(self);
325
326         g_return_val_if_fail (iface, NULL);
327         
328         if (!update_model (MODEST_TNY_FOLDER_TREE_VIEW(self), iface))
329                 g_warning ("failed or update model");
330
331         sel = gtk_tree_view_get_selection(GTK_TREE_VIEW(self));
332         g_signal_connect (sel, "changed",
333                           G_CALLBACK(selection_changed), self);
334                 
335         return GTK_WIDGET(self);
336 }
337
338
339
340
341 static gboolean
342 update_model_empty (ModestTnyFolderTreeView *self)
343 {
344         GtkTreeIter  iter;
345         GtkTreeStore *store;
346         ModestTnyFolderTreeViewPrivate *priv;
347         
348         g_return_val_if_fail (self, FALSE);
349         
350         store = gtk_tree_store_new (1, G_TYPE_STRING);
351         gtk_tree_store_append (store, &iter, NULL);
352
353         gtk_tree_store_set (store, &iter, 0,
354                             _("(empty)"), -1);
355
356         gtk_tree_view_set_model (GTK_TREE_VIEW(self),
357                                  GTK_TREE_MODEL(store));
358         g_object_unref (store);
359
360         priv = MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(self);
361         priv->view_is_empty = TRUE;
362         
363         return TRUE;
364 }
365
366
367 static gboolean
368 update_model (ModestTnyFolderTreeView *self, TnyAccountStoreIface *iface)
369 {
370         const GList *accounts;
371         TnyAccountTreeModel *folder_model;
372         ModestTnyFolderTreeViewPrivate *priv;
373                 
374         g_return_val_if_fail (iface, FALSE);
375
376         priv =  MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(self);
377         priv->view_is_empty = TRUE;
378
379         accounts = tny_account_store_iface_get_store_accounts (iface);
380         if (!accounts) {
381                 g_warning ("no accounts have been defined yet");
382                 return update_model_empty (self);
383         }
384         
385         folder_model = tny_account_tree_model_new ();
386         if (!folder_model) {
387                 g_warning ("failed to get account tree model");
388                 return update_model_empty (self);
389         }
390         
391         while (accounts) {
392                 TnyStoreAccountIface *account =
393                         TNY_STORE_ACCOUNT_IFACE(accounts->data);
394                 if (!account) {
395                         g_warning ("invalid account");
396                         g_object_unref (folder_model);
397                         return update_model_empty (self);
398                 }
399                 tny_account_tree_model_add (TNY_ACCOUNT_TREE_MODEL (folder_model),
400                                             account);
401                 accounts = accounts->next;
402         }
403         
404         gtk_tree_view_set_model (GTK_TREE_VIEW(self),
405                                  GTK_TREE_MODEL(folder_model)); 
406         g_object_unref (G_OBJECT(folder_model));
407         
408         priv->view_is_empty = FALSE; /* were not empty anymore! */
409         return TRUE;
410 }
411
412
413 void
414 selection_changed (GtkTreeSelection *sel, gpointer user_data)
415 {
416         GtkTreeModel            *model;
417         TnyMsgFolderIface       *folder = NULL;
418         GtkTreeIter             iter;
419         ModestTnyFolderTreeView *tree_view;
420         ModestTnyFolderTreeViewPrivate *priv;
421
422         g_return_if_fail (sel);
423         g_return_if_fail (user_data);
424
425         priv = MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(user_data);
426
427         /* is_empty means that there is only the 'empty' item */
428         if (priv->view_is_empty)
429                 return;
430         
431         /* folder was _un_selected if true */
432         if (!gtk_tree_selection_get_selected (sel, &model, &iter))
433         {
434                 if (priv->cur_folder) 
435                         tny_msg_folder_iface_expunge (priv->cur_folder);
436                 priv->cur_folder = NULL;
437                 return; 
438         }
439
440         tree_view = MODEST_TNY_FOLDER_TREE_VIEW (user_data);
441
442         gtk_tree_model_get (model, &iter,
443                             TNY_ACCOUNT_TREE_MODEL_INSTANCE_COLUMN,
444                             &folder, -1);
445
446         if (priv->cur_folder) 
447                 tny_msg_folder_iface_expunge (priv->cur_folder);
448         priv->cur_folder = folder;
449
450         /* folder will not be defined if you click eg. on the root node */
451         if (folder)
452                 g_signal_emit (G_OBJECT(tree_view), signals[FOLDER_SELECTED_SIGNAL], 0,
453                                folder);
454 }
455
456
457 gboolean
458 modest_tny_folder_tree_view_update_model(ModestTnyFolderTreeView *self, 
459                                          TnyAccountStoreIface *iface)
460 {
461         g_return_val_if_fail (MODEST_IS_TNY_FOLDER_TREE_VIEW (self), FALSE);
462         
463         return update_model (self, iface);
464 }