Add tracker replace API

PBH-BTN · Jan 15, 2025 · 9c6662b · 9c6662b
1 parent e550b79
commit 9c6662b
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 1 deletion.
diff --git a/src/main/java/com/ghostchu/peerbanhelper/downloaderplug/biglybt/Plugin.java b/src/main/java/com/ghostchu/peerbanhelper/downloaderplug/biglybt/Plugin.java
@@ -16,12 +16,14 @@
 import com.biglybt.pif.peers.*;
 import com.biglybt.pif.tag.Tag;
 import com.biglybt.pif.torrent.Torrent;
+import com.biglybt.pif.torrent.TorrentAnnounceURLListSet;
 import com.biglybt.pif.ui.config.BooleanParameter;
 import com.biglybt.pif.ui.config.IntParameter;
 import com.biglybt.pif.ui.config.LabelParameter;
 import com.biglybt.pif.ui.config.StringParameter;
 import com.biglybt.pif.ui.model.BasicPluginConfigModel;
 import com.biglybt.pifimpl.local.clientid.ClientIDManagerImpl;
+import com.biglybt.pifimpl.local.download.DownloadImpl;
 import com.biglybt.pifimpl.local.peers.PeerImpl;
 import com.ghostchu.peerbanhelper.downloaderplug.biglybt.network.ConnectorData;
 import com.ghostchu.peerbanhelper.downloaderplug.biglybt.network.bean.clientbound.BanBean;
@@ -39,6 +41,8 @@
 import lombok.Getter;
 import lombok.extern.slf4j.Slf4j;
 
+import java.net.MalformedURLException;
+import java.net.URL;
 import java.util.*;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicLong;
@@ -219,12 +223,41 @@ private void initEndpoints(Javalin javalin) {
                 .get("/downloads", this::handleDownloads)
                 .get("/download/{infoHash}", this::handleDownload)
                 .get("/download/{infoHash}/peers", this::handlePeers)
+                .patch("/download/{infoHash}/trackers", this::handleTrackersSet)
                 .get("/bans", this::handleBans)
                 .post("/bans", this::handleBanListApplied)
                 .put("/bans", this::handleBanListReplacement)
                 .delete("/bans", this::handleBatchUnban);
     }
 
+    private void handleTrackersSet(Context ctx) {
+        String arg = ctx.pathParam("infoHash");
+        byte[] infoHash = hexToByteArray(arg);
+        String trackers = ctx.body();
+        try {
+            Download download = pluginInterface.getDownloadManager().getDownload(infoHash);
+            if (download == null) {
+                ctx.status(HttpStatus.NOT_FOUND);
+                return;
+            }
+            List<TorrentAnnounceURLListSet> sets = new ArrayList<>();
+            for (String tracker : trackers.split("\n\n")) {
+                String[] group = tracker.split("\n");
+                URL[] urls = new URL[group.length];
+                for (int i = 0; i < group.length; i++) {
+                    try {
+                        urls[i] = new URL(group[i]);
+                    } catch (MalformedURLException ignored) {
+                    }
+                }
+                sets.add(download.getTorrent().getAnnounceURLList().create(urls));
+            }
+            download.getTorrent().getAnnounceURLList().setSets(sets.toArray(new TorrentAnnounceURLListSet[0]));
+        } catch (DownloadException e) {
+            ctx.status(HttpStatus.NOT_FOUND);
+        }
+    }
+
     private void handleSetConnector(Context context) {
         this.connectorData = context.bodyAsClass(ConnectorData.class);
     }
@@ -431,6 +464,11 @@ private DownloadStatsRecord getDownloadStatsRecord(DownloadStats stats) {
 
     private DownloadRecord getDownloadRecord(Download download) {
         if (download == null) return null;
+        DownloadImpl downloadImpl = (DownloadImpl) download;
+        List<List<String>> trackers = new ArrayList<>();
+        for (TorrentAnnounceURLListSet set : downloadImpl.getTorrent().getAnnounceURLList().getSets()) {
+            trackers.add(Arrays.stream(set.getURLs()).map(Object::toString).collect(Collectors.toList()));
+        }
         DownloadStatsRecord downloadStatsRecord = getDownloadStatsRecord(download.getStats());
         TorrentRecord torrentRecord = getTorrentRecord(download.getTorrent());
         return new DownloadRecord(
@@ -450,7 +488,8 @@ private DownloadRecord getDownloadRecord(Download download) {
                 download.isChecking(),
                 download.isMoving(),
                 download.getDownloadPeerId() == null ? null : bytesToHex(download.getDownloadPeerId()).toLowerCase(Locale.ROOT),
-                download.isRemoved());
+                download.isRemoved(),
+                trackers);
     }
 
     private PeerManagerRecord getPeerManagerRecord(PeerManager peerManager) {

diff --git a/...ava/com/ghostchu/peerbanhelper/downloaderplug/biglybt/network/wrapper/DownloadRecord.java b/...ava/com/ghostchu/peerbanhelper/downloaderplug/biglybt/network/wrapper/DownloadRecord.java
@@ -29,4 +29,5 @@ public class DownloadRecord {
     private boolean moving;
     private String downloadPeerIdISO88591;
     private boolean removed;
+    private List<List<String>> trackers;
 }