revng-revng/include/revng/PTML/IndentedOstream.h

#pragma once

//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#include "llvm/Support/raw_ostream.h"

#include "revng/PTML/Tag.h"

namespace ptml {

class IndentedOstream : public llvm::raw_ostream {
private:
  const MarkupBuilder &B;

  int IndentSize;
  int IndentDepth;
  // If the buffer ends with a newline, we want to delay emitting indentation on
  // the next character, so that we can account for (de)indentations that could
  // happen in the meantime. This boolean tracks that.
  bool TrailingNewline;
  raw_ostream &OS;

public:
  explicit IndentedOstream(llvm::raw_ostream &OS,
                           const MarkupBuilder &B,
                           int IndentSize = 2) :
    B(B),
    IndentSize(IndentSize),
    IndentDepth(0),
    TrailingNewline(false),
    OS(OS) {
    SetUnbuffered();
  }

  struct Scope {
  private:
    IndentedOstream &OS;

  public:
    Scope(IndentedOstream &OS) : OS(OS) { OS.indent(); }

    ~Scope() { OS.unindent(); }
  };

  Scope scope() { return Scope(*this); }

  void indent() { IndentDepth = std::min(INT_MAX, IndentDepth + 1); }
  void unindent() { IndentDepth = std::max(0, IndentDepth - 1); }

  uint64_t currentIndentation() const { return IndentSize * IndentDepth; }

  const MarkupBuilder &getMarkupBuilder() const { return B; }

private:
  void write_impl(const char *Ptr, size_t Size) override;
  uint64_t current_pos() const override;
  void writeIndent();
};

} // namespace ptml