#ifndef GAME_SCRIPT_SCRIPTMANAGER_H
#define GAME_SCRIPT_SCRIPTMANAGER_H

#include <map>
#include <set>
#include <string>

#include <components/compiler/fileparser.hpp>
#include <components/compiler/streamerrorhandler.hpp>

#include <components/interpreter/interpreter.hpp>
#include <components/interpreter/types.hpp>

#include <components/esm/refid.hpp>

#include "../mwbase/scriptmanager.hpp"

#include "globalscripts.hpp"

namespace MWWorld
{
    class ESMStore;
}

namespace Compiler
{
    class Context;
}

namespace Interpreter
{
    class Context;
    class Interpreter;
}

namespace MWScript
{
    class ScriptManager : public MWBase::ScriptManager
    {
        Compiler::StreamErrorHandler mErrorHandler;
        const MWWorld::ESMStore& mStore;
        Compiler::Context& mCompilerContext;
        Compiler::FileParser mParser;
        Interpreter::Interpreter mInterpreter;

        struct CompiledScript
        {
            Interpreter::Program mProgram;
            Compiler::Locals mLocals;
            std::set<ESM::RefId> mInactive;

            explicit CompiledScript(Interpreter::Program&& program, const Compiler::Locals& locals)
                : mProgram(std::move(program))
                , mLocals(locals)
            {
            }
        };

        std::unordered_map<ESM::RefId, CompiledScript> mScripts;
        GlobalScripts mGlobalScripts;
        std::unordered_map<ESM::RefId, Compiler::Locals> mOtherLocals;
        std::vector<ESM::RefId> mScriptBlacklist;

    public:
        ScriptManager(const MWWorld::ESMStore& store, Compiler::Context& compilerContext, int warningsMode,
            const std::vector<ESM::RefId>& scriptBlacklist);

        void clear() override;

        bool run(const ESM::RefId& name, Interpreter::Context& interpreterContext) override;
        ///< Run the script with the given name (compile first, if not compiled yet)

        bool compile(const ESM::RefId& name) override;
        ///< Compile script with the given namen
        /// \return Success?

        std::pair<int, int> compileAll() override;
        ///< Compile all scripts
        /// \return count, success

        const Compiler::Locals& getLocals(const ESM::RefId& name) override;
        ///< Return locals for script \a name.

        GlobalScripts& getGlobalScripts() override;

        const Compiler::Extensions& getExtensions() const override;
    };
}

#endif