#include "echo.hpp"

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

#include <forstio/codec/transport.hpp>

#include <iostream>

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;
		}
	}

	auto echo_srv = network.listen(*addr);
	if(!echo_srv){
		return -2;
	}

	echo_srv->accept().then([&](saw::own<saw::io_stream> client){
		if(!client){
			keep_running = false;
			return;
		}
		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));

		std::cout<<"Connected client"<<std::endl;
		auto peer_str = echo_peer_stream_p.first.get();
		
		echo_peer_stream_p.second.then([&,peer_str](auto simp_resp) -> error_or<void> {
			data<sch::Echo> nat_resp;
			{
				auto eov = simple_codec.decode(simp_resp, nat_resp);
				if(eov.is_error()){
					std::cerr<<"Failed to decode"<<std::endl;
					return eov;
				}
			}
			for(uint64_t i = 0u; i < nat_resp.size().get(); ++i){
				std::cout<<nat_resp.at(i);
			}
			std::cout<<std::endl;

			{
				auto eo_send = peer_str->send(std::move(simp_resp));
				if(eo_send.is_error()){
					auto& err = eo_send.get_error();
					return std::move(err);
				}
			}
			return make_void();
		}).detach();
	
		peer_str->on_disconnected().attach(std::move(echo_peer_stream_p.first)).then([]() -> error_or<void>{
			std::cout<<"Disconnected client"<<std::endl;
			return make_error<err::critical>("Destroy pipeline on purpose :>");
		}).detach();
	}).detach();

	while(keep_running){
		wait.wait_for(1u*1000*1000);
	}

	return 0;
}