static void initialize(LuaStack& stack, rainback::protocol::AbstractLine& ptl)
{
lua::push(stack, lua::value::table);
auto methods = stack.saveAndPop();
methods["listen"] = std::function<void(LuaStack&)>(
[](LuaStack& stack) {
rainback::protocol::AbstractLine& ptl = stack.as<rainback::protocol::AbstractLine&>(1);
auto socket = stack.as<QAbstractSocket*>(2);
if (!socket) {
throw LuaException("A socket or other IO device must be provided");
}
// Introduce a reference visible from Lua to ensure the socket doesn't get prematurely GC'd
auto worker = stack.lua()(""
"return function(proto, socket)\n"
" proto.__extra.io = socket;\n"
"end;"
);
lua::push(stack, worker);
stack.insert(1);
stack.pushedInvoke(2);
ptl.listen(socket);
stack.clear();
}
);
methods["write"] = std::function<void(rainback::protocol::AbstractLine&, const QString&)>(
[](rainback::protocol::AbstractLine& ptl, const QString& data) {
if (!ptl.io()) {
throw LuaException("Write must have a socket or IO device to write to");
}
auto io = ptl.io();
if (!io->isOpen()) {
throw LuaException("Write cannot write to a closed socket or IO device");
}
if (qobject_cast<QAbstractSocket*>(io)) {
auto socket = qobject_cast<QAbstractSocket*>(io);
if (socket->state() != QAbstractSocket::ConnectedState) {
std::stringstream str;
str << "Write cannot write to an unconnected socket. Current state was ";
str << rainback::proxy::getSocketStateName(socket->state());
throw LuaException(str.str());
}
}
ptl.write(data);
}
);
rainback::proxy::wrapQObject(stack, ptl, methods);
rainback::proxy::observeToDestroy(stack, ptl, true);
}
