// Copyright (C) 2024, Fluendo S.A.
//      Author: Andoni Morales Alastruey <amorales@fluendo.com>
//
// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
// If a copy of the MPL was not distributed with this file, You can obtain one at
// <https://mozilla.org/MPL/2.0/>.
//
// SPDX-License-Identifier: MPL-2.0

use gst::prelude::*;
use serial_test::serial;
use std::sync::Arc;
use std::sync::Mutex;
use std::thread;

fn init() {
    use std::sync::Once;
    static INIT: Once = Once::new();

    INIT.call_once(|| {
        gst::init().unwrap();
        gstquinn::plugin_register_static().expect("WebTransport source sink send receive tests");
    });
}

fn make_buffer(content: &[u8]) -> gst::Buffer {
    let mut buf = gst::Buffer::from_slice(content.to_owned());
    buf.make_mut().set_pts(gst::ClockTime::from_mseconds(200));
    buf
}

fn send_receive(src_pipeline_props: &str, sink_pipeline_props: &str) {
    init();

    let content = "Hello, world!\n".as_bytes();

    let src_pipeline = format!(
        "quinnwtclientsrc {} secure-connection=false",
        src_pipeline_props
    );
    let sink_pipeline = format!(
        "quinnwtserversink {} server-name=localhost \
            address=127.0.0.1 secure-connection=false",
        sink_pipeline_props
    );
    let h1_orig = Arc::new(Mutex::new(gst_check::Harness::new_empty()));

    let h11 = h1_orig.clone();

    thread::spawn(move || {
        let mut h1 = h11.lock().unwrap();

        h1.add_parse(&sink_pipeline);

        h1.set_src_caps(gst::Caps::builder("text/plain").build());

        h1.play();

        assert!(h1.push(make_buffer(content)) == Ok(gst::FlowSuccess::Ok));

        // Wait a bit before sending Eos and shutting down the pipeline
        thread::sleep(std::time::Duration::from_secs(2));

        h1.push_event(gst::event::Eos::new());

        drop(h1);
    });

    let mut h2 = gst_check::Harness::new_empty();
    h2.add_parse(&src_pipeline);

    h2.play();

    let buf = h2.pull_until_eos().unwrap().unwrap();

    assert_eq!(
        content,
        buf.into_mapped_buffer_readable().unwrap().as_slice()
    );

    // Close the server now that the client has finished reading the data
    let h11 = h1_orig.clone();
    let h1 = h11.lock().unwrap();
    h1.element().unwrap().set_state(gst::State::Null).unwrap();
    drop(h1);

    h2.element().unwrap().set_state(gst::State::Null).unwrap();
    drop(h2);
}

#[test]
#[serial]
fn test_send_receive_without_datagram() {
    send_receive("url=https://127.0.0.1:7770", "port=7770");
}

#[test]
#[serial]
fn test_send_receive_with_datagram() {
    send_receive("url=https://127.0.0.1:7771", "use-datagram=true port=7771");
}

#[test]
#[serial]
#[ignore = "CI runners resolve localhost to an IPv6 address only which are not handled correctly yet"]
fn test_send_receive_with_hostname() {
    send_receive("url=https://localhost:7772", "port=7772");
}