如何在OCaml中通过Unix模块与raw模式检测ESC键？

问题分析 #

现有代码遇到\027（ESC字符）时，会强制读取两个后续字符，但单独按下ESC键时，终端仅发送单个\027字节，这会导致input_char阻塞等待输入，且无法识别单独的ESC键。

解决方案思路 #

利用非阻塞IO判断ESC之后是否有后续字符：

• 读取到ESC时，临时将stdin设置为非阻塞模式
• 尝试读取后续字符：若能读取到，说明是方向键/功能键的转义序列，按原逻辑处理；若读取失败（无后续字符），则判定为单独ESC键
• 处理完成后恢复stdin的原始模式，避免影响后续输入

修改后的代码 #

open Unix

(* 保存终端原始属性，用于后续恢复 *)
let original_termios = tcgetattr stdin

(* 启用终端Raw模式，关闭规范输入与回显 *)
let enable_raw_mode () =
  let termios = tcgetattr stdin in
  let new_termios = { termios with
    c_icanon = false;
    c_echo = false;
    c_vmin = 1;
    c_vtime = 0;
  } in
  tcsetattr stdin TCSANOW new_termios

(* 恢复终端原始模式 *)
let disable_raw_mode () =
  tcsetattr stdin TCSANOW original_termios

(* 非阻塞读取单个字符，返回option类型处理无输入情况 *)
let read_char_nonblock () =
  try
    let c = input_char (in_channel_of_descr stdin) in
    Some c
  with
  | End_of_file -> None
  | Unix.Unix_error (Unix.EAGAIN, _, _) -> None

(* 读取字符并识别特殊按键，包含单独ESC键检测 *)
let read_character () =
  let in_chan = in_channel_of_descr stdin in
  let c = input_char in_chan in
  if c = '\027' then
    begin
      (* 将stdin切换为非阻塞模式 *)
      let flags = fcntl stdin F_GETFL in
      fcntl stdin F_SETFL (flags lor O_NONBLOCK);

      let result =
        match read_char_nonblock () with
        | Some '[' ->
            begin
              match read_char_nonblock () with
              | Some 'A' -> "Up Arrow"
              | Some 'B' -> "Down Arrow"
              | Some 'C' -> "Right Arrow"
              | Some 'D' -> "Left Arrow"
              | Some 'H' -> "Home"
              | Some 'F' -> "End"
              | Some '3' ->
                  if read_char_nonblock () = Some '~' then "Delete" else "Ignore"
              | Some '2' ->
                  if read_char_nonblock () = Some '~' then "Insert" else "Ignore"
              | Some '5' ->
                  if read_char_nonblock () = Some '~' then "Page Up" else "Ignore"
              | Some '6' ->
                  if read_char_nonblock () = Some '~' then "Page Down" else "Ignore"
              | _ -> "Ignore"
            end
        | _ ->
            (* 无后续字符，判定为单独ESC键 *)
            "ESC"
      in

      (* 恢复stdin为阻塞模式 *)
      fcntl stdin F_SETFL flags;
      result
    end
  else
    match c with
    | '\n' -> "Enter"
    | '\127' -> "Backspace"
    | _ -> String.make 1 c

(* 示例主函数：启用Raw模式，循环检测按键，按ESC退出 *)
let () =
  enable_raw_mode ();
  try
    while true do
      let key = read_character () in
      Printf.printf "Pressed: %s\n" key;
      if key = "ESC" then exit 0;
    done
  with
  | _ -> disable_raw_mode ()

关键修改说明 #

• 新增read_char_nonblock函数，处理非阻塞读取的异常，返回option<char>类型区分有/无输入
• 遇到ESC时临时切换stdin为非阻塞模式，通过尝试读取后续字符判断是转义序列还是单独ESC键
• 处理完成后恢复stdin的阻塞模式，避免影响后续输入逻辑
• 补充终端Raw模式的启用与恢复函数，确保终端状态在程序退出时回归正常

内容的提问来源于stack exchange，提问作者Enio Lecoder