mirrors/Terra
1
0
Fork 0
mirror of https://github.com/PolyhedralDev/Terra.git synced 2026-02-16 10:30:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Made requested changes

Browse Source
This commit is contained in:
ishland
2021-07-04 13:40:28 +08:00
parent f2946d85aa
commit f1845b1ecd
2 changed files with 39 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
common/src/main/java/com/dfsek/terra/api/util/JarUtil.java
View File

@@ -6,21 +6,13 @@ import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Enumeration;
import java.util.UUID;
import java.util.concurrent.atomic.AtomicReference;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
public class JarUtil {
private static final AtomicReference<File> terraModJar = new AtomicReference<>(null);
public static void copyResourcesToDirectory(JarFile fromJar, String sourceDir, String destDir) throws IOException {
for(Enumeration<JarEntry> entries = fromJar.entries(); entries.hasMoreElements(); ) {
JarEntry entry = entries.nextElement();
@@ -46,28 +38,7 @@ public class JarUtil {
}
public static JarFile getJarFile() throws URISyntaxException, IOException {
final File _jarFile1 = terraModJar.get();
if(_jarFile1 != null) return new JarFile(_jarFile1);
synchronized(terraModJar) {
final File _jarFile2 = terraModJar.get();
if(_jarFile2 != null) return new JarFile(_jarFile2);
// uncached here
final URI jarURI = getJarURL().toURI();
if(jarURI.getScheme() == null || !jarURI.getScheme().equalsIgnoreCase("file")) {
final File tempFile = File.createTempFile("terra-jar-" + UUID.randomUUID(), ".jar");
try(final FileOutputStream out = new FileOutputStream(tempFile)) {
Files.copy(Paths.get(jarURI), out);
}
tempFile.deleteOnExit();
terraModJar.set(tempFile);
return new JarFile(tempFile);
} else {
final File jarFile = new File(jarURI);
terraModJar.set(jarFile);
return new JarFile(jarFile);
}
}
return new JarFile(new File(getJarURL().toURI()));
}
public static URL getJarURL() {

38
platforms/fabric/src/main/java/com/dfsek/terra/fabric/TerraFabricPlugin.java
View File

@@ -27,6 +27,7 @@ import com.dfsek.terra.api.registry.CheckedRegistry;
import com.dfsek.terra.api.registry.LockedRegistry;
import com.dfsek.terra.api.transform.Transformer;
import com.dfsek.terra.api.transform.Validator;
import com.dfsek.terra.api.util.JarUtil;
import com.dfsek.terra.api.util.generic.pair.Pair;
import com.dfsek.terra.api.util.logging.DebugLogger;
import com.dfsek.terra.api.util.logging.Logger;
@@ -73,10 +74,18 @@ import org.apache.commons.io.FileUtils;
import org.apache.logging.log4j.LogManager;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.net.URISyntaxException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.atomic.AtomicReference;
import java.util.jar.JarFile;
public class TerraFabricPlugin implements TerraPlugin, ModInitializer {
@@ -137,6 +146,35 @@ public class TerraFabricPlugin implements TerraPlugin, ModInitializer {
private File dataFolder;
private final CommandManager manager = new TerraCommandManager(this);
private final AtomicReference<File> terraModJar = new AtomicReference<>(null);
@Override
public JarFile getModJar() throws URISyntaxException, IOException {
if(!FabricLoader.getInstance().isDevelopmentEnvironment()) return TerraPlugin.super.getModJar();
final File _jarFile1 = terraModJar.get();
if(_jarFile1 != null) return new JarFile(_jarFile1);
synchronized(terraModJar) {
final File _jarFile2 = terraModJar.get();
if(_jarFile2 != null) return new JarFile(_jarFile2);
// uncached here
final URI jarURI = JarUtil.getJarURL().toURI();
if(jarURI.getScheme() == null || !jarURI.getScheme().equalsIgnoreCase("file")) {
final File tempFile = File.createTempFile("terra-jar-" + UUID.randomUUID(), ".jar");
try(final FileOutputStream out = new FileOutputStream(tempFile)) {
Files.copy(Paths.get(jarURI), out);
}
tempFile.deleteOnExit();
terraModJar.set(tempFile);
return new JarFile(tempFile);
} else {
final File jarFile = new File(jarURI);
terraModJar.set(jarFile);
return new JarFile(jarFile);
}
}
}
public CommandManager getManager() {
return manager;
}