use crate::{parse, RichAnnotation, RichDecorator};
use std::fmt::Write;
use std::io;
pub fn from_read_coloured<R, FMap>(
input: R,
width: usize,
colour_map: FMap,
) -> Result<String, crate::Error>
where
R: io::Read,
FMap: Fn(&[RichAnnotation], &str) -> String,
{
let lines = parse(input)?
.render(width, RichDecorator::new())?
.into_lines()?;
let mut result = String::new();
for line in lines {
for ts in line.tagged_strings() {
write!(result, "{}", colour_map(&ts.tag, &ts.s))?;
}
result.push('\n');
}
Ok(result)
}