* if we cannot get a valid folder type from the server, we can 'guess' the
[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         g_type_class_add_private (gobject_class,
85                                   sizeof(ModestTnyFolderTreeViewPrivate));
86         
87         signals[FOLDER_SELECTED_SIGNAL] = 
88                 g_signal_new ("folder_selected",
89                               G_TYPE_FROM_CLASS (gobject_class),
90                               G_SIGNAL_RUN_FIRST,
91                               G_STRUCT_OFFSET (ModestTnyFolderTreeViewClass,folder_selected),
92                               NULL, NULL,
93                               g_cclosure_marshal_VOID__POINTER,
94                               G_TYPE_NONE, 1, G_TYPE_POINTER); 
95 }
96
97
98
99 static void
100 text_cell_data  (GtkTreeViewColumn *column,  GtkCellRenderer *renderer,
101                    GtkTreeModel *tree_model,  GtkTreeIter *iter,  gpointer data)
102 {
103         GObject *rendobj;
104         gchar *fname;
105         guint unread;
106         TnyMsgFolderType type;
107         
108         gtk_tree_model_get (tree_model, iter,
109                             TNY_ACCOUNT_TREE_MODEL_NAME_COLUMN, &fname,
110                             TNY_ACCOUNT_TREE_MODEL_TYPE_COLUMN, &type,
111                             TNY_ACCOUNT_TREE_MODEL_UNREAD_COLUMN, &unread, -1);
112         rendobj = G_OBJECT(renderer);
113
114         if (unread > 0) {
115                 gchar *folder_title = g_strdup_printf ("%s (%d)", fname, unread);
116                 g_object_set (rendobj,"text", folder_title,  "weight", 800, NULL);
117                 g_free (folder_title);
118         } else 
119                 g_object_set (rendobj,"text", fname, "weight", 400, NULL);
120                 
121         g_free (fname);
122 }
123
124 /* FIXME: move these to TnyMail */
125 enum {
126
127         TNY_MSG_FOLDER_TYPE_NOTES = TNY_MSG_FOLDER_TYPE_SENT + 1, /* urgh */
128         TNY_MSG_FOLDER_TYPE_DRAFTS,
129         TNY_MSG_FOLDER_TYPE_CONTACTS,
130         TNY_MSG_FOLDER_TYPE_CALENDAR
131 };
132         
133 static TnyMsgFolderType
134 guess_folder_type (const gchar* name)
135 {
136         TnyMsgFolderType type;
137         gchar *folder;
138
139         g_return_val_if_fail (name, TNY_MSG_FOLDER_TYPE_NORMAL);
140         
141         type = TNY_MSG_FOLDER_TYPE_NORMAL;
142         folder = g_utf8_strdown (name, strlen(name));
143
144         if (strcmp (folder, "inbox") == 0 || strcmp (folder, _("inbox")) == 0)
145             type = TNY_MSG_FOLDER_TYPE_INBOX;
146         else if (strcmp (folder, "outbox") == 0 || strcmp (folder, _("outbox")) == 0)
147                 type = TNY_MSG_FOLDER_TYPE_OUTBOX;
148         else if (g_str_has_prefix(folder, "junk") || g_str_has_prefix(folder, _("junk")))
149                 type = TNY_MSG_FOLDER_TYPE_JUNK;
150         else if (g_str_has_prefix(folder, "trash") || g_str_has_prefix(folder, _("trash")))
151                 type = TNY_MSG_FOLDER_TYPE_JUNK;
152         else if (g_str_has_prefix(folder, "sent") || g_str_has_prefix(folder, _("sent")))
153                 type = TNY_MSG_FOLDER_TYPE_SENT;
154
155         /* these are not *really* TNY_ types */
156         else if (g_str_has_prefix(folder, "draft") || g_str_has_prefix(folder, _("draft")))
157                 type = TNY_MSG_FOLDER_TYPE_DRAFTS;
158         else if (g_str_has_prefix(folder, "notes") || g_str_has_prefix(folder, _("notes")))
159                 type = TNY_MSG_FOLDER_TYPE_NOTES;
160         else if (g_str_has_prefix(folder, "contacts") || g_str_has_prefix(folder, _("contacts")))
161                 type = TNY_MSG_FOLDER_TYPE_CONTACTS;
162         else if (g_str_has_prefix(folder, "calendar") || g_str_has_prefix(folder, _("calendar")))
163                 type = TNY_MSG_FOLDER_TYPE_CALENDAR;
164         
165         g_free (folder);
166         return type;
167 }
168
169
170 static void
171 icon_cell_data  (GtkTreeViewColumn *column,  GtkCellRenderer *renderer,
172                  GtkTreeModel *tree_model,  GtkTreeIter *iter, gpointer data)
173 {
174         GObject *rendobj;
175         GdkPixbuf *pixbuf;
176         TnyMsgFolderType type;
177         gchar *fname;
178         int unread;
179         
180         rendobj = G_OBJECT(renderer);
181         gtk_tree_model_get (tree_model, iter,
182                             TNY_ACCOUNT_TREE_MODEL_TYPE_COLUMN, &type,
183                             TNY_ACCOUNT_TREE_MODEL_NAME_COLUMN, &fname,
184                             TNY_ACCOUNT_TREE_MODEL_UNREAD_COLUMN, &unread, -1);
185         rendobj = G_OBJECT(renderer);
186         
187         if (type == TNY_MSG_FOLDER_TYPE_NORMAL)
188                 type = guess_folder_type (fname);
189
190         g_free (fname);
191
192         switch (type) {
193         case TNY_MSG_FOLDER_TYPE_INBOX:
194                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_INBOX);
195                 break;
196         case TNY_MSG_FOLDER_TYPE_OUTBOX:
197                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_OUTBOX);
198                 break;
199         case TNY_MSG_FOLDER_TYPE_JUNK:
200                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_JUNK);
201                 break;
202         case TNY_MSG_FOLDER_TYPE_SENT:
203                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_SENT);
204                 break;
205         case TNY_MSG_FOLDER_TYPE_DRAFTS:
206                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_DRAFTS);
207                 break;
208         case TNY_MSG_FOLDER_TYPE_NOTES:
209                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_NOTES);
210                 break;
211         case TNY_MSG_FOLDER_TYPE_CALENDAR:
212                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_CALENDAR);
213                 break;
214         case TNY_MSG_FOLDER_TYPE_CONTACTS:
215                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_CONTACTS);
216                 break;
217                 
218         case TNY_MSG_FOLDER_TYPE_NORMAL:
219         default:
220                 pixbuf = modest_icon_factory_get_icon (MODEST_FOLDER_ICON_NORMAL);
221                 break;
222         }
223
224         g_object_set (rendobj,
225                       "pixbuf-expander-open",
226                       modest_icon_factory_get_icon (MODEST_FOLDER_ICON_OPEN),
227                       "pixbuf-expander-closed",
228                       modest_icon_factory_get_icon (MODEST_FOLDER_ICON_CLOSED),
229                       "pixbuf", pixbuf,
230                       NULL);
231 }
232
233 static void
234 modest_tny_folder_tree_view_init (ModestTnyFolderTreeView *obj)
235 {
236         ModestTnyFolderTreeViewPrivate *priv;
237         GtkTreeViewColumn *column;
238         GtkCellRenderer *renderer;
239         GtkTreeSelection *sel;
240         
241         priv =  MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(obj);
242         
243         priv->view_is_empty     = TRUE;
244         priv->tny_account_store = NULL;
245         priv->cur_folder = NULL;
246
247         column = gtk_tree_view_column_new ();
248         gtk_tree_view_column_set_title (column,
249                                         _("All Mail Folders"));
250         
251         gtk_tree_view_append_column (GTK_TREE_VIEW(obj),
252                                      column);
253         
254         renderer = gtk_cell_renderer_pixbuf_new();
255         gtk_tree_view_column_pack_start (column, renderer, FALSE);
256         gtk_tree_view_column_set_cell_data_func(column, renderer,
257                                                 icon_cell_data, NULL, NULL);
258         
259         renderer = gtk_cell_renderer_text_new();
260         gtk_tree_view_column_pack_start (column, renderer, FALSE);
261         gtk_tree_view_column_set_cell_data_func(column, renderer,
262                                                 text_cell_data, NULL, NULL);
263         
264         sel = gtk_tree_view_get_selection (GTK_TREE_VIEW(obj));
265         gtk_tree_selection_set_mode (sel, GTK_SELECTION_SINGLE);
266
267         gtk_tree_view_column_set_spacing (column, 2);
268         gtk_tree_view_column_set_resizable (column, TRUE);
269         gtk_tree_view_column_set_fixed_width (column, TRUE);            
270         gtk_tree_view_set_headers_visible   (GTK_TREE_VIEW(obj), TRUE);
271         gtk_tree_view_set_headers_clickable (GTK_TREE_VIEW(obj), FALSE);
272
273 }
274
275
276 static void
277 modest_tny_folder_tree_view_finalize (GObject *obj)
278 {
279         ModestTnyFolderTreeViewPrivate *priv;
280
281         g_return_if_fail (obj);
282         
283         priv =  MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(obj);
284         if (priv->tny_account_store) {
285                 g_object_unref (G_OBJECT(priv->tny_account_store));
286                 priv->tny_account_store = NULL;
287         }
288
289         (*parent_class->finalize)(obj);
290 }
291
292
293 static void
294 modest_tny_folder_tree_view_set_account_store (TnySummaryWindowIface *self,
295                                                TnyAccountStoreIface *account_store)
296 {
297         ModestTnyFolderTreeViewPrivate *priv;
298
299         g_return_if_fail (self);
300         g_return_if_fail (account_store);
301         
302         priv = MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(self);
303         if (priv->tny_account_store) {
304                 g_object_unref (priv->tny_account_store);
305                 priv->tny_account_store = NULL;
306         }
307
308         g_object_ref (G_OBJECT(priv->tny_account_store = account_store));
309 }
310
311
312
313 GtkWidget*
314 modest_tny_folder_tree_view_new (TnyAccountStoreIface *iface)
315 {
316         GObject *self;
317         ModestTnyFolderTreeViewPrivate *priv;
318         GtkTreeSelection *sel;
319
320         self = G_OBJECT(g_object_new(MODEST_TYPE_TNY_FOLDER_TREE_VIEW, NULL));
321         priv = MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(self);
322
323         g_return_val_if_fail (iface, NULL);
324         
325         if (!update_model (MODEST_TNY_FOLDER_TREE_VIEW(self), iface))
326                 g_warning ("failed or update model");
327
328         sel = gtk_tree_view_get_selection(GTK_TREE_VIEW(self));
329         g_signal_connect (sel, "changed",
330                           G_CALLBACK(selection_changed), self);
331                 
332         return GTK_WIDGET(self);
333 }
334
335
336
337
338 static gboolean
339 update_model_empty (ModestTnyFolderTreeView *self)
340 {
341         GtkTreeIter  iter;
342         GtkTreeStore *store;
343         ModestTnyFolderTreeViewPrivate *priv;
344         
345         g_return_val_if_fail (self, FALSE);
346         
347         store = gtk_tree_store_new (1, G_TYPE_STRING);
348         gtk_tree_store_append (store, &iter, NULL);
349
350         gtk_tree_store_set (store, &iter, 0,
351                             _("(empty)"), -1);
352
353         gtk_tree_view_set_model (GTK_TREE_VIEW(self),
354                                  GTK_TREE_MODEL(store));
355         g_object_unref (store);
356
357         priv = MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(self);
358         priv->view_is_empty = TRUE;
359         
360         return TRUE;
361 }
362
363
364 static gboolean
365 update_model (ModestTnyFolderTreeView *self,TnyAccountStoreIface *iface)
366 {
367         const GList *accounts;
368         TnyAccountTreeModel *folder_model;
369         ModestTnyFolderTreeViewPrivate *priv;
370                 
371         g_return_val_if_fail (iface, FALSE);
372
373         priv =  MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(self);
374         priv->view_is_empty = TRUE;
375
376         accounts = tny_account_store_iface_get_store_accounts (iface);
377         if (!accounts) {
378                 g_warning ("no accounts have been defined yet");
379                 return update_model_empty (self);
380         }
381         
382         folder_model = tny_account_tree_model_new ();
383         if (!folder_model) {
384                 g_warning ("failed to get account tree model");
385                 return update_model_empty (self);
386         }
387         
388         while (accounts) {
389                 TnyStoreAccountIface *account =
390                         TNY_STORE_ACCOUNT_IFACE(accounts->data);
391                 if (!account) {
392                         g_warning ("invalid account");
393                         g_object_unref (folder_model);
394                         return update_model_empty (self);
395                 }
396                 tny_account_tree_model_add (TNY_ACCOUNT_TREE_MODEL (folder_model),
397                                             account);
398                 accounts = accounts->next;
399         }
400         
401         gtk_tree_view_set_model (GTK_TREE_VIEW(self),
402                                  GTK_TREE_MODEL(folder_model)); 
403         g_object_unref (G_OBJECT(folder_model));
404         
405         priv->view_is_empty = FALSE; /* were not empty anymore! */
406         return TRUE;
407 }
408
409
410 void
411 selection_changed (GtkTreeSelection *sel, gpointer user_data)
412 {
413         GtkTreeModel            *model;
414         TnyMsgFolderIface       *folder = NULL;
415         GtkTreeIter             iter;
416         ModestTnyFolderTreeView *tree_view;
417         ModestTnyFolderTreeViewPrivate *priv;
418
419         g_return_if_fail (sel);
420         g_return_if_fail (user_data);
421
422         priv = MODEST_TNY_FOLDER_TREE_VIEW_GET_PRIVATE(user_data);
423
424         /* is_empty means that there is only the 'empty' item */
425         if (priv->view_is_empty)
426                 return;
427         
428         /* folder was _un_selected if true */
429         if (!gtk_tree_selection_get_selected (sel, &model, &iter))
430         {
431                 if (priv->cur_folder) 
432                         tny_msg_folder_iface_expunge (priv->cur_folder);
433                 priv->cur_folder = NULL;
434                 return; 
435         }
436
437         tree_view = MODEST_TNY_FOLDER_TREE_VIEW (user_data);
438
439         gtk_tree_model_get (model, &iter,
440                             TNY_ACCOUNT_TREE_MODEL_INSTANCE_COLUMN,
441                             &folder, -1);
442
443         if (priv->cur_folder) 
444                 tny_msg_folder_iface_expunge (priv->cur_folder);
445         priv->cur_folder = folder;
446
447         /* folder will not be defined if you click eg. on the root node */
448         if (folder)
449                 g_signal_emit (G_OBJECT(tree_view), signals[FOLDER_SELECTED_SIGNAL], 0,
450                                folder);
451 }
452
453
454
455