mod data_consumer {
    use async_io::Timer;
    use futures_lite::future;
    use mediasoup::data_consumer::{DataConsumerOptions, DataConsumerType};
    use mediasoup::data_producer::{DataProducer, DataProducerOptions};
    use mediasoup::data_structures::{AppData, TransportListenIp};
    use mediasoup::direct_transport::DirectTransportOptions;
    use mediasoup::plain_transport::PlainTransportOptions;
    use mediasoup::router::{Router, RouterOptions};
    use mediasoup::sctp_parameters::SctpStreamParameters;
    use mediasoup::transport::{Transport, TransportGeneric};
    use mediasoup::webrtc_transport::{
        TransportListenIps, WebRtcTransport, WebRtcTransportOptions,
    };
    use mediasoup::worker::{Worker, WorkerSettings};
    use mediasoup::worker_manager::WorkerManager;
    use std::collections::{HashMap, HashSet};
    use std::env;
    use std::sync::atomic::{AtomicUsize, Ordering};
    use std::sync::Arc;
    use std::time::Duration;

    struct CustomAppData {
        baz: &'static str,
    }

    struct CustomAppData2 {
        hehe: &'static str,
    }

    fn data_producer_options() -> DataProducerOptions {
        let mut options = DataProducerOptions::new_sctp(
            SctpStreamParameters::new_unordered_with_life_time(12345, 5000),
        );

        options.label = "foo".to_string();
        options.protocol = "bar".to_string();

        options
    }

    async fn init() -> (Worker, Router, WebRtcTransport, DataProducer) {
        {
            let mut builder = env_logger::builder();
            if env::var(env_logger::DEFAULT_FILTER_ENV).is_err() {
                builder.filter_level(log::LevelFilter::Off);
            }
            let _ = builder.is_test(true).try_init();
        }

        let worker_manager = WorkerManager::new(
            env::var("MEDIASOUP_WORKER_BIN")
                .map(|path| path.into())
                .unwrap_or_else(|_| "../worker/out/Release/mediasoup-worker".into()),
        );

        let worker = worker_manager
            .create_worker(WorkerSettings::default())
            .await
            .expect("Failed to create worker");

        let router = worker
            .create_router(RouterOptions::new(vec![]))
            .await
            .expect("Failed to create router");

        let transport = router
            .create_webrtc_transport({
                let mut transport_options =
                    WebRtcTransportOptions::new(TransportListenIps::new(TransportListenIp {
                        ip: "127.0.0.1".parse().unwrap(),
                        announced_ip: None,
                    }));

                transport_options.enable_sctp = true;

                transport_options
            })
            .await
            .expect("Failed to create transport1");

        let data_producer = transport
            .produce_data(data_producer_options())
            .await
            .expect("Failed to create data producer");

        (worker, router, transport, data_producer)
    }

    #[test]
    fn consume_data_succeeds() {
        future::block_on(async move {
            let (_worker, router, _transport1, data_producer) = init().await;

            let transport2 = router
                .create_plain_transport({
                    let mut transport_options = PlainTransportOptions::new(TransportListenIp {
                        ip: "127.0.0.1".parse().unwrap(),
                        announced_ip: None,
                    });

                    transport_options.enable_sctp = true;

                    transport_options
                })
                .await
                .expect("Failed to create transport1");

            let new_data_consumer_count = Arc::new(AtomicUsize::new(0));

            transport2
                .on_new_data_consumer({
                    let new_data_consumer_count = Arc::clone(&new_data_consumer_count);

                    move |_data_consumer| {
                        new_data_consumer_count.fetch_add(1, Ordering::SeqCst);
                    }
                })
                .detach();

            let data_consumer = transport2
                .consume_data({
                    let mut options = DataConsumerOptions::new_sctp_unordered_with_life_time(
                        data_producer.id(),
                        4000,
                    );

                    options.app_data = AppData::new(CustomAppData { baz: "LOL" });

                    options
                })
                .await
                .expect("Failed to consume data");

            assert_eq!(new_data_consumer_count.load(Ordering::SeqCst), 1);
            assert_eq!(data_consumer.data_producer_id(), data_producer.id());
            assert_eq!(data_consumer.closed(), false);
            assert_eq!(data_consumer.r#type(), DataConsumerType::Sctp);
            {
                let sctp_stream_parameters = data_consumer.sctp_stream_parameters();
                assert!(sctp_stream_parameters.is_some());
                assert_eq!(sctp_stream_parameters.unwrap().ordered(), false);
                assert_eq!(
                    sctp_stream_parameters.unwrap().max_packet_life_time(),
                    Some(4000),
                );
                assert_eq!(sctp_stream_parameters.unwrap().max_retransmits(), None);
            }
            assert_eq!(data_consumer.label().as_str(), "foo");
            assert_eq!(data_consumer.protocol().as_str(), "bar");
            assert_eq!(
                data_consumer
                    .app_data()
                    .downcast_ref::<CustomAppData>()
                    .unwrap()
                    .baz,
                "LOL",
            );

            {
                let dump = router.dump().await.expect("Failed to dump router");

                assert_eq!(dump.map_data_producer_id_data_consumer_ids, {
                    let mut map = HashMap::new();
                    map.insert(data_producer.id(), {
                        let mut set = HashSet::new();
                        set.insert(data_consumer.id());
                        set
                    });
                    map
                });
                assert_eq!(dump.map_data_consumer_id_data_producer_id, {
                    let mut map = HashMap::new();
                    map.insert(data_consumer.id(), data_producer.id());
                    map
                });
            }

            {
                let dump = transport2.dump().await.expect("Failed to dump transport");

                assert_eq!(dump.id, transport2.id());
                assert_eq!(dump.data_producer_ids, vec![]);
                assert_eq!(dump.data_consumer_ids, vec![data_consumer.id()]);
            }
        });
    }

    #[test]
    fn dump_succeeds() {
        future::block_on(async move {
            let (_worker, _router, transport, data_producer) = init().await;

            let data_consumer = transport
                .consume_data({
                    let mut options = DataConsumerOptions::new_sctp_unordered_with_life_time(
                        data_producer.id(),
                        4000,
                    );

                    options.app_data = AppData::new(CustomAppData { baz: "LOL" });

                    options
                })
                .await
                .expect("Failed to consume data");

            let dump = data_consumer
                .dump()
                .await
                .expect("Data consumer dump failed");

            assert_eq!(dump.id, data_consumer.id());
            assert_eq!(dump.data_producer_id, data_consumer.data_producer_id());
            assert_eq!(dump.r#type, DataConsumerType::Sctp);
            {
                let sctp_stream_parameters = dump.sctp_stream_parameters;
                assert!(sctp_stream_parameters.is_some());
                assert_eq!(
                    sctp_stream_parameters.unwrap().stream_id(),
                    data_consumer.sctp_stream_parameters().unwrap().stream_id(),
                );
                assert_eq!(sctp_stream_parameters.unwrap().ordered(), false);
                assert_eq!(
                    sctp_stream_parameters.unwrap().max_packet_life_time(),
                    Some(4000),
                );
                assert_eq!(sctp_stream_parameters.unwrap().max_retransmits(), None);
            }
            assert_eq!(dump.label.as_str(), "foo");
            assert_eq!(dump.protocol.as_str(), "bar");
        });
    }

    #[test]
    fn get_stats_succeeds() {
        future::block_on(async move {
            let (_worker, _router, transport, data_producer) = init().await;

            let data_consumer = transport
                .consume_data({
                    let mut options = DataConsumerOptions::new_sctp_unordered_with_life_time(
                        data_producer.id(),
                        4000,
                    );

                    options.app_data = AppData::new(CustomAppData { baz: "LOL" });

                    options
                })
                .await
                .expect("Failed to consume data");

            let stats = data_consumer
                .get_stats()
                .await
                .expect("Failed to get data consumer stats");

            assert_eq!(stats.len(), 1);
            assert_eq!(&stats[0].label, data_consumer.label());
            assert_eq!(&stats[0].protocol, data_consumer.protocol());
            assert_eq!(stats[0].messages_sent, 0);
            assert_eq!(stats[0].bytes_sent, 0);
        });
    }

    #[test]
    fn consume_data_on_direct_transport_succeeds() {
        future::block_on(async move {
            let (_worker, router, _transport1, data_producer) = init().await;

            let transport3 = router
                .create_direct_transport(DirectTransportOptions::default())
                .await
                .expect("Failed to create Direct transport");

            let new_data_consumer_count = Arc::new(AtomicUsize::new(0));

            transport3
                .on_new_data_consumer({
                    let new_data_consumer_count = Arc::clone(&new_data_consumer_count);

                    move |_data_consumer| {
                        new_data_consumer_count.fetch_add(1, Ordering::SeqCst);
                    }
                })
                .detach();

            let data_consumer = transport3
                .consume_data({
                    let mut options = DataConsumerOptions::new_direct(data_producer.id());

                    options.app_data = AppData::new(CustomAppData2 { hehe: "HEHE" });

                    options
                })
                .await
                .expect("Failed to consume data");

            assert_eq!(new_data_consumer_count.load(Ordering::SeqCst), 1);
            assert_eq!(data_consumer.data_producer_id(), data_producer.id());
            assert_eq!(data_consumer.closed(), false);
            assert_eq!(data_consumer.r#type(), DataConsumerType::Direct);
            assert_eq!(data_consumer.sctp_stream_parameters(), None);
            assert_eq!(data_consumer.label().as_str(), "foo");
            assert_eq!(data_consumer.protocol().as_str(), "bar");
            assert_eq!(
                data_consumer
                    .app_data()
                    .downcast_ref::<CustomAppData2>()
                    .unwrap()
                    .hehe,
                "HEHE",
            );

            {
                let dump = transport3.dump().await.expect("Failed to dump transport");

                assert_eq!(dump.id, transport3.id());
                assert_eq!(dump.data_producer_ids, vec![]);
                assert_eq!(dump.data_consumer_ids, vec![data_consumer.id()]);
            }
        });
    }

    #[test]
    fn dump_on_direct_transport_succeeds() {
        future::block_on(async move {
            let (_worker, router, _transport1, data_producer) = init().await;

            let transport3 = router
                .create_direct_transport(DirectTransportOptions::default())
                .await
                .expect("Failed to create Direct transport");

            let data_consumer = transport3
                .consume_data({
                    let mut options = DataConsumerOptions::new_direct(data_producer.id());

                    options.app_data = AppData::new(CustomAppData2 { hehe: "HEHE" });

                    options
                })
                .await
                .expect("Failed to consume data");

            let dump = data_consumer
                .dump()
                .await
                .expect("Data consumer dump failed");

            assert_eq!(dump.id, data_consumer.id());
            assert_eq!(dump.data_producer_id, data_consumer.data_producer_id());
            assert_eq!(dump.r#type, DataConsumerType::Direct);
            assert_eq!(dump.sctp_stream_parameters, None);
            assert_eq!(dump.label.as_str(), "foo");
            assert_eq!(dump.protocol.as_str(), "bar");
        });
    }

    #[test]
    fn get_stats_on_direct_transport_succeeds() {
        future::block_on(async move {
            let (_worker, router, _transport1, data_producer) = init().await;

            let transport3 = router
                .create_direct_transport(DirectTransportOptions::default())
                .await
                .expect("Failed to create Direct transport");

            let data_consumer = transport3
                .consume_data({
                    let mut options = DataConsumerOptions::new_direct(data_producer.id());

                    options.app_data = AppData::new(CustomAppData2 { hehe: "HEHE" });

                    options
                })
                .await
                .expect("Failed to consume data");

            let stats = data_consumer
                .get_stats()
                .await
                .expect("Failed to get data consumer stats");

            assert_eq!(stats.len(), 1);
            assert_eq!(&stats[0].label, data_consumer.label());
            assert_eq!(&stats[0].protocol, data_consumer.protocol());
            assert_eq!(stats[0].messages_sent, 0);
            assert_eq!(stats[0].bytes_sent, 0);
        });
    }

    #[test]
    fn close_event() {
        future::block_on(async move {
            let (_worker, router, _transport1, data_producer) = init().await;

            let transport2 = router
                .create_plain_transport({
                    let mut transport_options = PlainTransportOptions::new(TransportListenIp {
                        ip: "127.0.0.1".parse().unwrap(),
                        announced_ip: None,
                    });

                    transport_options.enable_sctp = true;

                    transport_options
                })
                .await
                .expect("Failed to create transport1");

            let data_consumer = transport2
                .consume_data({
                    let mut options = DataConsumerOptions::new_sctp_unordered_with_life_time(
                        data_producer.id(),
                        4000,
                    );

                    options.app_data = AppData::new(CustomAppData { baz: "LOL" });

                    options
                })
                .await
                .expect("Failed to consume data");

            {
                let (tx, rx) = async_oneshot::oneshot::<()>();
                let _handler = data_consumer.on_close(move || {
                    let _ = tx.send(());
                });
                drop(data_consumer);

                rx.await.expect("Failed to receive close event");
            }

            // Drop is async, give consumer a bit of time to finish
            Timer::after(Duration::from_millis(200)).await;

            {
                let dump = router.dump().await.expect("Failed to dump router");

                assert_eq!(dump.map_data_producer_id_data_consumer_ids, {
                    let mut map = HashMap::new();
                    map.insert(data_producer.id(), HashSet::new());
                    map
                });
                assert_eq!(dump.map_data_consumer_id_data_producer_id, HashMap::new());
            }

            {
                let dump = transport2.dump().await.expect("Failed to dump transport");

                assert_eq!(dump.data_producer_ids, vec![]);
                assert_eq!(dump.data_consumer_ids, vec![]);
            }
        });
    }

    #[test]
    fn data_producer_close_event() {
        future::block_on(async move {
            let (_worker, router, _transport1, data_producer) = init().await;

            let transport2 = router
                .create_plain_transport({
                    let mut transport_options = PlainTransportOptions::new(TransportListenIp {
                        ip: "127.0.0.1".parse().unwrap(),
                        announced_ip: None,
                    });

                    transport_options.enable_sctp = true;

                    transport_options
                })
                .await
                .expect("Failed to create transport1");

            let data_consumer = transport2
                .consume_data({
                    let mut options = DataConsumerOptions::new_sctp_unordered_with_life_time(
                        data_producer.id(),
                        4000,
                    );

                    options.app_data = AppData::new(CustomAppData { baz: "LOL" });

                    options
                })
                .await
                .expect("Failed to consume data");

            let (close_tx, close_rx) = async_oneshot::oneshot::<()>();
            let _handler = data_consumer.on_close(move || {
                let _ = close_tx.send(());
            });

            let (data_producer_close_tx, data_producer_close_rx) = async_oneshot::oneshot::<()>();
            let _handler = data_consumer.on_data_producer_close(move || {
                let _ = data_producer_close_tx.send(());
            });
            drop(data_producer);

            data_producer_close_rx
                .await
                .expect("Failed to receive data_producer_close event");

            close_rx.await.expect("Failed to receive close event");

            assert_eq!(data_consumer.closed(), true);
        });
    }

    #[test]
    fn transport_close_event() {
        future::block_on(async move {
            let (worker, router, _transport1, data_producer) = init().await;

            let transport2 = router
                .create_plain_transport({
                    let mut transport_options = PlainTransportOptions::new(TransportListenIp {
                        ip: "127.0.0.1".parse().unwrap(),
                        announced_ip: None,
                    });

                    transport_options.enable_sctp = true;

                    transport_options
                })
                .await
                .expect("Failed to create transport1");

            let data_consumer = transport2
                .consume_data({
                    let mut options = DataConsumerOptions::new_sctp_unordered_with_life_time(
                        data_producer.id(),
                        4000,
                    );

                    options.app_data = AppData::new(CustomAppData { baz: "LOL" });

                    options
                })
                .await
                .expect("Failed to consume data");

            let (close_tx, close_rx) = async_oneshot::oneshot::<()>();
            let _handler = data_consumer.on_close(move || {
                let _ = close_tx.send(());
            });

            let (transport_close_tx, transport_close_rx) = async_oneshot::oneshot::<()>();
            let _handler = data_consumer.on_transport_close(move || {
                let _ = transport_close_tx.send(());
            });

            unsafe {
                libc::kill(worker.pid() as i32, libc::SIGINT);
            }

            transport_close_rx
                .await
                .expect("Failed to receive transport_close event");
            close_rx.await.expect("Failed to receive close event");

            assert_eq!(data_consumer.closed(), true);
        });
    }
}
