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

#include <array>
#include <iostream>

#include "echo.hpp"

std::string message_content = "Hey there! Sending this message";

int main(){
	/**
	 * Create EventLoop
	 * Setup EventPort to the outside world
	 * And setup the io comms to the outside.
	 */
	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_scope{aio.event_loop};

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

	saw::own<saw::network_address<saw::net::Os>> net_addr = nullptr;
	saw::own<saw::async_io_stream<saw::net::Os>> async_rmt = nullptr;

	std::array<uint8_t, 32> read_data;
	uint64_t read_bytes = 0;
	auto& network = aio.io->get_network();
	network.resolve_address(saw::echo_address, saw::echo_port).then([&](auto addr){
		net_addr = std::move(addr);
		network.connect(*net_addr).then([&](auto rmt_srv){
			async_rmt = saw::heap<saw::async_io_stream<saw::net::Os>>(std::move(rmt_srv));
			async_rmt->write(&message_content[0], message_content.size());

			async_rmt->read(&read_data[0], message_content.size(), read_data.size()-1);

			async_rmt->read_done().then([&](size_t b){
	
					std::cout<<"Received bytes:\n";
					for(uint64_t i = 0; i < b; ++i){
						std::cout<<static_cast<char>(read_data[i]);
					}
					std::cout<<'\n'<<std::endl;

					keep_running = false;
			}).detach();

			async_rmt->on_read_disconnected().then([&](){
				keep_running = false;
				std::cout<<"Disconnected"<<std::endl;
			}).detach();
		}).detach([](auto err){
			std::cerr<<"Couldn't connect to address"<<std::endl;
			return err;
		});
	}).detach([](auto err){
		std::cerr<<"Couldn't resolve address"<<std::endl;
		return err;
	});

	wait_scope.poll();
	while(keep_running){
		wait_scope.wait();
	}

	std::cout<<"Shutting down echo client"<<std::endl;

	return 0;
}