diff options
Diffstat (limited to 'src/main/java/com/keuin/kbackupfabric/util')
4 files changed, 307 insertions, 0 deletions
diff --git a/src/main/java/com/keuin/kbackupfabric/util/ReflectionUtils.java b/src/main/java/com/keuin/kbackupfabric/util/ReflectionUtils.java new file mode 100644 index 0000000..f7bc351 --- /dev/null +++ b/src/main/java/com/keuin/kbackupfabric/util/ReflectionUtils.java @@ -0,0 +1,59 @@ +package com.keuin.kbackupfabric.util; + +import java.lang.reflect.Field; +import java.lang.reflect.InvocationTargetException; +import java.lang.reflect.Method; + +/** + * @Author 落叶飞翔的蜗牛 + * @Date 2018/3/10 + * @Description + */ +public class ReflectionUtils { + + /** + * 获取私有成员变量的值 + * + * @param instance + * @param filedName + * @return + */ + public static Object getPrivateField(Object instance, String filedName) throws NoSuchFieldException, IllegalAccessException { + Field field = instance.getClass().getDeclaredField(filedName); + field.setAccessible(true); + return field.get(instance); + } + + /** + * 设置私有成员的值 + * + * @param instance + * @param fieldName + * @param value + * @throws NoSuchFieldException + * @throws IllegalAccessException + */ + public static void setPrivateField(Object instance, String fieldName, Object value) throws NoSuchFieldException, IllegalAccessException { + Field field = instance.getClass().getDeclaredField(fieldName); + field.setAccessible(true); + field.set(instance, value); + } + + /** + * 访问私有方法 + * + * @param instance + * @param methodName + * @param classes + * @param objects + * @return + * @throws NoSuchMethodException + * @throws InvocationTargetException + * @throws IllegalAccessException + */ + public static Object invokePrivateMethod(Object instance, String methodName, Class[] classes, String objects) throws NoSuchMethodException, InvocationTargetException, IllegalAccessException { + Method method = instance.getClass().getDeclaredMethod(methodName, classes); + method.setAccessible(true); + return method.invoke(instance, objects); + } +}
\ No newline at end of file diff --git a/src/main/java/com/keuin/kbackupfabric/util/WorldUtil.java b/src/main/java/com/keuin/kbackupfabric/util/WorldUtil.java new file mode 100644 index 0000000..badc068 --- /dev/null +++ b/src/main/java/com/keuin/kbackupfabric/util/WorldUtil.java @@ -0,0 +1,15 @@ +package com.keuin.kbackupfabric.util; + +import net.minecraft.server.world.ThreadedAnvilChunkStorage; +import net.minecraft.world.World; + +import java.io.File; + +public class WorldUtil { + public static String getWorldDirectoryName(World world) throws NoSuchFieldException, IllegalAccessException { + File saveDir; + ThreadedAnvilChunkStorage threadedAnvilChunkStorage = (ThreadedAnvilChunkStorage) ReflectionUtils.getPrivateField(world.getChunkManager(), "threadedAnvilChunkStorage"); + saveDir = (File) ReflectionUtils.getPrivateField(threadedAnvilChunkStorage, "saveDir"); + return saveDir.getName(); + } +} diff --git a/src/main/java/com/keuin/kbackupfabric/util/ZipUtil.java b/src/main/java/com/keuin/kbackupfabric/util/ZipUtil.java new file mode 100644 index 0000000..d92ab12 --- /dev/null +++ b/src/main/java/com/keuin/kbackupfabric/util/ZipUtil.java @@ -0,0 +1,221 @@ +package com.keuin.kbackupfabric.util; + +import java.io.*; +import java.util.Enumeration; +import java.util.zip.*; + +public class ZipUtil { + + /** + * 递归压缩文件夹 + * + * @param srcRootDir 压缩文件夹根目录的子路径 + * @param file 当前递归压缩的文件或目录对象 + * @param zos 压缩文件存储对象 + * @throws Exception + */ + private static void zip(String srcRootDir, File file, ZipOutputStream zos) throws IOException { + if (file == null) { + return; + } + + // 如果是文件，则直接压缩该文件 + if (file.isFile()) { + int count, bufferLen = 1024; + byte[] data = new byte[bufferLen]; + + // 获取文件相对于压缩文件夹根目录的子路径 + String subPath = file.getAbsolutePath(); + int index = subPath.indexOf(srcRootDir); + if (index != -1) { + subPath = subPath.substring(srcRootDir.length() + File.separator.length()); + } + ZipEntry entry = new ZipEntry(subPath); + zos.putNextEntry(entry); + BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file)); + while ((count = bis.read(data, 0, bufferLen)) != -1) { + zos.write(data, 0, count); + } + bis.close(); + zos.closeEntry(); + } + // 如果是目录，则压缩整个目录 + else { + // 压缩目录中的文件或子目录 + File[] childFileList = file.listFiles(); + for (int n = 0; n < childFileList.length; n++) { + childFileList[n].getAbsolutePath().indexOf(file.getAbsolutePath()); + zip(srcRootDir, childFileList[n], zos); + } + } + } + + /** + * 对文件或文件目录进行压缩 + * + * @param srcPath 要压缩的源文件路径。如果是目录，则将递归压缩这个目录及其所有子文件、子目录树。 + * @param zipPath 压缩文件保存的路径。注意：zipPath不能是srcPath路径下的子文件夹 + * @param zipFileName 压缩文件名 + * @throws Exception + */ + public static void zip(String srcPath, String zipPath, String zipFileName) throws IOException, ZipUtilException { + if (srcPath.isEmpty() || zipPath.isEmpty() || zipFileName.isEmpty()) { + throw new ZipUtilException("Parameter for zip() contains null."); + } + CheckedOutputStream cos = null; + ZipOutputStream zos = null; + try { + File srcFile = new File(srcPath); + + //判断压缩文件保存的路径是否为源文件路径的子文件夹，如果是，则抛出异常（防止无限递归压缩的发生） + if (srcFile.isDirectory() && zipPath.indexOf(srcPath) != -1) { + throw new ZipUtilException("Detected loop recursion in directory structure, please check symlink linking to parent directories."); + } + + //判断压缩文件保存的路径是否存在，如果不存在，则创建目录 + File zipDir = new File(zipPath); + if (!zipDir.exists() || !zipDir.isDirectory()) { + zipDir.mkdirs(); + } + + //创建压缩文件保存的文件对象 + String zipFilePath = zipPath + File.separator + zipFileName; + File zipFile = new File(zipFilePath); + if (zipFile.exists()) { + //检测文件是否允许删除，如果不允许删除，将会抛出SecurityException + SecurityManager securityManager = new SecurityManager(); + securityManager.checkDelete(zipFilePath); + //删除已存在的目标文件 + zipFile.delete(); + } + + cos = new CheckedOutputStream(new FileOutputStream(zipFile), new CRC32()); + zos = new ZipOutputStream(cos); + + //如果只是压缩一个文件，则需要截取该文件的父目录 + String srcRootDir = srcPath; + if (srcFile.isFile() || true) { // Hack this stupid setting. We want to keep our least parent folder! + int index = srcPath.lastIndexOf(File.separator); + if (index != -1) { + srcRootDir = srcPath.substring(0, index); + } + } + //调用递归压缩方法进行目录或文件压缩 + zip(srcRootDir, srcFile, zos); + zos.flush(); + } catch (Exception e) { + throw e; + } finally { + try { + if (zos != null) { + zos.close(); + } + } catch (Exception e) { + e.printStackTrace(); + } + } + } + + /** + * 解压缩zip包 + * + * @param zipFilePath zip文件的全路径 + * @param unzipFilePath 解压后的文件保存的路径 + * @param includeZipFileName 解压后的文件保存的路径是否包含压缩文件的文件名。true-包含；false-不包含 + */ + @SuppressWarnings("unchecked") + public static void unzip(String zipFilePath, String unzipFilePath, boolean includeZipFileName) throws ZipUtilException, IOException { + if (zipFilePath.isEmpty() || unzipFilePath.isEmpty()) { + throw new ZipUtilException("Parameter for unzip() contains null."); + } + File zipFile = new File(zipFilePath); + // 如果解压后的文件保存路径包含压缩文件的文件名，则追加该文件名到解压路径 + if (includeZipFileName) { + String fileName = zipFile.getName(); + if (!fileName.isEmpty()) { + fileName = fileName.substring(0, fileName.lastIndexOf(".")); + } + unzipFilePath = unzipFilePath + File.separator + fileName; + } + // 创建解压缩文件保存的路径 + File unzipFileDir = new File(unzipFilePath); + if (!unzipFileDir.exists() || !unzipFileDir.isDirectory()) { + unzipFileDir.mkdirs(); + } + + // 开始解压 + ZipEntry entry = null; + String entryFilePath = null, entryDirPath = null; + File entryFile = null, entryDir = null; + int index = 0, count = 0, bufferSize = 1024; + byte[] buffer = new byte[bufferSize]; + BufferedInputStream bis = null; + BufferedOutputStream bos = null; + ZipFile zip = new ZipFile(zipFile); + Enumeration<ZipEntry> entries = (Enumeration<ZipEntry>) zip.entries(); + // 循环对压缩包里的每一个文件进行解压 + while (entries.hasMoreElements()) { + entry = entries.nextElement(); + // 构建压缩包中一个文件解压后保存的文件全路径 + entryFilePath = unzipFilePath + File.separator + entry.getName(); + // 构建解压后保存的文件夹路径 + index = entryFilePath.lastIndexOf(File.separator); + if (index != -1) { + entryDirPath = entryFilePath.substring(0, index); + } else { + entryDirPath = ""; + } + entryDir = new File(entryDirPath); + // 如果文件夹路径不存在，则创建文件夹 + if (!entryDir.exists() || !entryDir.isDirectory()) { + entryDir.mkdirs(); + } + + // 创建解压文件 + entryFile = new File(entryFilePath); + if (entryFile.exists()) { + // 检测文件是否允许删除，如果不允许删除，将会抛出SecurityException + SecurityManager securityManager = new SecurityManager(); + securityManager.checkDelete(entryFilePath); + // 删除已存在的目标文件 + entryFile.delete(); + } + + // 写入文件 + bos = new BufferedOutputStream(new FileOutputStream(entryFile)); + bis = new BufferedInputStream(zip.getInputStream(entry)); + while ((count = bis.read(buffer, 0, bufferSize)) != -1) { + bos.write(buffer, 0, count); + } + bos.flush(); + bos.close(); + } + zip.close(); + } + +// public static void main(String[] args) +// { +// String zipPath = "d:\\ziptest\\zipPath"; +// String dir = "d:\\ziptest\\rawfiles"; +// String zipFileName = "test.zip"; +// try +// { +// zip(dir, zipPath, zipFileName); +// } +// catch (Exception e) +// { +// e.printStackTrace(); +// } +// +// String zipFilePath = "D:\\ziptest\\zipPath\\test.zip"; +// String unzipFilePath = "D:\\ziptest\\zipPath"; +// try +// { +// unzip(zipFilePath, unzipFilePath, true); +// } +// catch (Exception e) +// { +// e.printStackTrace(); +// } +// } +}
\ No newline at end of file diff --git a/src/main/java/com/keuin/kbackupfabric/util/ZipUtilException.java b/src/main/java/com/keuin/kbackupfabric/util/ZipUtilException.java new file mode 100644 index 0000000..85bfec5 --- /dev/null +++ b/src/main/java/com/keuin/kbackupfabric/util/ZipUtilException.java @@ -0,0 +1,12 @@ +package com.keuin.kbackupfabric.util; + +public class ZipUtilException extends Exception { + private static final long serialVersionUID = 1L; + + public ZipUtilException() { + } + + public ZipUtilException(String gripe) { + super(gripe); + } +} |