#pragma once

#include "common.hpp"
#include "data.hpp"
#include <forstio/error.hpp>

namespace saw {
template<typename Schema, typename Encoding>
class data_server<Schema, Encoding, rmt::Sycl> {
private:
	/**
	 * Device context class
	 */
	device<rmt::Sycl>* device_;

	/**
	 * Store for the data the server manages.
	 */
	std::unordered_map<uint64_t, data<Schema, Encoding, rmt::Sycl>> values_;
public:
	/**
	 * Main constructor
	 */
	data_server(device<rmt::Sycl>& device__):
		device_{&device__}
	{}

	/**
	 * Receive data which we will store.
	 */
	error_or<void> send(const data<Schema, Encoding, storage::Default>& dat, id<Schema> store_id){
		auto eoval = device_->copy_to_device(dat);
		if(eoval.is_error()){
			auto& err = eoval.get_error();
			return std::move(err);
		}
		return make_error<err::not_implemented>();
	}

	error_or<data<Schema, Encoding, storage::Default>> receive(id<Schema> store_id){
		return make_error<err::not_implemented>();
	}
};

template<typename Schema, typename Encoding>
class data_client<Schema, Encoding, rmt::Sycl> {
private:
	/**
	 * Corresponding server for this client
	 */
	data_server<Schema, Encoding, rmt::Sycl>* srv_;

	/**
	 * The next id for identifying issues on the remote side.
	 */
	uint64_t next_id_;
public:
	/**
	 * Main constructor
	 */
	data_client(data_server<Schema, Encoding, rmt::Sycl>& srv__):
		srv_{&srv__},
		next_id_{0u}
	{}

	/**
	 * Send data to.
	 */
	error_or<id<Schema>> send(const data<Schema, Encoding, storage::Default>& dat){
		id<Schema> dat_id{next_id_};
		auto eov = srv_->send(dat, dat_id);
		if(eov.is_error()){
			auto& err = eov.get_error();
			return std::move(err);
		}

		++next_id_;
		return dat_id;
	}
};
}