#pragma once

#include <util/datetime/base.h>

#include <util/generic/ptr.h>

#include <util/stream/input.h>
#include <util/stream/output.h>

#include <util/thread/pool.h>

#include <functional>

class TInetStreamSocket;

// Simple server listens on the specified port and launches
// requestHandler in the separate thread for each incoming connection.
class TSimpleServer
{
public:
    using TRequestHandler = std::function<void(IInputStream* input, IOutputStream* output)>;

public:
    TSimpleServer(
        int port,
        TRequestHandler requestHandler,
        bool dontStartServer = false,
        TDuration acceptDelay = TDuration::Zero());

    ~TSimpleServer();

    void Stop();

    int GetPort() const;
    TString GetAddress() const;

private:
    const int Port_;
    std::unique_ptr<IThreadPool> ThreadPool_;
    THolder<IThreadFactory::IThread> ListenerThread_;
    std::unique_ptr<TInetStreamSocket> SendFinishSocket_;
};