1 files changed, 76 insertions, 35 deletions

diff --git a/src/peerinfo/gnunet-peerinfo-gtk.c b/src/peerinfo/gnunet-peerinfo-gtk.c
index 6d8569e8..85df3f3b 100644
--- a/src/peerinfo/gnunet-peerinfo-gtk.c
+++ b/src/peerinfo/gnunet-peerinfo-gtk.c
@@ -68,7 +68,17 @@ enum PEERINFO_ModelColumns
     /**
      * A gchararray
      */
-    PEERINFO_MC_ADDRESS_AS_STRING = 6
+    PEERINFO_MC_ADDRESS_AS_STRING = 6,
+
+    /**
+     * A GdkPixbuf
+     */
+    PEERINFO_MC_CONNECTIVITY_LED = 7,
+
+    /**
+     * A gboolean
+     */
+    PEERINFO_MC_CONNECTED_STATUS = 8
   };
 
 
@@ -124,6 +134,22 @@ static struct GNUNET_CONTAINER_MultiHashMap *peer2info;
  */
 static int tray_only;
 
+/**
+ * Green status led (connected)
+ */
+static GdkPixbuf *led_green;
+
+/**
+ * Red status led (disconnected)
+ */
+static GdkPixbuf *led_red;
+
+/**
+ * Main window list store.
+ */
+static GtkListStore *ls;
+
+
 
 /**
  * Get cfg.
@@ -212,19 +238,15 @@ static void
 peer_address_string_cb (void *cls, const char *address)
 {
   struct PeerInfo *info = cls;
-  GtkListStore *ls;
-  GtkTreeModel *tm;
   GtkTreeIter iter;
   GtkTreePath *path;
   char *country;
   const char *colon;
   const char *dot;
 
-  ls = GTK_LIST_STORE (get_object ("GNUNET_PEERINFO_GTK_list_store"));
-  tm = GTK_TREE_MODEL (ls);
   path = gtk_tree_row_reference_get_path (info->rr);
   GNUNET_assert (NULL != path);
-  GNUNET_assert (TRUE == gtk_tree_model_get_iter (tm, &iter, path));
+  GNUNET_assert (TRUE == gtk_tree_model_get_iter (GTK_TREE_MODEL (ls), &iter, path));
   gtk_tree_path_free (path);
   if (NULL == address)
   {
@@ -235,6 +257,8 @@ peer_address_string_cb (void *cls, const char *address)
                           PEERINFO_MC_COUNTRY_NAME, NULL, 
                           PEERINFO_MC_COUNTRY_FLAG, NULL, 
                           PEERINFO_MC_ADDRESS_AS_STRING, "<no address>", 
+                          PEERINFO_MC_CONNECTIVITY_LED, led_green,
+                          PEERINFO_MC_CONNECTED_STATUS, TRUE,
                           -1);
     info->tos = NULL;
     return;
@@ -255,6 +279,8 @@ peer_address_string_cb (void *cls, const char *address)
                       PEERINFO_MC_COUNTRY_NAME, country, 
                       PEERINFO_MC_COUNTRY_FLAG, GNUNET_PEERINFO_GTK_get_flag (country), 
                       PEERINFO_MC_ADDRESS_AS_STRING, address, 
+                      PEERINFO_MC_CONNECTIVITY_LED, led_green,
+                      PEERINFO_MC_CONNECTED_STATUS, TRUE,
                       -1);
   GNUNET_free (country);
   info->got_address = GNUNET_YES;
@@ -273,28 +299,28 @@ peer_address_cb (void *cls, const struct GNUNET_PeerIdentity *peer,
                  const struct GNUNET_HELLO_Address *address)
 {
   struct PeerInfo *info = cls;
-  GtkListStore *ls;
-  GtkTreeModel *tm;
   GtkTreeIter iter;
   GtkTreePath *path;
 
+  path = gtk_tree_row_reference_get_path (info->rr);
+  GNUNET_assert (NULL != path);
+  GNUNET_assert (TRUE == gtk_tree_model_get_iter (GTK_TREE_MODEL (ls), &iter, path));
+  gtk_tree_path_free (path);
   if (NULL == address)
   {
     /* disconnect */
-    ls = GTK_LIST_STORE (get_object ("GNUNET_PEERINFO_GTK_list_store"));
-    tm = GTK_TREE_MODEL (ls);
-    path = gtk_tree_row_reference_get_path (info->rr);
-    GNUNET_assert (NULL != path);
-    GNUNET_assert (TRUE == gtk_tree_model_get_iter (tm, &iter, path));
-    gtk_tree_path_free (path);
     gtk_list_store_set (ls, &iter, 
                         PEERINFO_MC_NUMBER_OF_ADDRESSES, (guint) 0,
-                        PEERINFO_MC_COUNTRY_NAME, NULL, 
-                        PEERINFO_MC_COUNTRY_FLAG, NULL, 
-                        PEERINFO_MC_ADDRESS_AS_STRING, "<disconnected>",
+                        PEERINFO_MC_CONNECTIVITY_LED, led_red,
+                        PEERINFO_MC_CONNECTED_STATUS, FALSE,
                         -1);
     return;
   }
+  gtk_list_store_set (ls, &iter, 
+                      PEERINFO_MC_NUMBER_OF_ADDRESSES, 1,
+                      PEERINFO_MC_CONNECTIVITY_LED, led_green,
+                      PEERINFO_MC_CONNECTED_STATUS, TRUE,
+                      -1);
   if (NULL != info->tos)
     GNUNET_TRANSPORT_address_to_string_cancel (info->tos);
   info->got_address = GNUNET_NO;
@@ -319,21 +345,12 @@ peerinfo_processor (void *cls, const struct GNUNET_PeerIdentity *peer,
                     const struct GNUNET_HELLO_Message *hello,
                     const char *err_msg)
 {
-  GtkListStore *ls;
-  GtkTreeModel *tm;
   GtkTreeIter iter;
   char *npid;
   struct GNUNET_CRYPTO_HashAsciiEncoded enc;
   struct PeerInfo *info;
   GtkTreePath *path;
 
-  ls = GTK_LIST_STORE (get_object ("GNUNET_PEERINFO_GTK_list_store"));
-  if (NULL == ls)
-  {
-    GNUNET_break (0);
-    return;
-  }
-  tm = GTK_TREE_MODEL (ls);
   info = GNUNET_CONTAINER_multihashmap_get (peer2info, &peer->hashPubKey);
   if (NULL == info)
   {
@@ -348,11 +365,13 @@ peerinfo_processor (void *cls, const struct GNUNET_PeerIdentity *peer,
                         PEERINFO_MC_COUNTRY_FLAG, NULL,
                         PEERINFO_MC_BANDWIDTH_IN, (guint64) 0,
                         PEERINFO_MC_BANDWIDTH_OUT, (guint64) 0,
-                        PEERINFO_MC_ADDRESS_AS_STRING, "<disconnected>",
+                        PEERINFO_MC_ADDRESS_AS_STRING, "",
+                        PEERINFO_MC_CONNECTIVITY_LED, led_red,
+                        PEERINFO_MC_CONNECTED_STATUS, FALSE,
                         -1);
-    path = gtk_tree_model_get_path (tm, &iter);
+    path = gtk_tree_model_get_path (GTK_TREE_MODEL (ls), &iter);
     info = GNUNET_malloc (sizeof (struct PeerInfo));
-    info->rr = gtk_tree_row_reference_new (tm, path);
+    info->rr = gtk_tree_row_reference_new (GTK_TREE_MODEL (ls), path);
     GNUNET_assert (NULL != info->rr);
     gtk_tree_path_free (path);
     GNUNET_CONTAINER_multihashmap_put (peer2info, &peer->hashPubKey, info,
@@ -388,19 +407,15 @@ status_cb (void *cls,
            uint32_t ats_count)
 {
   struct PeerInfo *info;
-  GtkListStore *ls;
-  GtkTreeModel *tm;
   GtkTreeIter iter;
   GtkTreePath *path;
   
   info = GNUNET_CONTAINER_multihashmap_get (peer2info, &address->peer.hashPubKey);
   if (NULL == info)
     return;                     /* should rarely happen... */
-  ls = GTK_LIST_STORE (get_object ("GNUNET_PEERINFO_GTK_list_store"));
-  tm = GTK_TREE_MODEL (ls);
   path = gtk_tree_row_reference_get_path (info->rr);
   GNUNET_assert (NULL != path);
-  GNUNET_assert (TRUE == gtk_tree_model_get_iter (tm, &iter, path));
+  GNUNET_assert (TRUE == gtk_tree_model_get_iter (GTK_TREE_MODEL (ls), &iter, path));
   gtk_tree_path_free (path);
   gtk_list_store_set (ls, &iter,
                       PEERINFO_MC_BANDWIDTH_IN, (guint64) ntohl (bandwidth_in.value__),
@@ -425,6 +440,29 @@ GNUNET_PEERINFO_GTK_quit_cb (GObject * object, gpointer user_data)
 
 
 /**
+ * Load LED image from resource file.
+ *
+ * @param color color of the LED to load
+ * @return the image as a GdkPixbuf
+ */
+static GdkPixbuf *
+load_led (const char *color)
+{
+  GdkPixbuf *pixbuf;
+  char *dir;
+  char *fn;
+
+  dir = GNUNET_GTK_installation_get_path (GNUNET_OS_IPK_DATADIR);
+  GNUNET_asprintf (&fn, "%s%s.png", dir, 
+                   color);
+  GNUNET_free (dir);
+  pixbuf = gdk_pixbuf_new_from_file (fn, NULL);
+  GNUNET_free (fn);
+  return pixbuf;
+}
+
+
+/**
  * Actual main function run right after GNUnet's scheduler
  * is initialized.  Initializes up GTK and Glade.
  *
@@ -442,7 +480,8 @@ run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   {
     return;
   }
-
+  led_green = load_led ("green");
+  led_red = load_led ("red");
   GNUNET_GTK_set_icon_search_path ();
   GNUNET_GTK_setup_nls ();
   peer2info = GNUNET_CONTAINER_multihashmap_create (256);
@@ -457,6 +496,8 @@ run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   ats = GNUNET_ATS_performance_init (get_configuration (), &status_cb, NULL);
   /* setup main window */
   main_window = GTK_WIDGET (get_object ("GNUNET_PEERINFO_GTK_main_window"));
+  ls = GTK_LIST_STORE (get_object ("GNUNET_PEERINFO_GTK_list_store"));
+  GNUNET_assert (NULL != ls);
   gtk_window_maximize (GTK_WINDOW (main_window));
   GNUNET_GTK_tray_icon_create (GTK_WINDOW (main_window),
                                "gnunet-gtk" /* FIXME: different icon? */ ,