path: root/src/torrent.c

/*
 * transmission-remote-gtk - A GTK RPC client to Transmission
 * Copyright (C) 2011  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-object.h>
#include <glib/gi18n.h>
#include <gtk/gtk.h>
#include <json-glib/json-glib.h>

#include "trg-client.h"
#include "torrent.h"
#include "protocol-constants.h"
#include "util.h"

JsonArray *torrent_get_peers(JsonObject * t)
{
    return json_object_get_array_member(t, FIELD_PEERS);
}

JsonObject *torrent_get_peersfrom(JsonObject * t)
{
    return json_object_get_object_member(t, FIELD_PEERSFROM);
}

JsonArray *torrent_get_wanted(JsonObject * t)
{
    return json_object_get_array_member(t, FIELD_WANTED);
}

JsonArray *torrent_get_priorities(JsonObject * t)
{
    return json_object_get_array_member(t, FIELD_PRIORITIES);
}

JsonArray *torrent_get_tracker_stats(JsonObject * t)
{
    return json_object_get_array_member(t, FIELD_TRACKER_STATS);
}

gint64 torrent_get_id(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_ID);
}

const gchar *torrent_get_download_dir(JsonObject * t)
{
    return json_object_get_string_member(t, FIELD_DOWNLOAD_DIR);
}

const gchar *torrent_get_name(JsonObject * t)
{
    return json_object_get_string_member(t, FIELD_NAME);
}

gint64 torrent_get_added_date(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_ADDED_DATE);
}

gboolean torrent_get_honors_session_limits(JsonObject * t)
{
    return json_object_get_boolean_member(t, FIELD_HONORS_SESSION_LIMITS);
}

gint64 torrent_get_bandwidth_priority(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_BANDWIDTH_PRIORITY);
}

gint64 torrent_get_upload_limit(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_UPLOAD_LIMIT);
}

gint64 torrent_get_peer_limit(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_PEER_LIMIT);
}

gboolean torrent_get_upload_limited(JsonObject * t)
{
    return json_object_get_boolean_member(t, FIELD_UPLOAD_LIMITED);
}

gint64 torrent_get_seed_ratio_mode(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_SEED_RATIO_MODE);
}

gdouble torrent_get_seed_ratio_limit(JsonObject * t)
{
    return json_object_get_double_member(t, FIELD_SEED_RATIO_LIMIT);
}

gint64 torrent_get_download_limit(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_DOWNLOAD_LIMIT);
}

gboolean torrent_get_download_limited(JsonObject * t)
{
    return json_object_get_boolean_member(t, FIELD_DOWNLOAD_LIMITED);
}

gint64 torrent_get_size(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_SIZEWHENDONE);
}

gint64 torrent_get_rate_down(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_RATEDOWNLOAD);
}

gint64 torrent_get_rate_up(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_RATEUPLOAD);
}

gint64 torrent_get_eta(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_ETA);
}

gint64 torrent_get_uploaded(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_UPLOADEDEVER);
}

gint64 torrent_get_have_valid(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_HAVEVALID);
}

gint64 torrent_get_have_unchecked(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_HAVEUNCHECKED);
}

gint64 torrent_get_downloaded(JsonObject * t)
{
    return torrent_get_have_valid(t) + torrent_get_have_unchecked(t);
}

gint64 torrent_get_status(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_STATUS);
}

gboolean torrent_get_is_finished(JsonObject * t)
{
    return torrent_get_left_until_done(t) <= 0;
}

gdouble torrent_get_percent_done(JsonObject * t)
{
    JsonNode *percentDone = json_object_get_member(t, FIELD_PERCENTDONE);
    GValue a = { 0 };
    json_node_get_value(percentDone, &a);
    switch (G_VALUE_TYPE(&a)) {
    case G_TYPE_INT64:
        return (gdouble) g_value_get_int64(&a) * 100.0;
    case G_TYPE_DOUBLE:
        return g_value_get_double(&a) * 100.0;
    default:
        return 0.0;
    }
}

guint32 torrent_get_flags(JsonObject * t, gint64 rpcv, gint64 status, gint64 downRate, gint64 upRate)
{
    guint32 flags = 0;
    if (rpcv >= NEW_STATUS_RPC_VERSION) {
        switch (status) {
        case TR_STATUS_STOPPED:
            flags |= TORRENT_FLAG_PAUSED;
            break;
        case TR_STATUS_CHECK_WAIT:
            flags |= TORRENT_FLAG_WAITING_CHECK;
            flags |= TORRENT_FLAG_CHECKING;
            break;
        case TR_STATUS_CHECK:
            flags |= TORRENT_FLAG_CHECKING;
            break;
        case TR_STATUS_DOWNLOAD_WAIT:
            flags |= TORRENT_FLAG_DOWNLOADING_WAIT;
            flags |= TORRENT_FLAG_QUEUED;
            break;
        case TR_STATUS_DOWNLOAD:
            flags |= TORRENT_FLAG_DOWNLOADING;
            break;
        case TR_STATUS_SEED_WAIT:
            flags |= TORRENT_FLAG_SEEDING_WAIT;
            break;
        case TR_STATUS_SEED:
            flags |= TORRENT_FLAG_SEEDING;
            break;
        }
    } else {
        switch (status) {
        case OLD_STATUS_DOWNLOADING:
            flags |= TORRENT_FLAG_DOWNLOADING;
            break;
        case OLD_STATUS_PAUSED:
            flags |= TORRENT_FLAG_PAUSED;
            break;
        case OLD_STATUS_SEEDING:
            flags |= TORRENT_FLAG_SEEDING;
            break;
        case OLD_STATUS_CHECKING:
            flags |= TORRENT_FLAG_CHECKING;
            break;
        case OLD_STATUS_WAITING_TO_CHECK:
            flags |= TORRENT_FLAG_WAITING_CHECK;
            flags |= TORRENT_FLAG_CHECKING;
            break;
        }
    }

    if (torrent_get_is_finished(t) == TRUE)
        flags |= TORRENT_FLAG_COMPLETE;
    else
        flags |= TORRENT_FLAG_INCOMPLETE;

    if (downRate > 0 || upRate > 0)
        flags |= TORRENT_FLAG_ACTIVE;

    if (strlen(torrent_get_errorstr(t)) > 0)
        flags |= TORRENT_FLAG_ERROR;

    return flags;
}

gchar *torrent_get_status_icon(gint64 rpcv, guint flags)
{
    if (flags & TORRENT_FLAG_ERROR)
        return g_strdup(GTK_STOCK_DIALOG_WARNING);
    else if (flags & TORRENT_FLAG_DOWNLOADING)
        return g_strdup(GTK_STOCK_GO_DOWN);
    else if (flags & TORRENT_FLAG_PAUSED)
        return g_strdup(GTK_STOCK_MEDIA_PAUSE);
    else if (flags & TORRENT_FLAG_SEEDING)
        return g_strdup(GTK_STOCK_GO_UP);
    else if (flags & TORRENT_FLAG_CHECKING)
        return g_strdup(GTK_STOCK_REFRESH);
    else
        return g_strdup(GTK_STOCK_DIALOG_QUESTION);
}

gint64 torrent_get_done_date(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_DONE_DATE);
}

const gchar *torrent_get_errorstr(JsonObject * t)
{
    return json_object_get_string_member(t, FIELD_ERRORSTR);
}

gchar *torrent_get_status_string(gint64 rpcv, gint64 value)
{
    if (rpcv >= NEW_STATUS_RPC_VERSION)
    {
        switch (value) {
        case TR_STATUS_DOWNLOAD:
            return g_strdup(_("Downloading"));
        case TR_STATUS_DOWNLOAD_WAIT:
            return g_strdup(_("Queued download"));
        case TR_STATUS_CHECK_WAIT:
            return g_strdup(_("Waiting To Check"));
        case TR_STATUS_CHECK:
            return g_strdup(_("Checking"));
        case TR_STATUS_SEED_WAIT:
            return g_strdup(_("Queued seed"));
        case TR_STATUS_SEED:
            return g_strdup(_("Seeding"));
        case TR_STATUS_STOPPED:
            return g_strdup(_("Paused"));
        }
    }
    else
    {
        switch (value) {
        case OLD_STATUS_DOWNLOADING:
            return g_strdup(_("Downloading"));
        case OLD_STATUS_PAUSED:
            return g_strdup(_("Paused"));
        case OLD_STATUS_SEEDING:
            return g_strdup(_("Seeding"));
        case OLD_STATUS_CHECKING:
            return g_strdup(_("Checking"));
        case OLD_STATUS_WAITING_TO_CHECK:
            return g_strdup(_("Waiting To Check"));
        }
    }

    g_warning("Unknown status: %ld", value);
    return g_strdup(_("Unknown"));
}

gboolean torrent_has_tracker(JsonObject * t, GRegex * rx, gchar * search)
{
    GList *trackers;
    GList *li;
    gboolean ret = FALSE;

    trackers = json_array_get_elements(torrent_get_tracker_stats(t));

    for (li = trackers; li; li = g_list_next(li)) {
        JsonObject *tracker = json_node_get_object((JsonNode *) li->data);
        const gchar *trackerAnnounce = tracker_stats_get_announce(tracker);
        gchar *trackerAnnounceHost =
            trg_gregex_get_first(rx, trackerAnnounce);
        int cmpResult = g_strcmp0(trackerAnnounceHost, search);
        g_free(trackerAnnounceHost);
        if (!cmpResult) {
            ret = TRUE;
            break;
        }
    }

    g_list_free(trackers);

    return ret;
}

gint64 torrent_get_left_until_done(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_LEFTUNTILDONE);
}

const gchar *tracker_stats_get_announce(JsonObject * t)
{
    return json_object_get_string_member(t, FIELD_ANNOUNCE);
}

const gchar *tracker_stats_get_scrape(JsonObject * t)
{
    return json_object_get_string_member(t, FIELD_SCRAPE);
}

JsonArray *get_torrents_removed(JsonObject * response)
{
    if (G_UNLIKELY(json_object_has_member(response, FIELD_REMOVED)))
        return json_object_get_array_member(response, FIELD_REMOVED);
    else
        return NULL;
}

JsonArray *get_torrents(JsonObject * response)
{
    return json_object_get_array_member(response, FIELD_TORRENTS);
}

JsonArray *torrent_get_files(JsonObject * args)
{
    return json_object_get_array_member(args, FIELD_FILES);
}

gint64 torrent_get_peers_connected(JsonObject *args)
{
    return json_object_get_int_member(args, FIELD_PEERS_CONNECTED);
}

gint64 torrent_get_peers_sending_to_us(JsonObject *args)
{
    return json_object_get_int_member(args, FIELD_PEERS_SENDING_TO_US);
}

gint64 torrent_get_peers_getting_from_us(JsonObject *args)
{
    return json_object_get_int_member(args, FIELD_PEERS_GETTING_FROM_US);
}

gint64 torrent_get_queue_position(JsonObject *args)
{
    if (json_object_has_member(args, FIELD_QUEUE_POSITION))
        return json_object_get_int_member(args, FIELD_QUEUE_POSITION);
    else
        return -1;
}

/* tracker stats */

gint64 tracker_stats_get_id(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_ID);
}

gint64 tracker_stats_get_tier(JsonObject * t)
{
    return json_object_get_int_member(t, FIELD_TIER);
}

gint64 tracker_stats_get_last_announce_peer_count(JsonObject *t)
{
    return json_object_get_int_member(t, FIELD_LAST_ANNOUNCE_PEER_COUNT);
}

gint64 tracker_stats_get_last_announce_time(JsonObject *t)
{
    return json_object_get_int_member(t, FIELD_LAST_ANNOUNCE_TIME);
}

gint64 tracker_stats_get_seeder_count(JsonObject *t)
{
    return json_object_get_int_member(t, FIELD_SEEDERCOUNT);
}

gint64 tracker_stats_get_leecher_count(JsonObject *t)
{
    return json_object_get_int_member(t, FIELD_LEECHERCOUNT);
}

const gchar *tracker_stats_get_announce_result(JsonObject *t)
{
    return json_object_get_string_member(t, FIELD_LAST_ANNOUNCE_RESULT);
}

const gchar *tracker_stats_get_host(JsonObject *t)
{
    return json_object_get_string_member(t, FIELD_HOST);
}