/*

 * Copyright (c) 2022, Ali Mohammad Pur <mpfard@serenityos.org>

 *

 * SPDX-License-Identifier: BSD-2-Clause

 */

#pragma once

#include <AK/ByteString.h>

#include <AK/GenericLexer.h>

#include <AK/HashMap.h>

#include <AK/Variant.h>

#include <AK/Vector.h>

#include <LibXML/FundamentalTypes.h>

namespace XML {

struct Attribute {

    Name name;

    ByteString value;

};

struct Node {

    struct Text {

        StringBuilder builder;

    };

    struct Comment {

        ByteString text;

    };

    struct Element {

        Name name;

        HashMap<Name, ByteString> attributes;

        Vector<NonnullOwnPtr<Node>> children;

    };

    bool operator==(Node const&) const;

    LineTrackingLexer::Position offset;

    Variant<Text, Comment, Element> content;

    Node* parent { nullptr };

    bool is_text() const { return content.has<Text>(); }

    Text const& as_text() const { return content.get<Text>(); }

    bool is_comment() const { return content.has<Comment>(); }

    Comment const& as_comment() const { return content.get<Comment>(); }

    bool is_element() const { return content.has<Element>(); }

    Element const& as_element() const { return content.get<Element>(); }

};

}