1 files changed, 99 insertions, 3 deletions

diff --git a/src/peerinfo/gnunet-peerinfo-gtk.c b/src/peerinfo/gnunet-peerinfo-gtk.c
index 12d66124..3971bee9 100644
--- a/src/peerinfo/gnunet-peerinfo-gtk.c
+++ b/src/peerinfo/gnunet-peerinfo-gtk.c
@@ -88,7 +88,12 @@ enum PEERINFO_ModelColumns
   /**
    * A gboolean
    */
-  PEERINFO_MC_IS_FRIEND = 9
+  PEERINFO_MC_IS_FRIEND = 9,
+
+  /**
+   * A `struct PeerInfo *`
+   */
+  PEERINFO_MC_PEERINFO = 10
 };
 
 
@@ -113,6 +118,11 @@ struct PeerInfo
   struct GNUNET_TRANSPORT_AddressToStringContext *tos;
 
   /**
+   * Identity of the peer for this entry.
+   */
+  struct GNUNET_PeerIdentity pid;
+
+  /**
    * Did we get any address?
    */
   int got_address;
@@ -379,6 +389,8 @@ peerinfo_processor (void *cls, const struct GNUNET_PeerIdentity *peer,
   info = GNUNET_CONTAINER_multipeermap_get (peer2info, peer);
   if (NULL == info)
   {
+    info = GNUNET_new (struct PeerInfo);
+    info->pid = *peer;
     gtk_list_store_append (ls, &iter);
     gtk_list_store_set (ls, &iter,
                         PEERINFO_MC_PEER_IDENTITY_STRING, GNUNET_i2s (peer),
@@ -392,9 +404,9 @@ peerinfo_processor (void *cls, const struct GNUNET_PeerIdentity *peer,
                         PEERINFO_MC_CONNECTED_STATUS, FALSE,
                         PEERINFO_MC_IS_FRIEND, GNUNET_CONTAINER_multipeermap_contains (friends,
                                                                                        peer),
+                        PEERINFO_MC_PEERINFO, info,
                         -1);
     path = gtk_tree_model_get_path (GTK_TREE_MODEL (ls), &iter);
-    info = GNUNET_malloc (sizeof (struct PeerInfo));
     info->rr = gtk_tree_row_reference_new (GTK_TREE_MODEL (ls), path);
     GNUNET_assert (NULL != info->rr);
     gtk_tree_path_free (path);
@@ -454,6 +466,51 @@ status_cb (void *cls,
 
 
 /**
+ * Write a friend to the friends file.
+ *
+ * @param cls the `struct GNUNET_FRIENDS_Writer`
+ * @param friend friend to write to file
+ * @param value unused
+ * @return #GNUNET_OK if the writing succeeded
+ */
+static int
+write_friend (void *cls,
+              const struct GNUNET_PeerIdentity *friend,
+              void *value)
+{
+  struct GNUNET_FRIENDS_Writer *w = cls;
+
+  return GNUNET_FRIENDS_write (w, friend);
+}
+
+
+/**
+ * Write an updated friends file out to disk.
+ */
+static void
+write_friends ()
+{
+  struct GNUNET_FRIENDS_Writer *w;
+
+  w = GNUNET_FRIENDS_write_start (get_configuration ());
+  if (NULL == w)
+  {
+    GNUNET_break (0);
+    return;
+  }
+  GNUNET_CONTAINER_multipeermap_iterate (friends,
+                                         &write_friend,
+                                         w);
+  if (GNUNET_OK !=
+      GNUNET_FRIENDS_write_stop (w))
+  {
+    GNUNET_break (0);
+    return;
+  }
+}
+
+
+/**
  * The user has toggled the 'is friend' column for one of the peers.
  * Update everything.
  *
@@ -466,7 +523,46 @@ GNUNET_PEERINFO_GTK_main_window_friends_cellrenderertoggle_toggled_cb (GtkCellRe
                                                                        gchar *path,
                                                                        gpointer user_data)
 {
-  GNUNET_break (0); // not implemented
+  GtkListStore *ls;
+  GtkTreeIter old;
+  struct PeerInfo *info;
+  gboolean oldvalue;
+
+  ls = GTK_LIST_STORE (get_object ("GNUNET_PEERINFO_GTK_list_store"));
+  if (NULL == ls)
+  {
+    GNUNET_break (0);
+    return;
+  }
+  if (TRUE !=
+      gtk_tree_model_get_iter_from_string (GTK_TREE_MODEL (ls), &old, path))
+  {
+    GNUNET_break (0);
+    return;
+  }
+  gtk_tree_model_get (GTK_TREE_MODEL (ls), &old,
+                      PEERINFO_MC_PEERINFO, &info,
+                      PEERINFO_MC_IS_FRIEND, &oldvalue,
+                      -1);
+  gtk_list_store_set (ls, &old,
+                      PEERINFO_MC_IS_FRIEND, ! oldvalue,
+                      -1);
+  if (oldvalue)
+  {
+    GNUNET_break (1 ==
+                  GNUNET_CONTAINER_multipeermap_remove_all (friends,
+                                                            &info->pid));
+  }
+  else
+  {
+    GNUNET_break (GNUNET_YES ==
+                  GNUNET_CONTAINER_multipeermap_put (friends,
+                                                     &info->pid,
+                                                     "true",
+                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
+
+  }
+  write_friends ();
 }