pajlada
77 lines
2.4 KiB
C++
77 lines
2.4 KiB
C++
#include "IrcConnection2.hpp"
|
|
|
|
#include "common/Version.hpp"
|
|
|
|
namespace chatterino {
|
|
|
|
namespace {
|
|
|
|
const auto payload = QString("chatterino/" CHATTERINO_VERSION);
|
|
|
|
} // namespace
|
|
|
|
IrcConnection::IrcConnection(QObject *parent)
|
|
: Communi::IrcConnection(parent)
|
|
{
|
|
// send ping every x seconds
|
|
this->pingTimer_.setInterval(5000);
|
|
this->pingTimer_.start();
|
|
QObject::connect(&this->pingTimer_, &QTimer::timeout, [this] {
|
|
if (this->isConnected())
|
|
{
|
|
if (this->waitingForPong_.load())
|
|
{
|
|
// Not sending another ping as we haven't received the matching pong yet
|
|
return;
|
|
}
|
|
|
|
if (!this->recentlyReceivedMessage_.load())
|
|
{
|
|
this->sendRaw("PING " + payload);
|
|
this->reconnectTimer_.start();
|
|
this->waitingForPong_ = true;
|
|
}
|
|
this->recentlyReceivedMessage_ = false;
|
|
}
|
|
});
|
|
|
|
// reconnect after x seconds without receiving a message
|
|
this->reconnectTimer_.setInterval(5000);
|
|
this->reconnectTimer_.setSingleShot(true);
|
|
QObject::connect(&this->reconnectTimer_, &QTimer::timeout, [this] {
|
|
if (this->isConnected())
|
|
{
|
|
reconnectRequested.invoke();
|
|
}
|
|
});
|
|
|
|
QObject::connect(this, &Communi::IrcConnection::connected, this, [this] {
|
|
this->waitingForPong_ = false;
|
|
});
|
|
|
|
QObject::connect(this, &Communi::IrcConnection::pongMessageReceived,
|
|
[this](Communi::IrcPongMessage *message) {
|
|
if (message->argument() == payload)
|
|
{
|
|
this->waitingForPong_ = false;
|
|
}
|
|
});
|
|
|
|
QObject::connect(
|
|
this, &Communi::IrcConnection::messageReceived,
|
|
[this](Communi::IrcMessage *message) {
|
|
this->recentlyReceivedMessage_ = true;
|
|
|
|
if (this->reconnectTimer_.isActive())
|
|
{
|
|
this->reconnectTimer_.stop();
|
|
|
|
// The reconnect timer had started, that means we were waiting for a pong.
|
|
// Since we received a message, this means that any pong response is meaningless, so we can just stop waiting for the pong and send another ping soon:tm:
|
|
this->waitingForPong_ = false;
|
|
}
|
|
});
|
|
}
|
|
|
|
} // namespace chatterino
|