diff options
Diffstat (limited to 'src/trg-stats-dialog.c')
| -rw-r--r-- | src/trg-stats-dialog.c | 376 |
1 files changed, 376 insertions, 0 deletions
diff --git a/src/trg-stats-dialog.c b/src/trg-stats-dialog.c new file mode 100644 index 0000000..3ae0336 --- /dev/null +++ b/src/trg-stats-dialog.c @@ -0,0 +1,376 @@ +/* + * 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/gi18n.h> +#include <gtk/gtk.h> +#include <json-glib/json-glib.h> +#include <curl/curl.h> + +#include "hig.h" +#include "requests.h" +#include "json.h" +#include "util.h" +#include "trg-client.h" +#include "trg-stats-dialog.h" +#include "trg-main-window.h" +#include "trg-tree-view.h" + +enum { + STATCOL_STAT, + STATCOL_SESSION, + STATCOL_CUMULAT, + STATCOL_COLUMNS +}; + +enum { + PROP_0, + PROP_PARENT, + PROP_CLIENT +}; + +#define STATS_UPDATE_INTERVAL 5 + +G_DEFINE_TYPE(TrgStatsDialog, trg_stats_dialog, GTK_TYPE_DIALOG) +#define TRG_STATS_DIALOG_GET_PRIVATE(o) \ + (G_TYPE_INSTANCE_GET_PRIVATE ((o), TRG_TYPE_STATS_DIALOG, TrgStatsDialogPrivate)) +typedef struct _TrgStatsDialogPrivate TrgStatsDialogPrivate; + +struct _TrgStatsDialogPrivate { + TrgClient *client; + TrgMainWindow *parent; + GtkWidget *tv; + GtkListStore *model; + GtkTreeRowReference *rr_up, *rr_down, *rr_files_added, + *rr_session_count, *rr_active, *rr_version; +}; + +static GObject *instance = NULL; +static gboolean trg_update_stats_timerfunc(gpointer data); +static gboolean on_stats_reply(gpointer data); + +static void +trg_stats_dialog_get_property(GObject * object, guint property_id, + GValue * value, GParamSpec * pspec) +{ + G_OBJECT_WARN_INVALID_PROPERTY_ID(object, property_id, pspec); +} + +static void +trg_stats_dialog_set_property(GObject * object, guint property_id, + const GValue * value, GParamSpec * pspec) +{ + TrgStatsDialogPrivate *priv = TRG_STATS_DIALOG_GET_PRIVATE(object); + switch (property_id) { + case PROP_CLIENT: + priv->client = g_value_get_pointer(value); + break; + case PROP_PARENT: + priv->parent = g_value_get_object(value); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID(object, property_id, pspec); + break; + } +} + +static void +trg_stats_response_cb(GtkDialog * dlg, gint res_id, + gpointer data G_GNUC_UNUSED) +{ + gtk_widget_destroy(GTK_WIDGET(dlg)); + instance = NULL; +} + +static GtkTreeRowReference *stats_dialog_add_statistic(GtkListStore * + model, gchar * name) +{ + GtkTreeIter iter; + GtkTreePath *path; + GtkTreeRowReference *rr; + + gtk_list_store_append(model, &iter); + gtk_list_store_set(model, &iter, STATCOL_STAT, name, -1); + path = gtk_tree_model_get_path(GTK_TREE_MODEL(model), &iter); + rr = gtk_tree_row_reference_new(GTK_TREE_MODEL(model), path); + + gtk_tree_path_free(path); + + return rr; +} + +static void +update_statistic(GtkTreeRowReference * rr, gchar * session, + gchar * cumulat) +{ + GtkTreePath *path = gtk_tree_row_reference_get_path(rr); + GtkTreeModel *model = gtk_tree_row_reference_get_model(rr); + GtkTreeIter iter; + + gtk_tree_model_get_iter(model, &iter, path); + + gtk_list_store_set(GTK_LIST_STORE(model), &iter, STATCOL_SESSION, + session, STATCOL_CUMULAT, cumulat, -1); + + gtk_tree_path_free(path); +} + +static JsonObject *get_session_arg(JsonObject * args) +{ + return json_object_get_object_member(args, "current-stats"); +} + +static JsonObject *get_cumulat_arg(JsonObject * args) +{ + return json_object_get_object_member(args, "cumulative-stats"); +} + +static void +update_int_stat(JsonObject * args, GtkTreeRowReference * rr, + gchar * jsonKey) +{ + gchar session_val[32]; + gchar cumulat_val[32]; + + g_snprintf(session_val, sizeof(session_val), "%" G_GINT64_FORMAT, + json_object_get_int_member(get_session_arg(args), jsonKey)); + g_snprintf(cumulat_val, sizeof(cumulat_val), "%" G_GINT64_FORMAT, + json_object_get_int_member(get_cumulat_arg(args), jsonKey)); + + update_statistic(rr, session_val, cumulat_val); +} + +static void +update_size_stat(JsonObject * args, GtkTreeRowReference * rr, + gchar * jsonKey) +{ + gchar session_val[32]; + gchar cumulat_val[32]; + + trg_strlsize(cumulat_val, + json_object_get_int_member(get_cumulat_arg(args), + jsonKey)); + trg_strlsize(session_val, + json_object_get_int_member(get_session_arg(args), + jsonKey)); + + update_statistic(rr, session_val, cumulat_val); +} + +static void +update_time_stat(JsonObject * args, GtkTreeRowReference * rr, + gchar * jsonKey) +{ + gchar session_val[32]; + gchar cumulat_val[32]; + + tr_strltime_long(session_val, + json_object_get_int_member(get_session_arg(args), + jsonKey), + sizeof(session_val)); + tr_strltime_long(cumulat_val, + json_object_get_int_member(get_cumulat_arg(args), + jsonKey), + sizeof(cumulat_val)); + + update_statistic(rr, session_val, cumulat_val); +} + +static gboolean on_stats_reply(gpointer data) +{ + trg_response *response = (trg_response *) data; + TrgStatsDialogPrivate *priv; + JsonObject *args; + char versionStr[32]; + + if (!TRG_IS_STATS_DIALOG(response->cb_data)) { + trg_response_free(response); + return FALSE; + } + + priv = TRG_STATS_DIALOG_GET_PRIVATE(response->cb_data); + + if (response->status == CURLE_OK) { + args = get_arguments(response->obj); + + g_snprintf(versionStr, sizeof(versionStr), "Transmission %s", + trg_client_get_version_string(priv->client)); + update_statistic(priv->rr_version, versionStr, ""); + + update_size_stat(args, priv->rr_up, "uploadedBytes"); + update_size_stat(args, priv->rr_down, "downloadedBytes"); + update_int_stat(args, priv->rr_files_added, "filesAdded"); + update_int_stat(args, priv->rr_session_count, "sessionCount"); + update_time_stat(args, priv->rr_active, "secondsActive"); + + if (trg_client_is_connected(priv->client)) + g_timeout_add_seconds(STATS_UPDATE_INTERVAL, + trg_update_stats_timerfunc, + response->cb_data); + } else { + trg_error_dialog(GTK_WINDOW(data), response); + } + + trg_response_free(response); + return FALSE; +} + +static gboolean trg_update_stats_timerfunc(gpointer data) +{ + TrgStatsDialogPrivate *priv; + + if (TRG_IS_STATS_DIALOG(data)) { + priv = TRG_STATS_DIALOG_GET_PRIVATE(data); + if (trg_client_is_connected(priv->client)) + dispatch_async(priv->client, session_stats(), on_stats_reply, + data); + } + + return FALSE; +} + +static void +trg_stats_add_column(GtkTreeView * tv, gint index, gchar * title, + gint width) +{ + GtkCellRenderer *renderer = gtk_cell_renderer_text_new(); + GtkTreeViewColumn *column = + gtk_tree_view_column_new_with_attributes(title, renderer, + "text", index, NULL); + gtk_tree_view_column_set_sizing(column, GTK_TREE_VIEW_COLUMN_FIXED); + gtk_tree_view_column_set_fixed_width(column, width); + + gtk_tree_view_append_column(tv, column); +} + +static GObject *trg_stats_dialog_constructor(GType type, + guint + n_construct_properties, + GObjectConstructParam * + construct_params) +{ + GtkWidget *tv; + + GObject *obj = G_OBJECT_CLASS + (trg_stats_dialog_parent_class)->constructor(type, + n_construct_properties, + construct_params); + TrgStatsDialogPrivate *priv = TRG_STATS_DIALOG_GET_PRIVATE(obj); + + gtk_window_set_title(GTK_WINDOW(obj), _("Statistics")); + gtk_window_set_transient_for(GTK_WINDOW(obj), + GTK_WINDOW(priv->parent)); + gtk_window_set_destroy_with_parent(GTK_WINDOW(obj), TRUE); + gtk_dialog_add_button(GTK_DIALOG(obj), GTK_STOCK_CLOSE, + GTK_RESPONSE_CLOSE); + + gtk_container_set_border_width(GTK_CONTAINER(obj), GUI_PAD); + + gtk_dialog_set_default_response(GTK_DIALOG(obj), GTK_RESPONSE_CLOSE); + + g_signal_connect(G_OBJECT(obj), + "response", G_CALLBACK(trg_stats_response_cb), NULL); + + priv->model = + gtk_list_store_new(STATCOL_COLUMNS, G_TYPE_STRING, G_TYPE_STRING, + G_TYPE_STRING); + + priv->rr_version = + stats_dialog_add_statistic(priv->model, _("Version")); + priv->rr_down = + stats_dialog_add_statistic(priv->model, _("Download Total")); + priv->rr_up = + stats_dialog_add_statistic(priv->model, _("Upload Total")); + priv->rr_files_added = + stats_dialog_add_statistic(priv->model, _("Files Added")); + priv->rr_session_count = + stats_dialog_add_statistic(priv->model, _("Session Count")); + priv->rr_active = + stats_dialog_add_statistic(priv->model, _("Time Active")); + + tv = priv->tv = trg_tree_view_new(); + gtk_widget_set_sensitive(tv, TRUE); + + trg_stats_add_column(GTK_TREE_VIEW(tv), STATCOL_STAT, _("Statistic"), + 200); + trg_stats_add_column(GTK_TREE_VIEW(tv), STATCOL_SESSION, _("Session"), + 200); + trg_stats_add_column(GTK_TREE_VIEW(tv), STATCOL_CUMULAT, + _("Cumulative"), 200); + + gtk_tree_view_set_model(GTK_TREE_VIEW(tv), + GTK_TREE_MODEL(priv->model)); + + gtk_container_set_border_width(GTK_CONTAINER(tv), GUI_PAD); + gtk_box_pack_start(GTK_BOX(gtk_bin_get_child(GTK_BIN(obj))), tv, TRUE, + TRUE, 0); + + dispatch_async(priv->client, session_stats(), on_stats_reply, obj); + + return obj; +} + +static void trg_stats_dialog_class_init(TrgStatsDialogClass * klass) +{ + GObjectClass *object_class = G_OBJECT_CLASS(klass); + + g_type_class_add_private(klass, sizeof(TrgStatsDialogPrivate)); + + object_class->get_property = trg_stats_dialog_get_property; + object_class->set_property = trg_stats_dialog_set_property; + object_class->constructor = trg_stats_dialog_constructor; + + g_object_class_install_property(object_class, + PROP_PARENT, + g_param_spec_object + ("parent-window", "Parent window", + "Parent window", + TRG_TYPE_MAIN_WINDOW, + G_PARAM_READWRITE | + G_PARAM_CONSTRUCT_ONLY | + G_PARAM_STATIC_NAME | + G_PARAM_STATIC_NICK | + G_PARAM_STATIC_BLURB)); + + g_object_class_install_property(object_class, + PROP_CLIENT, + g_param_spec_pointer + ("trg-client", "TClient", + "Client", + G_PARAM_READWRITE | + G_PARAM_CONSTRUCT_ONLY | + G_PARAM_STATIC_NAME | + G_PARAM_STATIC_NICK | + G_PARAM_STATIC_BLURB)); +} + +static void trg_stats_dialog_init(TrgStatsDialog * self) +{ +} + +TrgStatsDialog *trg_stats_dialog_get_instance(TrgMainWindow * parent, + TrgClient * client) +{ + if (instance == NULL) { + instance = g_object_new(TRG_TYPE_STATS_DIALOG, + "trg-client", client, + "parent-window", parent, NULL); + } + + return TRG_STATS_DIALOG(instance); +} |