server.cpp

#include "pch.hpp"

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

int main(int argc, char* argv[]) {
	if (argc < 2) {
		std::cerr << "Usage: server <port>\n";
		return 1;
	}

	try {
		asio::io_context io;
		Server serv(io, atoi(argv[1]));
		io.run();
	} catch (std::exception& e) {
		std::cerr << e.what() << std::endl;
	}

	return 0;
}

Server::Server(asio::io_context& io, int port)
    : io_(io),
      acceptor_(io, tcp::endpoint(tcp::v4(), port))
{
	start_accept();
}

void Server::broadcast(const std::string& msg, std::shared_ptr<Session> sender) {
	for (auto& client : clients) {
		if (client != sender) {
				client->deliver(msg);
		}
	}
}

void Server::leave(std::shared_ptr<Session> session) {
	clients.erase(session);
}

void Server::start_accept() {
	auto new_connection = Session::create(io_, *this);

	acceptor_.async_accept(new_connection->socket(),
		[this, new_connection](std::error_code ec) {
			if (!ec) {
				clients.insert(new_connection);
				new_connection->start();
				std::cout << "Someone connected\n";
			}

			start_accept(); // accept next client
		});
}