use std::{collections::HashMap, sync::Arc};

use polywrap_client_builder::{ClientConfig, ClientConfigBuilder};
use polywrap_core::{macros::uri, package::WrapPackage, uri::Uri, wrapper::Wrapper};
use polywrap_msgpack_serde::to_vec;
use polywrap_tests_utils::mocks::{
    get_different_mock_package, get_different_mock_wrapper, get_mock_invoker, get_mock_package,
    get_mock_wrapper, DifferentMockResolver, MockResolver,
};
use serde::Serialize;

#[derive(Serialize)]
struct EnvOne {
    a: String,
    b: String,
}

#[derive(Serialize)]
struct EnvTwo {
    d: String,
}

#[test]
fn test_env_methods() {
    let mut builder = ClientConfig::new();
    let uri = uri!("wrap://mock/wrapper");

    assert!(builder.envs.is_none());

    builder.add_env(uri.clone(), to_vec(&EnvTwo { d: "d".to_string() }).unwrap());

    let current_env = builder.envs.clone().unwrap();
    let env_from_builder = current_env.get(&uri);

    assert!(env_from_builder.is_some());
    assert_eq!(
        env_from_builder.unwrap(),
        &to_vec(&EnvTwo { d: "d".to_string() }).unwrap()
    );

    let mut envs = HashMap::new();
    envs.insert(
        uri.clone(),
        to_vec(&EnvOne {
            a: "a".to_string(),
            b: "b".to_string(),
        })
        .unwrap(),
    );

    builder.add_envs(envs);

    let current_env = builder.envs.clone().unwrap();
    let env_from_builder = current_env.get(&uri);
    assert_eq!(
        env_from_builder.unwrap(),
        &to_vec(&EnvOne {
            a: "a".to_string(),
            b: "b".to_string(),
        })
        .unwrap()
    );

    builder.remove_env(&uri);

    assert!(builder.envs.is_none());
}

#[test]
fn test_interface_implementation_methods() {
    let interface_uri = uri!("wrap://mock/interface");
    let implementation_a_uri = uri!("wrap://mock/implementation-a");
    let implementation_b_uri = uri!("wrap://mock/implementation-b");

    let another_interface_uri = uri!("wrap://mock/another-interface");

    let mut builder = ClientConfig::new();
    assert!(builder.interfaces.is_none());
    builder.add_interface_implementation(interface_uri.clone(), implementation_a_uri.clone());
    builder.add_interface_implementation(another_interface_uri, implementation_b_uri.clone());

    assert!(builder.interfaces.is_some());
    assert!(builder.interfaces.unwrap().len() == 2);

    // Recreate builder again to test add interfaces implementations
    let mut builder = ClientConfig::new();

    assert!(builder.interfaces.is_none());

    builder.add_interface_implementations(
        interface_uri.clone(),
        vec![implementation_a_uri.clone(), implementation_b_uri.clone()],
    );

    let interfaces = builder.interfaces.clone().unwrap();
    let implementations = interfaces.get(&interface_uri).unwrap();
    assert!(builder.interfaces.is_some());
    assert_eq!(
        implementations,
        &vec![implementation_a_uri.clone(), implementation_b_uri.clone()]
    );

    let implementation_c_uri = uri!("wrap://mock/implementation-c");
    builder.add_interface_implementation(interface_uri.clone(), implementation_c_uri.clone());

    let interfaces = builder.interfaces.clone().unwrap();
    let implementations = interfaces.get(&interface_uri).unwrap();
    assert_eq!(
        implementations,
        &vec![
            implementation_a_uri.clone(),
            implementation_b_uri.clone(),
            implementation_c_uri.clone()
        ]
    );

    builder.remove_interface_implementation(&interface_uri, &implementation_b_uri);
    let interfaces = builder.interfaces.clone().unwrap();
    let implementations = interfaces.get(&interface_uri).unwrap();
    assert_eq!(
        implementations,
        &vec![implementation_a_uri, implementation_c_uri]
    );
}

#[test]
fn test_redirects() {
    let mut builder = ClientConfig::new();
    assert!(builder.redirects.is_none());

    let a_uri = uri!("mock/a");
    let b_uri = uri!("mock/b");
    let c_uri = uri!("mock/c");
    let d_uri = uri!("mock/d");
    let f_uri = uri!("mock/f");
    let g_uri = uri!("mock/g");

    let redirects = HashMap::from([
        (c_uri.clone(), d_uri.clone()),
        (f_uri.clone(), g_uri.clone()),
    ]);

    builder.add_redirects(redirects.clone());

    assert!(builder.redirects.is_some());
    let builder_redirects = builder.redirects.unwrap();
    assert_eq!(builder_redirects, redirects);

    let mut builder = ClientConfig::new();
    assert!(builder.redirects.is_none());

    builder.add_redirect(a_uri.clone(), b_uri.clone());
    assert!(builder.redirects.is_some());

    builder.add_redirects(HashMap::from([(a_uri.clone(), c_uri.clone())]));
    let redirects = builder.redirects.clone().unwrap();
    let a_uri_redirect = redirects.get(&a_uri);
    assert_eq!(Some(&c_uri), a_uri_redirect);

    builder.remove_redirect(&a_uri);
    assert!(builder.redirects.is_none());
}

#[test]
fn test_packages() {
    let mut builder = ClientConfig::new();
    assert!(builder.packages.is_none());

    let uri_a: Uri = String::from("wrap://package/a").try_into().unwrap();
    let uri_b: Uri = String::from("wrap://package/b").try_into().unwrap();

    let uri_package_a = (uri_a, get_mock_package());

    let uri_package_b = (uri_b.clone(), get_mock_package());

    let uri_package_c = (
        String::from("wrap://package/c").try_into().unwrap(),
        get_mock_package(),
    );

    builder.add_packages(vec![uri_package_a, uri_package_b, uri_package_c]);
    assert!(builder.packages.is_some());

    let builder_packages = builder.packages.unwrap();
    assert_eq!(builder_packages.len(), 3);

    let package: &Arc<dyn WrapPackage> = &builder_packages[1].1;
    let wrapper = package.create_wrapper().unwrap();

    let result_package_a = wrapper.invoke("foo", None, None, get_mock_invoker());
    assert_eq!(result_package_a.unwrap(), vec![195]);

    // We need to recreate the builder because when we do builder.packages.unwrap
    // the ownership is given, not allowing us to call the builder again
    let mut builder = ClientConfig::new();

    let modified_uri_package_b = (uri_b.clone(), get_different_mock_package());

    builder.add_packages(builder_packages);
    builder.add_package(modified_uri_package_b.0, modified_uri_package_b.1);
    builder.remove_package(&String::from("wrap://package/c").try_into().unwrap());

    let builder_packages = builder.packages.unwrap();
    assert_eq!(builder_packages.len(), 2);
    let b_package = builder_packages
        .into_iter()
        .find(|(uri, _)| uri == &uri_b)
        .unwrap();
    let wrapper = b_package.1.create_wrapper().unwrap();
    let result_package_a = wrapper.invoke("bar", None, None, get_mock_invoker());
    assert_eq!(result_package_a.unwrap(), vec![195]);
}

#[test]
fn test_wrappers() {
    let mut builder = ClientConfig::new();
    assert!(builder.wrappers.is_none());

    let uri_wrapper_a = (
        String::from("wrap://wrapper/a").try_into().unwrap(),
        get_mock_wrapper(),
    );

    let uri_wrapper_b = (
        String::from("wrap://wrapper/b").try_into().unwrap(),
        get_mock_wrapper(),
    );

    let uri_wrapper_c = (
        String::from("wrap://wrapper/c").try_into().unwrap(),
        get_mock_wrapper(),
    );

    builder.add_wrappers(vec![uri_wrapper_a, uri_wrapper_b, uri_wrapper_c]);
    assert!(builder.wrappers.is_some());
    let builder_wrappers = builder.wrappers.unwrap();
    assert_eq!(builder_wrappers.len(), 3);

    let wrapper: &Arc<dyn Wrapper> = &builder_wrappers[1].1;
    let result_package_a = wrapper.invoke("foo", None, None, get_mock_invoker());
    assert_eq!(result_package_a.unwrap(), vec![195]);

    // We need to recreate the builder because when we do builder.wrappers.unwrap
    // the ownership is given, not allowing us to call the builder again
    let mut builder = ClientConfig::new();

    let modified_uri_wrapper_b = (
        String::from("wrap://wrapper/b").try_into().unwrap(),
        get_different_mock_wrapper(),
    );

    builder.add_wrappers(builder_wrappers);
    builder.add_wrapper(modified_uri_wrapper_b.0, modified_uri_wrapper_b.1);
    builder.remove_wrapper(&String::from("wrap://wrapper/c").try_into().unwrap());

    let builder_wrappers = builder.wrappers.unwrap();
    assert_eq!(builder_wrappers.len(), 2);

    let wrapper_uri = String::from("wrap://wrapper/b").try_into().unwrap();
    let b_wrapper = builder_wrappers
        .into_iter()
        .find(|(uri, _)| uri == &wrapper_uri)
        .unwrap();

    let result_package_b = b_wrapper.1.invoke("bar", None, None, get_mock_invoker());
    assert_eq!(result_package_b.unwrap(), [195]);
}

#[test]
fn test_resolvers() {
    let mut builder = ClientConfig::new();
    assert!(builder.resolvers.is_none());
    builder.add_resolvers(vec![Arc::new(MockResolver {})]);
    assert!(builder.resolvers.is_some());
    builder.add_resolver(Arc::new(DifferentMockResolver {}));
    assert_eq!(builder.resolvers.unwrap().len(), 2);
}