rip-starttls 0.1.0

//! # Tokio
//!
//! This module contains the async I/O connector based on [`tokio`]
//! for [`RipStarttls`](super::RipStarttls).

use std::io::Result;

use tokio::{
    io::{AsyncBufReadExt, AsyncWriteExt, BufStream},
    net::TcpStream,
};

use super::{Event, State};

#[derive(Clone, Debug, Default, Eq, PartialEq)]
pub struct RipStarttls {
    state: super::RipStarttls,
}

impl RipStarttls {
    pub fn new(handshake_discarded: bool) -> Self {
        let state = super::RipStarttls::new(handshake_discarded);
        Self { state }
    }

    pub async fn do_starttls_prefix(mut self, stream: TcpStream) -> Result<TcpStream> {
        let mut stream = BufStream::new(stream);
        let mut event = None;

        while let Some(output) = self.state.resume(event.take()) {
            match output {
                State::DiscardHandshake => {
                    let mut line = String::new();
                    stream.read_line(&mut line).await?;
                    event = Some(Event::HandshakeDiscarded(line));
                }
                State::WriteStarttlsCommand => {
                    let cmd = super::RipStarttls::COMMAND;
                    let count = stream.write(cmd.as_bytes()).await?;
                    stream.flush().await?;
                    event = Some(Event::StarttlsCommandWrote(count));
                }
                State::DiscardResponse => {
                    let mut line = String::new();
                    stream.read_line(&mut line).await?;
                    event = Some(Event::ResponseDiscarded(line));
                }
            }
        }

        Ok(stream.into_inner())
    }
}