#include "Acceptor.h"
#include "InetAddress.h"
#include "TcpStream.h"

#include <thread>
#include <string.h>

// a thread-per-connection current echo server
int main(int argc, char* argv[])
{
  bool nodelay = false;
  bool ipv6 = false;
  for (int i = 1; i < argc; ++i) {
    if (strcmp(argv[i], "-6") == 0)
      ipv6 = true;
    if (strcmp(argv[i], "-D") == 0)
      nodelay = true;
  }
  InetAddress listenAddr(3007, ipv6);
  Acceptor acceptor(listenAddr);
  printf("Listen on port 3007\n");
  printf("Accepting... Ctrl-C to exit\n");
  int count = 0;

  while (true)
  {
    TcpStreamPtr tcpStream = acceptor.accept();
    printf("accepted no. %d client\n", ++count);
    if (nodelay)
      tcpStream->setTcpNoDelay(true);

    // C++11 doesn't allow capturing unique_ptr in lambda, C++14 allows.
    std::thread thr([count] (TcpStreamPtr stream) {
      printf("thread for no. %d client started.\n", count);
      char buf[4096];
      int nr = 0;
      while ( (nr = stream->receiveSome(buf, sizeof(buf))) > 0)
      {
        int nw = stream->sendAll(buf, nr);
        if (nw < nr)
        {
          break;
        }
      }
      printf("thread for no. %d client ended.\n", count);
    }, std::move(tcpStream));
    thr.detach();
  }
}