supercilex / io-adapters Goto Github PK

use std::fmt::Formatter;
use std::io::{Error, ErrorKind, Result, Write};
use std::str::from_utf8;

pub struct FmtToIoAdapter<'a, 'b> {
  formatter: &'a mut std::fmt::Formatter<'b>,
}

impl<'a, 'b> From<&'a mut std::fmt::Formatter<'b>> for FmtToIoAdapter<'a, 'b> {
  fn from(formatter: &'a mut Formatter<'b>) -> Self {
    FmtToIoAdapter { formatter }
  }
}

impl Write for FmtToIoAdapter<'_, '_> {
  fn write(&mut self, buf: &[u8]) -> Result<usize> {
    let str = from_utf8(buf).map_err(|e| Error::new(ErrorKind::InvalidData, e))?;
    self.formatter.write_str(str).map_err(|e| Error::new(ErrorKind::Other, e))?;
    Ok(buf.len())
  }

  fn flush(&mut self) -> Result<()> {
    Ok(())
  }
}