cpp11chat/chat_networking.hpp

112 lines
3.4 KiB
C++
Raw Normal View History

#include <iostream>
#include <string>
#include <functional>
#include <thread>
#include <boost/asio.hpp>
#include <algorithm>
#include "chat_messages.hpp"
namespace chat
{
using namespace chat;
using namespace boost;
using boost::asio::ip::tcp;
class client_network_manager
{
private:
asio::io_service& _ios;
tcp::socket _socket;
2016-12-03 12:22:51 +00:00
asio::streambuf _isb;//<-------.
asio::streambuf _osb;//<---. |
std::istream _is;//--------|---'
std::ostream _os;//--------'
std::string _login;
static void throw_if_error(boost::system::error_code& ec)
{
if (ec)
throw std::runtime_error("Networking error: " + ec.message());
}
std::string receive_message_sync()
{
std::string data;
asio::read_until(_socket, _isb, byte(message::TERM));
std::getline(_is, data, byte(message::TERM));
return data;
}
void handshake(boost::system::error_code ec, tcp::resolver::iterator)
{
throw_if_error(ec);
std::cout << receive_message_sync() << '\n';
_os << chat_message(message::HELLO);
_os << chat_message(message::NEPTUN, _login);
std::reverse(_login.begin(), _login.end());
_os << chat_message(message::PASSW, _login);
asio::write(_socket, _osb);
std::cout << receive_message_sync() << '\n';
std::cout << receive_message_sync() << '\n';
std::cout << receive_message_sync() << '\n';
receive();
}
void receive()
{
asio::async_read_until(_socket, _isb, byte(message::TERM),
[this](boost::system::error_code ec, size_t)
{
throw_if_error(ec);
std::string data;
std::getline(_is, data, byte(message::TERM));
switch (data[0])
{
// TODO: handle stuff
}
receive();
});
}
2016-12-03 12:22:51 +00:00
void connect(tcp::resolver::iterator epit)
{
2016-12-03 12:22:51 +00:00
asio::async_connect
(
_socket, epit,
std::bind(&client_network_manager::handshake,
this, std::placeholders::_1,
std::placeholders::_2)
);
}
public:
2016-12-03 12:22:51 +00:00
client_network_manager(asio::io_service& ioservice,
tcp::resolver::iterator epit,
std::string login)
: _ios(ioservice), _socket(_ios), _isb(), _osb(), _is(&_isb), _os(&_osb), _login(login)
{ connect(epit); }
void send(chat_message message)
{
_ios.post([this, message]
{
2016-12-03 12:22:51 +00:00
asio::async_write(_socket, asio::buffer(message.get()),
[](boost::system::error_code, size_t)
{});
});
}
};
}