#pragma once

#include <forstio/async/async.hpp>
#include <forstio/buffer.hpp>
#include <forstio/io/io.hpp>
#include <forstio/codec/data.hpp>

namespace saw {

template <typename Incoming, typename Outgoing,
		  typename Encoding,
		  typename BufferT = ring_buffer>
class streaming_io_peer {
public:
	/**
	 * Constructor with the option to provide a custom codec, in and out buffer
	 */
	streaming_io_peer(
		own<conveyor_feeder<data<Incoming, Encoding>>> feed,
		own<async_io_stream> stream, codec<Encoding> codec, BufferT in, BufferT out);

	/**
	 * Constructor
	 */
	streaming_io_peer(
		own<conveyor_feeder<data<Incoming, Encoding>>> feed,
		own<async_io_stream> stream);

	/**
	 * Deleted copy and move constructors
	 */
	SAW_FORBID_COPY(streaming_io_peer);
	SAW_FORBID_MOVE(streaming_io_peer);

	/**
	 * Send a message to the remote peer
	 */
	error_or<void> send(data<Outgoing, Encoding> builder);

	/**
	 * A phantom conveyor feeder. Meant for interfacing with other components
	 */
	conveyor_feeder<data<Outgoing, Encoding>> &feeder();

	conveyor<void> on_read_disconnected();

private:
	/// @unimplemented
	class peer_conveyor_feeder final
		: public conveyor_feeder<data<Outgoing, Encoding>> {
	public:
		peer_conveyor_feeder(
			streaming_io_peer<Incoming, Outgoing, Encoding, BufferT> &peer_)
			: peer_{peer_} {}

		void feed(data<Outgoing, Encoding> &&data_) override {
			(void)data_;
		}

		void fail(error &&err) override { (void)err; }

		size_t space() const override { return 0; }

		size_t queued() const override { return 0; }

	private:
		streaming_io_peer<Incoming, Outgoing, Encoding,
						  BufferT> &peer_;
	};

private:
	own<conveyor_feeder<data<Incoming, Encoding>>>
		incoming_feeder_ = nullptr;

	own<async_io_stream> io_stream_;

	codec<Encoding> codec_;

	BufferT in_buffer_;
	BufferT out_buffer_;

	conveyor_sink sink_read_;
	conveyor_sink sink_write_;

	peer_conveyor_feeder conveyor_feeder_;
};

/**
 * Setup new streaming io peer with the provided network protocols.
 * This is a convenience wrapper intended for a faster setup of this class
 */
template <typename Incoming, typename Outgoing,
		  typename Encoding,
		  typename BufferT = ring_buffer>
std::pair<own<streaming_io_peer<Incoming, Outgoing, Encoding, BufferT>>,
		  conveyor<data<Incoming, Encoding>>>
new_streaming_io_peer(own<async_io_stream> stream);

} // namespace saw

#include "io_peer.tmpl.hpp"