Introduce indicate_indicator_set_server() so that server can be defined for

message indicators too.

2 files changed, 20 insertions, 4 deletions

diff --git a/libindicate/indicator.c b/libindicate/indicator.c
index cb257eb..a526ce7 100644
--- a/libindicate/indicator.c
+++ b/libindicate/indicator.c
@@ -220,16 +220,29 @@ indicate_indicator_new_with_server (IndicateServer * server)
 
 	IndicateIndicator * indicator = g_object_new(INDICATE_TYPE_INDICATOR, NULL);
 
+	indicate_indicator_set_server (indicator, server);
+
+	return indicator;
+}
+
+void
+indicate_indicator_set_server (IndicateIndicator * indicator, IndicateServer * server)
+{
 	IndicateIndicatorPrivate * priv = INDICATE_INDICATOR_GET_PRIVATE(indicator);
+
+	if (server != NULL) {
+		g_object_ref(server);
+	}
+
 	if (priv->server != NULL) {
+		indicate_server_remove_indicator (priv->server, indicator);
 		g_object_unref(priv->server);
-		priv->server = NULL;
 	}
 
 	priv->server = server;
-	g_object_ref(priv->server);
-
-	return indicator;
+	if (server != NULL) {
+		indicate_server_add_indicator (server, indicator);
+	}
 }
 
 
diff --git a/libindicate/indicator.h b/libindicate/indicator.h
index 20b998d..b2e33c0 100644
--- a/libindicate/indicator.h
+++ b/libindicate/indicator.h
@@ -119,6 +119,9 @@ void indicate_indicator_hide (IndicateIndicator * indicator);
 
 gboolean indicate_indicator_is_visible (IndicateIndicator * indicator);
 
+/* Define this indicator server */
+void indicate_indicator_set_server (IndicateIndicator * indicator, IndicateServer * server);
+
 /* Every entry has an ID, here's how to get it */
 guint indicate_indicator_get_id (IndicateIndicator * indicator);