#include <iostream>

#include <SFML/Network.hpp>

int main(int argc, char* argv[])
{
    sf::IpAddress ip = sf::IpAddress::getLocalAddress();
    sf::TcpSocket socket;
    char connectionType, mode;

    std::string text = "Connected to: ";

    std::cout << "Enter (s) for Server or (c) for Client" << std::endl;
    std::cin >> connectionType;

    if (connectionType == 's')
    {
        sf::TcpListener listener;
        listener.listen(1234);
        listener.accept(socket);

        text += "Server";
        mode = 's';
    }
    else if (connectionType == 'c')
    {
        socket.connect(ip, 1234);
        text += "Client";
        mode = 'r';
    }
    else
        return 1;

    bool done = false;

    while (!done)
    {
        if (mode == 's')
        {
            std::getline(std::cin, text);

            sf::Packet packet;
            packet << text;
            socket.send(packet);

            packet.clear();
            mode = 'r';
        }
        else
        {
            sf::Packet packet;
            packet.clear();

            if (socket.receive(packet) == sf::Socket::Disconnected)
            {
                std::cout << "Disconnected!" << std::endl;
                break;
            }

            //This next line fixes the error just need to comment it out
            //text.reserve(1);
            packet >> text;
            packet.clear();

            std::cout << text << std::endl;

            mode = 's';
        }
    }

    system("pause");

    return 0;
}