forked from qt-creator/qt-creator
Lua: Change fetch module access to async
Change-Id: I32bd760e347e2bd4465bd028e31fb3c4ff220aa5 Reviewed-by: David Schulz <david.schulz@qt.io>
This commit is contained in:
@@ -36,7 +36,11 @@ static QString opToString(QNetworkAccessManager::Operation op)
|
|||||||
|
|
||||||
void addFetchModule()
|
void addFetchModule()
|
||||||
{
|
{
|
||||||
LuaEngine::registerProvider("__fetch", [](sol::state_view lua) -> sol::object {
|
LuaEngine::registerProvider(
|
||||||
|
"Fetch", [](sol::state_view lua) -> sol::object {
|
||||||
|
sol::table async = lua.script("return require('async')", "_fetch_").get<sol::table>();
|
||||||
|
sol::function wrap = async["wrap"];
|
||||||
|
|
||||||
sol::table fetch = lua.create_table();
|
sol::table fetch = lua.create_table();
|
||||||
|
|
||||||
auto networkReplyType = lua.new_usertype<QNetworkReply>(
|
auto networkReplyType = lua.new_usertype<QNetworkReply>(
|
||||||
@@ -53,20 +57,22 @@ void addFetchModule()
|
|||||||
.arg(r->error());
|
.arg(r->error());
|
||||||
});
|
});
|
||||||
|
|
||||||
fetch["fetch_cb"] = [](const sol::table &options, const sol::function &callback, const sol::this_state &thisState) {
|
fetch["fetch_cb"] = [](const sol::table &options,
|
||||||
|
const sol::function &callback,
|
||||||
|
const sol::this_state &thisState) {
|
||||||
auto url = options.get<QString>("url");
|
auto url = options.get<QString>("url");
|
||||||
|
|
||||||
auto method = (options.get_or<QString>("method", "GET")).toLower();
|
auto method = (options.get_or<QString>("method", "GET")).toLower();
|
||||||
auto headers = options.get_or<sol::table>("headers", {});
|
auto headers = options.get_or<sol::table>("headers", {});
|
||||||
auto data = options.get_or<QString>("body", {});
|
auto data = options.get_or<QString>("body", {});
|
||||||
bool convertToTable = options.get<std::optional<bool>>("convertToTable").value_or(false);
|
bool convertToTable
|
||||||
|
= options.get<std::optional<bool>>("convertToTable").value_or(false);
|
||||||
|
|
||||||
QNetworkRequest request((QUrl(url)));
|
QNetworkRequest request((QUrl(url)));
|
||||||
if (headers && !headers.empty()) {
|
if (headers && !headers.empty()) {
|
||||||
for (const auto &[k, v] : headers) {
|
for (const auto &[k, v] : headers)
|
||||||
request.setRawHeader(k.as<QString>().toUtf8(), v.as<QString>().toUtf8());
|
request.setRawHeader(k.as<QString>().toUtf8(), v.as<QString>().toUtf8());
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
QNetworkReply *reply = nullptr;
|
QNetworkReply *reply = nullptr;
|
||||||
if (method == "get")
|
if (method == "get")
|
||||||
@@ -77,11 +83,16 @@ void addFetchModule()
|
|||||||
throw std::runtime_error("Unknown method: " + method.toStdString());
|
throw std::runtime_error("Unknown method: " + method.toStdString());
|
||||||
|
|
||||||
if (convertToTable) {
|
if (convertToTable) {
|
||||||
QObject::connect(reply, &QNetworkReply::finished, reply, [reply, thisState, callback]() {
|
QObject::connect(
|
||||||
|
reply,
|
||||||
|
&QNetworkReply::finished,
|
||||||
|
&LuaEngine::instance(),
|
||||||
|
[reply, thisState, callback]() {
|
||||||
reply->deleteLater();
|
reply->deleteLater();
|
||||||
|
|
||||||
if (reply->error() != QNetworkReply::NoError) {
|
if (reply->error() != QNetworkReply::NoError) {
|
||||||
callback(QString("%1 (%2)").arg(reply->errorString()).arg(reply->error()));
|
callback(
|
||||||
|
QString("%1 (%2)").arg(reply->errorString()).arg(reply->error()));
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -103,7 +114,8 @@ void addFetchModule()
|
|||||||
});
|
});
|
||||||
|
|
||||||
} else {
|
} else {
|
||||||
QObject::connect(reply, &QNetworkReply::finished, reply, [reply, callback]() {
|
QObject::connect(
|
||||||
|
reply, &QNetworkReply::finished, &LuaEngine::instance(), [reply, callback]() {
|
||||||
// We don't want the network reply to be deleted by the manager, but
|
// We don't want the network reply to be deleted by the manager, but
|
||||||
// by the Lua GC
|
// by the Lua GC
|
||||||
reply->setParent(nullptr);
|
reply->setParent(nullptr);
|
||||||
@@ -112,24 +124,10 @@ void addFetchModule()
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
fetch["fetch"] = wrap(fetch["fetch_cb"]);
|
||||||
|
|
||||||
return fetch;
|
return fetch;
|
||||||
});
|
});
|
||||||
|
|
||||||
LuaEngine::registerProvider("Fetch", [](sol::state_view lua) -> sol::object {
|
|
||||||
return lua
|
|
||||||
.script(
|
|
||||||
R"(
|
|
||||||
local f = require("__fetch")
|
|
||||||
local a = require("async")
|
|
||||||
|
|
||||||
return {
|
|
||||||
fetch_cb = f.fetch_cb,
|
|
||||||
fetch = a.wrap(f.fetch_cb)
|
|
||||||
}
|
|
||||||
)",
|
|
||||||
"_fetch_")
|
|
||||||
.get<sol::table>();
|
|
||||||
});
|
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace Lua::Internal
|
} // namespace Lua::Internal
|
||||||
|
Reference in New Issue
Block a user