
net.codestory.http.io.Resources Maven / Gradle / Ivy
/**
* Copyright (C) 2013 [email protected]
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License
*/
package net.codestory.http.io;
import java.io.*;
import java.net.*;
import java.nio.charset.*;
import java.nio.file.*;
import java.nio.file.attribute.*;
import java.util.*;
import net.codestory.http.types.*;
import org.reflections.*;
import org.reflections.scanners.*;
public class Resources {
private static final String ROOT = "app";
private Resources() {
// Static utility class
}
public static Set list() {
Set paths = new TreeSet<>();
try {
if (new File("target/classes").exists() && !Boolean.getBoolean("http.disable.classpath")) {
new Reflections(ROOT, new ResourcesScanner()).getResources(name -> true)
.forEach(resource -> paths.add(relativeName(resource)));
}
if (!Boolean.getBoolean("http.disable.filesystem")) {
Files.walkFileTree(Paths.get(ROOT), new SimpleFileVisitor() {
@Override
public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {
paths.add(relativeName(file.toString()));
return FileVisitResult.CONTINUE;
}
});
}
} catch (IOException e) {
// Ignore
}
paths.remove("");
return paths;
}
private static String relativeName(String path) {
return Strings.substringAfter(path, ROOT + '/');
}
public static boolean isPublic(Path path) {
for (Path part : path) {
if (part.toString().equals("..") || part.toString().startsWith("_")) {
return false;
}
}
return Resources.exists(path);
}
public static Path findExistingPath(String uri) {
if (uri.endsWith("/")) {
return findExistingPath(uri + "index");
}
for (String extension : ContentTypes.TEMPLATE_EXTENSIONS) {
Path templatePath = Paths.get(uri + extension);
if (exists(templatePath)) {
return templatePath;
}
}
return null;
}
public static boolean exists(Path path) {
String pathWithPrefix = withPrefix(path);
return existsInFileSystem(pathWithPrefix) || existsInClassPath(pathWithPrefix);
}
public static String read(Path path, Charset charset) throws IOException {
String pathWithPrefix = withPrefix(path);
return existsInFileSystem(pathWithPrefix) ? readFile(pathWithPrefix, charset) : readClasspath(pathWithPrefix, charset);
}
public static byte[] readBytes(Path path) throws IOException {
String pathWithPrefix = withPrefix(path);
return existsInFileSystem(pathWithPrefix) ? readFileBytes(pathWithPrefix) : readClasspathBytes(pathWithPrefix);
}
private static String withPrefix(Path path) {
return ROOT + (path.toString().startsWith("/") ? "" : "/") + path;
}
private static boolean existsInClassPath(String path) {
URL url = ClassLoader.getSystemResource(path);
if (url == null) {
return false;
}
File file = fileForClasspath(url);
return (file == null) || file.isFile();
}
private static boolean existsInFileSystem(String path) {
return new File(path).isFile();
}
private static String readClasspath(String path, Charset charset) throws IOException {
URL url = ClassLoader.getSystemResource(path);
if (url == null) {
throw new IllegalArgumentException("Classpath resource not found classpath:" + path);
}
File file = fileForClasspath(url);
if (file != null) {
if (!file.isFile()) {
throw new IllegalArgumentException("Invalid file classpath: " + path);
}
return readFile(file.getAbsolutePath(), charset);
}
try (InputStream from = url.openStream()) {
return InputStreams.readString(from, charset);
}
}
private static byte[] readClasspathBytes(String path) throws IOException {
URL url = ClassLoader.getSystemResource(path);
if (url == null) {
throw new IllegalArgumentException("Invalid file classpath: " + path);
}
File file = fileForClasspath(url);
if (file != null) {
if (!file.isFile()) {
throw new IllegalArgumentException("Invalid file classpath: " + path);
}
return readFileBytes(file.getAbsolutePath());
}
try (InputStream from = url.openStream()) {
return InputStreams.readBytes(from);
}
}
private static String readFile(String path, Charset charset) throws IOException {
return new String(readFileBytes(path), charset);
}
private static byte[] readFileBytes(String path) throws IOException {
if (!new File(path).isFile()) {
throw new IllegalArgumentException("Invalid file path: " + path);
}
return Files.readAllBytes(Paths.get(path));
}
private static File fileForClasspath(URL url) {
String filename = url.getFile();
if ((filename == null) || filename.contains(".jar!")) {
return null;
}
try {
return new File(URLDecoder.decode(filename, "US-ASCII").replace("/target/classes/", "/src/main/resources/"));
} catch (UnsupportedEncodingException e) {
throw new IllegalArgumentException("Invalid filename classpath: " + url, e);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy