Тема: File tree. Візуалізація вмісту папки
Добрий день. Знову прошу вашої допомоги. Ось те що я отримав https://github.com/npogoncuk/file-tree
Ось опис завдання
File Tree
Description
Build a String representation of directory hierarchy under a given path
Details
Implement tree method in FileTreeImpl class.
Input parameter is a path.
If a given path does not exist, return an empty Optional.
If a given path refers to a file, return a String with its name and size like this:
some-file.txt 128 bytes
IF a given path refers to a directory, return a String with its name, total size and its full hierarchy:
some-dir 100 bytes
├─ some-inner-dir 50 bytes
│  ├─ some-file.txt 20 bytes    
│  └─ some-other-file.txt 30 bytes
└─ some-one-more-file.txt 50 bytes
Use pseudo graphic symbols to format output.
Compute directory size as a sum of all its contents.
Sort content in following way:
directories go first,
directories and files are sorted in lexicographic order (case-insensitive).Моя реалізація
import java.io.File;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;
public class FileTreeImpl implements FileTree {
    @Override
    public Optional<String> tree(Path path) {
        File file = new File(String.valueOf(path));
        if ( !file.exists()) return Optional.empty();
        if ( file.isFile()) {
            return Optional.of(file.getName() + " " + file.length() + " bytes");
        }
        if (file.isDirectory()) {
            return Optional.of(directoryTree(file, 0, false));
        }
        return Optional.empty();
    }
    private String directoryTree(File folder, int depth, boolean lastFolder) {
        String directory = folder.getName() + " " + folderSize(folder);
        if (depth != 0) directory = ((!lastFolder) ? "├─ " : "└─ ") + directory;
        File[] files = folder.listFiles();
        int count = files.length;
        files = sortFiles(files);
        for (int i = 0; i < count; i++) {
            directory += "\n";
            if (depth != 0) directory += "│  ";
            if (files[i].isFile()) {
                directory += (lastFolder ? "   " : "│  ") + (i + 1 == count ? "└" : "├") + "─ " +
                        files[i].getName() + " " + files[i].length() + " bytes";
            } else {
                directory += directoryTree(files[i], depth + 1, i + 1 == numberOfFolders(files));
            }
        }
        System.out.println(directory);
        return directory;
    }
    private long getFolderSize(File folder) {
        long size = 0;
        File[] files = folder.listFiles();
        int count = files.length;
        for (int i = 0; i < count; i++) {
            if (files[i].isFile()) {
                size += files[i].length();
            } else {
                size += getFolderSize(files[i]);
            }
        }
        return size;
    }
    private String folderSize(File folder) {
        return getFolderSize(folder) + " bytes";
    }
    private File[] sortFiles(File[] folder) {
        // сортує за іменами
        Arrays.sort(folder);
        List<File> sorted = new ArrayList<>();
        // спочатку вибираю папки
        for (int i = 0; i < folder.length; i++) {
            if (folder[i].isDirectory()) sorted.add(folder[i]);
        }
        // тепер вибираю файли
        for (int i = 0; i < folder.length; i++) {
            if (folder[i].isFile()) sorted.add(folder[i]);
        }
        return sorted.toArray(new File[sorted.size()]);
    }
    private int numberOfFolders(File[] files) {
        int folders = 0;
        for (File file : files)
            if (file.isDirectory()) folders++;
        return folders;
    }
}Ось тести
import org.junit.Test;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.stream.Collectors;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
public class FileTreeTest {
    @Test
    public void testEmptyCases() {
        assertTrue("null case failure", new FileTreeImpl().tree(null).isEmpty());
        assertTrue("nonexistent path case failure", new FileTreeImpl().tree(Paths.get("some", " non", "existent", "path")).isEmpty());
    }
    @Test
    public void testFileCases() {
        testCase("some.txt 0 bytes", tree("test1", "some.txt"));
        testCase("some1.txt 4 bytes", tree("test1", "some1.txt"));
        testCase("some2.txt 8 bytes", tree("test1", "some2.txt"));
    }
    @Test
    public void testDirCase1() throws IOException {
        testDirCase("test1");
    }
    @Test
    public void testDirCase2() throws IOException {
        testDirCase("test2");
    }
    @Test
    public void testDirCase3() throws IOException {
        testDirCase("test3");
    }
    @Test
    public void testDirCase4() throws IOException {
        testDirCase("test4");
    }
    @Test
    public void testDirCase5() throws IOException {
        testDirCase("test5");
    }
    @Test
    public void testDirCase6() throws IOException {
        testDirCase("test6");
    }
    @Test
    public void testDirCase7() throws IOException {
        testDirCase("test7");
    }
    @Test
    public void testDirCase8() throws IOException {
        testDirCase("test8");
    }
    private void testDirCase(final String caseName) throws IOException {
        testCase(expectedFile(caseName), tree(caseName));
    }
    private void testCase(final String expected, final String actual) {
        assertEquals(expected.trim(), actual.trim());
    }
    private String expectedFile(String caseName) throws IOException {
        return Files.lines(Paths.get("src/test/resources", caseName + ".txt"))
                .collect(Collectors.joining("\n"));
    }
    private String tree(final String... subPath) {
        return new FileTreeImpl().tree(Paths.get("src/test/resources", subPath)).orElse(null);
    }
}Перші два тести проходить. А далі я заплутався як малювати вертикальні лінії. Хотів вставити скрін і показати що вийшло, бо все ж таки щось подібне я отримав(проте буває надто багато вертикальних ліній або відступів), але це теж мені не виходить.
Підкажіть, будь ласка, що виправити в коді або новий алгоритм.
Дякую за витрачений час і буду радий будь-яким підказкам.