zzz/muwire
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

server side of browse host

Browse Source
This commit is contained in:
Zlatin Balevsky
2019-12-07 23:35:16 +00:00
parent 7fd7444dbf
commit 7d9ebb5b0b
7 changed files with 220 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
webui/src/main/java/com/muwire/webui/BrowseManager.java Normal file
View File

@ -0,0 +1,75 @@
package com.muwire.webui;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import com.muwire.core.Core;
import com.muwire.core.Persona;
import com.muwire.core.search.BrowseStatus;
import com.muwire.core.search.BrowseStatusEvent;
import com.muwire.core.search.UIBrowseEvent;
import com.muwire.core.search.UIResultEvent;
public class BrowseManager {
private final Core core;
private final Map<Persona, Browse> browses = new ConcurrentHashMap<>();
public BrowseManager(Core core) {
this.core = core;
}
public void onBrowseStatusEvent(BrowseStatusEvent e) {
Browse browse = browses.get(e.getHost());
if (browse == null)
return; // hmm
browse.status = e.getStatus();
if (browse.status == BrowseStatus.FETCHING)
browse.totalResults = e.getTotalResults();
}
public void onUIResultEvent(UIResultEvent e) {
Browse browse = browses.get(e.getSender());
if (browse == null)
return;
browse.results.add(e);
}
void browse(Persona p) {
Browse browse = new Browse(p);
browses.put(p, browse);
UIBrowseEvent event = new UIBrowseEvent();
event.setHost(p);
core.getEventBus().publish(event);
}
Map<Persona, Browse> getBrowses(){
return browses;
}
static class Browse {
private final Persona persona;
private volatile BrowseStatus status;
private volatile int totalResults;
private final List<UIResultEvent> results = Collections.synchronizedList(new ArrayList<>());
Browse(Persona persona) {
this.persona = persona;
}
public BrowseStatus getStatus() {
return status;
}
public int getTotalResults() {
return totalResults;
}
public List<UIResultEvent> getResults() {
return results;
}
}
}

116
webui/src/main/java/com/muwire/webui/BrowseServlet.java Normal file
View File

@ -0,0 +1,116 @@
package com.muwire.webui;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import com.muwire.core.Persona;
import com.muwire.webui.BrowseManager.Browse;
import net.i2p.data.Base64;
import net.i2p.data.DataHelper;
public class BrowseServlet extends HttpServlet {
private BrowseManager browseManager;
@Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
String section = req.getParameter("section");
if (section == null) {
resp.sendError(403, "Bad section param");
return;
}
StringBuilder sb = new StringBuilder();
sb.append("<?xml version='1.0' encoding='UTF-8'?>");
if (section.equals("status")) {
sb.append("<Status>");
browseManager.getBrowses().forEach( (persona, browse) -> {
sb.append("<Browse>");
sb.append("<Host>").append(Util.escapeHTMLinXML(persona.getHumanReadableName())).append("</Host>");
sb.append("<HostB64>").append(persona.toBase64()).append("</HostB64>");
sb.append("<BrowseStatus>").append(browse.getStatus()).append("</BrowseStatus>");
sb.append("<TotalResults>").append(browse.getTotalResults()).append("</TotalResults>");
sb.append("<ResultsCount>").append(browse.getResults().size()).append("</ResultsCount>");
sb.append("</Browse>");
});
sb.append("</Status>");
} else if (section.equals("results")) {
String hostB64 = req.getParameter("host");
if (hostB64 == null) {
resp.sendError(403,"Bad param");
return;
}
Persona host;
try {
host = new Persona(new ByteArrayInputStream(Base64.decode(hostB64)));
} catch (Exception bad) {
resp.sendError(403, "Bad param");
return;
}
Browse browse = browseManager.getBrowses().get(host);
if (browse == null)
return; // hmm
sb.append("<Results>");
browse.getResults().forEach(result -> {
sb.append("<Result>");
sb.append("<Name>").append(Util.escapeHTMLinXML(result.getName())).append("</Name>");
sb.append("<Size>").append(DataHelper.formatSize2Decimal(result.getSize(), false)).append("B").append("</Size>");
if (result.getComment() != null) {
sb.append("<Comment>").append(Util.escapeHTMLinXML(result.getComment())).append("</Comment>");
}
// TODO: add more fields
sb.append("</Result>");
});
sb.append("</Results>");
}
resp.setContentType("text/xml");
resp.setCharacterEncoding("UTF-8");
resp.setDateHeader("Expires", 0);
resp.setHeader("Pragma", "no-cache");
resp.setHeader("Cache-Control", "no-store, max-age=0, no-cache, must-revalidate");
byte[] out = sb.toString().getBytes("UTF-8");
resp.setContentLength(out.length);
resp.getOutputStream().write(out);
}
@Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
String action = req.getParameter("action");
if (action == null) {
resp.sendError(403,"Bad param");
return;
}
if (action.equals("browse")) {
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;
}
browseManager.browse(host);
resp.sendRedirect("/MuWire/BrowseHost.jsp");
} // TODO: implement canceling of browse
}
@Override
public void init(ServletConfig cfg) throws ServletException {
browseManager = (BrowseManager) cfg.getServletContext().getAttribute("browseManager");
}
}

7
webui/src/main/java/com/muwire/webui/MuWireClient.java
View File

@ -26,7 +26,9 @@ import com.muwire.core.files.FileDownloadedEvent;
import com.muwire.core.files.FileHashedEvent;
import com.muwire.core.files.FileHashingEvent;
import com.muwire.core.files.FileLoadedEvent;
import com.muwire.core.search.BrowseStatusEvent;
import com.muwire.core.search.UIResultBatchEvent;
import com.muwire.core.search.UIResultEvent;
import net.i2p.app.ClientAppManager;
import net.i2p.app.ClientAppState;
@ -133,10 +135,15 @@ public class MuWireClient {
core.getEventBus().register(FileDownloadedEvent.class, fileManager);
core.getEventBus().register(FileHashingEvent.class, fileManager);
BrowseManager browseManager = new BrowseManager(core);
core.getEventBus().register(BrowseStatusEvent.class, browseManager);
core.getEventBus().register(UIResultEvent.class, browseManager);
servletContext.setAttribute("searchManager", searchManager);
servletContext.setAttribute("downloadManager", downloadManager);
servletContext.setAttribute("connectionCounter", connectionCounter);
servletContext.setAttribute("fileManager", fileManager);
servletContext.setAttribute("browseManager", browseManager);
core.getEventBus().publish(new UILoadedEvent());
}

10
webui/templates/web.xml.template
View File

@ -31,6 +31,11 @@
<servlet-class>com.muwire.webui.FilesServlet</servlet-class>
</servlet>
<servlet>
<servlet-name>com.muwire.webui.BrowseServlet</servlet-name>
<servlet-class>com.muwire.webui.BrowseServlet</servlet-class>
</servlet>
<servlet-mapping>
<servlet-name>com.muwire.webui.MuWireServlet</servlet-name>
<url-pattern>/index.jsp</url-pattern>
@ -51,6 +56,11 @@
<url-pattern>/Files</url-pattern>
</servlet-mapping>
<servlet-mapping>
<servlet-name>com.muwire.webui.BrowseServlet</servlet-name>
<url-pattern>/Browse</url-pattern>
</servlet-mapping>
__JASPER__
</web-app>