/*
Minetest
Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation; either version 2.1 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/

#include "cpp_api/s_base.h"
#include "cpp_api/s_internal.h"
#include "cpp_api/s_security.h"
#include "lua_api/l_object.h"
#include "common/c_converter.h"
#include "server/player_sao.h"
#include "filesys.h"
#include "content/mods.h"
#include "porting.h"
#include "util/string.h"
#include "server.h"
#ifndef SERVER
#include "client/client.h"
#endif


extern "C" {
#include "lualib.h"
#if USE_LUAJIT
	#include "luajit.h"
#endif
}

#include <cstdio>
#include <cstdarg>
#include "script/common/c_content.h"
#include <sstream>


class ModNameStorer
{
private:
	lua_State *L;
public:
	ModNameStorer(lua_State *L_, const std::string &mod_name):
		L(L_)
	{
		// Store current mod name in registry
		lua_pushstring(L, mod_name.c_str());
		lua_rawseti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_CURRENT_MOD_NAME);
	}
	~ModNameStorer()
	{
		// Clear current mod name from registry
		lua_pushnil(L);
		lua_rawseti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_CURRENT_MOD_NAME);
	}
};


/*
	ScriptApiBase
*/

ScriptApiBase::ScriptApiBase(ScriptingType type):
		m_type(type)
{
#ifdef SCRIPTAPI_LOCK_DEBUG
	m_lock_recursion_count = 0;
#endif

	m_luastack = luaL_newstate();
	FATAL_ERROR_IF(!m_luastack, "luaL_newstate() failed");

	lua_atpanic(m_luastack, &luaPanic);

	if (m_type == ScriptingType::Client)
		clientOpenLibs(m_luastack);
	else
		luaL_openlibs(m_luastack);

	// Make the ScriptApiBase* accessible to ModApiBase
#if INDIRECT_SCRIPTAPI_RIDX
	*(void **)(lua_newuserdata(m_luastack, sizeof(void *))) = this;
#else
	lua_pushlightuserdata(m_luastack, this);
#endif
	lua_rawseti(m_luastack, LUA_REGISTRYINDEX, CUSTOM_RIDX_SCRIPTAPI);

	// Add and save an error handler
	lua_getglobal(m_luastack, "debug");
	lua_getfield(m_luastack, -1, "traceback");
	lua_rawseti(m_luastack, LUA_REGISTRYINDEX, CUSTOM_RIDX_BACKTRACE);
	lua_pop(m_luastack, 1); // pop debug

	// If we are using LuaJIT add a C++ wrapper function to catch
	// exceptions thrown in Lua -> C++ calls
#if USE_LUAJIT
	lua_pushlightuserdata(m_luastack, (void*) script_exception_wrapper);
	luaJIT_setmode(m_luastack, -1, LUAJIT_MODE_WRAPCFUNC | LUAJIT_MODE_ON);
	lua_pop(m_luastack, 1);
#endif

	// Add basic globals
	lua_newtable(m_luastack);
	lua_setglobal(m_luastack, "core");

	if (m_type == ScriptingType::Client)
		lua_pushstring(m_luastack, "/");
	else
		lua_pushstring(m_luastack, DIR_DELIM);
	lua_setglobal(m_luastack, "DIR_DELIM");

	lua_pushstring(m_luastack, porting::getPlatformName());
	lua_setglobal(m_luastack, "PLATFORM");

	// Make sure Lua uses the right locale
	setlocale(LC_NUMERIC, "C");
}

ScriptApiBase::~ScriptApiBase()
{
	lua_close(m_luastack);
}

int ScriptApiBase::luaPanic(lua_State *L)
{
	std::ostringstream oss;
	oss << "LUA PANIC: unprotected error in call to Lua API ("
		<< readParam<std::string>(L, -1) << ")";
	FATAL_ERROR(oss.str().c_str());
	// NOTREACHED
	return 0;
}

void ScriptApiBase::clientOpenLibs(lua_State *L)
{
	static const std::vector<std::pair<std::string, lua_CFunction>> m_libs = {
		{ "", luaopen_base },
		{ LUA_TABLIBNAME,  luaopen_table   },
		{ LUA_OSLIBNAME,   luaopen_os      },
		{ LUA_STRLIBNAME,  luaopen_string  },
		{ LUA_MATHLIBNAME, luaopen_math    },
		{ LUA_DBLIBNAME,   luaopen_debug   },
#if USE_LUAJIT
		{ LUA_JITLIBNAME,  luaopen_jit     },
#endif
	};

	for (const std::pair<std::string, lua_CFunction> &lib : m_libs) {
	    lua_pushcfunction(L, lib.second);
	    lua_pushstring(L, lib.first.c_str());
	    lua_call(L, 1, 0);
	}
}

void ScriptApiBase::loadMod(const std::string &script_path,
		const std::string &mod_name)
{
	ModNameStorer mod_name_storer(getStack(), mod_name);

	loadScript(script_path);
}

void ScriptApiBase::loadScript(const std::string &script_path)
{
	verbosestream << "Loading and running script from " << script_path << std::endl;

	lua_State *L = getStack();

	int error_handler = PUSH_ERROR_HANDLER(L);

	bool ok;
	if (m_secure) {
		ok = ScriptApiSecurity::safeLoadFile(L, script_path.c_str());
	} else {
		ok = !luaL_loadfile(L, script_path.c_str());
	}
	ok = ok && !lua_pcall(L, 0, 0, error_handler);
	if (!ok) {
		const char *error_msg = lua_tostring(L, -1);
		if (!error_msg)
			error_msg = "(error object is not a string)";
		lua_pop(L, 2); // Pop error message and error handler
		throw ModError("Failed to load and run script from " +
				script_path + ":\n" + error_msg);
	}
	lua_pop(L, 1); // Pop error handler
}

#ifndef SERVER
void ScriptApiBase::loadModFromMemory(const std::string &mod_name)
{
	ModNameStorer mod_name_storer(getStack(), mod_name);

	sanity_check(m_type == ScriptingType::Client);

	const std::string init_filename = mod_name + ":init.lua";
	const std::string chunk_name = "@" + init_filename;

	const std::string *contents = getClient()->getModFile(init_filename);
	if (!contents)
		throw ModError("Mod \"" + mod_name + "\" lacks init.lua");

	verbosestream << "Loading and running script " << chunk_name << std::endl;

	lua_State *L = getStack();

	int error_handler = PUSH_ERROR_HANDLER(L);

	bool ok = ScriptApiSecurity::safeLoadString(L, *contents, chunk_name.c_str());
	if (ok)
		ok = !lua_pcall(L, 0, 0, error_handler);
	if (!ok) {
		const char *error_msg = lua_tostring(L, -1);
		if (!error_msg)
			error_msg = "(error object is not a string)";
		lua_pop(L, 2); // Pop error message and error handler
		throw ModError("Failed to load and run mod \"" +
				mod_name + "\":\n" + error_msg);
	}
	lua_pop(L, 1); // Pop error h<span class="hl com">/*</span>
<span class="hl com">Minetest</span>
<span class="hl com">Copyright (C) 2013 celeron55, Perttu Ahola &lt;celeron55&#64;gmail.com&gt;</span>
<span class="hl com"></span>
<span class="hl com">This program is free software; you can redistribute it and/or modify</span>
<span class="hl com">it under the terms of the GNU Lesser General Public License as published by</span>
<span class="hl com">the Free Software Foundation; either version 2.1 of the License, or</span>
<span class="hl com">(at your option) any later version.</span>
<span class="hl com"></span>
<span class="hl com">This program is distributed in the hope that it will be useful,</span>
<span class="hl com">but WITHOUT ANY WARRANTY; without even the implied warranty of</span>
<span class="hl com">MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the</span>
<span class="hl com">GNU Lesser General Public License for more details.</span>
<span class="hl com"></span>
<span class="hl com">You should have received a copy of the GNU Lesser General Public License along</span>
<span class="hl com">with this program; if not, write to the Free Software Foundation, Inc.,</span>
<span class="hl com">51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.</span>
<span class="hl com">*/</span>

<span class="hl ppc">#include</span> <span class="hl pps">&quot;scripting_mainmenu.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;content/mods.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;cpp_api/s_internal.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;lua_api/l_base.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;lua_api/l_http.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;lua_api/l_mainmenu.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;lua_api/l_sound.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;lua_api/l_util.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;lua_api/l_settings.h&quot;</span><span class="hl ppc"></span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;log.h&quot;</span><span class="hl ppc"></span>

<span class="hl kwc">extern</span> <span class="hl str">&quot;C&quot;</span> <span class="hl opt">{</span>
<span class="hl ppc">#include</span> <span class="hl pps">&quot;lualib.h&quot;</span><span class="hl ppc"></span>
<span class="hl opt">}</span>
<span class="hl ppc">#define MAINMENU_NUM_ASYNC_THREADS 4</span>


<span class="hl kwc">MainMenuScripting</span><span class="hl opt">::</span><span class="hl kwd">MainMenuScripting</span><span class="hl opt">(</span>GUIEngine<span class="hl opt">*</span> guiengine<span class="hl opt">):</span>
		<span class="hl kwd">ScriptApiBase</span><span class="hl opt">(</span><span class="hl kwc">ScriptingType</span><span class="hl opt">::</span>MainMenu<span class="hl opt">)</span>
<span class="hl opt">{</span>
	<span class="hl kwd">setGuiEngine</span><span class="hl opt">(</span>guiengine<span class="hl opt">);</span>

	SCRIPTAPI_PRECHECKHEADER

	<span class="hl kwd">lua_getglobal</span><span class="hl opt">(</span>L<span class="hl opt">,</span> <span class="hl str">&quot;core&quot;</span><span class="hl opt">);</span>
	<span class="hl kwb">int</span> top <span class="hl opt">=</span> <span class="hl kwd">lua_gettop</span><span class="hl opt">(</span>L<span class="hl opt">);</span>

	<span class="hl kwd">lua_newtable</span><span class="hl opt">(</span>L<span class="hl opt">);</span>
	<span class="hl kwd">lua_setglobal</span><span class="hl opt">(</span>L<span class="hl opt">,</span> <span class="hl str">&quot;gamedata&quot;</span><span class="hl opt">);</span>

	<span class="hl slc">// Initialize our lua_api modules</span>
	<span class="hl kwd">initializeModApi</span><span class="hl opt">(</span>L<span class="hl opt">,</span> top<span class="hl opt">);</span>
	<span class="hl kwd">lua_pop</span><span class="hl opt">(</span>L<span class="hl opt">,</span> <span class="hl num">1</span><span class="hl opt">);</span>

	<span class="hl slc">// Push builtin initialization type</span>
	<span class="hl kwd">lua_pushstring</span><span class="hl opt">(</span>L<span class="hl opt">,</span> <span class="hl str">&quot;mainmenu&quot;</span><span class="hl opt">);</span>
	<span class="hl kwd">lua_setglobal</span><span class="hl opt">(</span>L<span class="hl opt">,</span> <span class="hl str">&quot;INIT&quot;</span><span class="hl opt">);</span>

	infostream <span class="hl opt">&lt;&lt;</span> <span class="hl str">&quot;SCRIPTAPI: Initialized main menu modules&quot;</span> <span class="hl opt">&lt;&lt;</span> <span class="hl kwc">std</span><span class="hl opt">::</span>endl<span class="hl opt">;</span>
<span class="hl opt">}</span>

<span class="hl com">/******************************************************************************/</span>
<span class="hl kwb">void</span> <span class="hl kwc">MainMenuScripting</span><span class="hl opt">::</span><span class="hl kwd">initializeModApi</span><span class="hl opt">(</span>lua_State <span class="hl opt">*</span>L<span class="hl opt">,</span> <span class="hl kwb">int</span> top<span class="hl opt">)</span>
<span class="hl opt">{</span>
	<span class="hl kwd">registerLuaClasses</span><span class="hl opt">(</span>L<span class="hl opt">,</span> top<span class="hl opt">);</span>

	<span class="hl slc">// Initialize mod API modules</span>
	<span class="hl kwc">ModApiMainMenu</span><span class="hl opt">::</span><span class="hl kwd">Initialize</span><span class="hl opt">(</span>L<span class="hl opt">,</span> top<span class="hl opt">);</span>
	<span class="hl kwc">ModApiUtil</span><span class="hl opt">::</span><span class="hl kwd">Initialize</span><span class="hl opt">(</span>L<span class="hl opt">,</span> top<span class="hl opt">);</span>
	<span class="hl kwc">ModApiSound</span><span class="hl opt">::</span><span class="hl kwd">Initialize</span><span class="hl opt">(</span>L<span class="hl opt">,</span> top<span class="hl