#include "echo.hpp"

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

#include <iostream>

#include <forstio/codec/transport.hpp>

int main(){
	using namespace saw;

	auto eo_aio = saw::setup_async_io();
	if(eo_aio.is_error()){
		auto& err = eo_aio.get_error();
		std::cerr<<err.get_message()<<std::endl;
		return err.get_id();
	}
	auto& aio = eo_aio.get_value();
	/**
	 * Make the event loop the current event loop on this thread
	 */
	saw::wait_scope wait{aio.event_loop};

	bool keep_running = true;
	aio.event_port.on_signal(saw::Signal::Terminate).then([&keep_running](){
		keep_running = false;
	}).detach();

	auto& network = aio.io->get_network();

	auto eo_addr = network.resolve_address(saw::echo_address, saw::echo_port).take();
	if(eo_addr.is_error()){
		return -1;
	}
	auto& addr = eo_addr.get_value();

	data<sch::Echo> nat_echo{"hello"};
	data<sch::Echo, encode::KelSimple> simple_echo;
	codec<sch::Echo, encode::KelSimple> simple_codec;

	{
		auto eov = simple_codec.encode(nat_echo, simple_echo);
		if(eov.is_error()){
			return -1;
		}
	}

	network.connect(*addr).then([&](saw::own<saw::io_stream> client){
		auto echo_stream = saw::heap<saw::async_io_stream>(std::move(client));
		auto echo_peer_stream_p = saw::new_streaming_io_peer<sch::Echo, sch::Echo, trans::FixedLength<8u>, encode::KelSimple, ring_buffer>(std::move(echo_stream));

		{
			auto eo_send = echo_peer_stream_p.first->send(std::move(simple_echo));
			if(eo_send.is_error()){
				return;
			}
		}
		{
			simple_echo = {};
			auto eov = simple_codec.encode(nat_echo, simple_echo);
			if(eov.is_error()){
				return ;
			}
		}
		
		data<sch::Echo> nat_resp;
		echo_peer_stream_p.second.then([&](auto simp_resp){
			auto eov = simple_codec.decode(simp_resp, nat_resp);
			std::cout<<"Answer:\n";
			for(uint64_t i = 0u; i < nat_resp.size(); ++i){
				std::cout<<nat_resp.at(i);
			}
			std::cout<<std::endl;
			return eov;
		}).detach();
		
		echo_peer_stream_p.first->on_read_disconnected().attach(std::move(echo_peer_stream_p.first)).then([]() -> error_or<void>{
			return make_error<err::critical>("Destroy pipeline on purpose :>");
		}).detach();
		
	}).detach();

	while(keep_running){
		wait.wait(std::chrono::seconds{1u});
	}

	return 0;
}