From ccf8773b180a59c9467cfeede55db5b5ab1e47a4 Mon Sep 17 00:00:00 2001 From: CalDescent Date: Sun, 19 Dec 2021 16:40:31 +0000 Subject: [PATCH] Added code that was left out from last commit. --- .../java/org/qortal/arbitrary/ArbitraryDataRenderer.java | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/main/java/org/qortal/arbitrary/ArbitraryDataRenderer.java b/src/main/java/org/qortal/arbitrary/ArbitraryDataRenderer.java index 486e6143..a096a078 100644 --- a/src/main/java/org/qortal/arbitrary/ArbitraryDataRenderer.java +++ b/src/main/java/org/qortal/arbitrary/ArbitraryDataRenderer.java @@ -93,11 +93,12 @@ public class ArbitraryDataRenderer { if (HTMLParser.isHtmlFile(filename)) { // HTML file - needs to be parsed byte[] data = Files.readAllBytes(Paths.get(filePath)); // TODO: limit file size that can be read into memory - HTMLParser htmlParser = new HTMLParser(resourceId, inPath, prefix, usePrefix); - data = htmlParser.replaceRelativeLinks(filename, data); + HTMLParser htmlParser = new HTMLParser(resourceId, inPath, prefix, usePrefix, data); + htmlParser.injectJavascript(); + htmlParser.replaceRelativeLinks(filename); response.setContentType(context.getMimeType(filename)); response.setContentLength(data.length); - response.getOutputStream().write(data); + response.getOutputStream().write(htmlParser.getData()); } else { // Regular file - can be streamed directly