import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;

// 假設(shè)這是您動態(tài)編譯并加載的類
class UserProvidedCode {
    public static void main(String[] args) {
        System.out.println("Hello from user code!");
        System.err.println("An error message from user code."); 
        System.out.print("Another line of output.");
    }

    public String calculateSum(int a, int b) {
        int sum = a + b;
        System.out.println("Calculating sum: " + a + " + " + b);
        return "Sum is: " + sum;
    }
}

public class DynamicOutputCapturer {

    /**
     * 捕獲指定方法執(zhí)行時的System.out和System.err輸出。
     *
     * @param targetClass 目標(biāo)類
     * @param methodName 目標(biāo)方法名
     * @param paramTypes 目標(biāo)方法的參數(shù)類型數(shù)組
     * @param args 目標(biāo)方法的參數(shù)值數(shù)組
     * @return 包含標(biāo)準(zhǔn)輸出和標(biāo)準(zhǔn)錯誤的字符串
     * @throws NoSuchMethodException 如果找不到指定方法
     * @throws InvocationTargetException 如果被調(diào)用的方法拋出異常
     * @throws IllegalAccessException 如果無法訪問指定方法
     */
    public static String captureMethodOutput(Class<?> targetClass, String methodName, Class<?>[] paramTypes, Object[] args)
            throws NoSuchMethodException, InvocationTargetException, IllegalAccessException {

        // 保存原始的System.out和System.err流
        PrintStream originalOut = System.out;
        PrintStream originalErr = System.err;

        // 創(chuàng)建用于捕獲輸出的字節(jié)數(shù)組輸出流
        ByteArrayOutputStream baosOut = new ByteArrayOutputStream();
        ByteArrayOutputStream baosErr = new ByteArrayOutputStream();

        // 創(chuàng)建新的PrintStream，將其指向字節(jié)數(shù)組輸出流
        PrintStream newOut = new PrintStream(baosOut, true); // true表示自動刷新
        PrintStream newErr = new PrintStream(baosErr, true);

        try {
            // 重定向System.out和System.err
            System.setOut(newOut);
            System.setErr(newErr);

            // 獲取并調(diào)用目標(biāo)方法
            Method method = targetClass.getMethod(methodName, paramTypes);
            Object instance = null;

            // 如果方法不是靜態(tài)的，需要創(chuàng)建類的實(shí)例
            if (!Modifier.isStatic(method.getModifiers())) {
                try {
                    instance = targetClass.newInstance(); // 嘗試調(diào)用無參構(gòu)造函數(shù)
                } catch (InstantiationException e) {
                    // 如果類沒有公共的無參構(gòu)造函數(shù)，將無法創(chuàng)建實(shí)例
                    System.err.println("Error: Cannot create instance for non-static method. " +
                                       "Class " + targetClass.getName() + " may lack a public no-arg constructor.");
                    throw new RuntimeException("Failed to instantiate class: " + targetClass.getName(), e);
                }
            }

            // 調(diào)用目標(biāo)方法
            Object returnValue = method.invoke(instance, args);

            // 確保所有緩沖區(qū)內(nèi)容被寫入
            newOut.flush();
            newErr.flush();

            // 獲取捕獲到的輸出內(nèi)容
            String capturedStandardOutput = baosOut.toString();
            String capturedErrorOutput = baosErr.toString();

            StringBuilder result = new StringBuilder();
            if (returnValue != null) {
                result.append("Method Return Value: ").append(returnValue.toString()).append("\n");
            }
            if (!capturedStandardOutput.isEmpty()) {
                result.append("Standard Output:\n").append(capturedStandardOutput);
            }
            if (!capturedErrorOutput.isEmpty()) {
                result.append("Error Output:\n").append(capturedErrorOutput);
            }

            return result.toString();

        } finally {
            // 恢復(fù)原始的System.out和System.err
            System.setOut(originalOut);
            System.setErr(originalErr);
            // 關(guān)閉新的PrintStream
            newOut.close();
            newErr.close();
            // ByteArrayOutputStream不需要手動關(guān)閉，因?yàn)樗趦?nèi)存
        }
    }

    public static void main(String[] args) {
        try {
            // 示例1: 調(diào)用UserProvidedCode的main方法并捕獲輸出
            System.out.println("--- 捕獲 UserProvidedCode.main() 的輸出 ---");
            String mainOutput = captureMethodOutput(UserProvidedCode.class, "main", new Class[]{String[].class}, new Object[]{new String[0]});
            System.out.println("捕獲結(jié)果:\n" + mainOutput);

            // 示例2: 調(diào)用UserProvidedCode的calculateSum方法并捕獲輸出
            System.out.println("\n--- 捕獲 UserProvidedCode.calculateSum(5, 3) 的輸出 ---");
            String sumOutput = captureMethodOutput(UserProvidedCode.class, "calculateSum", new Class[]{int.class, int.class}, new Object[]{5, 3});
            System.out.println("捕獲結(jié)果:\n" + sumOutput);

            // 驗(yàn)證原始的System.out是否已恢復(fù)
            System.out.println("\n--- 原始 System.out 已恢復(fù) ---");
            System.out.println("這條消息應(yīng)該正常打印到控制臺。");

        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}