#ifndef OSTREAM_WRAPPER_H_62B23520_7C8E_11DE_8A39_0800200C9A66

#define OSTREAM_WRAPPER_H_62B23520_7C8E_11DE_8A39_0800200C9A66

#if defined(_MSC_VER) ||                                            \

    (defined(__GNUC__) && (__GNUC__ == 3 && __GNUC_MINOR__ >= 4) || \

     (__GNUC__ >= 4))  // GCC supports "pragma once" correctly since 3.4

#pragma once

#endif

#include <string>

#include <vector>

#include "yaml-cpp/dll.h"

namespace YAML {

class YAML_CPP_API ostream_wrapper {

 public:

  ostream_wrapper();

  explicit ostream_wrapper(std::ostream& stream);

  ostream_wrapper(const ostream_wrapper&) = delete;

  ostream_wrapper(ostream_wrapper&&) = delete;

  ostream_wrapper& operator=(const ostream_wrapper&) = delete;

  ostream_wrapper& operator=(ostream_wrapper&&) = delete;

  ~ostream_wrapper();

  void write(const std::string& str);

  void write(const char* str, std::size_t size);

  void set_comment() { m_comment = true; }

  const char* str() const {

    if (m_pStream) {

      return nullptr;

    } else {

      m_buffer[m_pos] = '\0';

      return &m_buffer[0];

    }

  }

  std::size_t row() const { return m_row; }

  std::size_t col() const { return m_col; }

  std::size_t pos() const { return m_pos; }

  bool comment() const { return m_comment; }

 private:

  void update_pos(char ch);

 private:

  mutable std::vector<char> m_buffer;

  std::ostream* const m_pStream;

  std::size_t m_pos;

  std::size_t m_row, m_col;

  bool m_comment;

};

template <std::size_t N>

inline ostream_wrapper& operator<<(ostream_wrapper& stream,

                                   const char (&str)[N]) {

  stream.write(str, N - 1);

  return stream;

}

inline ostream_wrapper& operator<<(ostream_wrapper& stream,

                                   const std::string& str) {

  stream.write(str);

  return stream;

}

inline ostream_wrapper& operator<<(ostream_wrapper& stream, char ch) {

  stream.write(&ch, 1);

  return stream;

}

}  // namespace YAML

#endif  // OSTREAM_WRAPPER_H_62B23520_7C8E_11DE_8A39_0800200C9A66