读取管道输入后重置stdin，解决termbg终端主题检测失效问题

我需要从标准输入（stdin）读取Markdown内容，然后以终端友好的带彩色格式输出。为此我使用termbg crate来检测终端的明暗主题。

读取管道输入的代码如下：

fn read_from_stdin() -> String {
  let stdin = io::stdin();
  let stdin_lock = stdin.lock();
  let buffer = stdin_lock.lines().filter_map(|l| l.ok()).fold("".to_string(), |acc, s| { format!("{acc}\n{s}")});
  return buffer;
}

读取完stdin后，我尝试用termbg检测终端主题：

let timeout = std::time::Duration::from_millis(100);
let theme = termbg::theme(timeout);
match theme {
  Ok(t) => {
    println!("Detected theme: {:?}", t);
  },
  _ => {
    println!("No theme detected!"); // 在tmux中会出现这种情况
  }
}

问题在于，termbg通过OSC查询来检测主题，这类查询需要使用stdin/stdout。只有当我用硬编码字符串替代stdin读取、且不通过管道输入内容时，主题检测才能正常工作。显然termbg需要干净的stdin环境，但我无法确保这一点。

如何在检测终端主题前，确保stdin处于termbg可正常工作的空状态？

完整可测试代码：

use std::{io::{self, BufRead}}

fn main() -> io::Result<()> {
  let buffer = read_from_stdin();
  
  let timeout = std::time::Duration::from_millis(100);
  let theme = termbg::theme(timeout);
  match theme { // 大多输出错误主题或完全检测不到
    Ok(t) => {
      println!("Detected theme: {:?}", t);
    },
    _ => {
      println!("No theme detected!"); // 在tmux中会出现这种情况
    }
  }
  
  // 解析和格式化从stdin读取的Markdown的代码在此处，不过和问题无关
  
  return Ok(());
}

fn read_from_stdin() -> String {
  let stdin = io::stdin();
  let stdin_lock = stdin.lock();
  let buffer = stdin_lock.lines().filter_map(|l| l.ok()).fold("".to_string(), |acc, s| { format!("{acc}\n{s}")});
  return buffer;
}

1. 优先执行主题检测，再读取stdin #

termbg的主题检测依赖未被占用的stdin/stdout资源，因此最直接的解决方式是先完成主题检测，再读取stdin内容。这样检测操作执行时，stdin还未被锁定或读取，能保证环境符合termbg的要求。

修改后的main函数示例：

fn main() -> io::Result<()> {
  // 先检测终端主题
  let timeout = std::time::Duration::from_millis(100);
  let theme = termbg::theme(timeout);
  match theme {
    Ok(t) => {
      println!("Detected theme: {:?}", t);
    },
    _ => {
      println!("No theme detected!");
    }
  }
  
  // 再读取stdin中的Markdown内容
  let buffer = read_from_stdin();
  
  // 后续解析、格式化逻辑...
  
  Ok(())
}

2. 确保stdin锁完全释放 #

如果业务逻辑要求必须先读取stdin，需要保证在执行主题检测前完全释放stdin的锁。你的read_from_stdin函数中，stdin_lock是局部变量，函数结束后会自动释放锁，但如果存在其他未释放的stdin锁，会干扰termbg的检测。

另外，若stdin已经被管道输入填满，termbg的OSC查询会被输入内容干扰，这种情况下优先检测主题是唯一可靠的方案。

3. 处理tmux环境的兼容问题 #

tmux会拦截OSC查询指令，导致termbg无法直接检测主题。可以通过两种方式解决：

• 在tmux配置文件中添加set -g allow-passthrough on，允许终端控制指令穿透到外层终端；
• 添加 fallback 逻辑：当termbg检测失败时，读取用户手动设置的环境变量（如TERM_THEME=light）来指定主题。

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