forked from HMCL-dev/HMCL
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'official/main' into prs
- Loading branch information
Showing
3 changed files
with
117 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
99 changes: 99 additions & 0 deletions
99
HMCLCore/src/main/java/org/jackhuang/hmcl/util/io/DirectoryStructurePrinter.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,99 @@ | ||
| package org.jackhuang.hmcl.util.io; | ||
|
|
||
| import java.io.IOException; | ||
| import java.nio.file.FileVisitResult; | ||
| import java.nio.file.FileVisitor; | ||
| import java.nio.file.Files; | ||
| import java.nio.file.Path; | ||
| import java.nio.file.attribute.BasicFileAttributes; | ||
|
|
||
| final class DirectoryStructurePrinter { | ||
| private DirectoryStructurePrinter() { | ||
| } | ||
|
|
||
| public static String list(Path path, int maxDepth) throws IOException { | ||
| StringBuilder output = new StringBuilder(128); | ||
| list(path, maxDepth, output); | ||
| output.setLength(output.length() - 1); | ||
| return output.toString(); | ||
| } | ||
|
|
||
| private static void list(Path path, int maxDepth, StringBuilder output) throws IOException { | ||
| output.append("Filesystem structure of: ").append(path).append('\n'); | ||
|
|
||
| if (!Files.exists(path)) { | ||
| pushMessage(output, "nonexistent path", 1); | ||
| return; | ||
| } | ||
| if (Files.isRegularFile(path)) { | ||
| pushMessage(output, "regular file path", 1); | ||
| return; | ||
| } | ||
| if (Files.isDirectory(path)) { | ||
| Files.walkFileTree(path, new FileVisitor<Path>() { | ||
| private boolean isFolderEmpty; | ||
|
|
||
| private int depth = 1; | ||
|
|
||
| @Override | ||
| public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) { | ||
| isFolderEmpty = true; | ||
|
|
||
| pushFile(output, dir, depth); | ||
| if (depth == maxDepth) { | ||
| pushMessage(output, "too deep", depth); | ||
| return FileVisitResult.SKIP_SUBTREE; | ||
| } | ||
|
|
||
| depth++; | ||
| return FileVisitResult.CONTINUE; | ||
| } | ||
|
|
||
| @Override | ||
| public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) { | ||
| isFolderEmpty = false; | ||
|
|
||
| pushFile(output, file, depth); | ||
| return FileVisitResult.CONTINUE; | ||
| } | ||
|
|
||
| @Override | ||
| public FileVisitResult visitFileFailed(Path file, IOException exc) { | ||
| visitFile(file, null); | ||
|
|
||
| pushMessage(output, exc.toString(), depth); | ||
| return FileVisitResult.CONTINUE; | ||
| } | ||
|
|
||
| @Override | ||
| public FileVisitResult postVisitDirectory(Path dir, IOException exc) { | ||
| if (isFolderEmpty) { | ||
| pushMessage(output, "empty directory", depth); | ||
| } | ||
|
|
||
| depth--; | ||
| return FileVisitResult.CONTINUE; | ||
| } | ||
| }); | ||
| return; | ||
| } | ||
|
|
||
| pushMessage(output, "unknown file type", 1); | ||
| } | ||
|
|
||
| private static void pushFile(StringBuilder output, Path file, int depth) { | ||
| output.append("|"); | ||
| for (int i = 1; i < depth; i++) { | ||
| output.append(" |"); | ||
| } | ||
| output.append("-> ").append(FileUtils.getName(file)).append('\n'); | ||
| } | ||
|
|
||
| private static void pushMessage(StringBuilder output, String message, int depth) { | ||
| output.append("| "); | ||
| for (int i = 1; i < depth; i++) { | ||
| output.append(" | "); | ||
| } | ||
| output.append('<').append(message).append(">\n"); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters