// Copyright 2017 Parity Technologies (UK) Ltd.
// 
// Permission is hereby granted, free of charge, to any person obtaining a 
// copy of this software and associated documentation files (the "Software"), 
// to deal in the Software without restriction, including without limitation 
// the rights to use, copy, modify, merge, publish, distribute, sublicense, 
// and/or sell copies of the Software, and to permit persons to whom the 
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in 
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS 
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER 
// DEALINGS IN THE SOFTWARE.

extern crate bytes;
extern crate futures;
extern crate libp2p_secio as secio;
extern crate libp2p_swarm as swarm;
extern crate libp2p_tcp_transport as tcp;
extern crate ring;
extern crate tokio_core;
extern crate tokio_io;
extern crate untrusted;

use bytes::Bytes;
use futures::future::{Future, FutureResult, IntoFuture, loop_fn, Loop};
use futures::{Stream, Sink};
use ring::signature::RSAKeyPair;
use std::io::Error as IoError;
use std::iter;
use std::sync::Arc;
use swarm::{Transport, ConnectionUpgrade};
use tcp::Tcp;
use tokio_core::reactor::Core;
use tokio_io::{AsyncRead, AsyncWrite};
use tokio_io::codec::length_delimited;
use untrusted::Input;

fn main() {
    let mut core = Core::new().unwrap();
    let tcp = Tcp::new(core.handle()).unwrap();

    let with_secio = tcp
        .with_upgrade(swarm::PlainText)
        .or_upgrade({
            let private_key = {
                let pkcs8 = include_bytes!("test-private-key.pk8");
                Arc::new(RSAKeyPair::from_pkcs8(Input::from(&pkcs8[..])).unwrap())
            };
            let public_key = include_bytes!("test-public-key.der").to_vec();

            secio::SecioConnUpgrade {
                local_public_key: public_key,
                local_private_key: private_key,
            }
        });

    let with_echo = with_secio.with_upgrade(Echo);

    let future = with_echo.listen_on(swarm::multiaddr::Multiaddr::new("/ip4/0.0.0.0/tcp/10333").unwrap())
        .map_err(|_| panic!())
        .unwrap()
        .for_each(|socket| {
            loop_fn(socket, |socket| {
                socket.into_future()
                    .map_err(|(err, _)| err)
                    .and_then(|(msg, rest)| {
                        if let Some(msg) = msg {
                            Box::new(rest.send(msg).map(|m| Loop::Continue(m))) as Box<Future<Item = _, Error = _>>
                        } else {
                            Box::new(Ok(Loop::Break(())).into_future()) as Box<Future<Item = _, Error = _>>
                        }
                    })
            })
        });

    core.run(future).unwrap();
}

// TODO: copy-pasted from echo-dialer
#[derive(Debug, Copy, Clone)]
pub struct Echo;
impl<C> ConnectionUpgrade<C> for Echo
    where C: AsyncRead + AsyncWrite
{
    type NamesIter = iter::Once<(Bytes, Self::UpgradeIdentifier)>;
    type UpgradeIdentifier = ();

    #[inline]
    fn protocol_names(&self) -> Self::NamesIter {
        iter::once(("/echo/1.0.0".into(), ()))
    }

    type Output = length_delimited::Framed<C>;
    type Future = FutureResult<Self::Output, IoError>;

    #[inline]
    fn upgrade(self, socket: C, _: Self::UpgradeIdentifier) -> Self::Future {
        Ok(length_delimited::Framed::new(socket)).into_future()
    }
}