deboa-extras 0.0.9-beta.11

use bytes::{Buf, Bytes};
use std::io::{Read, Write};

use brotli::CompressorWriter;
use deboa::{
    errors::{DeboaError, IoError},
    fs::io::{Compressor, Decompressor},
    request::DeboaRequest,
    response::DeboaResponse,
    Result,
};

#[derive(PartialEq)]
/// Brotli compression implementation for Deboa.
pub struct BrotliCompressor;

#[deboa::async_trait]
impl Compressor for BrotliCompressor {
    fn name(&self) -> String {
        "br".to_string()
    }

    async fn compress_body(&self, request: &DeboaRequest) -> Result<Bytes> {
        let mut writer = CompressorWriter::new(Vec::new(), 0, 11, 22);
        let result = writer.write_all(
            request
                .raw_body()
                .as_ref(),
        );

        if let Err(e) = result {
            return Err(DeboaError::Io(IoError::Compress { message: e.to_string() }));
        }

        let result = writer.flush();

        if let Err(e) = result {
            return Err(DeboaError::Io(IoError::Compress { message: e.to_string() }));
        }

        Ok(Bytes::from(writer.into_inner()))
    }
}

#[derive(PartialEq)]
/// Brotli decompression implementation for Deboa.
pub struct BrotliDecompressor;

#[deboa::async_trait]
impl Decompressor for BrotliDecompressor {
    fn name(&self) -> String {
        "br".to_string()
    }

    async fn decompress_body(&self, response: &mut DeboaResponse) -> Result<()> {
        let body = response
            .raw_body()
            .await;
        let mut reader = brotli::Decompressor::new(body.reader(), 0);
        let mut buffer = Vec::<u8>::new();
        let result = reader.read_to_end(&mut buffer);

        if let Err(e) = result {
            return Err(DeboaError::Io(IoError::Decompress { message: e.to_string() }));
        }

        response.set_raw_body(Bytes::from(buffer));
        Ok(())
    }
}