#include <thread>
#define WIN32_LEAN_AND_MEAN 
#include <windows.h>

#ifndef LUA_LIB
#define LUA_LIB
#endif // !LUA_LIB
#ifndef LUA_BUILD_AS_DLL
#define LUA_BUILD_AS_DLL
#endif // !LUA_BUILD_AS_DLL
#include <Lua5.1/lua.hpp>
#pragma comment(lib, "Lua5.1.lib")

bool bContinueRunning;

inline void QLuaRegisterCallback(lua_State * L, lua_CFunction pfCallback, const char * pcNameToRegister)
{
    lua_pushcclosure(L, pfCallback, 0);
    lua_setfield(L, LUA_GLOBALSINDEX, pcNameToRegister);
}

int TestThreadProc(void * pvContext)
{
    lua_State * Lt = (lua_State *)pvContext;
    while (bContinueRunning)
    {
        lua_getfield(Lt, LUA_GLOBALSINDEX, "message");
        lua_pushstring(Lt, "Hello from thread");
        lua_pushinteger(Lt, 1);
        lua_pcall(Lt, 2, 0, NULL);
        ::Sleep(1000);
    }
    return 0;
}

int TestThreadProc1(void * pvContext)
{
    lua_State * Lt = (lua_State *)pvContext;
    while (bContinueRunning)
    {
        lua_getfield(Lt, LUA_GLOBALSINDEX, "message");
        lua_pushstring(Lt, "Hello from thread 1");
        lua_pushinteger(Lt, 1);
        lua_pcall(Lt, 2, 0, NULL);
        ::Sleep(1000);
    }
    return 0;
}

int TestThreadProc2(void * pvContext)
{
    lua_State * Lt = (lua_State *)pvContext;
    while (bContinueRunning)
    {
        lua_getfield(Lt, LUA_GLOBALSINDEX, "message");
        lua_pushstring(Lt, "Hello from thread 2");
        lua_pushinteger(Lt, 1);
        lua_pcall(Lt, 2, 0, NULL);
        ::Sleep(1000);
    }
    return 0;
}


int QLuaOnInit(lua_State * L)
{
    bContinueRunning = true;
    return 0;
}

int QLuaOnStop(lua_State * L)
{
    bContinueRunning = false;
    lua_pushinteger(L, 10000);
    return 1;
}

int QLuaMain(lua_State * L)
{

    lua_State * Lt = lua_newthread(L);
    ::std::thread TestThread(TestThreadProc, (void *)Lt);
    Lt = lua_newthread(L);
    ::std::thread TestThread1(TestThreadProc1, (void *)Lt);
    Lt = lua_newthread(L);
    ::std::thread TestThread2(TestThreadProc2, (void *)Lt);

    while (bContinueRunning)
    {
        lua_getfield(L, LUA_GLOBALSINDEX, "message");
        lua_pushstring(L, "Hello from main");
        lua_pushinteger(L, 1);
        lua_pcall(L, 2, 0, NULL);
        ::Sleep(1000);
    }
    TestThread.join();
    TestThread1.join();
    TestThread2.join();
    return 0;
}

static struct luaL_reg ls_lib[] = {
{ NULL, NULL }
};

extern "C" LUALIB_API int luaopen_QTest(lua_State *L)
{
    luaL_openlib(L, "QTest", ls_lib, 0);

    QLuaRegisterCallback(L, QLuaMain, "main");

    QLuaRegisterCallback(L, QLuaOnInit, "OnInit");
    QLuaRegisterCallback(L, QLuaOnStop, "OnStop");

    return 0;
}

BOOL APIENTRY DllMain(HMODULE hModule,
    DWORD  ul_reason_for_call,
    LPVOID lpReserved
)
{
    switch (ul_reason_for_call)
    {
    case DLL_PROCESS_ATTACH:
    case DLL_THREAD_ATTACH:
    case DLL_THREAD_DETACH:
    case DLL_PROCESS_DETACH:
        break;
    }
    return TRUE;
}