#pragma once

#include "remote.hpp"

#include <map>

namespace saw {
namespace rmt {
struct Loopback {};
}

template<>
class remote<rmt::Loopback> {
private:
	struct key_t {
		std::array<uint64_t,3> data;

		bool operator<(const std::array<uint64_t,3>& rhs) const {
			for(uint64_t i = 0u; i < 3; ++i){
				if(data[i] != rhs.data[i]){
					return data[i] < rhs.data[i];
				}
			}
			return false;
		}
	};

	std::map<key_t, ptr<i_data_server<rmt::Loopback>>> registered_data_servers_;
public:
	/**
	 * Resolves an address for the remote
	 */
	error_or<own<remote_address<rmt::Loopback>>> parse_address(data<schema::UInt64> id){
		return heap<remote_address<rmt::Loopback>>(id);
	}

	/**
	 * Connect to a remote
	 */
	template<typename Iface, typename Encode, typename Storage>
	conveyor<rpc_client<Iface, Encode, Storage, rmt::Loopback>> connect(const remote_address<rmt::Loopback>& addr);

	/**
	 * Start listening
	 */
	template<typename Iface, typename Encode, typename Storage>
	rpc_server<Iface, Encode, Storage, rmt::Loopback> listen(const remote_address<rmt::Loopback>& addr, typename rpc_server<Iface,Encode,Storage,rmt::Loopback>::InterfaceT iface){
		return {addr, std::move(iface)};
	}

	/**
	 * Start data server
	 */
	template<typename Schema, typename Encode>
	error_or<own<data_server<Schema, Encode, rmt::Loopback>>> data_listen(const remote_address<rmt::Loopback>& addr){
		auto find = registered_data_servers_.find(addr.get_address_id());
		if(find == registered_data_servers_.end()){
			return make_error<err::already_exists>("Server already bound to this address");
		}
		return {addr,*this};
	}

	/**
	 * Connect to a data server
	 */
	template<typename Schema, typename Encode>
	conveyor<data_client<Schema, Encode, rmt::Loopback>> data_connect(const remote_address<rmt::Loopback>& addr){
		auto class_id = srv.get_class_id();
		key_t key;
		key.data = {addr.get_address_id(), class_id.first, class_id.second};

		auto find = registered_data_servers_.find(key);
		if(find == registered_data_servers_.end()){
			return make_error<err::not_found>("Server not found");
		}

		
	}

	/**
	 * Internal function
	 */
	error_or<void> register_data_server(const remote_address<rmt::Loopback>& addr, i_data_server<rmt::Loopback>& srv){

		auto class_id = srv.get_class_id();
		key_t key;
		key.data = {addr.get_address_id(), class_id.first, class_id.second};
		auto insert = registered_data_servers_.emplace(std::make_pair(key, {srv}));
		if(insert.second){
			return make_error<err::already_exists>("Server already bound to this address");
		}

		return make_void();
	}
};
}