diff --git a/src/main/java/ru/mcs/udk/UDKSearcher.java b/src/main/java/ru/mcs/udk/UDKSearcher.java index 91c9dfd..5a3aa72 100644 --- a/src/main/java/ru/mcs/udk/UDKSearcher.java +++ b/src/main/java/ru/mcs/udk/UDKSearcher.java @@ -12,15 +12,12 @@ import java.io.BufferedReader; import java.io.BufferedWriter; -import java.io.ByteArrayOutputStream; import java.io.File; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.FileWriter; import java.io.IOException; import java.io.PrintStream; -import java.io.RandomAccessFile; -import java.nio.channels.FileChannel; import java.nio.charset.StandardCharsets; import java.nio.file.Files; import java.nio.file.Path; @@ -59,7 +56,7 @@ } // Поиск файлов - List foundFiles = findFilesByExtension(new File(directoryPath), fileFormat.toLowerCase()); + List foundFiles = findFilesByExtension(new File(directoryPath), List.of(fileFormat.toLowerCase().split(","))); // Запись результатов в файл if (foundFiles.isEmpty()) { @@ -80,7 +77,7 @@ } } - private static List findFilesByExtension(File directory, String extension) { + private static List findFilesByExtension(File directory, List extensions) { List foundFiles = new ArrayList<>(); if (directory.isDirectory()) { File[] files = directory.listFiles(); @@ -88,10 +85,13 @@ for (File file : files) { if (file.isDirectory()) { // Рекурсивный поиск в подпапках - foundFiles.addAll(findFilesByExtension(file, extension)); - } else if (file.getName().toLowerCase().endsWith("." + extension)) { - // Проверка расширения файла - foundFiles.add(file); + foundFiles.addAll(findFilesByExtension(file, extensions)); + } else { + for (String extension : extensions) { + if (file.getName().toLowerCase().endsWith("." + extension)) { + foundFiles.add(file); + } + } } } }