EndMinecraftPlusV2/src/cn/serendipityr/EndMinecraftPlusV2/Tools/ProxyUtil.java

package cn.serendipityr.EndMinecraftPlusV2.Tools;

import java.io.*;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;

public class ProxyUtil {
    public static List<String> proxies = new ArrayList<>();
    public static List<Proxy> workingProxiesList = new ArrayList<>();

    public static void getProxies() {
        String getMethod;

        switch (ConfigUtil.ProxyGetType) {
            case 2:
                getMethod = "通过本地文件获取";
                getProxiesFromFile(false, true);
                break;
            case 3:
                getMethod = "通过API+本地文件获取";
                getProxiesFromFile(false, true);
                getProxiesFromAPIs(false, false);
                break;
            case 1:
            default:
                getMethod = "通过API获取";
                getProxiesFromAPIs(false, true);
        }

        LogUtil.doLog(0, "获取代理完成! (" + getMethod + " | 数量: " + proxies.size() + "个)", "ProxyUtil");
        LogUtil.emptyLog();
    }

    public static void getProxiesFromAPIs(boolean async, boolean replace) {
        if (async) {
            List<String> newProxies = proxies;

            if (replace) {
                newProxies = new ArrayList<>();
            }

            for (String url:ConfigUtil.ProxyAPIs) {
                String ips = HTTPUtil.sendGet(url);
                Matcher matcher = OtherUtils.matches(ips, "\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\:\\d{1,5}");
                while (matcher.find()) {
                    String ip = matcher.group();
                    newProxies.add(ip);
                }
            }

            proxies = newProxies;
        } else {
            for (String url:ConfigUtil.ProxyAPIs) {
                if (replace) {
                    proxies = new ArrayList<>();
                }

                String ips = HTTPUtil.sendGet(url);
                Matcher matcher = OtherUtils.matches(ips, "\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\:\\d{1,5}");
                while (matcher.find()) {
                    String ip = matcher.group();
                    proxies.add(ip);
                }
            }
        }
    }


    public static void getProxiesFromFile(boolean async, boolean replace) {
        try {
            if (!ConfigUtil.ProxyFile.exists()) {
                LogUtil.doLog(1, "无法从文件读取代理! 文件不存在。", null);
                return;
            }

            BufferedReader reader = new BufferedReader(new FileReader(ConfigUtil.ProxyFile));
            String tempString;

            if (async) {
                List<String> newProxies = proxies;

                if (replace) {
                    newProxies = new ArrayList<>();
                }

                while ((tempString = reader.readLine()) != null) {
                    newProxies.add(tempString);
                }

                proxies = newProxies;
            } else {
                if (replace) {
                    proxies = new ArrayList<>();
                }

                while ((tempString = reader.readLine()) != null) {
                    proxies.add(tempString);
                }
            }

            reader.close();
        } catch (IOException e) {
            LogUtil.doLog(1, "无法从文件读取代理! IO异常: " + e, null);
        }
    }

    public static void runUpdateProxiesTask(int time) {
        new Thread(() -> {
            while (true) {
                OtherUtils.doSleep(time * 1000L);

                switch (ConfigUtil.ProxyGetType) {
                    case 1:
                        getProxiesFromAPIs(true, true);
                        LogUtil.doLog(0, "代理更新完毕! (通过API获取 | 数量: " + proxies.size() + "个)", "ProxyUtil");
                        break;
                    case 2:
                        getProxiesFromFile(true, true);
                        LogUtil.doLog(0, "代理更新完毕! (通过本地文件获取 | 数量: " + proxies.size() + "个)", "ProxyUtil");
                        break;
                    case 3:
                        getProxiesFromFile(true, true);
                        getProxiesFromAPIs(true, false);
                        LogUtil.doLog(0, "代理更新完毕! (通过API+本地文件获取 | 数量: " + proxies.size() + "个)", "ProxyUtil");
                        break;
                }
            }
        }).start();
    }

    public static void saveWorkingProxy(Proxy proxy) {
        File workingProxies = new File("working-proxies.txt");
        InetSocketAddress inetSocketAddress = (InetSocketAddress) proxy.address();

        List<Proxy> tempList = workingProxiesList;

        if (!tempList.contains(proxy)) {
            try {
                FileWriter fileWriter = new FileWriter(workingProxies, true);
                String proxyAddress = (inetSocketAddress.getAddress() + ":" + inetSocketAddress.getPort() + "\n").replace("/","");
                fileWriter.write(proxyAddress);
                fileWriter.close();
                workingProxiesList.add(proxy);
            } catch (IOException e) {
                LogUtil.doLog(1, "保存有效代理失败! IO异常: " + e.getMessage(), null);
            }
        }
    }
}