use crate::{FromStream, ToStream};

use std::io::{Error, ErrorKind, Read, Result, Write};

impl<S> ToStream<S> for bool {
    fn to_stream<W: Write>(&self, stream: &mut W) -> Result<()> {
        let bytes = [if *self { 1 } else { 0 }];
        stream.write_all(&bytes)?;
        Ok(())
    }
}

impl<S> FromStream<S> for bool {
    fn from_stream<R: Read>(stream: &mut R) -> Result<Self> {
        let mut bytes = [0; 1];

        stream.read_exact(&mut bytes)?;

        match bytes[0] {
            0 => Ok(false),
            1 => Ok(true),
            _ => Err(Error::new(
                ErrorKind::InvalidData,
                "Invalid byte representation",
            )),
        }
    }
}