tty.hpp

#pragma once


#include <utility>
#include <memory>
#include <uv.h>
#include "stream.hpp"
#include "util.hpp"


namespace uvw {


namespace details {


struct ResetModeMemo {
    ~ResetModeMemo() {
        uv_tty_reset_mode();
    }
};


enum class UVTTYModeT: std::underlying_type_t<uv_tty_mode_t> {
    NORMAL = UV_TTY_MODE_NORMAL,
    RAW = UV_TTY_MODE_RAW,
    IO = UV_TTY_MODE_IO
};


}


class TTYHandle final: public StreamHandle<TTYHandle, uv_tty_t> {
    static auto resetModeMemo() {
        static std::weak_ptr<details::ResetModeMemo> weak;
        auto shared = weak.lock();
        if(!shared) { weak = shared = std::make_shared<details::ResetModeMemo>(); }
        return shared;
    }

public:
    using Mode = details::UVTTYModeT;

    explicit TTYHandle(ConstructorAccess ca, std::shared_ptr<Loop> ref, FileHandle desc, bool readable)
        : StreamHandle{ca, std::move(ref)},
          memo{resetModeMemo()},
          fd{desc},
          rw{readable}
    {}

    bool init() {
        return initialize(&uv_tty_init, fd, rw);
    }

    bool mode(Mode m) {
        return (0 == uv_tty_set_mode(get(), static_cast<std::underlying_type_t<Mode>>(m)));
    }

    bool reset() noexcept {
        return (0 == uv_tty_reset_mode());
    }

    WinSize getWinSize() {
        WinSize size;

        if(0 != uv_tty_get_winsize(get(), &size.width, &size.height)) {
            size.width = -1;
            size.height = -1;
        }

        return size;
    }

private:
    std::shared_ptr<details::ResetModeMemo> memo;
    FileHandle::Type fd;
    int rw;
};


}