hello world!

1 files changed, 369 insertions, 0 deletions

diff --git a/src/trg-files-tree-view.c b/src/trg-files-tree-view.c
new file mode 100644
index 0000000..5486a08
--- /dev/null
+++ b/src/trg-files-tree-view.c
@@ -0,0 +1,369 @@
+/*
+ * transmission-remote-gtk - Transmission RPC client for GTK
+ * Copyright (C) 2010  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 <gtk/gtk.h>
+
+#include "trg-client.h"
+#include "trg-tree-view.h"
+#include "trg-files-tree-view.h"
+#include "trg-files-model.h"
+#include "trg-cell-renderer-wanted.h"
+#include "trg-cell-renderer-priority.h"
+#include "trg-main-window.h"
+#include "requests.h"
+#include "json.h"
+#include "dispatch.h"
+#include "protocol-constants.h"
+
+G_DEFINE_TYPE(TrgFilesTreeView, trg_files_tree_view, TRG_TYPE_TREE_VIEW)
+#define TRG_FILES_TREE_VIEW_GET_PRIVATE(o) \
+  (G_TYPE_INSTANCE_GET_PRIVATE ((o), TRG_TYPE_FILES_TREE_VIEW, TrgFilesTreeViewPrivate))
+typedef struct _TrgFilesTreeViewPrivate TrgFilesTreeViewPrivate;
+
+struct _TrgFilesTreeViewPrivate {
+    trg_client *client;
+    TrgMainWindow *win;
+};
+
+static void trg_files_tree_view_class_init(TrgFilesTreeViewClass * klass)
+{
+    g_type_class_add_private(klass, sizeof(TrgFilesTreeViewPrivate));
+}
+
+static void set_unwanted_foreachfunc(GtkTreeModel * model,
+				     GtkTreePath * path,
+				     GtkTreeIter * iter, gpointer data)
+{
+    gtk_list_store_set(GTK_LIST_STORE(model), iter, FILESCOL_WANTED,
+		       FALSE, FILESCOL_ICON, GTK_STOCK_CANCEL, -1);
+}
+
+static void set_wanted_foreachfunc(GtkTreeModel * model,
+				   GtkTreePath * path, GtkTreeIter * iter,
+				   gpointer data)
+{
+    gtk_list_store_set(GTK_LIST_STORE(model), iter, FILESCOL_WANTED,
+		       TRUE, FILESCOL_ICON, GTK_STOCK_FILE, -1);
+}
+
+static void set_low_foreachfunc(GtkTreeModel * model, GtkTreePath * path,
+				GtkTreeIter * iter, gpointer data)
+{
+    gtk_list_store_set(GTK_LIST_STORE(model), iter, FILESCOL_PRIORITY,
+		       T_PRIORITY_LOW, -1);
+}
+
+static void set_normal_foreachfunc(GtkTreeModel * model,
+				   GtkTreePath * path, GtkTreeIter * iter,
+				   gpointer data)
+{
+    gtk_list_store_set(GTK_LIST_STORE(model), iter, FILESCOL_PRIORITY,
+		       T_PRIORITY_NORMAL, -1);
+}
+
+static void set_high_foreachfunc(GtkTreeModel * model, GtkTreePath * path,
+				 GtkTreeIter * iter, gpointer data)
+{
+    gtk_list_store_set(GTK_LIST_STORE(model), iter, FILESCOL_PRIORITY,
+		       T_PRIORITY_HIGH, -1);
+}
+
+static void add_file_id_to_array(JsonObject * args, gchar * key,
+				 gint index)
+{
+    JsonArray *array = json_object_get_array_member(args, key);
+    json_array_add_int_element(array, index);
+}
+
+static void send_updated_file_prefs_foreachfunc(GtkTreeModel * model,
+						GtkTreePath * path,
+						GtkTreeIter * iter,
+						gpointer data)
+{
+    JsonObject *args = (JsonObject *) data;
+    gint priority, id;
+    gboolean wanted;
+
+    gtk_tree_model_get(model, iter, FILESCOL_WANTED, &wanted,
+		       FILESCOL_PRIORITY, &priority, FILESCOL_ID, &id, -1);
+
+    if (wanted == FALSE)
+	add_file_id_to_array(args, FIELD_FILES_UNWANTED, id);
+    else
+	add_file_id_to_array(args, FIELD_FILES_WANTED, id);
+
+    if (priority == T_PRIORITY_LOW)
+	add_file_id_to_array(args, FIELD_FILES_PRIORITY_LOW, id);
+    else if (priority == T_PRIORITY_HIGH)
+	add_file_id_to_array(args, FIELD_FILES_PRIORITY_HIGH, id);
+    else
+	add_file_id_to_array(args, FIELD_FILES_PRIORITY_NORMAL, id);
+}
+
+static void remove_array_if_empty(JsonObject * args, gchar * key)
+{
+    JsonArray *array = json_object_get_array_member(args, key);
+    if (json_array_get_length(array) == 0)
+	json_object_remove_member(args, key);
+}
+
+static void send_updated_file_prefs(TrgFilesTreeView * tv)
+{
+    TrgFilesTreeViewPrivate *priv = TRG_FILES_TREE_VIEW_GET_PRIVATE(tv);
+    JsonNode *req;
+    JsonObject *args;
+    GtkTreeSelection *selection;
+    GtkTreeModel *model;
+    gint targetId;
+    JsonArray *targetIdArray;
+
+    selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(tv));
+    model = gtk_tree_view_get_model(GTK_TREE_VIEW(tv));
+    targetId = trg_files_model_get_torrent_id(TRG_FILES_MODEL(model));
+    targetIdArray = json_array_new();
+    json_array_add_int_element(targetIdArray, targetId);
+
+    req = torrent_set(targetIdArray);
+    args = node_get_arguments(req);
+
+    json_object_set_array_member(args, FIELD_FILES_WANTED,
+				 json_array_new());
+    json_object_set_array_member(args, FIELD_FILES_UNWANTED,
+				 json_array_new());
+    json_object_set_array_member(args, FIELD_FILES_PRIORITY_HIGH,
+				 json_array_new());
+    json_object_set_array_member(args, FIELD_FILES_PRIORITY_NORMAL,
+				 json_array_new());
+    json_object_set_array_member(args, FIELD_FILES_PRIORITY_LOW,
+				 json_array_new());
+
+    gtk_tree_selection_selected_foreach(selection,
+					send_updated_file_prefs_foreachfunc,
+					args);
+
+    remove_array_if_empty(args, FIELD_FILES_WANTED);
+    remove_array_if_empty(args, FIELD_FILES_UNWANTED);
+    remove_array_if_empty(args, FIELD_FILES_PRIORITY_HIGH);
+    remove_array_if_empty(args, FIELD_FILES_PRIORITY_NORMAL);
+    remove_array_if_empty(args, FIELD_FILES_PRIORITY_LOW);
+
+    trg_files_model_set_update_barrier(TRG_FILES_MODEL(model),
+				       priv->client->updateSerial + 1);
+    dispatch_async(priv->client, req, on_generic_interactive_action,
+		   priv->win);
+}
+
+static void set_low(GtkWidget * w, gpointer data)
+{
+    TrgFilesTreeView *tv = TRG_FILES_TREE_VIEW(data);
+    GtkTreeSelection *selection =
+	gtk_tree_view_get_selection(GTK_TREE_VIEW(data));
+    gtk_tree_selection_selected_foreach(selection, set_low_foreachfunc,
+					NULL);
+    send_updated_file_prefs(tv);
+}
+
+static void set_normal(GtkWidget * w, gpointer data)
+{
+    TrgFilesTreeView *tv = TRG_FILES_TREE_VIEW(data);
+    GtkTreeSelection *selection =
+	gtk_tree_view_get_selection(GTK_TREE_VIEW(data));
+    gtk_tree_selection_selected_foreach(selection,
+					set_normal_foreachfunc, NULL);
+    send_updated_file_prefs(tv);
+}
+
+static void set_high(GtkWidget * w, gpointer data)
+{
+    TrgFilesTreeView *tv = TRG_FILES_TREE_VIEW(data);
+    GtkTreeSelection *selection =
+	gtk_tree_view_get_selection(GTK_TREE_VIEW(data));
+    gtk_tree_selection_selected_foreach(selection,
+					set_high_foreachfunc, NULL);
+    send_updated_file_prefs(tv);
+}
+
+static void set_unwanted(GtkWidget * w, gpointer data)
+{
+    TrgFilesTreeView *tv = TRG_FILES_TREE_VIEW(data);
+    GtkTreeSelection *selection =
+	gtk_tree_view_get_selection(GTK_TREE_VIEW(data));
+    gtk_tree_selection_selected_foreach(selection,
+					set_unwanted_foreachfunc, NULL);
+    send_updated_file_prefs(tv);
+}
+
+static void set_wanted(GtkWidget * w, gpointer data)
+{
+    TrgFilesTreeView *tv = TRG_FILES_TREE_VIEW(data);
+    GtkTreeSelection *selection =
+	gtk_tree_view_get_selection(GTK_TREE_VIEW(data));
+    gtk_tree_selection_selected_foreach(selection,
+					set_wanted_foreachfunc, NULL);
+    send_updated_file_prefs(tv);
+}
+
+static void
+view_popup_menu(GtkWidget * treeview, GdkEventButton * event,
+		gpointer data)
+{
+    GtkWidget *menu, *menuitem;
+
+    menu = gtk_menu_new();
+
+    menuitem = gtk_menu_item_new_with_label("High Priority");
+    g_signal_connect(menuitem, "activate", G_CALLBACK(set_high), treeview);
+    gtk_menu_shell_append(GTK_MENU_SHELL(menu), menuitem);
+
+    menuitem = gtk_menu_item_new_with_label("Normal Priority");
+    g_signal_connect(menuitem, "activate", G_CALLBACK(set_normal),
+		     treeview);
+    gtk_menu_shell_append(GTK_MENU_SHELL(menu), menuitem);
+
+    menuitem = gtk_menu_item_new_with_label("Low Priority");
+    g_signal_connect(menuitem, "activate", G_CALLBACK(set_low), treeview);
+    gtk_menu_shell_append(GTK_MENU_SHELL(menu), menuitem);
+
+    gtk_menu_shell_append(GTK_MENU_SHELL(menu),
+			  gtk_separator_menu_item_new());
+
+    menuitem = gtk_image_menu_item_new_with_label(GTK_STOCK_FILE);
+    gtk_image_menu_item_set_use_stock(GTK_IMAGE_MENU_ITEM(menuitem), TRUE);
+    gtk_image_menu_item_set_always_show_image(GTK_IMAGE_MENU_ITEM
+					      (menuitem), TRUE);
+    gtk_menu_item_set_label(GTK_MENU_ITEM(menuitem), "Download");
+    g_signal_connect(menuitem, "activate", G_CALLBACK(set_wanted),
+		     treeview);
+    gtk_menu_shell_append(GTK_MENU_SHELL(menu), menuitem);
+
+    menuitem = gtk_image_menu_item_new_with_label(GTK_STOCK_STOP);
+    gtk_image_menu_item_set_use_stock(GTK_IMAGE_MENU_ITEM(menuitem), TRUE);
+    gtk_image_menu_item_set_always_show_image(GTK_IMAGE_MENU_ITEM
+					      (menuitem), TRUE);
+    gtk_menu_item_set_label(GTK_MENU_ITEM(menuitem), "Skip");
+    g_signal_connect(menuitem, "activate", G_CALLBACK(set_unwanted),
+		     treeview);
+    gtk_menu_shell_append(GTK_MENU_SHELL(menu), menuitem);
+
+    gtk_widget_show_all(menu);
+
+    gtk_menu_popup(GTK_MENU(menu), NULL, NULL, NULL, NULL,
+		   (event != NULL) ? event->button : 0,
+		   gdk_event_get_time((GdkEvent *) event));
+}
+
+static gboolean
+view_onButtonPressed(GtkWidget * treeview, GdkEventButton * event,
+		     gpointer userdata)
+{
+    GtkTreeSelection *selection;
+    GtkTreePath *path;
+
+    if (event->type == GDK_BUTTON_PRESS && event->button == 3) {
+	selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(treeview));
+
+	if (gtk_tree_view_get_path_at_pos(GTK_TREE_VIEW(treeview),
+					  (gint) event->x,
+					  (gint) event->y, &path,
+					  NULL, NULL, NULL)) {
+	    if (!gtk_tree_selection_path_is_selected(selection, path)) {
+		gtk_tree_selection_unselect_all(selection);
+		gtk_tree_selection_select_path(selection, path);
+	    }
+	    gtk_tree_path_free(path);
+
+	    view_popup_menu(treeview, event, userdata);
+	    return TRUE;
+	}
+    }
+
+    return FALSE;
+}
+
+static gboolean view_onPopupMenu(GtkWidget * treeview, gpointer userdata)
+{
+    view_popup_menu(treeview, NULL, userdata);
+    return TRUE;
+}
+
+static void trg_files_tree_view_add_priority_column(TrgTreeView * tv,
+						    char *title, int index,
+						    int width)
+{
+    GtkCellRenderer *renderer;
+    GtkTreeViewColumn *column;
+
+    renderer = trg_cell_renderer_priority_new();
+    column = gtk_tree_view_column_new_with_attributes(title, renderer,
+						      "priority-value",
+						      index, NULL);
+
+    trg_tree_view_std_column_setup(column, index, width);
+    gtk_tree_view_append_column(GTK_TREE_VIEW(tv), column);
+}
+
+static void trg_files_tree_view_add_wanted_column(TrgTreeView * tv,
+						  char *title, int index,
+						  int width)
+{
+    GtkCellRenderer *renderer;
+    GtkTreeViewColumn *column;
+
+    renderer = trg_cell_renderer_wanted_new();
+    column = gtk_tree_view_column_new_with_attributes(title, renderer,
+						      "wanted-value",
+						      index, NULL);
+
+    trg_tree_view_std_column_setup(column, index, width);
+    gtk_tree_view_append_column(GTK_TREE_VIEW(tv), column);
+}
+
+static void trg_files_tree_view_init(TrgFilesTreeView * self)
+{
+    trg_tree_view_add_pixbuf_text_column(TRG_TREE_VIEW(self),
+					 FILESCOL_ICON, FILESCOL_NAME,
+					 "Name", -1);
+    trg_tree_view_add_column(TRG_TREE_VIEW(self), "Size", FILESCOL_SIZE);
+    trg_tree_view_add_prog_column(TRG_TREE_VIEW(self), "Progress",
+				  FILESCOL_PROGRESS, -1);
+    trg_files_tree_view_add_wanted_column(TRG_TREE_VIEW(self),
+					  "Wanted", FILESCOL_WANTED, -1);
+    trg_files_tree_view_add_priority_column(TRG_TREE_VIEW(self),
+					    "Priority",
+					    FILESCOL_PRIORITY, -1);
+
+    g_signal_connect(self, "button-press-event",
+		     G_CALLBACK(view_onButtonPressed), NULL);
+    g_signal_connect(self, "popup-menu", G_CALLBACK(view_onPopupMenu),
+		     NULL);
+}
+
+TrgFilesTreeView *trg_files_tree_view_new(TrgFilesModel * model,
+					  TrgMainWindow * win,
+					  trg_client * client)
+{
+    GObject *obj = g_object_new(TRG_TYPE_FILES_TREE_VIEW, NULL);
+    TrgFilesTreeViewPrivate *priv = TRG_FILES_TREE_VIEW_GET_PRIVATE(obj);
+
+    gtk_tree_view_set_model(GTK_TREE_VIEW(obj), GTK_TREE_MODEL(model));
+    priv->client = client;
+    priv->win = win;
+
+    return TRG_FILES_TREE_VIEW(obj);
+}