AtomsScriptLang.h

#pragma once
// ===========================================================================
// Copyright (c) 2015-2024 Toolchefs Ltd. All rights reserved.
//
// Use of this software is subject to the terms of the Toolchefs license
// agreement provided at the time of installation or download, or which
// otherwise accompanies this software in either electronic or hard copy form.
// ===========================================================================
 
#include "Globals.h"
#include <memory>
#include <functional>
#include <vector>
#include <string>
 
namespace Atoms
{
    class ATOMSJIT_EXPORT AtomsScriptLang
    {
    public:
 
        static AtomsScriptLang& instance();
 
        void initialize();
 
        std::vector<std::pair<std::string, std::string>> testCode(const std::string& source);
 
        bool testCodeWithLog(const std::string& source, const std::function<void(const char*, const char*)>& externalLog);
 
        bool addCode(const std::string& name, const std::string& source);
 
        bool addCodeWithArgs(const std::string& name, const std::string& source, const std::string& args);
 
        void removeCode(const std::string& name);
 
        template<typename T>
        std::function<T> lookup(const std::string& name)
        {
            return std::function<T>(reinterpret_cast<T*>(_lookup(name)));
        }
 
        template<typename T>
        std::function<T> lookup(const std::string& moduleName, const std::string& name)
        {
            return std::function<T>(reinterpret_cast<T*>(_lookup(moduleName, name)));
        }
 
        uintptr_t lookupPtr(const std::string& moduleName, const std::string& name)
        {
            return _lookup(moduleName, name);
        }
 
        void clear();
 
        std::string generateUniqueName();
 
        void setLog(void (*callback)(const char*)) { m_log = callback; }
 
        void log(const char* msg) { if (m_log) m_log(msg); }
 
        void setDefaultArgs(const std::vector<std::string>& args);
 
        void addDefaultArgs(const std::vector<std::string>& args);
 
    private:
 
        AtomsScriptLang();
 
        ~AtomsScriptLang();
 
        AtomsScriptLang(const AtomsScriptLang&) = delete;
 
        AtomsScriptLang& operator=(const AtomsScriptLang&) = delete;
 
        uintptr_t _lookup(const std::string& name);
 
        uintptr_t _lookup(const std::string& moduleName, const std::string& name);
 
    private:
 
#ifdef ENABLE_ATOMS_LLVM
        std::unique_ptr<class AtomsJIT> m_jit;
#endif
 
        size_t m_unique_identifier;
 
        void (*m_log)(const char*);
    };
 
}