You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.0 KiB
60 lines
2.0 KiB
2 years ago
|
#ifdef _DEBUG
|
||
|
|
#include "lua/repl.hpp"
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <string_view>
|
||
|
|
|
||
|
|
#include <expected.hpp>
|
||
|
|
#include <sol/sol.hpp>
|
||
|
|
#include <sol/utility/to_string.hpp>
|
||
|
|
|
||
|
|
#include "lua/lua.hpp"
|
||
|
|
#include "utils/str_cat.hpp"
|
||
|
|
|
||
|
|
namespace devilution {
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
sol::protected_function_result TryRunLuaAsExpressionThenStatement(std::string_view code)
|
||
|
|
{
|
||
|
|
// Try to compile as an expression first. This also how the `lua` repl is implemented.
|
||
|
|
const sol::state &lua = LuaState();
|
||
|
|
std::string expression = StrCat("return ", code, ";");
|
||
|
|
sol::detail::typical_chunk_name_t basechunkname = {};
|
||
|
|
sol::load_status status = static_cast<sol::load_status>(
|
||
|
|
luaL_loadbufferx(lua.lua_state(), expression.data(), expression.size(),
|
||
|
|
sol::detail::make_chunk_name(expression, sol::detail::default_chunk_name(), basechunkname), "text"));
|
||
|
|
if (status != sol::load_status::ok) {
|
||
|
|
// Try as a statement:
|
||
|
|
status = static_cast<sol::load_status>(
|
||
|
|
luaL_loadbufferx(lua.lua_state(), code.data(), code.size(),
|
||
|
|
sol::detail::make_chunk_name(code, sol::detail::default_chunk_name(), basechunkname), "text"));
|
||
|
|
if (status != sol::load_status::ok) {
|
||
|
|
return sol::protected_function_result(
|
||
|
|
lua.lua_state(), sol::absolute_index(lua.lua_state(), -1), 0, 1, static_cast<sol::call_status>(status));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
sol::stack_aligned_protected_function fn(lua.lua_state(), -1);
|
||
|
|
return fn();
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
tl::expected<std::string, std::string> RunLuaReplLine(std::string_view code)
|
||
|
|
{
|
||
|
|
const sol::protected_function_result result = TryRunLuaAsExpressionThenStatement(code);
|
||
|
|
if (!result.valid()) {
|
||
|
|
if (result.get_type() == sol::type::string) {
|
||
|
|
return tl::make_unexpected(result.get<std::string>());
|
||
|
|
}
|
||
|
|
return tl::make_unexpected("Unknown Lua error");
|
||
|
|
}
|
||
|
|
if (result.get_type() == sol::type::none) {
|
||
|
|
return std::string {};
|
||
|
|
}
|
||
|
|
return sol::utility::to_string(sol::stack_object(result));
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace devilution
|
||
|
|
#endif // _DEBUG
|