StreamConsole.h

#pragma once

#include <iosfwd>

#include <Poco/AtomicCounter.h>
#include <Poco/Semaphore.h>
#include <Poco/SharedPtr.h>

#include "io/Console.h"

namespace BeeeOn {

class StreamConsoleSessionImpl : public ConsoleSessionImpl {
public:
        typedef Poco::SharedPtr<StreamConsoleSessionImpl> Ptr;

        StreamConsoleSessionImpl(
                Poco::Semaphore &semaphore,
                std::istream &in,
                std::ostream &out);
        ~StreamConsoleSessionImpl();

        std::string readUntil(const char c) override;
        std::string readBytes(const unsigned int length) override;
        void print(const std::string &text, bool newline = true) override;
        bool eof() override;

private:
        Poco::Semaphore &m_semaphore;
        std::istream &m_in;
        std::ostream &m_out;
};

class StreamConsole : public Console {
public:
        typedef Poco::SharedPtr<StreamConsole> Ptr;

        StreamConsole(std::istream &in, std::ostream &out);

        void close() override;

protected:
        ConsoleSessionImpl::Ptr openSession() override;

private:
        Poco::Semaphore m_semaphore;
        Poco::AtomicCounter m_close;

        std::istream &m_in;
        std::ostream &m_out;
};

}