// =================================================
// 文件：src/main/java/com/example/filerenamer/FileFilterStrategy.java
// =================================================
package com.example.filerenamer;
 
import java.nio.file.Path;
 
/**
 * 文件過(guò)濾策略接口
 */
public interface FileFilterStrategy {
    /**
     * 判斷是否接受該文件進(jìn)行重命名
     * @param path 文件路徑
     * @return true 則重命名
     */
    boolean accept(Path path);
}
 
// =================================================
// 文件：src/main/java/com/example/filerenamer/SuffixFilterStrategy.java
// =================================================
package com.example.filerenamer;
 
import java.nio.file.Path;
import java.util.Set;
 
/**
 * 按文件后綴過(guò)濾策略
 */
public class SuffixFilterStrategy implements FileFilterStrategy {
    private final Set<String> sourceSuffixes;
    private final boolean ignoreCase;
 
    public SuffixFilterStrategy(Set<String> suffixes, boolean ignoreCase) {
        this.sourceSuffixes = suffixes;
        this.ignoreCase = ignoreCase;
    }
 
    @Override
    public boolean accept(Path path) {
        String name = path.getFileName().toString();
        int idx = name.lastIndexOf('.');
        if (idx < 0) return false;
        String ext = name.substring(idx + 1);
        return sourceSuffixes.stream()
            .anyMatch(s -> ignoreCase
                ? s.equalsIgnoreCase(ext)
                : s.equals(ext));
    }
}
 
// =================================================
// 文件：src/main/java/com/example/filerenamer/RenameHook.java
// =================================================
package com.example.filerenamer;
 
import java.nio.file.Path;
 
/**
 * 重命名鉤子接口
 */
public interface RenameHook {
    /**
     * 重命名前回調(diào)
     * @param oldPath 原始文件路徑
     */
    void before(Path oldPath);
 
    /**
     * 重命名后回調(diào)
     * @param newPath 新文件路徑
     */
    void after(Path newPath);
}
 
// =================================================
// 文件：src/main/java/com/example/filerenamer/BatchResult.java
// =================================================
package com.example.filerenamer;
 
import java.nio.file.Path;
import java.util.List;
 
/**
 * 批量重命名結(jié)果
 */
public class BatchResult {
    private final List<Path> succeeded;
    private final List<Path> failed;
    public BatchResult(List<Path> succ, List<Path> fail) {
        this.succeeded = succ; this.failed = fail;
    }
    public List<Path> getSucceeded() { return succeeded; }
    public List<Path> getFailed() { return failed; }
}
 
// =================================================
// 文件：src/main/java/com/example/filerenamer/FileTypeBatchRenamer.java
// =================================================
package com.example.filerenamer;
 
import java.io.IOException;
import java.nio.file.*;
import java.util.*;
import java.util.concurrent.ForkJoinPool;
import java.util.stream.Collectors;
import java.util.stream.Stream;
 
/**
 * 批量修改文件類(lèi)型工具類(lèi)
 */
public class FileTypeBatchRenamer {
 
    private final Path rootDir;
    private final String targetSuffix;
    private final boolean backup;
    private final FileFilterStrategy filter;
    private final List<RenameHook> hooks;
 
    public FileTypeBatchRenamer(Path rootDir,
                                String targetSuffix,
                                boolean backup,
                                FileFilterStrategy filter,
                                List<RenameHook> hooks) {
        this.rootDir = rootDir;
        this.targetSuffix = targetSuffix.startsWith(".")
            ? targetSuffix.substring(1) : targetSuffix;
        this.backup = backup;
        this.filter = filter;
        this.hooks = hooks != null ? hooks : Collections.emptyList();
    }
 
    /**
     * 同步批量重命名
     */
    public BatchResult renameAll() {
        List<Path> succ = new ArrayList<>();
        List<Path> fail = new ArrayList<>();
        try (Stream<Path> stream = Files.walk(rootDir)) {
            for (Path p : (Iterable<Path>) stream::iterator) {
                if (!Files.isRegularFile(p) || !filter.accept(p)) continue;
                try {
                    hooks.forEach(h -> h.before(p));
                    Path renamed = doRename(p);
                    hooks.forEach(h -> h.after(renamed));
                    succ.add(renamed);
                } catch (IOException ex) {
                    fail.add(p);
                }
            }
        } catch (IOException e) {
            // 根目錄無(wú)法訪問(wèn)
        }
        return new BatchResult(succ, fail);
    }
 
    /**
     * 并行批量重命名
     */
    public BatchResult renameAllParallel() {
        ForkJoinPool pool = new ForkJoinPool();
        try {
            List<Path> all = Files.walk(rootDir)
                .filter(Files::isRegularFile)
                .filter(filter::accept)
                .collect(Collectors.toList());
            List<Path> succ = Collections.synchronizedList(new ArrayList<>());
            List<Path> fail = Collections.synchronizedList(new ArrayList<>());
            pool.submit(() ->
                all.parallelStream().forEach(p -> {
                    try {
                        hooks.forEach(h -> h.before(p));
                        Path r = doRename(p);
                        hooks.forEach(h -> h.after(r));
                        succ.add(r);
                    } catch (IOException ex) {
                        fail.add(p);
                    }
                })
            ).get();
            return new BatchResult(succ, fail);
        } catch (Exception e) {
            return new BatchResult(Collections.emptyList(), Collections.emptyList());
        } finally {
            pool.shutdown();
        }
    }
 
    // 執(zhí)行單個(gè)文件重命名（含備份邏輯）
    private Path doRename(Path p) throws IOException {
        String name = p.getFileName().toString();
        int idx = name.lastIndexOf('.');
        String base = idx < 0 ? name : name.substring(0, idx);
        if (backup) {
            Path bak = p.resolveSibling(base + "." + targetSuffix + ".bak");
            Files.copy(p, bak, StandardCopyOption.REPLACE_EXISTING);
        }
        Path dest = p.resolveSibling(base + "." + targetSuffix);
        return Files.move(p, dest, StandardCopyOption.REPLACE_EXISTING);
    }
 
    /**
     * 示例 main
     */
    public static void main(String[] args) {
        Path dir = Paths.get("C:/data/files");
        Set<String> srcSuffix = Set.of("jpg","png","bmp");
        FileFilterStrategy filter = new SuffixFilterStrategy(srcSuffix, true);
        FileTypeBatchRenamer renamer = new FileTypeBatchRenamer(
            dir, "webp", true, filter, null
        );
        BatchResult result = renamer.renameAllParallel();
        System.out.println("成功：" + result.getSucceeded().size());
        System.out.println("失敗：" + result.getFailed().size());
    }
}