session.cpp

#include "session.hpp"
#include "server.hpp"

Session::pointer Session::create(asio::io_context& io, Server& server) {
	return pointer(new Session(io, server));
}

Session::Session(asio::io_context& io, Server& server)
	: socket_(io),
		server_(server) {}

tcp::socket& Session::socket() {
	return socket_;
}

void Session::start() {
	do_read();
}

void Session::do_read() {
	auto self = shared_from_this();

	socket_.async_read_some(asio::buffer(buff_),
		[this, self](std::error_code ec, std::size_t length) {
			if (!ec) {
				std::string msg(buff_.data(), length);
				server_.broadcast(msg, self);

				do_read();
			} else {
				server_.leave(self);
			}
		});
}

void Session::deliver(const std::string& msg) {
	// if the queue is NOT empty
	bool write_in_progress = !write_msgs_.empty();

	// add a new message
	write_msgs_.push_back(msg);

	// Only start writing if there was nothing in the queue before
	if (!write_in_progress) {
		do_write();
	}
}

void Session::do_write() {
	auto self = shared_from_this();

	asio::async_write(socket_,
		asio::buffer(write_msgs_.front()),
		[this, self](std::error_code ec, std::size_t /*length*/) {
			if (!ec) {
				write_msgs_.pop_front();

				if (!write_msgs_.empty()) {
					do_write();
				}
			} else {
				server_.leave(self);
			}
		});
}