#pragma once

// Faster than ostringstream--returns its string by ref
#include <ostream>
#include "c11log/details/stack_buf.h"

namespace c11log
{
namespace details
{

class stack_devicebuf:public std::streambuf
{
public:
    using Base = std::streambuf;
    stack_devicebuf() = default;
    ~stack_devicebuf() = default;

    stack_devicebuf(const stack_devicebuf& other) = delete;
    stack_devicebuf(stack_devicebuf&& other) = delete;
    stack_devicebuf& operator=(const stack_devicebuf&) = delete;
    stack_devicebuf& operator=(stack_devicebuf&&) = delete;

    bufpair_t buf()
    {
        return _stackbuf.get();
    }

    std::size_t size()
    {
        return _stackbuf.size();
    }

    void clear()
    {
        _stackbuf.clear();
    }

protected:
    // copy the give buffer into the accumulated fast buffer
    std::streamsize xsputn(const char_type* s, std::streamsize count) override
    {
        _stackbuf.append(s, static_cast<unsigned int>(count));
        return count;
    }

    int_type overflow(int_type ch) override
    {
        if (traits_type::not_eof(ch))
        {
            char c = traits_type::to_char_type(ch);
            xsputn(&c, 1);
        }
        return ch;
    }
private:
    stack_buf<192> _stackbuf;
};

class stack_oss:public std::ostream
{
public:
    stack_oss():std::ostream(&_dev) {}
    ~stack_oss() = default;

    stack_oss(const stack_oss& other) = delete;
    stack_oss(stack_oss&& other) = delete;
    stack_oss& operator=(const stack_oss& other) = delete;

    bufpair_t buf()
    {
        return _dev.buf();
    }

    std::size_t size()
    {
        return _dev.size();
    }

    void clear()
    {
        _dev.clear();
    }

private:
    stack_devicebuf _dev;
};
}
}