/*
 * MRustC - Rust Compiler
 * - By John Hodge (Mutabah/thePowersGang)
 *
 * parse/ttstrea.hpp
 * - Token tree streams (for post-lex parsing)
 */
#pragma once

#include "tokentree.hpp"
#include "tokenstream.hpp"

/// Borrowed TTStream
class TTStream:
    public TokenStream
{
    ::std::vector< ::std::pair<unsigned int, const TokenTree*> > m_stack;
    ::std::shared_ptr<Span> m_parent_span;
    const Ident::Hygiene*   m_hygiene_ptr = nullptr;
public:
    TTStream(Span parent, const TokenTree& input_tt);
    ~TTStream();

    TTStream& operator=(const TTStream& x) { m_stack = x.m_stack; return *this; }

    Position getPosition() const override;
    ::std::shared_ptr<Span> outerSpan() const override { return m_parent_span; }

protected:
    Ident::Hygiene realGetHygiene() const override;
    Token realGetToken() override;
};

/// Owned TTStream
class TTStreamO:
    public TokenStream
{
    Position    m_last_pos;
    TokenTree   m_input_tt;
    ::std::vector< ::std::pair<unsigned int, TokenTree*> > m_stack;
    const Ident::Hygiene*   m_hygiene_ptr = nullptr;
public:
    ::std::shared_ptr<Span> m_parent_span;
    TTStreamO(Span parent, TokenTree input_tt);
    TTStreamO(TTStreamO&& x) = default;
    ~TTStreamO();

    TTStreamO& operator=(const TTStreamO& x) { m_stack = x.m_stack; return *this; }
    TTStreamO& operator=(TTStreamO&& x) = default;

    Position getPosition() const override;
    ::std::shared_ptr<Span> outerSpan() const override { return m_parent_span; }

protected:
    Ident::Hygiene realGetHygiene() const override;
    Token realGetToken() override;
};