Java在Windows CMD终端实现ANSI颜色输出的策略与实践

import java.io.IOException;

public class ConsoleOutput {
    // 定义ANSI颜色代码
    public static final String ANSI_RESET_ALL = "\33[0m"; // 重置所有属性
    public static final String ANSI_YELLOW_FG = "\33[33m"; // 黄色前景色
    public static final String ANSI_RESET_FG = "\33[39m"; // 重置前景色

    public static void main(String[] args) {
        // 使用自定义的println方法输出彩色文本
        println(ANSI_YELLOW_FG + "This text is yellow" + ANSI_RESET_FG);
        println("This is normal color"); // 输出普通颜色文本
    }

    /**
     * 通过执行外部cmd命令来输出字符串，支持ANSI颜色。
     * @param s 待输出的字符串，可包含ANSI转义序列。
     */
    static void println(String s) {
        try {
            // 构建并执行cmd命令：cmd /c echo [字符串]
            // /c 参数表示执行完命令后关闭cmd窗口
            new ProcessBuilder("cmd", "/c", "echo " + s)
                .inheritIO() // 将子进程的标准输入、输出、错误流重定向到当前Java进程
                .start()     // 启动子进程
                .waitFor();  // 等待子进程执行完毕
        } catch (InterruptedException | IOException e) {
            // 捕获中断异常或IO异常
            throw new RuntimeException("Failed to execute external command for console output.", e);
        }
    }
}

import java.lang.foreign.*;
import java.lang.invoke.MethodHandle;

public class ConsoleOutputFFM {
    // 定义ANSI颜色代码
    public static final String ANSI_RESET_ALL = "\33[0m";
    public static final String ANSI_YELLOW_FG = "\33[33m";
    public static final String ANSI_RESET_FG = "\33[39m";

    // 定义Windows API常量
    static final int STD_OUTPUT_HANDLE = -11; // 标准输出句柄
    static final int ENABLE_PROCESSED_OUTPUT = 0x0001; // 启用处理过的输出
    static final int ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004; // 启用虚拟终端处理

    public static void main(String[] args) {
        // 尝试初始化ANSI支持
        if (initANSI()) {
            System.out.println("ANSI support initialized successfully.");
        } else {
            System.out.println("Failed to initialize ANSI support, falling back to default.");
        }

        // 输出彩色文本
        System.out.println(ANSI_YELLOW_FG + "This text is yellow" + ANSI_RESET_FG);
        System.out.println("This is normal color");
    }

    /**
     * 使用Java FFM API调用Windows API来启用控制台的虚拟终端处理模式。
     * @return 如果成功启用ANSI支持，则返回true；否则返回false。
     */
    static boolean initANSI() {
        try (Arena arena = Arena.ofConfined()) { // 创建一个受限的内存区域
            // 加载kernel32.dll库
            SymbolLookup sl = SymbolLookup.libraryLookup("kernel32.dll", arena);
            Linker linker = Linker.nativeLinker();

            // 获取GetStdHandle函数句柄
            MethodHandle GetStdHandle = linker.downcallHandle(
                sl.find("GetStdHandle").orElseThrow(() -> new NoSuchMethodException("GetStdHandle not found")),
                FunctionDescriptor.of(ValueLayout.ADDRESS, ValueLayout.JAVA_INT)
            );

            // 获取SetConsoleMode函数句柄
            MethodHandle SetConsoleMode = linker.downcallHandle(
                sl.find("SetConsoleMode").orElseThrow(() -> new NoSuchMethodException("SetConsoleMode not found")),
                FunctionDescriptor.of(ValueLayout.JAVA_BOOLEAN, ValueLayout.ADDRESS, ValueLayout.JAVA_INT)
            );

            // 调用GetStdHandle获取标准输出句柄
            MemorySegment consoleHandle = (MemorySegment) GetStdHandle.invokeExact(STD_OUTPUT_HANDLE);

            // 调用SetConsoleMode启用虚拟终端处理
            return (boolean) SetConsoleMode.invokeExact(
                consoleHandle,
                ENABLE_PROCESSED_OUTPUT | ENABLE_VIRTUAL_TERMINAL_PROCESSING
            );
        } catch (Throwable e) {
            // 捕获所有可能的异常并打印堆栈信息
            System.err.println("Error enabling ANSI support via FFM API: " + e.getMessage());
            e.printStackTrace();
            return false;
        }
    }
}