All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.graylog2.web.resources.WebInterfaceAssetsResource Maven / Gradle / Ivy
/**
* This file is part of Graylog.
*
* Graylog is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Graylog is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Graylog. If not, see plugins;
private final LoadingCache fileSystemCache;
@Inject
public WebInterfaceAssetsResource(IndexHtmlGenerator indexHtmlGenerator, Set plugins, MimetypesFileTypeMap mimeTypes) {
this.indexHtmlGenerator = indexHtmlGenerator;
this.plugins = plugins;
this.mimeTypes = requireNonNull(mimeTypes);
this.fileSystemCache = CacheBuilder.newBuilder()
.maximumSize(1024)
.build(new CacheLoader() {
@Override
public FileSystem load(@Nonnull URI key) throws Exception {
try {
return FileSystems.getFileSystem(key);
} catch (FileSystemNotFoundException e) {
try {
return FileSystems.newFileSystem(key, Collections.emptyMap());
} catch (FileSystemAlreadyExistsException f) {
return FileSystems.getFileSystem(key);
}
}
}
});
}
@Path("assets/plugin/{plugin}/{filename}")
@GET
public Response get(@Context Request request,
@PathParam("plugin") String pluginName,
@PathParam("filename") String filename) {
final Plugin plugin = getPluginForName(pluginName)
.orElseThrow(() -> new NotFoundException("Couldn't find plugin " + pluginName));
try {
final URL resourceUrl = getResourceUri(true, filename, plugin.metadata().getClass());
return getResponse(request, filename, resourceUrl, true);
} catch (URISyntaxException | IOException e) {
throw new NotFoundException("Couldn't find " + filename + " in plugin " + pluginName, e);
}
}
private Optional getPluginForName(String pluginName) {
return this.plugins.stream().filter(plugin -> plugin.metadata().getUniqueId().equals(pluginName)).findFirst();
}
@Path("assets/{filename: .*}")
@GET
public Response get(@Context Request request,
@Context HttpHeaders headers,
@PathParam("filename") String filename) {
if (filename == null || filename.isEmpty() || "/".equals(filename) || "index.html".equals(filename)) {
return getDefaultResponse(headers);
}
try {
final URL resourceUrl = getResourceUri(false, filename, this.getClass());
return getResponse(request, filename, resourceUrl, false);
} catch (IOException | URISyntaxException e) {
return getDefaultResponse(headers);
}
}
@GET
@Path("index.html")
public Response getIndex(@Context HttpHeaders headers) {
return getDefaultResponse(headers);
}
@GET
public Response getIndex(@Context ContainerRequest request, @Context HttpHeaders headers) {
final URI originalLocation = request.getRequestUri();
if (originalLocation.getPath().endsWith("/")) {
return get(request, headers, originalLocation.getPath());
}
final URI redirect = UriBuilder.fromPath(originalLocation.getPath() + "/").build();
return Response.temporaryRedirect(redirect).build();
}
private Response getResponse(Request request, String filename,
URL resourceUrl, boolean fromPlugin) throws IOException, URISyntaxException {
final URI uri = resourceUrl.toURI();
final java.nio.file.Path path;
final byte[] fileContents;
switch (resourceUrl.getProtocol()) {
case "file": {
path = Paths.get(uri);
fileContents = Files.readAllBytes(path);
break;
}
case "jar": {
final FileSystem fileSystem = fileSystemCache.getUnchecked(uri);
path = fileSystem.getPath(pluginPrefixFilename(fromPlugin, filename));
fileContents = Resources.toByteArray(resourceUrl);
break;
}
default:
throw new IllegalArgumentException("Not a JAR or local file: " + resourceUrl);
}
final FileTime lastModifiedTime = Files.getLastModifiedTime(path);
final Date lastModified = Date.from(lastModifiedTime.toInstant());
final HashCode hashCode = Hashing.sha256().hashBytes(fileContents);
final EntityTag entityTag = new EntityTag(hashCode.toString());
final Response.ResponseBuilder response = request.evaluatePreconditions(lastModified, entityTag);
if (response != null) {
return response.build();
}
final String contentType = firstNonNull(mimeTypes.getContentType(filename), MediaType.APPLICATION_OCTET_STREAM);
final CacheControl cacheControl = new CacheControl();
cacheControl.setMaxAge((int) TimeUnit.DAYS.toSeconds(365));
cacheControl.setNoCache(false);
cacheControl.setPrivate(false);
return Response
.ok(fileContents, contentType)
.tag(entityTag)
.cacheControl(cacheControl)
.lastModified(lastModified)
.build();
}
private URL getResourceUri(boolean fromPlugin, String filename,
Class aClass) throws URISyntaxException, FileNotFoundException {
final URL resourceUrl = aClass.getResource(pluginPrefixFilename(fromPlugin, filename));
if (resourceUrl == null) {
throw new FileNotFoundException("Resource file " + filename + " not found.");
}
return resourceUrl;
}
@Nonnull
private String pluginPrefixFilename(boolean fromPlugin, String filename) {
if (fromPlugin) {
return "/" + filename;
} else {
return "/" + PluginAssets.pathPrefix + "/" + filename;
}
}
private Response getDefaultResponse(HttpHeaders headers) {
return Response
.ok(indexHtmlGenerator.get(headers.getRequestHeaders()))
.header(HttpHeaders.CONTENT_TYPE, MediaType.TEXT_HTML)
.header("X-UA-Compatible", "IE=edge")
.build();
}
}
