use std::io::{self, Read, Stderr, Write};
use io_tee::{TeeReader, TeeWriter};
#[cfg(feature = "plugin")]
use io_tee::{ReadExt, WriteExt};
pub enum DebugReader<R: Read> {
Off(R),
On(TeeReader<R, Stderr>),
}
impl<R: Read> DebugReader<R> {
#[cfg(feature = "plugin")]
#[cfg_attr(docsrs, doc(cfg(feature = "plugin")))]
pub(crate) fn new(reader: R, debug_enabled: bool) -> Self {
if debug_enabled {
DebugReader::On(reader.tee_dbg())
} else {
DebugReader::Off(reader)
}
}
}
impl<R: Read> Read for DebugReader<R> {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
match self {
Self::Off(reader) => reader.read(buf),
Self::On(reader) => reader.read(buf),
}
}
}
pub enum DebugWriter<W: Write> {
Off(W),
On(TeeWriter<W, Stderr>),
}
impl<W: Write> DebugWriter<W> {
#[cfg(feature = "plugin")]
#[cfg_attr(docsrs, doc(cfg(feature = "plugin")))]
pub(crate) fn new(writer: W, debug_enabled: bool) -> Self {
if debug_enabled {
DebugWriter::On(writer.tee_dbg())
} else {
DebugWriter::Off(writer)
}
}
}
impl<W: Write> Write for DebugWriter<W> {
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
match self {
Self::Off(writer) => writer.write(buf),
Self::On(writer) => writer.write(buf),
}
}
fn flush(&mut self) -> io::Result<()> {
match self {
Self::Off(writer) => writer.flush(),
Self::On(writer) => writer.flush(),
}
}
}