0

AssetsManager: add getAsString + refactoring

This commit is contained in:
2020-12-27 20:58:00 +03:00
parent 96ee1eb3c7
commit 45f780f194
3 changed files with 88 additions and 41 deletions

View File

@@ -3,13 +3,13 @@ package ghast;
import lombok.experimental.UtilityClass;
import org.bukkit.plugin.Plugin;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.*;
import java.net.URL;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Scanner;
import static java.text.MessageFormat.format;
@@ -18,25 +18,10 @@ import static java.text.MessageFormat.format;
public class AssetsManager {
private static final String ERROR_NOT_FOUND = "Asset \"{0}\" not found";
private static final String ERROR_OPEN = "Error open asset \"{0}\": {1}";
public void saveTo(String resourceName, Path targetPath) {
URL resourceUrl = getResourceUrl(resourceName);
Path saveToPath;
if (Files.isDirectory(targetPath)) {
saveToPath = targetPath.resolve(resourceName);
} else {
saveToPath = targetPath;
}
doSaveTo(resourceUrl, saveToPath);
}
public void saveTo(String resourceName, File targetPath) {
saveTo(resourceName, targetPath.toPath());
}
public InputStream loadResource(String resourceName, String defaultResourceName, boolean saveDefault) {
//region getAsInputStream methods
public InputStream getAsInputStream(String resourceName, String defaultResourceName, boolean saveDefault) {
Plugin plugin = GhastTools.getPlugin();
InputStream inputStream;
@@ -59,18 +44,77 @@ public class AssetsManager {
return inputStream;
}
public InputStream loadResource(String resourceName, String defaultResourceName) {
return loadResource(resourceName, defaultResourceName, true);
public InputStream getAsInputStream(String resourceName, String defaultResourceName) {
return getAsInputStream(resourceName, defaultResourceName, true);
}
public InputStream loadResource(String resourceName, boolean saveDefault) {
return loadResource(resourceName, resourceName, saveDefault);
public InputStream getAsInputStream(String resourceName, boolean saveDefault) {
return getAsInputStream(resourceName, resourceName, saveDefault);
}
public InputStream loadResource(String resourceName) {
return loadResource(resourceName, resourceName, true);
public InputStream getAsInputStream(String resourceName) {
return getAsInputStream(resourceName, resourceName, true);
}
//endregion
//region getAsReader methods
public Reader getAsReader(String resourceName, String defaultResourceName, boolean saveDefault) {
return new InputStreamReader(getAsInputStream(resourceName, defaultResourceName, saveDefault));
}
public Reader getAsReader(String resourceName, String defaultResourceName) {
return new InputStreamReader(getAsInputStream(resourceName, defaultResourceName, true));
}
public Reader getAsReader(String resourceName, boolean saveDefault) {
return new InputStreamReader(getAsInputStream(resourceName, resourceName, saveDefault));
}
public Reader getAsReader(String resourceName) {
return new InputStreamReader(getAsInputStream(resourceName, resourceName, true));
}
//endregion
//region getAsString methods
public String getAsString(String resourceName, String defaultResourceName, Charset charset, boolean saveDefault) {
try (InputStream inputStream = getAsInputStream(resourceName, defaultResourceName, saveDefault);
Scanner scanner = new Scanner(inputStream, charset.name()).useDelimiter("\\A")) {
return scanner.next();
} catch (IOException e) {
throw new AssetsException(format(ERROR_OPEN, resourceName, e.getMessage()), e);
}
}
public String getAsString(String resourceName, String defaultResourceName, Charset charset) {
return getAsString(resourceName, defaultResourceName, charset, true);
}
public String getAsString(String resourceName, String defaultResourceName, boolean saveDefault) {
return getAsString(resourceName, defaultResourceName, StandardCharsets.UTF_8, saveDefault);
}
public String getAsString(String resourceName, String defaultResourceName) {
return getAsString(resourceName, defaultResourceName, StandardCharsets.UTF_8, true);
}
public String getAsString(String resourceName, Charset charset, boolean saveDefault) {
return getAsString(resourceName, resourceName, charset, saveDefault);
}
public String getAsString(String resourceName, Charset charset) {
return getAsString(resourceName, resourceName, charset, true);
}
public String getAsString(String resourceName, boolean saveDefault) {
return getAsString(resourceName, resourceName, StandardCharsets.UTF_8, saveDefault);
}
public String getAsString(String resourceName) {
return getAsString(resourceName, resourceName, StandardCharsets.UTF_8, true);
}
//endregion
private URL getResourceUrl(String resourceName) {
URL resourceUrl = AssetsManager.class.getClassLoader().getResource(resourceName);
if (resourceUrl == null) {
@@ -81,13 +125,17 @@ public class AssetsManager {
}
private void doSaveTo(URL resourceUrl, Path saveToPath) {
try (InputStream inputStream = resourceUrl.openStream();
OutputStream outputStream = Files.newOutputStream(saveToPath)) {
try {
Files.createDirectories(saveToPath.getParent());
byte[] buffer = new byte[8192];
int count;
while ((count = inputStream.read(buffer)) != -1) {
outputStream.write(buffer, 0, count);
try (InputStream inputStream = resourceUrl.openStream();
OutputStream outputStream = Files.newOutputStream(saveToPath)) {
byte[] buffer = new byte[8192];
int count;
while ((count = inputStream.read(buffer)) != -1) {
outputStream.write(buffer, 0, count);
}
}
} catch (IOException e) {
throw new AssetsException(format("Error save asset \"{0}\" to \"{1}\": {2}",
@@ -99,7 +147,7 @@ public class AssetsManager {
try {
return Files.newInputStream(pathToResource);
} catch (IOException e) {
throw new AssetsException(format("Error open asset \"{0}\": {1}", pathToResource, e.getMessage()), e);
throw new AssetsException(format(ERROR_OPEN, pathToResource, e.getMessage()), e);
}
}
@@ -107,7 +155,7 @@ public class AssetsManager {
try {
return resourceUrl.openStream();
} catch (IOException e) {
throw new AssetsException(format("Error open asset \"{0}\": {1}", resourceUrl, e.getMessage()), e);
throw new AssetsException(format(ERROR_OPEN, resourceUrl, e.getMessage()), e);
}
}
}

View File

@@ -5,8 +5,7 @@ import org.bukkit.configuration.file.YamlConfiguration;
import org.bukkit.plugin.Plugin;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.lang.ref.WeakReference;
@UtilityClass
@@ -30,8 +29,8 @@ public class GhastTools {
getPlugin().saveDefaultConfig();
}
try (InputStream inputStream = AssetsManager.loadResource("config.yml", saveDefault)) {
return YamlConfiguration.loadConfiguration(new InputStreamReader(inputStream));
try (Reader reader = AssetsManager.getAsReader("config.yml", saveDefault)) {
return YamlConfiguration.loadConfiguration(reader);
} catch (IOException e) {
throw new RuntimeException("Error load config: " + e.getMessage(), e);
}