qBittorrent/src/webui/www/private/scripts/prop-peers.js

/*
 * Bittorrent Client using Qt and libtorrent.
 * Copyright (C) 2018  Thomas Piccirello <thomas.piccirello@gmail.com>
 *
 * 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.
 *
 * In addition, as a special exception, the copyright holders give permission to
 * link this program with the OpenSSL project's "OpenSSL" library (or with
 * modified versions of it that use the same license as the "OpenSSL" library),
 * and distribute the linked executables. You must obey the GNU General Public
 * License in all respects for all of the code used other than "OpenSSL".  If you
 * modify file(s), you may extend this exception to your version of the file(s),
 * but you are not obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 */

"use strict";

window.qBittorrent ??= {};
window.qBittorrent.PropPeers ??= (() => {
    const exports = () => {
        return {
            updateData: updateData,
            clear: clear
        };
    };

    const torrentPeersTable = new window.qBittorrent.DynamicTable.TorrentPeersTable();
    let loadTorrentPeersTimer = -1;
    let syncTorrentPeersLastResponseId = 0;
    let show_flags = true;

    const loadTorrentPeersData = () => {
        if (document.hidden)
            return;
        if ($("propPeers").classList.contains("invisible")
            || $("propertiesPanel_collapseToggle").classList.contains("panel-expand")) {
            syncTorrentPeersLastResponseId = 0;
            torrentPeersTable.clear();
            return;
        }
        const current_hash = torrentsTable.getCurrentTorrentID();
        if (current_hash === "") {
            syncTorrentPeersLastResponseId = 0;
            torrentPeersTable.clear();
            clearTimeout(loadTorrentPeersTimer);
            return;
        }
        const url = new URL("api/v2/sync/torrentPeers", window.location);
        url.search = new URLSearchParams({
            hash: current_hash,
            rid: syncTorrentPeersLastResponseId,
        });
        fetch(url, {
                method: "GET",
                cache: "no-store"
            })
            .then(async (response) => {
                if (!response.ok)
                    return;

                const responseJSON = await response.json();

                $("error_div").textContent = "";
                if (responseJSON) {
                    const full_update = (responseJSON["full_update"] === true);
                    if (full_update)
                        torrentPeersTable.clear();
                    if (responseJSON["rid"])
                        syncTorrentPeersLastResponseId = responseJSON["rid"];
                    if (responseJSON["peers"]) {
                        for (const key in responseJSON["peers"]) {
                            if (!Object.hasOwn(responseJSON["peers"], key))
                                continue;

                            responseJSON["peers"][key]["rowId"] = key;
                            torrentPeersTable.updateRowData(responseJSON["peers"][key]);
                        }
                    }
                    if (responseJSON["peers_removed"]) {
                        responseJSON["peers_removed"].each((hash) => {
                            torrentPeersTable.removeRow(hash);
                        });
                    }
                    torrentPeersTable.updateTable(full_update);

                    if (responseJSON["show_flags"]) {
                        if (show_flags !== responseJSON["show_flags"]) {
                            show_flags = responseJSON["show_flags"];
                            torrentPeersTable.columns["country"].force_hide = !show_flags;
                            torrentPeersTable.updateColumn("country");
                        }
                    }
                }
                else {
                    torrentPeersTable.clear();
                }

            })
            .finally(() => {
                clearTimeout(loadTorrentPeersTimer);
                loadTorrentPeersTimer = loadTorrentPeersData.delay(window.qBittorrent.Client.getSyncMainDataInterval());
            });
    };

    const updateData = () => {
        clearTimeout(loadTorrentPeersTimer);
        loadTorrentPeersTimer = -1;
        loadTorrentPeersData();
    };

    const clear = () => {
        torrentPeersTable.clear();
    };

    const torrentPeersContextMenu = new window.qBittorrent.ContextMenu.ContextMenu({
        targets: "#torrentPeersTableDiv",
        menu: "torrentPeersMenu",
        actions: {
            addPeer: (element, ref) => {
                const hash = torrentsTable.getCurrentTorrentID();
                if (!hash)
                    return;

                new MochaUI.Window({
                    id: "addPeersPage",
                    icon: "images/qbittorrent-tray.svg",
                    title: "QBT_TR(Add Peers)QBT_TR[CONTEXT=PeersAdditionDialog]",
                    loadMethod: "iframe",
                    contentURL: `addpeers.html?hash=${hash}`,
                    scrollbars: false,
                    resizable: false,
                    maximizable: false,
                    paddingVertical: 0,
                    paddingHorizontal: 0,
                    width: 350,
                    height: 260
                });
            },
            banPeer: (element, ref) => {
                const selectedPeers = torrentPeersTable.selectedRowsIds();
                if (selectedPeers.length === 0)
                    return;

                if (confirm("QBT_TR(Are you sure you want to permanently ban the selected peers?)QBT_TR[CONTEXT=PeerListWidget]")) {
                    fetch("api/v2/transfer/banPeers", {
                        method: "POST",
                        body: new URLSearchParams({
                            hash: torrentsTable.getCurrentTorrentID(),
                            peers: selectedPeers.join("|")
                        })
                    });
                }
            }
        },
        offsets: {
            x: 0,
            y: 2
        },
        onShow: function() {
            const selectedPeers = torrentPeersTable.selectedRowsIds();

            if (selectedPeers.length >= 1) {
                this.showItem("copyPeer");
                this.showItem("banPeer");
            }
            else {
                this.hideItem("copyPeer");
                this.hideItem("banPeer");
            }
        }
    });

    new ClipboardJS("#CopyPeerInfo", {
        text: (trigger) => {
            return torrentPeersTable.selectedRowsIds().join("\n");
        }
    });

    torrentPeersTable.setup("torrentPeersTableDiv", "torrentPeersTableFixedHeaderDiv", torrentPeersContextMenu, true);

    return exports();
})();
Object.freeze(window.qBittorrent.PropPeers);