From ca33535630c862c6c71ca0f5e776f4855c0e4836 Mon Sep 17 00:00:00 2001
From: Zlatin Balevsky <zlatinb@gmail.com>
Date: Thu, 12 Mar 2020 22:46:48 +0000
Subject: [PATCH] POST hook for downloading feed items

---
 .../java/com/muwire/webui/FeedServlet.java    | 52 +++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/webui/src/main/java/com/muwire/webui/FeedServlet.java b/webui/src/main/java/com/muwire/webui/FeedServlet.java
index 92a750d4..2f847f8a 100644
--- a/webui/src/main/java/com/muwire/webui/FeedServlet.java
+++ b/webui/src/main/java/com/muwire/webui/FeedServlet.java
@@ -1,10 +1,12 @@
 package com.muwire.webui;
 
 import java.io.ByteArrayInputStream;
+import java.io.File;
 import java.io.IOException;
 import java.text.Collator;
 import java.util.Comparator;
 import java.util.List;
+import java.util.Optional;
 import java.util.stream.Collectors;
 
 import javax.servlet.ServletConfig;
@@ -14,9 +16,11 @@ import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
 import com.muwire.core.Core;
+import com.muwire.core.InfoHash;
 import com.muwire.core.Persona;
 import com.muwire.core.filefeeds.Feed;
 import com.muwire.core.filefeeds.FeedItem;
+import com.muwire.core.filefeeds.UIDownloadFeedItemEvent;
 import com.muwire.core.util.DataUtil;
 import com.muwire.webui.FeedManager.RemoteFeed;
 
@@ -133,6 +137,54 @@ public class FeedServlet extends HttpServlet {
             }
             feedManager.unsubscribe(host);
             Util.pause();
+        } else if (action.equals("download")) {
+            if (core == null) {
+                resp.sendError(403, "Not initialized");
+                return;
+            }
+            String personaB64 = req.getParameter("host");
+            if (personaB64 == null) {
+                resp.sendError(403,"Bad param");
+                return;
+            }
+            Persona host;
+            try {
+                host = new Persona(new ByteArrayInputStream(Base64.decode(personaB64)));
+            } catch (Exception bad) {
+                resp.sendError(403,"Bad param");
+                return;
+            }
+            String infoHashB64 = req.getParameter("infoHash");
+            if (infoHashB64 == null) {
+                resp.sendError(403, "Bad param");
+                return;
+            }
+            final InfoHash infoHash;
+            try {
+                infoHash = new InfoHash(Base64.decode(infoHashB64));
+            } catch (Exception bad) {
+                resp.sendError(403, "Bad param");
+                return;
+            }
+            
+            Feed feed = core.getFeedManager().getFeed(host);
+            if (feed == null)
+                return;
+            
+            Optional<FeedItem> itemOptional = core.getFeedManager().getFeedItems(host).
+                    stream().filter(item -> item.getInfoHash().equals(infoHash)).findFirst();
+            if (itemOptional.isEmpty())
+                return;
+            FeedItem item = itemOptional.get();
+            
+            File target = new File(core.getMuOptions().getDownloadLocation(), item.getName());
+            
+            UIDownloadFeedItemEvent event = new UIDownloadFeedItemEvent();
+            event.setItem(item);
+            event.setSequential(feed.isSequential());
+            event.setTarget(target);
+            core.getEventBus().publish(event);
+            Util.pause();
         }
     }