1 #import <ObjXMPP/namespaces.h>
4 #import "JubGtkRosterUI.h"
5 #import "JubGObjectMap.h"
6 #import "JubGtkChatUI.h"
7 #import "JubGtkHelper.h"
9 static void roster_row_activated(GtkTreeView *tree_view, GtkTreePath *path,
10 GtkTreeViewColumn *column, gpointer data)
12 JubGtkRosterUI *roster = data;
14 GtkTreeModel *tree_model;
17 OFAutoreleasePool *pool;
19 tree_model = gtk_tree_view_get_model(tree_view);
20 gtk_tree_model_get_iter(tree_model, &row_iter, path);
21 gtk_tree_model_get(tree_model, &row_iter, 1, &jid_s, -1);
23 // This was a group row
26 pool = [OFAutoreleasePool new];
27 jid = [XMPPJID JIDWithString: [OFString stringWithUTF8String: jid_s]];
29 [roster performSelectorOnMainThread: @selector(chatForJID:)
35 static void presence_changed(GtkComboBox *combo_box, gpointer data)
38 XMPPConnection *connection = data;
39 OFAutoreleasePool *pool = [OFAutoreleasePool new];
41 const char *status = gtk_combo_box_get_active_id(combo_box);
43 if (!strcmp(status, "unavailable"))
44 pres = [XMPPPresence presenceWithType: @"unavailable"];
46 pres = [XMPPPresence presence];
47 if (strcmp(status, "available"))
48 [pres setShow: @(status)];
51 [connection sendStanza: pres];
56 static gboolean filter_roster_by_presence(GtkTreeModel *model,
57 GtkTreeIter *iter, gpointer data)
61 OFCountedSet *presences = data;
63 gtk_tree_model_get(model, iter, 1, &jid_s, -1);
69 jid = [[OFString alloc] initWithUTF8String: jid_s];
73 int num = [presences countForObject: jid];
83 @implementation JubGtkRosterUI
84 - initWithClient: (JubChatClient*)client;
89 GtkTreeView *roster_view;
92 groupMap = [[OFMapTable alloc]
93 initWithKeyFunctions: keyFunctions
94 valueFunctions: rowRefFunctions];
95 contactMap = [[OFMutableDictionary alloc] init];
96 chatMap = [[OFMutableDictionary alloc] init];
97 presences = [[OFCountedSet alloc] init];
98 connection = [client.connection retain];
100 [connection addDelegate: self];
101 [client.roster addDelegate: self];
103 builder = gtk_builder_new();
104 gtk_builder_add_from_file(builder, "data/gtk/roster.ui", NULL);
105 gtk_builder_connect_signals(builder, NULL);
108 GTK_WIDGET(gtk_builder_get_object(builder, "RosterWindow"));
110 gtk_widget_show(roster_window);
112 roster_model = GTK_TREE_STORE(gtk_builder_get_object(builder,
115 roster_filter = GTK_TREE_MODEL_FILTER(
116 gtk_builder_get_object(builder, "RosterTreeModelFilter"));
118 gtk_tree_model_filter_set_visible_func(roster_filter,
119 filter_roster_by_presence, presences, NULL);
121 roster_view = GTK_TREE_VIEW(gtk_builder_get_object(builder,
124 g_signal_connect(roster_view, "row_activated",
125 G_CALLBACK(roster_row_activated), self);
127 presence_combo = GTK_COMBO_BOX(gtk_builder_get_object(builder,
128 "PresenceComboBox"));
130 presence_combo_changed_handler_id =
131 g_signal_connect(presence_combo, "changed",
132 G_CALLBACK(presence_changed), connection);
134 g_object_unref(G_OBJECT(builder));
146 [contactMap release];
149 [connection release];
151 gtk_widget_destroy(roster_window);
156 /* Presence handling */
157 - (void)connection: (XMPPConnection*)connection
158 didReceivePresence: (XMPPPresence*)presence
160 if ([presence.type isEqual: @"available"])
161 [presences addObject: [presence.from bareJID]];
162 else if ([presence.type isEqual: @"unavailable"])
163 [presences removeObject: [presence.from bareJID]];
166 gtk_tree_model_filter_refilter(roster_filter);
170 // FIXME: This needs to move somewhere else
171 - (JubGtkChatUI*)chatForJID: (XMPPJID*)jid
173 OFAutoreleasePool *pool = [OFAutoreleasePool new];
175 [chatMap objectForKey: [jid bareJID]];
177 OFString * title = [@"Chat with " stringByAppendingString:
180 chat = [[[JubGtkChatUI alloc]
183 [chatMap removeObjectForKey: [jid bareJID]];
185 sendBlock: ^(OFString *text) {
187 [XMPPMessage messageWithType: @"chat"];
190 [connection sendStanza: msg];
194 [chatMap setObject: chat
195 forKey: [jid bareJID]];
203 - (void)connection: (XMPPConnection*)connection
204 didReceiveMessage: (XMPPMessage*)message
206 JubGtkChatUI *chat = [self chatForJID: message.from];
207 [chat addMessage: message.body
208 sender: [message.from bareJID]];
211 /* Roster Delegate methods */
212 - (void)Jub_addRosterItem: (XMPPRosterItem*)item
213 group: (OFString*)group
216 GtkTreeIter group_iter, contact_iter;
217 GtkTreeRowReference *group_ref, *contact_ref;
218 GtkTreePath *group_path, *contact_path;
219 OFMapTable *contactRows =
220 [contactMap objectForKey: [item.JID bareJID]];
222 group_ref = [groupMap valueForKey: group];
225 // Create new group row
226 gtk_tree_store_append(roster_model, &group_iter, NULL);
227 gtk_tree_store_set(roster_model, &group_iter,
228 0, [group UTF8String], -1);
230 group_path = gtk_tree_model_get_path(GTK_TREE_MODEL(
231 roster_model), &group_iter);
233 group_ref = gtk_tree_row_reference_new(GTK_TREE_MODEL(
234 roster_model), group_path);
236 [groupMap setValue: group_ref
239 // Get iter for existing group row
240 group_path = gtk_tree_row_reference_get_path(group_ref);
242 gtk_tree_model_get_iter(GTK_TREE_MODEL(roster_model),
243 &group_iter, group_path);
245 gtk_tree_path_free(group_path);
247 // Create new contact row
248 gtk_tree_store_append(roster_model, &contact_iter, &group_iter);
250 gtk_tree_store_set(roster_model, &contact_iter,
251 0, [item.name UTF8String],
252 1, [[item.JID bareJID] UTF8String], -1);
254 gtk_tree_store_set(roster_model, &contact_iter,
255 0, [item.JID.node UTF8String],
256 1, [[item.JID bareJID] UTF8String], -1);
258 contact_path = gtk_tree_model_get_path(GTK_TREE_MODEL(
259 roster_model), &contact_iter);
261 contact_ref = gtk_tree_row_reference_new(GTK_TREE_MODEL(
262 roster_model), contact_path);
264 gtk_tree_path_free(contact_path);
266 [contactRows setValue: contact_ref
271 - (void)Jub_removeRosterItem: (XMPPRosterItem*)item
272 group: (OFString*)group
275 GtkTreeIter contact_iter, group_iter;
276 GtkTreePath *contact_path, *group_path;
277 GtkTreeRowReference *contact_ref, *group_ref;
278 OFMapTable *contactRows =
279 [contactMap objectForKey: [item.JID bareJID]];
281 contact_ref = [contactRows valueForKey: group];
282 contact_path = gtk_tree_row_reference_get_path(contact_ref);
283 gtk_tree_model_get_iter(GTK_TREE_MODEL(roster_model),
284 &contact_iter, contact_path);
286 gtk_tree_store_remove(roster_model, &contact_iter);
288 group_ref = [groupMap valueForKey: group];
289 group_path = gtk_tree_row_reference_get_path(group_ref);
290 gtk_tree_model_get_iter(GTK_TREE_MODEL(roster_model),
291 &group_iter, group_path);
293 if (!gtk_tree_model_iter_has_child(GTK_TREE_MODEL(roster_model),
295 gtk_tree_store_remove(roster_model, &group_iter);
296 [groupMap removeValueForKey: group];
299 gtk_tree_path_free(group_path);
303 - (void)rosterWasReceived: (XMPPRoster*)roster_
305 of_log(@"Handling roster");
306 [[roster_ rosterItems] enumerateKeysAndObjectsUsingBlock:
307 ^(OFString *bareJID, XMPPRosterItem *item, BOOL *stop) {
309 OFMapTable *contactRows = [OFMapTable
310 mapTableWithKeyFunctions: keyFunctions
311 valueFunctions: rowRefFunctions];
313 [contactMap setObject: contactRows
316 if (item.groups != nil)
317 groups = item.groups;
319 groups = @[@"General"];
321 for (OFString *group in groups)
322 [self Jub_addRosterItem: item
327 - (void)roster: (XMPPRoster*)roster_
328 didReceiveRosterItem: (XMPPRosterItem*)item
331 XMPPRosterItem *oldItem =
332 [roster_.rosterItems objectForKey: [item.JID bareJID]];
335 if (oldItem.groups != nil)
336 groups = oldItem.groups;
338 groups = @[@"General"];
340 for (OFString *group in groups)
341 [self Jub_removeRosterItem: oldItem
344 [contactMap removeObjectForKey: [item.JID bareJID]];
347 if (![item.subscription isEqual: @"remove"]) {
348 OFMapTable *contactRows = [OFMapTable
349 mapTableWithKeyFunctions: keyFunctions
350 valueFunctions: rowRefFunctions];
352 [contactMap setObject: contactRows
353 forKey: [item.JID bareJID]];
355 if (item.groups != nil)
356 groups = item.groups;
358 groups = @[@"General"];
360 for (OFString *group in groups)
361 [self Jub_addRosterItem: item
366 - (void)client: (JubChatClient*)client
367 didChangePresence: (XMPPPresence*)presence
369 OFString *tooltip = @"";
371 [[presence elementForName: @"show"
372 namespace: XMPP_NS_CLIENT] stringValue];
374 [[presence elementForName: @"status"
375 namespace: XMPP_NS_CLIENT] stringValue];
378 tooltip = [@"<b>Status:</b> " stringByAppendingString: status];
381 // Block the PresenceComboBox's changed handler, so it doesn't
382 // fire and resend presence
383 g_signal_handler_block(presence_combo,
384 presence_combo_changed_handler_id);
386 if ([presence.type isEqual: @"unavailable"])
387 gtk_combo_box_set_active_id(presence_combo,
389 else if (show == nil)
390 gtk_combo_box_set_active_id(presence_combo,
393 gtk_combo_box_set_active_id(presence_combo,
396 // Unblock the changed handler
397 g_signal_handler_unblock(presence_combo,
398 presence_combo_changed_handler_id);
400 gtk_widget_set_tooltip_markup(GTK_WIDGET(presence_combo),
401 [tooltip UTF8String]);