asp/keep
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: Use io::copy to simplify StripAnsiFilter

Browse Source 
Co-authored-by: aider (openai/andrew/openrouter/deepseek/deepseek-chat-v3.1) <aider@aider.chat>
This commit is contained in:
Andrew Phillips
2025-09-02 11:58:22 -03:00
parent dce89e2f67
commit 2773855cec
1 changed files with 1 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/filter_plugin/strip_ansi.rs
View File

@@ -1,7 +1,5 @@
use std::io::{Result, Read, Write}; use std::io::{Result, Read, Write};
use strip_ansi_escapes::Writer; use strip_ansi_escapes::Writer;
use crate::common::PIPESIZE;
use super::FilterPlugin; use super::FilterPlugin;
pub struct StripAnsiFilter; pub struct StripAnsiFilter;
@@ -14,16 +12,8 @@ impl StripAnsiFilter {
impl FilterPlugin for StripAnsiFilter { impl FilterPlugin for StripAnsiFilter {
fn filter<R: Read, W: Write>(&mut self, reader: &mut R, writer: &mut W) -> Result<()> { fn filter<R: Read, W: Write>(&mut self, reader: &mut R, writer: &mut W) -> Result<()> {
let mut buffer = vec![0; PIPESIZE];
let mut ansi_writer = Writer::new(writer); let mut ansi_writer = Writer::new(writer);
std::io::copy(reader, &mut ansi_writer)?;
loop {
let bytes_read = reader.read(&mut buffer)?;
if bytes_read == 0 {
break;
}
ansi_writer.write_all(&buffer[..bytes_read])?;
}
ansi_writer.flush()?; ansi_writer.flush()?;
Ok(()) Ok(())
} }