1 files changed, 114 insertions, 0 deletions
diff --git a/src/trg-model.c b/src/trg-model.c
new file mode 100644
index 0000000..fa67d08
--- /dev/null
+++ b/src/trg-model.c
@@ -0,0 +1,114 @@
+/*
+ * transmission-remote-gtk - A GTK RPC client to Transmission
+ * Copyright (C) 2011-2013  Alan Fitton
+
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include <glib.h>
+#include <gtk/gtk.h>
+#include <json-glib/json-glib.h>
+
+/* An extension of GtkListStore which provides some functions for looking up
+ * an entry by ID. Also for removing entries which have an old update serial,
+ * which means it needs removing.
+ */
+
+struct trg_model_remove_removed_foreachfunc_args {
+    gint64 currentSerial;
+    gint serial_column;
+    GList *toRemove;
+};
+
+gboolean
+trg_model_remove_removed_foreachfunc(GtkTreeModel * model,
+                                     GtkTreePath * path G_GNUC_UNUSED,
+                                     GtkTreeIter * iter, gpointer data)
+{
+    struct trg_model_remove_removed_foreachfunc_args *args =
+        (struct trg_model_remove_removed_foreachfunc_args *) data;
+    gint64 rowSerial;
+    gtk_tree_model_get(model, iter, args->serial_column, &rowSerial, -1);
+    if (rowSerial != args->currentSerial)
+        args->toRemove =
+            g_list_append(args->toRemove, gtk_tree_iter_copy(iter));
+
+    return FALSE;
+}
+
+guint
+trg_model_remove_removed(GtkListStore * model, gint serial_column,
+                         gint64 currentSerial)
+{
+    struct trg_model_remove_removed_foreachfunc_args args;
+    GList *li;
+    guint removed = 0;
+
+    args.toRemove = NULL;
+    args.currentSerial = currentSerial;
+    args.serial_column = serial_column;
+    gtk_tree_model_foreach(GTK_TREE_MODEL(model),
+                           trg_model_remove_removed_foreachfunc, &args);
+    if (args.toRemove != NULL) {
+        for (li = g_list_last(args.toRemove); li != NULL;
+             li = g_list_previous(li)) {
+            gtk_list_store_remove(model, (GtkTreeIter *) li->data);
+            gtk_tree_iter_free((GtkTreeIter *) li->data);
+            removed++;
+        }
+        g_list_free(args.toRemove);
+    }
+
+    return removed;
+}
+
+struct find_existing_item_foreach_args {
+    gint64 id;
+    gint search_column;
+    GtkTreeIter *iter;
+    gboolean found;
+};
+
+static gboolean
+find_existing_item_foreachfunc(GtkTreeModel * model,
+                               GtkTreePath * path G_GNUC_UNUSED,
+                               GtkTreeIter * iter, gpointer data)
+{
+    struct find_existing_item_foreach_args *args =
+        (struct find_existing_item_foreach_args *) data;
+    gint64 currentId;
+
+    gtk_tree_model_get(model, iter, args->search_column, &currentId, -1);
+    if (currentId == args->id) {
+        args->iter = iter;
+        return args->found = TRUE;
+    }
+
+    return FALSE;
+}
+
+gboolean
+find_existing_model_item(GtkTreeModel * model, gint search_column,
+                         gint64 id, GtkTreeIter * iter)
+{
+    struct find_existing_item_foreach_args args;
+    args.id = id;
+    args.found = FALSE;
+    args.search_column = search_column;
+    gtk_tree_model_foreach(model, find_existing_item_foreachfunc, &args);
+    if (args.found == TRUE)
+        *iter = *(args.iter);
+    return args.found;
+}