src/main/java/com/example/githubreader/service/DirectoryContentService.java - master - KolchenkoAleksey/github-reader

src/main/java/com/example/githubreader/service/DirectoryContentService.java
129 строк6 KB
Aleksey
add DirectoryConfig, DirectoryContentService, DirectoryController
23 фев 2025, 00:05
23 фев 2025, 00:05b27eb50
// File: src/main/java/com/example/githubreader/service/DirectoryContentService.java
package com.example.githubreader.service;

import com.example.githubreader.config.DirectoryConfig;
import com.example.githubreader.config.GithubConfig;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.stream.Collectors;

/**
 * Сервис для работы с содержимым локальных директорий.
 */
@Slf4j
@Service
@RequiredArgsConstructor
public class DirectoryContentService {

    private final DirectoryConfig directoryConfig;
    private final GithubConfig githubConfig;  // Добавляем GithubConfig для паттернов
    private final ExecutorService executorService = Executors.newVirtualThreadPerTaskExecutor();

    /**
     * Считывает содержимое всех файлов из директории и сохраняет в один файл.
     *
     * @param directoryPath Путь к директории (опционально, если не указан — используется defaultPath)
     */
    public void saveAllContentsToFile(String directoryPath) {
        String effectivePath = directoryPath != null ? directoryPath : directoryConfig.getDefaultPath();
        if (effectivePath == null) {
            throw new IllegalArgumentException("Directory path is not provided and defaultPath is not configured");
        }

        try {
            Path dirPath = Paths.get(effectivePath);
            if (!Files.isDirectory(dirPath)) {
                throw new IllegalArgumentException("Provided path is not a directory: " + effectivePath);
            }

            List<String> contents = new ArrayList<>();
            List<Future<?>> futures = new ArrayList<>();

            Files.walk(dirPath)
                    .filter(Files::isRegularFile)
                    .filter(this::matchesPatterns)  // Добавляем фильтр по паттернам
                    .forEach(filePath -> {
                        Future<?> future = executorService.submit(() -> {
                            try {
                                String content = Files.readString(filePath);
                                String relativePath = dirPath.relativize(filePath).toString();
                                String formattedContent = "File: " + relativePath + "\n" + content + "\n";
                                synchronized (contents) {
                                    contents.add(formattedContent);
                                }
                                log.debug("Processed file: {}", relativePath);
                            } catch (IOException e) {
                                log.error("Failed to read file {}: {}", filePath, e.getMessage());
                            }
                        });
                        futures.add(future);
                    });

            for (Future<?> future : futures) {
                future.get();
            }

            String dirName = dirPath.getFileName().toString();
            String outputFileName = "all_contents_from_" + dirName + ".txt";
            Path outputPath = Paths.get("output", outputFileName);

            Files.createDirectories(outputPath.getParent());
            String allContent = contents.stream().collect(Collectors.joining("\n"));
            Files.writeString(outputPath, allContent);

            log.info("Saved all directory contents to file: {}", outputPath);

        } catch (Exception e) {
            log.error("Failed to process directory {}: {}", effectivePath, e.getMessage());
            throw new RuntimeException("Failed to save directory contents to file", e);
        }
    }

    /**
     * Проверяет, соответствует ли путь include и exclude паттернам.
     *
     * @param filePath Путь к файлу
     * @return true, если файл должен быть включен
     */
    private boolean matchesPatterns(Path filePath) {
        String path = filePath.toString().replace("\\", "/"); // Нормализуем слеши для совместимости
        for (String exclude : githubConfig.getExcludePatterns()) {
            if (matchesPattern(path, exclude)) {
                return false;
            }
        }
        for (String include : githubConfig.getIncludePatterns()) {
            if (matchesPattern(path, include)) {
                return true;
            }
        }
        return false; // Если не соответствует включающим паттернам, исключаем
    }

    /**
     * Проверяет, соответствует ли путь заданному паттерну.
     *
     * @param path Путь к файлу
     * @param pattern Паттерн для проверки
     * @return true, если путь соответствует паттерну
     */
    private boolean matchesPattern(String path, String pattern) {
        String regex = pattern
                .replace(".", "\\.")
                .replace("*", ".*")
                .replace("?", ".")
                .replace("/", "[/\\\\]"); // Учитываем оба типа слешей
        return path.matches(".*" + regex + ".*"); // Добавляем .* для проверки частичного совпадения
    }
}