#include <kapi/cio.hpp>

#include "kernel/test_support/log_buffer.hpp"

#include <iostream>
#include <string>
#include <string_view>

namespace kapi::cio
{

  namespace
  {

    class test_output_device : public output_device
    {
    public:
      test_output_device() = default;

      auto write(output_stream stream, std::string_view text) -> void override
      {
        auto & standard_stream = stream == output_stream::stdout ? std::cout : std::cerr;
        standard_stream << text;
        if (text != "\n")
        {
          kernel::tests::log_buffer::append(std::string{text});
        }
      }
    } device{};

  }  // namespace

  auto init() -> void
  {
    set_output_device(device);
  }

  auto reset() -> void
  {
    kernel::tests::log_buffer::clear();
  }

}  // namespace kapi::cio