#include "ScriptTypes.h"
#include "ScriptSystem.h"
#include "ScriptWorker.h"

#pragma once

class CScriptInstance
{
private:
    JSInstanceName  m_InstanceName;
    CScriptSystem*  m_System;
    duk_context*    m_Ctx;
    size_t          m_RefCount;
    uint64_t        m_ExecTimeout;
    uint64_t        m_ExecStartTime;
    std::ifstream   m_SourceFile;
    char*           m_SourceCode;
    JSAppCallbackID m_CurExecCallbackId;
    std::vector<CScriptWorker*> m_Workers;
    bool            m_bStopping;
    bool            m_bAborting;

public:
    CScriptInstance(CScriptSystem* sys, const char* name);
    ~CScriptInstance();

    JSInstanceName& Name();
    CScriptSystem*  System();
    CDebuggerUI*    Debugger();
    JSAppCallbackID CallbackId();

    bool   Run(const char* path);
    void   SetExecTimeout(uint64_t timeout);
    bool   IsTimedOut();

    inline size_t GetRefCount() { return m_RefCount; }
    void   IncRefCount();
    void   DecRefCount();
    void   SetStopping(bool bStopping);
    inline bool IsStopping() { return m_bStopping; }
    bool   PrepareAbort();

    bool RegisterWorker(CScriptWorker* worker);
    void UnregisterWorker(CScriptWorker* worker);
    void StopRegisteredWorkers();

    inline void RawInvokeAppCallback(JSAppCallback& cb, void* _hookEnv)
    {
        m_CurExecCallbackId = cb.m_CallbackId;

        RawCall(cb.m_DukFuncHeapPtr, cb.m_DukArgSetupFunc, _hookEnv);

        if (cb.m_CleanupFunc != nullptr)
        {
            cb.m_CleanupFunc(m_Ctx, _hookEnv);
        }

        m_CurExecCallbackId = JS_INVALID_CALLBACK;
    }

    inline void RawCall(void *dukFuncHeapPtr, JSDukArgSetupFunc argSetupFunc, void *param = nullptr)
    {
        m_ExecStartTime = Timestamp();
        duk_push_heapptr(m_Ctx, dukFuncHeapPtr);
        duk_idx_t nargs = argSetupFunc ? argSetupFunc(m_Ctx, param) : 0;

        try
        {
            if (duk_pcall(m_Ctx, nargs) == DUK_EXEC_ERROR)
            {
                duk_get_prop_string(m_Ctx, -1, "stack");
                m_System->ConsoleLog("%s", duk_safe_to_string(m_Ctx, -1));
                duk_pop(m_Ctx);
            }

            duk_pop(m_Ctx);
        }
        catch (std::runtime_error& exc)
        {
            FatalHandler(exc);
        }
    }

    void   RawCMethodCall(void* dukThisHeapPtr, duk_c_function func,
                          JSDukArgSetupFunc argSetupFunc = nullptr,
                          void* argSetupParam = nullptr);

    void   PostCMethodCall(void* dukThisHeapPtr, duk_c_function func,
                           JSDukArgSetupFunc argSetupFunc = nullptr,
                           void* argSetupParam = nullptr, size_t argSetupParamSize = 0);

    void   RawConsoleInput(const char* code);

private:
    void FatalHandler(std::runtime_error& exc);
    static uint64_t Timestamp();
    void Cleanup();
};