#include "dl_lua.h"

#define LUA_USE_APICHECK
#include <lua.hpp>

#include "dl_lua_register.h"

#include "io/dl_file.h"

namespace dl
{
	LuaScript* g_lua;

	LuaScript::LuaScript()
	{
		L = luaL_newstate();
		luaL_openlibs(L);
	}

	LuaScript::~LuaScript()
	{
		lua_close(L);
	}

	bool LuaScript::DoString(std::string_view str)
	{
		int ret = luaL_loadstring(L, str.data());
		if (ret)
		{
			log_warn("lua 解析失败[{}]：\n{}", lua_tostring(L, -1), str);
			lua_pop(L, 1);
			return false;
		}
		ret = lua_pcall(L, 0, 0, 0);
		if (ret)
		{
			log_warn("lua 执行失败[{}]：\n{}", lua_tostring(L, -1), str);
			lua_pop(L, 1);
			return false;
		}
		return true;
	}

	bool LuaScript::DoFile(std::string_view path_name)
	{
		Buffer buf = File::GetBuffer(path_name);
		int ret = luaL_loadbuffer(L, (const char*)buf.data(), buf.size(), path_name.data());
		if (ret)
		{
			log_warn("lua 解析（加载）失败[{}]：{}", path_name, lua_tostring(L, -1));
			lua_pop(L, 1);
			return false;
		}
		ret = lua_pcall(L, 0, 0, 0);
		if (ret)
		{
			log_warn("lua 执行失败：{}", lua_tostring(L, -1));
			lua_pop(L, 1);
			return false;
		}
		log_msg("执行lua脚本成功：{}", path_name);
		return true;
	}

	bool LuaScript::DoFileEx(std::string_view path_name, LuaRef& ret_ref)
	{
		Buffer buf = File::GetBuffer(path_name);
		int ret = luaL_loadbuffer(L, (const char*)buf.data(), buf.size(), path_name.data());
		if (ret)
		{
			log_warn("lua 解析（加载）失败[{}]：{}", path_name, lua_tostring(L, -1));
			lua_pop(L, 1);
			return false;
		}
		ret = lua_pcall(L, 0, 1, 0);
		if (ret)
		{
			log_warn("lua 执行失败：{}", lua_tostring(L, -1));
			lua_pop(L, 1);
			return false;
		}
		ret_ref = LuaRef::fromStack(L, -1);
		lua_pop(L, 1);

		log_msg("执行lua脚本成功：{}", path_name);
		return true;
	}

	bool LuaScript::CallFunction(luabridge::LuaRef func)
	{
		if (!func)
		{
			log_warn("lua 执行函数失败：为空");
			return false;
		}

		const char* traceback = nullptr;
		auto handler = [&traceback](lua_State* L) -> int
			{
				const char* msg = luaL_tolstring(L, 1, nullptr);
				luaL_traceback(L, L, msg, 1);
				lua_pop(L, 1);

				const char* tb = lua_tostring(L, -1);
				traceback = tb ? tb : "Unknown Lua error";

				return 1;
			};

		auto result = func.callWithHandler(handler);
		if (!result)
		{
			log_warn("lua 执行函数失败：{}", traceback);
			return false;
		}
		return true;
	}

	std::string LuaScript::GetRefString(LuaRef ref)
	{
		if (!ref)
			return {};
		if (ref.isCallable())
		{
			auto ret = ref.call<std::string>();
			if (ret)
				return ret.value();
		}
		else if (ref.isString())
		{
			return ref.cast<std::string>().value();
		}
		return {};
	}

	void LuaScript::RegisterAll()
	{
		lua_register_all(L);
	}

}