implemented WebResourceServlet, which loads resources from the UberWebResourceLoader

This commit is contained in:
Sebastian Sdorra
2018-08-21 15:23:54 +02:00
parent e2fa2388f1
commit be21c35bf8
5 changed files with 206 additions and 3 deletions

View File

@@ -40,9 +40,13 @@ import java.util.List;
* This class collects and manages {@link Resource} * This class collects and manages {@link Resource}
* which are used by the web interface. * which are used by the web interface.
* *
* TODO remove before 2.0.0
*
* @author Sebastian Sdorra * @author Sebastian Sdorra
* @since 1.12 * @since 1.12
* @deprecated unnecessary for new ui
*/ */
@Deprecated
public interface ResourceManager public interface ResourceManager
{ {

View File

@@ -0,0 +1,77 @@
package sonia.scm;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.io.Resources;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import sonia.scm.filter.WebElement;
import sonia.scm.plugin.PluginLoader;
import sonia.scm.plugin.UberWebResourceLoader;
import sonia.scm.util.HttpUtil;
import javax.inject.Inject;
import javax.inject.Singleton;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URL;
/**
* WebResourceServlet serves resources from the {@link UberWebResourceLoader}.
*
* @since 2.0.0
*/
@Singleton
@WebElement(value = WebResourceServlet.PATTERN, regex = true)
public class WebResourceServlet extends HttpServlet {
/**
* exclude api requests and the old frontend servlets.
*
* TODO remove old frontend servlets
*/
@VisibleForTesting
static final String PATTERN = "/(?!api/|index.html|error.html|plugins/resources).+";
private static final Logger LOG = LoggerFactory.getLogger(WebResourceServlet.class);
private final UberWebResourceLoader webResourceLoader;
@Inject
public WebResourceServlet(PluginLoader pluginLoader) {
this.webResourceLoader = pluginLoader.getUberWebResourceLoader();
}
@Override
protected void doGet(HttpServletRequest request, HttpServletResponse response) {
String uri = normalizeUri(request);
LOG.trace("try to load {}", uri);
URL url = webResourceLoader.getResource(uri);
if (url != null) {
serveResource(response, url);
} else {
handleResourceNotFound(response);
}
}
private String normalizeUri(HttpServletRequest request) {
return HttpUtil.getStrippedURI(request);
}
private void serveResource(HttpServletResponse response, URL url) {
// TODO lastModifiedDate, if-... ???
try (OutputStream output = response.getOutputStream()) {
Resources.copy(url, output);
} catch (IOException ex) {
LOG.warn("failed to serve resource: {}", url);
response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
}
}
private void handleResourceNotFound(HttpServletResponse response) {
response.setStatus(HttpServletResponse.SC_NOT_FOUND);
}
}

View File

@@ -42,6 +42,7 @@ import com.google.common.collect.ImmutableList.Builder;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
import sonia.scm.util.HttpUtil;
//~--- JDK imports ------------------------------------------------------------ //~--- JDK imports ------------------------------------------------------------
@@ -185,7 +186,7 @@ public class DefaultUberWebResourceLoader implements UberWebResourceLoader
*/ */
private URL find(String path) private URL find(String path)
{ {
URL resource = null; URL resource;
try try
{ {

View File

@@ -49,6 +49,7 @@ import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Map.Entry; import java.util.Map.Entry;
import java.util.Set; import java.util.Set;
import java.util.stream.Collectors;
import javax.servlet.ServletContext; import javax.servlet.ServletContext;
import sonia.scm.plugin.PluginWrapper; import sonia.scm.plugin.PluginWrapper;
@@ -172,7 +173,12 @@ public abstract class AbstractResourceManager implements ResourceManager
if (scriptResources != null) if (scriptResources != null)
{ {
resources.addAll(scriptResources); // filter new resources and keep only the old ones, which are starting with a /
List<String> filtered = scriptResources.stream()
.filter((res) -> res.startsWith("/"))
.collect(Collectors.toList());
resources.addAll(filtered);
} }
} }
} }

View File

@@ -0,0 +1,115 @@
package sonia.scm;
import com.google.common.base.Charsets;
import com.google.common.io.Files;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import sonia.scm.plugin.PluginLoader;
import sonia.scm.plugin.UberWebResourceLoader;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
@RunWith(MockitoJUnitRunner.class)
public class WebResourceServletTest {
@Rule
public TemporaryFolder temporaryFolder = new TemporaryFolder();
@Mock
private HttpServletRequest request;
@Mock
private HttpServletResponse response;
@Mock
private PluginLoader pluginLoader;
@Mock
private UberWebResourceLoader webResourceLoader;
private WebResourceServlet servlet;
@Before
public void setUpMocks() {
when(pluginLoader.getUberWebResourceLoader()).thenReturn(webResourceLoader);
when(request.getContextPath()).thenReturn("/scm");
servlet = new WebResourceServlet(pluginLoader);
}
@Test
public void testPattern() {
assertTrue("/some/resource".matches(WebResourceServlet.PATTERN));
assertTrue("/favicon.ico".matches(WebResourceServlet.PATTERN));
assertTrue("/other.html".matches(WebResourceServlet.PATTERN));
assertFalse("/api/v2/repositories".matches(WebResourceServlet.PATTERN));
// exclude old style ui template servlets
assertFalse("/".matches(WebResourceServlet.PATTERN));
assertFalse("/index.html".matches(WebResourceServlet.PATTERN));
assertFalse("/error.html".matches(WebResourceServlet.PATTERN));
assertFalse("/plugins/resources/js/sonia/scm/hg.config-wizard.js".matches(WebResourceServlet.PATTERN));
}
@Test
public void testDoGetWithNonExistingResource() {
when(request.getRequestURI()).thenReturn("/scm/awesome.jpg");
servlet.doGet(request, response);
verify(response).setStatus(HttpServletResponse.SC_NOT_FOUND);
}
@Test
public void testDoGet() throws IOException {
when(request.getRequestURI()).thenReturn("/scm/README.txt");
TestingOutputServletOutputStream output = new TestingOutputServletOutputStream();
when(response.getOutputStream()).thenReturn(output);
File file = temporaryFolder.newFile();
Files.write("hello".getBytes(Charsets.UTF_8), file);
when(webResourceLoader.getResource("/README.txt")).thenReturn(file.toURI().toURL());
servlet.doGet(request, response);
assertEquals("hello", output.buffer.toString());
}
@Test
public void testDoGetWithError() throws IOException {
when(request.getRequestURI()).thenReturn("/scm/README.txt");
ServletOutputStream output = mock(ServletOutputStream.class);
when(response.getOutputStream()).thenReturn(output);
File file = temporaryFolder.newFile();
assertTrue(file.delete());
when(webResourceLoader.getResource("/README.txt")).thenReturn(file.toURI().toURL());
servlet.doGet(request, response);
verify(output, never()).write(anyInt());
verify(response).setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
}
private static class TestingOutputServletOutputStream extends ServletOutputStream {
private ByteArrayOutputStream buffer = new ByteArrayOutputStream();
@Override
public void write(int b) {
buffer.write(b);
}
}
}