adding search result rankings to model

4 files changed, 51 insertions, 4 deletions

diff --git a/src/plugins/fs/fs.c b/src/plugins/fs/fs.c
index c14113d0..24550810 100644
--- a/src/plugins/fs/fs.c
+++ b/src/plugins/fs/fs.c
@@ -122,6 +122,7 @@ saveEventProcessor (void *cls)
 {
   const GNUNET_FSUI_Event *event = cls;
   void *ret;
+  unsigned int i;
 
   ret = NULL;
   switch (event->type)
@@ -157,6 +158,12 @@ saveEventProcessor (void *cls)
                                event->data.SearchResumed.fisSize,
                                event->data.SearchResumed.fis,
                                event->data.SearchResumed.state);
+      for (i=0;i<event->data.SearchResumed.fisSize;i++)
+        fs_search_update(ret,
+                         &event->data.SearchResumed.fis[i],
+                         event->data.SearchResumed.availability_rank[i],
+                         event->data.SearchResumed.availability_certainty[i],
+                         event->data.SearchResumed.applicability_rank[i]);
       break;
     case GNUNET_FSUI_search_stopped:
       fs_search_stopped (event->data.SearchStopped.sc.cctx);
@@ -165,6 +172,7 @@ saveEventProcessor (void *cls)
       fs_search_update (event->data.SearchUpdate.sc.cctx,
                         &event->data.SearchUpdate.fi,
                         event->data.SearchUpdate.availability_rank,
+                        event->data.SearchUpdate.availability_certainty,
                         event->data.SearchUpdate.applicability_rank);
       break;
       /* download events */
diff --git a/src/plugins/fs/fs.h b/src/plugins/fs/fs.h
index e41aefe9..62262d34 100644
--- a/src/plugins/fs/fs.h
+++ b/src/plugins/fs/fs.h
@@ -64,6 +64,9 @@ enum
   SEARCH_INTERNAL,
   SEARCH_INTERNAL_PARENT,
   SEARCH_STATUS,
+  SEARCH_AVAILABILITY_RANK,
+  SEARCH_AVAILABILITY_CERTAINTY,
+  SEARCH_APPLICABILITY_RANK,
   SEARCH_NUM,
 };
 
diff --git a/src/plugins/fs/search.c b/src/plugins/fs/search.c
index 9628a6bb..1b33b5af 100644
--- a/src/plugins/fs/search.c
+++ b/src/plugins/fs/search.c
@@ -172,7 +172,7 @@ fs_search_result_received (SearchList * searchContext,
       (GNUNET_YES == GNUNET_GC_get_configuration_value_yesno (cfg,
                                                               "GNUNET-GTK",
                                                               "DISABLE-OWN",
-                                                              GNUNET_YES)))
+                                                              GNUNET_NO)))
     return;
   model = GTK_TREE_STORE (gtk_tree_view_get_model (searchContext->treeview));
   /* Check that the entry does not already exist (for resume!) */
@@ -205,9 +205,39 @@ fs_search_result_received (SearchList * searchContext,
 void fs_search_update (SearchList * searchContext,
                        const GNUNET_ECRS_FileInfo * info,
                        int availability_rank,
+                       unsigned int availability_certainty,
                        unsigned int applicability_rank)
 {
-  /* FIXME */
+  GtkTreeStore *model;
+  GtkTreeIter iter;
+  struct GNUNET_ECRS_URI *have;
+
+  model = GTK_TREE_STORE (gtk_tree_view_get_model (searchContext->treeview));
+  /* find existing entry */
+  if (gtk_tree_model_get_iter_first (GTK_TREE_MODEL (model), &iter))
+    {
+      do
+        {
+          have = NULL;
+          gtk_tree_model_get (GTK_TREE_MODEL (model),
+                              &iter, SEARCH_URI, &have, -1);
+          if ( (have != NULL) && 
+               (GNUNET_ECRS_uri_test_equal (have, info->uri)) )
+            {
+              /* gotcha! */
+              gtk_tree_store_set(searchContext->tree,
+                                 &iter,
+                                 SEARCH_AVAILABILITY_RANK, availability_rank,
+                                 SEARCH_AVAILABILITY_CERTAINTY, availability_certainty,
+                                 SEARCH_APPLICABILITY_RANK, applicability_rank,
+                                 -1);
+              return; /* done! */
+            }
+        }
+      while (gtk_tree_model_iter_next (GTK_TREE_MODEL (model), &iter));
+    }
+  /* not found!? */
+  GNUNET_GE_BREAK(NULL, 0);
 }
 
 
@@ -466,7 +496,8 @@ fs_search_started (struct GNUNET_FSUI_SearchList * fsui_list,
   g_signal_connect_swapped (list->treeview,
                             "button-press-event",
                             G_CALLBACK (search_click_handler), list);
-  list->tree = gtk_tree_store_new (SEARCH_NUM, G_TYPE_STRING,   /* name */
+  list->tree = gtk_tree_store_new (SEARCH_NUM,
+                                   G_TYPE_STRING,       /* name */
                                    G_TYPE_UINT64,       /* size */
                                    G_TYPE_STRING,       /* human-readable size */
                                    G_TYPE_STRING,       /* mime-type */
@@ -478,7 +509,11 @@ fs_search_started (struct GNUNET_FSUI_SearchList * fsui_list,
                                    G_TYPE_STRING,       /* fg-color */
                                    G_TYPE_POINTER,      /* internal: search list */
                                    G_TYPE_POINTER,      /* internal: download parent list */
-                                   G_TYPE_STRING);      /* status */
+                                   G_TYPE_STRING,       /* status */
+                                   G_TYPE_INT,          /* availability rank */
+                                   G_TYPE_UINT,         /* availability certainty */
+                                   G_TYPE_UINT          /* applicability rank */
+                                   );     
 
   gtk_tree_view_set_model (list->treeview, GTK_TREE_MODEL (list->tree));
   gtk_tree_selection_set_mode (gtk_tree_view_get_selection (list->treeview),
diff --git a/src/plugins/fs/search.h b/src/plugins/fs/search.h
index bf280397..546c7bfd 100644
--- a/src/plugins/fs/search.h
+++ b/src/plugins/fs/search.h
@@ -64,6 +64,7 @@ void fs_search_result_received (SearchList * searchContext,
 void fs_search_update (SearchList * searchContext,
                        const GNUNET_ECRS_FileInfo * info,
                        int availability_rank,
+                       unsigned int availability_certainty,
                        unsigned int applicability_rank);
 
 /**