client.cpp

#include "pch.hpp"

using asio::ip::tcp;

void send_loop(tcp::socket& socket, const std::string& username) {
	std::string line;

	while (std::getline(std::cin, line)) {
		std::string msg = username + ": " + line + "\n";

		std::error_code ec;
		asio::write(socket, asio::buffer(msg), ec);

		if (ec) {
			std::cerr << "Send error: " << ec.message() << std::endl;
			break;
		}
	}
}

void receive_loop(tcp::socket& socket) {
	std::array<char, 1024> buf;
	std::error_code ec;

	while (true) {
		size_t len = socket.read_some(asio::buffer(buf), ec);

		if (ec == asio::error::eof) {
			std::cout << "Disconnected from server\n";
			break;
		} else if (ec) {
			std::cerr << "Receive error: " << ec.message() << "\n";
			break;
		}

		std::cout.write(buf.data(), len);
		std::cout << std::endl;
	}
}

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

	try {
		asio::io_context io;
		tcp::resolver resolver(io);
		auto endpoints =
			resolver.resolve("localhost", argv[1]);
		tcp::socket socket(io);
		asio::connect(socket, endpoints);

		std::thread sender(send_loop, std::ref(socket), argv[2]);
		receive_loop(socket);
		
		sender.join();
	} catch (std::exception& e) {
		std::cerr << e.what() << std::endl;
	}

	return 0;
}