package com.wang.you.util;

import java.io.File;
import java.util.Arrays;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

public class FileRenamer {

    private static final String NEW_PREFIX = "ic_wangyou_color";
    private static final String FOLDER_PATH = "/storage/emulated/0/wangyou";

    public static void rename() {
        File folder = new File(FOLDER_PATH);
        if (!folder.exists() || !folder.isDirectory()) {
            System.out.println("指定的路徑不是一個有效的文件夾！");
            return;
        }

        String[] subFolders = {"xhdpi", "xxhdpi", "xxxhdpi"};

        for (String subFolderName : subFolders) {
            File subFolder = new File(folder, subFolderName);
            if (subFolder.exists() && subFolder.isDirectory()) {
                processSubFolder(subFolder);
            } else {
                System.out.println("子文件夾 " + subFolderName + " 不存在或不是一個有效的文件夾！");
            }
        }
    }

    private static void processSubFolder(File subFolder) {
        File[] files = subFolder.listFiles((dir, name) -> name.toLowerCase().endsWith(".png"));
        if (files == null || files.length == 0) {
            System.out.println("在文件夾 " + subFolder.getName() + " 中沒有找到任何 .png 文件！");
            return;
        }

        for (File oldFile : files) {
            String oldFileName = oldFile.getName();
            Pattern pattern = Pattern.compile("(\\d+)_(\\w+\\.png)");
            Matcher matcher = pattern.matcher(oldFileName);

            if (matcher.matches()) {
                String numberPart = matcher.group(1);
                String colorValueWithExt = matcher.group(2);
                String colorValue = colorValueWithExt.split("\\.")[0];

                String newFileName = NEW_PREFIX + numberPart + "_" + colorValue + ".png";
                File newFile = new File(subFolder, newFileName);

                if (oldFile.renameTo(newFile)) {
                    System.out.println("Renamed: " + oldFileName + " -> " + newFileName);
                } else {
                    System.out.println("Failed to rename: " + oldFileName);
                }
            } else {
                System.out.println("Invalid file name format: " + oldFileName);
            }
        }

        outputCodeSnippet(subFolder);
    }

    private static void outputCodeSnippet(File folder) {
        Pattern pattern = Pattern.compile("(?<=color)(\\d+)(?=_)");

        List<File> allPngFiles = Arrays.stream(folder.listFiles())
                .filter(file -> file.getName().toLowerCase().endsWith(".png"))
                .collect(Collectors.toList());

        List<File> renamedFiles = allPngFiles.stream()
                .filter(file -> file.getName().startsWith(NEW_PREFIX))
                .sorted((f1, f2) -> {
                    Matcher m1 = pattern.matcher(f1.getName());
                    Matcher m2 = pattern.matcher(f2.getName());
                    if (m1.find() && m2.find()) {
                        return Integer.compare(Integer.parseInt(m1.group()), Integer.parseInt(m2.group()));
                    }
                    return 0;
                })
                .collect(Collectors.toList());

        System.out.println("準備輸出字符串renamedFiles: " + renamedFiles.size());

        StringBuilder outputString = new StringBuilder("private final int[] wangyou_colorArr_" + folder.getName() + " = new int[]{\n");

        for (File file : renamedFiles) {
            String fileNameWithoutExtension = file.getName().replaceFirst("[.][^.]+$", "");
            outputString.append("            R.mipmap.")
                    .append(fileNameWithoutExtension)
                    .append(",\n");
        }

        if (outputString.toString().endsWith(",\n")) {
            outputString.setLength(outputString.length() - 2);
        }
        outputString.append("\n};");

        System.out.println(outputString.toString());

        List<File> unmatchedFiles = allPngFiles.stream()
                .filter(file -> !file.getName().startsWith(NEW_PREFIX))
                .collect(Collectors.toList());

        if (!unmatchedFiles.isEmpty()) {
            System.out.println("以下文件未被包含：");
            unmatchedFiles.forEach(file -> System.out.println("    " + file.getName()));
        }
    }
}