extern crate std;

use core::{
    future::Future,
    pin::Pin,
    task::{Context, Poll},
};
use std::{io, net::UdpSocket};

use super::{WriteExactFuture, WriteFuture};

impl<'a> Future for WriteFuture<'a, UdpSocket> {
    type Output = Result<usize, io::Error>;

    fn poll(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Self::Output> {
        let me = self.get_mut();
        me.0.set_nonblocking(true)
            .expect("unable to set nonblocking-mode.");
        let r = me.0.send(me.1);
        me.0.set_nonblocking(false)
            .expect("unable to clear nonblocking-mode.");

        Poll::Ready(r)
    }
}

impl<'a> Future for WriteExactFuture<'a, UdpSocket> {
    type Output = Result<(), io::Error>;

    fn poll(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Self::Output> {
        let me = self.get_mut();
        me.0.set_nonblocking(true)
            .expect("unable to set nonblocking-mode.");
        let mut n = 0;
        while n != me.1.len() {
            n += match me.0.send(&(me.1)[n..]) {
                Ok(x) => x,
                Err(e) => return Poll::Ready(Err(e)),
            }
        }
        me.0.set_nonblocking(false)
            .expect("unable to clear nonblocking-mode.");

        Poll::Ready(Ok(()))
    }
}