#include <forstio/test/suite.hpp>

#include "../c++/remote_loopback.hpp"

namespace {
namespace sch {
using namespace saw::schema;

using TestInterface = Interface<
	Member<Function<UInt32, Int64>, "foo">
>;

}

SAW_TEST("Remote Loopback RPC"){
	using namespace saw;

	event_loop loop;
	wait_scope wait{loop};

	remote<rmt::Loopback> rmt;

	auto eov = rmt.parse_address(0u);
	SAW_EXPECT(eov.is_value(), "Didn't parse correctly");
	auto& addr = eov.get_value();

	auto eo_u32_srv = rmt.template data_listen<sch::UInt32, encode::Native>(*addr);
	SAW_EXPECT(eo_u32_srv.is_value(), std::string{"Couldn't listen: "} + std::string{eo_u32_srv.get_error().get_category()});
	auto& u32_srv = eo_u32_srv.get_value();
	
	auto eo_i64_srv = rmt.template data_listen<sch::Int64, encode::Native>(*addr);
	SAW_EXPECT(eo_i64_srv.is_value(), std::string{"Couldn't listen: "} + std::string{eo_i64_srv.get_error().get_category()});
	auto& i64_srv = eo_i64_srv.get_value();

	auto foo_func = [](data<sch::UInt32, encode::Native> input) -> error_or<data<sch::Int64, encode::Native>> {
		data<sch::Int64, encode::Native> rv;
		rv.set(input.get() * 2);
		return rv;
	};

	auto iface = interface_factory<sch::TestInterface, encode::Native>::create(std::move(foo_func));
	
	auto eo_rpc_srv = rmt.template rpc_listen<sch::TestInterface, encode::Native>(*addr, std::move(iface));
	SAW_EXPECT(eo_rpc_srv.is_value(), std::string{"Couldn't listen: "} + std::string{eo_rpc_srv.get_error().get_category()});
	auto& rpc_srv = eo_rpc_srv.get_value();

	id<sch::UInt32> id32{0u};
	id<sch::Int64> id64{0u};

	data<sch::UInt32, encode::Native> inp{21};

	{
		auto eo_send = u32_srv->send(inp, id32);
		SAW_EXPECT(eo_send.is_value(), "Failed send.");
	}
	{
		auto eo_called = rpc_srv->template call<"foo">(id32, id64);
		SAW_EXPECT(eo_called.is_value(), (std::string{"Failed call: "} + std::string{eo_called.get_error().get_category()} ));
	}
	{
		auto eo_recv = i64_srv->receive(id64);
		SAW_EXPECT(eo_recv.is_value(), "Failed receive.");
		auto& recv = eo_recv.get_value();
		SAW_EXPECT((recv.get() == 42), (std::string{"Wrong value received: "} + std::to_string(recv.get())));
	}
}

SAW_TEST("Remote Loopback Data"){
	using namespace saw;

	event_loop loop;
	wait_scope wait{loop};

	remote<rmt::Loopback> rmt;

	using Schema = sch::Array<sch::Float64>;

	auto eov = rmt.parse_address(0u);
	SAW_EXPECT(eov.is_value(), "Didn't parse correctly");
	auto& addr = eov.get_value();

	auto eo_srv = rmt.template data_listen<Schema, encode::Native>(*addr);
	SAW_EXPECT(eo_srv.is_value(), std::string{"Couldn't listen: "} + std::string{eo_srv.get_error().get_category()});
	auto& srv = eo_srv.get_value();

	auto cvr_client = rmt.template data_connect<Schema, encode::Native>(*addr);
	auto eo_client = cvr_client.take();
	SAW_EXPECT(eo_client.is_value(), "Couldn't connect.");
	auto& client = eo_client.get_value();

	data<Schema> foo{4};
	for(uint64_t i = 0; i < foo.size().get(); ++i){
		foo.at(i).set(i * 2.0);
	}
	id<Schema> sent_id = [&](){
		auto eov = client.send(foo);
		SAW_EXPECT(eov.is_value(), "Failed send.");
		return eov.get_value();
	}();

	{
		auto conv = client.receive(sent_id);
		auto eov = conv.take();

		SAW_EXPECT(eov.is_value(), "Failed receive.");
		// SAW_EXPECT(eov.get_value() == foo, "Wrong received value.");
	}
	{
		auto eov = client.find(sent_id);
		SAW_EXPECT(eov.is_value(), "Failed find.");
		auto& f_val = eov.get_value();
		// SAW_EXPECT(f_val.is_valid(), "Nullptr in find.");
		// SAW_EXPECT(*f_val == foo, "Wrong received value.");
	}
	{
		auto eov = client.erase(sent_id);
		SAW_EXPECT(eov.is_value(), "Failed erase.");
	}
	{
		auto conv = client.receive(sent_id);
		auto eov = conv.take();
		SAW_EXPECT(!eov.is_value(), "Failed receive. Value should already be erased.");
	}
	
	id<Schema> alloc_id = [&](){
		auto eov = client.allocate(data<sch::FixedArray<sch::UInt64,1>>{{4}});
		SAW_EXPECT(eov.is_value(), "Failed send.");
		return eov.get_value();
	}();
	{
		auto eov = client.find(alloc_id);
		SAW_EXPECT(eov.is_value(), "Failed find.");
		auto& f_val = eov.get_value();
		// SAW_EXPECT(f_val.is_valid(), "Nullptr in find.");
		// f_val->set(5u);
	}
	{
		auto conv = client.receive(alloc_id);
		auto eov = conv.take();
		SAW_EXPECT(eov.is_value(), "Failed receive.");
		SAW_EXPECT(eov.get_value().size().get() == 4u, "Wrong received value.");
	}
}

}