openmw/apps/openmw_test_suite/mwscript/test_utils.hpp

#ifndef MWSCRIPT_TESTING_UTIL_H
#define MWSCRIPT_TESTING_UTIL_H

#include <optional>
#include <string>
#include <utility>
#include <vector>

#include <components/compiler/context.hpp>
#include <components/compiler/errorhandler.hpp>
#include <components/compiler/exception.hpp>
#include <components/compiler/extensions.hpp>
#include <components/compiler/extensions0.hpp>
#include <components/compiler/fileparser.hpp>
#include <components/compiler/opcodes.hpp>
#include <components/compiler/scanner.hpp>

#include <components/interpreter/context.hpp>
#include <components/interpreter/installopcodes.hpp>
#include <components/interpreter/interpreter.hpp>
#include <components/interpreter/opcodes.hpp>

#include <components/misc/stringops.hpp>

namespace
{
    class TestCompilerContext : public Compiler::Context
    {
    public:
        bool canDeclareLocals() const override { return true; }
        char getGlobalType(const std::string& name) const override { return ' '; }
        std::pair<char, bool> getMemberType(const std::string& name, const std::string& id) const override { return {' ', false}; }
        bool isId(const std::string& name) const override { return Misc::StringUtils::ciEqual(name, "player"); }
    };

    class TestErrorHandler : public Compiler::ErrorHandler
    {
        std::vector<std::pair<std::string, Compiler::TokenLoc>> mErrors;

        void report(const std::string& message, const Compiler::TokenLoc& loc, Compiler::ErrorHandler::Type type) override
        {
            if(type == Compiler::ErrorHandler::ErrorMessage)
                mErrors.emplace_back(message, loc);
        }

        void report(const std::string& message, Compiler::ErrorHandler::Type type) override
        {
            report(message, {}, type);
        }

    public:
        void reset() override
        {
            Compiler::ErrorHandler::reset();
            mErrors.clear();
        }

        const std::vector<std::pair<std::string, Compiler::TokenLoc>>& getErrors() const { return mErrors; }
    };

    class LocalVariables
    {
        std::vector<int> mShorts;
        std::vector<int> mLongs;
        std::vector<float> mFloats;

        template<class T>
        T getLocal(std::size_t index, const std::vector<T>& vector) const
        {
            if(index < vector.size())
                return vector[index];
            return {};
        }

        template<class T>
        void setLocal(T value, std::size_t index, std::vector<T>& vector)
        {
            if(index >= vector.size())
                vector.resize(index + 1);
            vector[index] = value;
        }
    public:
        void clear()
        {
            mShorts.clear();
            mLongs.clear();
            mFloats.clear();
        }

        int getShort(std::size_t index) const { return getLocal(index, mShorts); };

        int getLong(std::size_t index) const { return getLocal(index, mLongs); };

        float getFloat(std::size_t index) const { return getLocal(index, mFloats); };

        void setShort(std::size_t index, int value) { setLocal(value, index, mShorts); };

        void setLong(std::size_t index, int value) { setLocal(value, index, mLongs); };

        void setFloat(std::size_t index, float value) { setLocal(value, index, mFloats); };
    };

    class GlobalVariables
    {
        std::map<std::string, int, std::less<>> mShorts;
        std::map<std::string, int, std::less<>> mLongs;
        std::map<std::string, float, std::less<>> mFloats;

        template<class T>
        T getGlobal(std::string_view name, const std::map<std::string, T, std::less<>>& map) const
        {
            auto it = map.find(name);
            if(it != map.end())
                return it->second;
            return {};
        }
    public:
        void clear()
        {
            mShorts.clear();
            mLongs.clear();
            mFloats.clear();
        }

        int getShort(std::string_view name) const { return getGlobal(name, mShorts); };

        int getLong(std::string_view name) const { return getGlobal(name, mLongs); };

        float getFloat(std::string_view name) const { return getGlobal(name, mFloats); };

        void setShort(std::string_view name, int value) { mShorts[std::string(name)] = value; };

        void setLong(std::string_view name, int value) { mLongs[std::string(name)] = value; };

        void setFloat(std::string_view name, float value) { mFloats[std::string(name)] = value; };
    };

    class TestInterpreterContext : public Interpreter::Context
    {
        LocalVariables mLocals;
        std::map<std::string, GlobalVariables, std::less<>> mMembers;
    public:
        std::string getTarget() const override { return {}; };

        int getLocalShort(int index) const override { return mLocals.getShort(index); };

        int getLocalLong(int index) const override { return mLocals.getLong(index); };

        float getLocalFloat(int index) const override { return mLocals.getFloat(index); };

        void setLocalShort(int index, int value) override { mLocals.setShort(index, value); };

        void setLocalLong(int index, int value) override { mLocals.setLong(index, value); };

        void setLocalFloat(int index, float value) override { mLocals.setFloat(index, value); };

        void messageBox(const std::string& message, const std::vector<std::string>& buttons) override {};

        void report(const std::string& message) override {};

        int getGlobalShort(std::string_view name) const override { return {}; };

        int getGlobalLong(std::string_view name) const override { return {}; };

        float getGlobalFloat(std::string_view name) const override { return {}; };

        void setGlobalShort(std::string_view name, int value) override {};

        void setGlobalLong(std::string_view name, int value) override {};

        void setGlobalFloat(std::string_view name, float value) override {};

        std::vector<std::string> getGlobals() const override { return {}; };

        char getGlobalType(std::string_view name) const override { return ' '; };

        std::string getActionBinding(std::string_view action) const override { return {}; };

        std::string getActorName() const override { return {}; };

        std::string getNPCRace() const override { return {}; };

        std::string getNPCClass() const override { return {}; };

        std::string getNPCFaction() const override { return {}; };

        std::string getNPCRank() const override { return {}; };

        std::string getPCName() const override { return {}; };

        std::string getPCRace() const override { return {}; };

        std::string getPCClass() const override { return {}; };

        std::string getPCRank() const override { return {}; };

        std::string getPCNextRank() const override { return {}; };

        int getPCBounty() const override { return {}; };

        std::string getCurrentCellName() const override { return {}; };

        int getMemberShort(std::string_view id, std::string_view name, bool global) const override
        {
            auto it = mMembers.find(id);
            if(it != mMembers.end())
                return it->second.getShort(name);
            return {};
        };

        int getMemberLong(std::string_view id, std::string_view name, bool global) const override
        {
            auto it = mMembers.find(id);
            if(it != mMembers.end())
                return it->second.getLong(name);
            return {};
        };

        float getMemberFloat(std::string_view id, std::string_view name, bool global) const override
        {
            auto it = mMembers.find(id);
            if(it != mMembers.end())
                return it->second.getFloat(name);
            return {};
        };

        void setMemberShort(std::string_view id, std::string_view name, int value, bool global) override { mMembers[std::string(id)].setShort(name, value); };

        void setMemberLong(std::string_view id, std::string_view name, int value, bool global) override { mMembers[std::string(id)].setLong(name, value); };

        void setMemberFloat(std::string_view id, std::string_view name, float value, bool global) override { mMembers[std::string(id)].setFloat(name, value); };
    };

    struct CompiledScript
    {
        std::vector<Interpreter::Type_Code> mByteCode;
        Compiler::Locals mLocals;

        CompiledScript(const std::vector<Interpreter::Type_Code>& code, const Compiler::Locals& locals) : mByteCode(code), mLocals(locals) {}
    };
}

#endif