RawText.h

Go to the documentation of this file.

 
#pragma once
 
#include <AST/ASTNode.h>
 
namespace AST {
 
class RawText : public ASTNode {
    protected:
        AST::Token<std::string> m_TEXT;
        std::string getEscapedText() const {
            std::string escaped;
            for (char c : m_TEXT.getValue()) {
                switch (c) {
                    case '\n': escaped += "\\n"; break;
                    case '\t': escaped += "\\t"; break;
                    case '\r': escaped += "\\r"; break;
                    default: escaped += c; break;
                }
            }
            return escaped;
        }
    public:
        static constexpr AST::NodeType static_type = AST::NodeType::RawText;
        constexpr AST::NodeType getType() const override { return static_type; }
 
        const AST::Token<std::string>& TEXT() const {
            return m_TEXT;
        }
        void setText(const AST::Token<std::string>& text) {
            m_TEXT = text;
        }
        void appendText(const std::string& text) {
            m_TEXT += text;
        }
 
        std::ostream& prettyPrint(std::ostream& os, int indentation_level = 0) const override {
            std::string indent(indentation_level * PRETTYPRINT_INDENTATION_AMOUNT, ' ');
            os << indent << "(RawText " << getEscapedText();
            for (const auto& child : children) {
                os << std::endl;
                child->prettyPrint(os, indentation_level + 1);
            }
            os << ")" << std::flush;
            return os;
        }
};
 
} // namespace AST