Add ability to add and ban a peer from the Web UI

6 years ago · 0fa28f233f
8 changed files with 197 additions and 1 deletions
--- a/src/webui/api/torrentscontroller.cpp
+++ b/src/webui/api/torrentscontroller.cpp
@ -40,6 +40,7 @@
				@@ -40,6 +40,7 @@
 #include <QUrl>

 #include "base/bittorrent/downloadpriority.h"
+#include "base/bittorrent/peeraddress.h"
 #include "base/bittorrent/peerinfo.h"
 #include "base/bittorrent/session.h"
 #include "base/bittorrent/torrenthandle.h"
@ -682,6 +683,42 @@ void TorrentsController::removeTrackersAction()
				@@ -682,6 +683,42 @@ void TorrentsController::removeTrackersAction()
        torrent->forceReannounce();
 }

+void TorrentsController::addPeersAction()
+{
+    checkParams({"hashes", "peers"});
+
+    const QStringList hashes = params()["hashes"].split('|');
+    const QStringList peers = params()["peers"].split('|');
+
+    QVector<BitTorrent::PeerAddress> peerList;
+    peerList.reserve(peers.size());
+    for (const QString &peer : peers) {
+        const BitTorrent::PeerAddress addr = BitTorrent::PeerAddress::parse(peer.trimmed());
+        if (!addr.ip.isNull())
+            peerList.append(addr);
+    }
+
+    if (peerList.isEmpty())
+        throw APIError(APIErrorType::BadParams, "No valid peers were specified");
+
+    QJsonObject results;
+
+    applyToTorrents(hashes, [peers, peerList, &results](BitTorrent::TorrentHandle *const torrent)
+    {
+        const int peersAdded = std::count_if(peerList.cbegin(), peerList.cend(), [torrent](const BitTorrent::PeerAddress &peer)
+        {
+            return torrent->connectPeer(peer);
+        });
+
+        results[torrent->hash()] = QJsonObject {
+            {"added", peersAdded},
+            {"failed", (peers.size() - peersAdded)}
+        };
+    });
+
+    setResult(results);
+}
+
 void TorrentsController::pauseAction()
 {
    checkParams({"hashes"});
--- a/src/webui/api/torrentscontroller.h
+++ b/src/webui/api/torrentscontroller.h
@ -66,6 +66,7 @@ private slots:
				@@ -66,6 +66,7 @@ private slots:
    void addTrackersAction();
    void editTrackerAction();
    void removeTrackersAction();
+    void addPeersAction();
    void filePrioAction();
    void uploadLimitAction();
    void downloadLimitAction();
--- a/src/webui/api/transfercontroller.cpp
+++ b/src/webui/api/transfercontroller.cpp
@ -29,8 +29,13 @@
				@@ -29,8 +29,13 @@
 #include "transfercontroller.h"

 #include <QJsonObject>
+#include <QVector>

+#include "base/bittorrent/peeraddress.h"
+#include "base/bittorrent/peerinfo.h"
 #include "base/bittorrent/session.h"
+#include "base/global.h"
+#include "apierror.h"

 const char KEY_TRANSFER_DLSPEED[] = "dl_info_speed";
 const char KEY_TRANSFER_DLDATA[] = "dl_info_data";
@ -111,3 +116,15 @@ void TransferController::speedLimitsModeAction()
				@@ -111,3 +116,15 @@ void TransferController::speedLimitsModeAction()
 {
    setResult(QString::number(BitTorrent::Session::instance()->isAltGlobalSpeedLimitEnabled()));
 }
+
+void TransferController::banPeersAction()
+{
+    checkParams({"peers"});
+
+    const QStringList peers = params()["peers"].split('|');
+    for (const QString &peer : peers) {
+        const BitTorrent::PeerAddress addr = BitTorrent::PeerAddress::parse(peer.trimmed());
+        if (!addr.ip.isNull())
+            BitTorrent::Session::instance()->banIP(addr.ip.toString());
+    }
+}
--- a/src/webui/api/transfercontroller.h
+++ b/src/webui/api/transfercontroller.h
@ -46,4 +46,5 @@ private slots:
				@@ -46,4 +46,5 @@ private slots:
    void downloadLimitAction();
    void setUploadLimitAction();
    void setDownloadLimitAction();
+    void banPeersAction();
 };
--- a/src/webui/www/private/addpeers.html
+++ b/src/webui/www/private/addpeers.html
@ -0,0 +1,69 @@
				@@ -0,0 +1,69 @@
+<!DOCTYPE html>
+<html lang="${LANG}">
+    <head>
+        <meta charset="UTF-8" />
+        <title>QBT_TR(Add Peers)QBT_TR[CONTEXT=PeersAdditionDialog]</title>
+        <link rel="stylesheet" href="css/style.css?v=${CACHEID}" type="text/css" />
+        <script src="scripts/lib/mootools-1.2-core-yc.js"></script>
+        <script src="scripts/lib/mootools-1.2-more.js"></script>
+        <script>
+            'use strict';
+
+            new Keyboard({
+                defaultEventType: 'keydown',
+                events: {
+                    'Escape': function(event) {
+                        window.parent.closeWindows();
+                        event.preventDefault();
+                    },
+                    'Esc': function(event) {
+                        window.parent.closeWindows();
+                        event.preventDefault();
+                    }
+                }
+            }).activate();
+
+            window.addEvent('domready', function() {
+                const hash = new URI().getData('hash');
+                if (!hash)
+                    return false;
+
+                $('peers').focus();
+
+                $('addPeersOk').addEvent('click', function(e) {
+                    new Event(e).stop();
+
+                    const peers = $('peers').get('value').trim().split(/[\r\n]+/);
+                    if (peers.length === 0)
+                        return
+
+                    new Request({
+                        url: 'api/v2/torrents/addPeers',
+                        method: 'post',
+                        data: {
+                            hashes: hash,
+                            peers: peers.join(';')
+                        },
+                        onFailure: function() {
+                            alert("QBT_TR(Unable to add peers. Please ensure you are adhering to the IP:port format.)QBT_TR[CONTEXT=HttpServer]");
+                        },
+                        onSuccess: function() {
+                            window.parent.closeWindows();
+                        }
+                    }).send();
+                });
+            });
+        </script>
+    </head>
+
+    <body>
+        <div style="padding: 10px 10px 0px 10px;">
+            <p>QBT_TR(List of peers to add (one IP per line):)QBT_TR[CONTEXT=PeersAdditionDialog]</p>
+            <textarea id="peers" rows="10" style="width: 100%;" placeholder="QBT_TR(Format: IPv4:port / [IPv6]:port)QBT_TR[CONTEXT=PeersAdditionDialog]"></textarea>
+            <div style="margin-top: 10px; text-align: center;">
+                <button onclick="window.parent.closeWindows();">QBT_TR(Cancel)QBT_TR[CONTEXT=PeersAdditionDialog]</button>
+                <button id="addPeersOk">QBT_TR(Ok)QBT_TR[CONTEXT=PeersAdditionDialog]</button>
+            </div>
+        </div>
+    </body>
+</html>
--- a/src/webui/www/private/index.html
+++ b/src/webui/www/private/index.html
@ -188,6 +188,11 @@
				@@ -188,6 +188,11 @@
        <li><a href="#RemoveTracker"><img src="images/qbt-theme/list-remove.svg" alt="QBT_TR(Remove tracker)QBT_TR[CONTEXT=TrackerListWidget]"/> QBT_TR(Remove tracker)QBT_TR[CONTEXT=TrackerListWidget]</a></li>
        <li><a href="#CopyTrackerUrl" id="CopyTrackerUrl"><img src="images/qbt-theme/edit-copy.svg" alt="QBT_TR(Copy tracker URL)QBT_TR[CONTEXT=TrackerListWidget]"/> QBT_TR(Copy tracker URL)QBT_TR[CONTEXT=TrackerListWidget]</a></li>
    </ul>
+    <ul id="torrentPeersMenu" class="contextMenu">
+        <li><a href="#addPeer"><img src="images/qbt-theme/list-add.svg" alt="QBT_TR(Add a new peer...)QBT_TR[CONTEXT=PeerListWidget]"/> QBT_TR(Add a new peer...)QBT_TR[CONTEXT=PeerListWidget]</a></li>
+        <li><a href="#copyPeer" id="CopyPeerInfo"><img src="images/qbt-theme/edit-copy.svg" alt="QBT_TR(Copy IP:port)QBT_TR[CONTEXT=PeerListWidget]"/> QBT_TR(Copy IP:port)QBT_TR[CONTEXT=PeerListWidget]</a></li>
+        <li class="separator"><a href="#banPeer"><img src="images/qbt-theme/user-group-delete.svg" alt="QBT_TR(Ban peer permanently)QBT_TR[CONTEXT=PeerListWidget]"/> QBT_TR(Ban peer permanently)QBT_TR[CONTEXT=PeerListWidget]</a></li>
+    </ul>
    <ul id="torrentFilesMenu" class="contextMenu">
        <li class="separator">
            <a href="#FilePrio" class="arrow-right"><span style="display: inline-block; width: 16px;"></span> QBT_TR(Priority)QBT_TR[CONTEXT=PropertiesWidget]</a>
--- a/src/webui/www/private/scripts/prop-peers.js
+++ b/src/webui/www/private/scripts/prop-peers.js
@ -103,4 +103,69 @@ updateTorrentPeersData = function() {
				@@ -103,4 +103,69 @@ updateTorrentPeersData = function() {
    loadTorrentPeersData();
 };

-torrentPeersTable.setup('torrentPeersTableDiv', 'torrentPeersTableFixedHeaderDiv', null);
+const torrentPeersContextMenu = new ContextMenu({
+    targets: '#torrentPeersTableDiv',
+    menu: 'torrentPeersMenu',
+    actions: {
+        addPeer: function(element, ref) {
+            const hash = torrentsTable.getCurrentTorrentHash();
+            if (!hash)
+                return;
+
+            new MochaUI.Window({
+                id: 'addPeersPage',
+                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: 240
+            });
+        },
+        banPeer: function(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]')) {
+                new Request({
+                    url: 'api/v2/torrents/banPeers',
+                    noCache: true,
+                    method: 'post',
+                    data: {
+                        hash: torrentsTable.getCurrentTorrentHash(),
+                        peers: selectedPeers.join('|')
+                    }
+                }).send();
+            }
+        }
+    },
+    offsets: {
+        x: -15,
+        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: function(trigger) {
+        return torrentPeersTable.selectedRowsIds().join("\n");
+    }
+});
+
+torrentPeersTable.setup('torrentPeersTableDiv', 'torrentPeersTableFixedHeaderDiv', torrentPeersContextMenu);
--- a/src/webui/www/webui.qrc
+++ b/src/webui/www/webui.qrc
@ -2,6 +2,7 @@
				@@ -2,6 +2,7 @@
    <qresource prefix="/www">
        <file>private/about.html</file>
        <file>private/aboutToolbar.html</file>
+        <file>private/addpeers.html</file>
        <file>private/addtrackers.html</file>
        <file>private/confirmdeletion.html</file>
        <file>private/css/Core.css</file>