#pragma once

#include "Parsing/IParserLineStream.h"

#include <unordered_map>
#include <vector>

class MockParserLineStream final : public IParserLineStream
{
public:
    static constexpr const char* MOCK_FILENAME = "MockFile";

private:
    class IncludePos
    {
    public:
        std::shared_ptr<std::string> m_filename;
        const std::vector<std::string>& m_lines;
        unsigned m_pos;

        IncludePos(std::string filename, const std::vector<std::string>& lines);
    };

    std::unordered_map<std::string, std::vector<std::string>> m_include_lines;
    std::vector<IncludePos> m_include_positions;

public:
    explicit MockParserLineStream(const std::vector<std::string>& lines);

    void AddIncludeLines(const std::string& filename, const std::vector<std::string>& lines);
    ParserLine NextLine() override;
    bool IncludeFile(const std::string& filename) override;
    void PopCurrentFile() override;
    _NODISCARD bool IsOpen() const override;
    _NODISCARD bool Eof() const override;
};