Aurelien's patch to add a set_server function to indicators.

1 files changed, 26 insertions, 3 deletions

diff --git a/libindicate/indicator.c b/libindicate/indicator.c
index cb257eb..18cdc4a 100644
--- a/libindicate/indicator.c
+++ b/libindicate/indicator.c
@@ -220,16 +220,39 @@ indicate_indicator_new_with_server (IndicateServer * server)
 
 	IndicateIndicator * indicator = g_object_new(INDICATE_TYPE_INDICATOR, NULL);
 
+	indicate_indicator_set_server (indicator, server);
+
+	return indicator;
+}
+
+/**
+	indicate_indicator_set_server:
+	@indicator: a #IndicateIndicator to act on
+	@server: the #IndicateServer which should be associated with @indicator
+
+	Defines which server this indicator must be associated to.  Also, adds a
+	reference to the server.
+*/
+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);
+	if (server != NULL) {
+		indicate_server_add_indicator (server, indicator);
+	}
 
-	return indicator;
+	return;
 }